vector

type ChromaConfig struct {
	// Host is the Chroma server hostname.
	Host string `yaml:"host"`

	// Port is the Chroma HTTP port (default: 8000).
	Port int `yaml:"port,omitempty"`

	// APIKey for authenticated access (optional).
	APIKey string `yaml:"api_key,omitempty"`

	// UseTLS enables HTTPS connections.
	UseTLS bool `yaml:"use_tls,omitempty"`
}

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

type ChromemConfig struct {
	// PersistPath for file persistence (optional).
	// If empty, vectors are stored in memory only.
	// Directory will be created if it doesn't exist.
	PersistPath string `yaml:"persist_path,omitempty"`

	// Compress enables gzip compression for persistence.
	// Reduces file size but increases CPU usage.
	Compress bool `yaml:"compress,omitempty"`
}

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

type Config struct {
	// Type identifies the provider implementation.
	// Values: "chromem", "qdrant", "chroma", "pinecone", "milvus", "weaviate"
	Type string `yaml:"type"`

	// Collection is the default collection name.
	// Individual operations can override this.
	Collection string `yaml:"collection,omitempty"`
}

type MilvusConfig struct {
	// Host is the Milvus server hostname.
	Host string `yaml:"host"`

	// Port is the Milvus HTTP port (default: 19530).
	Port int `yaml:"port,omitempty"`

	// APIKey for authenticated access (optional).
	APIKey string `yaml:"api_key,omitempty"`

	// UseTLS enables HTTPS connections.
	UseTLS bool `yaml:"use_tls,omitempty"`
}

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

type NilProvider struct{}

type PineconeConfig struct {
	// APIKey is required for Pinecone authentication.
	APIKey string `yaml:"api_key"`

	// Host is the Pinecone API host (optional, defaults to https://api.pinecone.io).
	Host string `yaml:"host,omitempty"`

	// IndexName is the default index to use.
	IndexName string `yaml:"index_name"`

	// Environment is the Pinecone environment (e.g., "us-west1-gcp").
	Environment string `yaml:"environment,omitempty"`
}

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

type Provider interface {
	// Upsert adds or updates a document with its vector embedding.
	//
	// If a document with the same ID exists, it will be updated.
	// The vector dimension must match the collection's configured dimension.
	//
	// Parameters:
	//   - collection: logical grouping (e.g., "memory", "documents")
	//   - id: unique document identifier
	//   - vector: embedding vector from the embedder
	//   - metadata: additional searchable/filterable attributes
	Upsert(ctx context.Context, collection string, id string, vector []float32, metadata map[string]any) error

	// Search finds the most similar vectors in a collection.
	//
	// Returns results ordered by similarity score (highest first).
	// The query vector should be generated by the same embedder
	// used for indexing.
	//
	// Parameters:
	//   - collection: which collection to search
	//   - vector: query embedding vector
	//   - topK: maximum number of results to return
	Search(ctx context.Context, collection string, vector []float32, topK int) ([]Result, error)

	// SearchWithFilter combines vector similarity with metadata filtering.
	//
	// The filter map supports equality matching on metadata fields.
	// Filter semantics:
	//   - {"field": "value"} - exact match
	//   - {"field1": "v1", "field2": "v2"} - AND of all conditions
	//
	// Example:
	//   filter := map[string]any{"user_id": "user123", "type": "document"}
	SearchWithFilter(ctx context.Context, collection string, vector []float32, topK int, filter map[string]any) ([]Result, error)

	// Delete removes a document from a collection by ID.
	Delete(ctx context.Context, collection string, id string) error

	// DeleteByFilter removes all documents matching the filter.
	//
	// Use with caution - this can delete many documents at once.
	DeleteByFilter(ctx context.Context, collection string, filter map[string]any) error

	// CreateCollection creates a new collection with specified vector dimension.
	//
	// Some backends create collections implicitly on first Upsert.
	// This method allows explicit creation with configuration.
	CreateCollection(ctx context.Context, collection string, vectorDimension int) error

	// DeleteCollection removes a collection and all its documents.
	DeleteCollection(ctx context.Context, collection string) error

	// Name returns the provider implementation name (e.g., "chromem", "qdrant").
	Name() string

	// Close releases resources held by the provider.
	// Should be called when the provider is no longer needed.
	io.Closer
}

type ProviderConfig struct {
	// Type identifies which provider to create.
	Type ProviderType `yaml:"type"`

	// Chromem configuration (used when Type == "chromem").
	Chromem *ChromemConfig `yaml:"chromem,omitempty"`

	// Qdrant configuration (used when Type == "qdrant").
	Qdrant *QdrantConfig `yaml:"qdrant,omitempty"`

	// Pinecone configuration (used when Type == "pinecone").
	Pinecone *PineconeConfig `yaml:"pinecone,omitempty"`

	// Weaviate configuration (used when Type == "weaviate").
	Weaviate *WeaviateConfig `yaml:"weaviate,omitempty"`

	// Milvus configuration (used when Type == "milvus").
	Milvus *MilvusConfig `yaml:"milvus,omitempty"`

	// Chroma configuration (used when Type == "chroma").
	Chroma *ChromaConfig `yaml:"chroma,omitempty"`
}

type ProviderType string

type QdrantConfig struct {
	// Host is the Qdrant server hostname.
	Host string `yaml:"host"`

	// Port is the Qdrant gRPC port (default: 6334).
	Port int `yaml:"port"`

	// APIKey for authenticated access (optional).
	APIKey string `yaml:"api_key,omitempty"`

	// UseTLS enables TLS connections.
	UseTLS bool `yaml:"use_tls,omitempty"`
}

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

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

type Result struct {
	// ID is the unique document identifier.
	ID string

	// Score represents similarity/relevance (higher is better).
	Score float32

	// Content is the original text content (if stored in metadata).
	Content string

	// Vector is the document's embedding (optional, not always returned).
	Vector []float32

	// Metadata contains the document's metadata fields.
	Metadata map[string]any
}

type WeaviateConfig struct {
	// Host is the Weaviate server hostname.
	Host string `yaml:"host"`

	// Port is the Weaviate HTTP port (default: 8080).
	Port int `yaml:"port,omitempty"`

	// APIKey for authenticated access (optional).
	APIKey string `yaml:"api_key,omitempty"`

	// UseTLS enables HTTPS connections.
	UseTLS bool `yaml:"use_tls,omitempty"`
}

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