vector

type CodeChunkService struct {
	// contains filtered or unexported fields
}

type DistanceMetric string

type EmbeddingModel interface {
	// GenerateEmbedding generates a vector embedding for the given text
	GenerateEmbedding(ctx context.Context, text string) ([]float32, error)

	// GenerateEmbeddings generates vector embeddings for multiple texts (batch operation)
	GenerateEmbeddings(ctx context.Context, texts []string) ([][]float32, error)

	// GetDimension returns the dimension of the embedding vectors
	GetDimension() int

	// GetModelName returns the name of the embedding model being used
	GetModelName() string
}

type MethodSignatureData struct {
	MethodName     string
	ClassName      string
	ReturnType     string
	ParameterTypes []string
	ParameterNames []string
	FilePath       string
	StartLine      int
	EndLine        int
	FileID         int32
}

type OllamaEmbedding struct {
	// contains filtered or unexported fields
}

type OllamaEmbeddingConfig struct {
	APIURL    string // e.g., "http://localhost:11434"
	APIKey    string // Optional API key for authentication
	Model     string // e.g., "nomic-embed-text", "all-minilm"
	Dimension int    // Dimension of the embedding vector
}

type QdrantDatabase struct {
	// contains filtered or unexported fields
}

type VectorDatabase interface {
	// CreateCollection creates a new collection with the specified dimension and distance metric
	CreateCollection(ctx context.Context, collectionName string, vectorDim int, distance DistanceMetric) error

	// DeleteCollection deletes a collection
	DeleteCollection(ctx context.Context, collectionName string) error

	// CollectionExists checks if a collection exists
	CollectionExists(ctx context.Context, collectionName string) (bool, error)

	// UpsertChunks inserts or updates code chunks in the vector database
	UpsertChunks(ctx context.Context, collectionName string, chunks []*model.CodeChunk) error

	// SearchSimilar finds similar code chunks using vector similarity search
	SearchSimilar(ctx context.Context, collectionName string, queryVector []float32, limit int, filter map[string]interface{}) ([]*model.CodeChunk, []float32, error)

	// GetChunkByID retrieves a specific chunk by its ID
	GetChunkByID(ctx context.Context, collectionName string, chunkID string) (*model.CodeChunk, error)

	// DeleteChunk deletes a chunk by its ID
	DeleteChunk(ctx context.Context, collectionName string, chunkID string) error

	// GetChunksByFilePath retrieves all chunks for a specific file path
	GetChunksByFilePath(ctx context.Context, collectionName string, filePath string) ([]*model.CodeChunk, error)

	// Close closes the database connection
	Close() error

	// Health checks the health of the vector database
	Health(ctx context.Context) error
}