vectordb

type ClientFactory interface {
	// CreateClient creates a new vector database client
	CreateClient(config *Config) (VectorDBClient, error)

	// GetSupportedTypes returns the list of supported database types
	GetSupportedTypes() []VectorDBType

	// ValidateConfig validates the configuration for the specific database type
	ValidateConfig(config *Config) error
}

type CollectionInfo struct {
	Name        string `json:"name"`
	Description string `json:"description,omitempty"`
	Count       int64  `json:"count"`
	Vectorizer  string `json:"vectorizer,omitempty"`
}

type CollectionOperations interface {
	// CreateCollection creates a new collection with the given schema
	CreateCollection(ctx context.Context, name string, schema *CollectionSchema) error

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

	// ListCollections returns a list of all collections
	ListCollections(ctx context.Context) ([]CollectionInfo, error)

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

	// GetCollectionCount returns the number of documents in a collection
	GetCollectionCount(ctx context.Context, name string) (int64, error)
}

type CollectionSchema struct {
	Class      string           `json:"class"`
	Vectorizer string           `json:"vectorizer,omitempty"`
	Properties []SchemaProperty `json:"properties"`
}

type Config struct {
	Type         VectorDBType `yaml:"type"`
	URL          string       `yaml:"url,omitempty"`
	APIKey       string       `yaml:"api_key,omitempty"`
	OpenAIAPIKey string       `yaml:"openai_api_key,omitempty"`
	Timeout      int          `yaml:"timeout,omitempty"`

	// Mock-specific configuration
	Enabled            bool `yaml:"enabled,omitempty"`
	SimulateEmbeddings bool `yaml:"simulate_embeddings,omitempty"`
	EmbeddingDimension int  `yaml:"embedding_dimension,omitempty"`

	// Supabase-specific configuration
	DatabaseURL string `yaml:"database_url,omitempty"`
	DatabaseKey string `yaml:"database_key,omitempty"`

	// Milvus-specific configuration
	Address  string `yaml:"address,omitempty"`
	Username string `yaml:"username,omitempty"`
	Password string `yaml:"password,omitempty"`
	Database string `yaml:"database,omitempty"`
	Tenant   string `yaml:"tenant,omitempty"`

	// MongoDB-specific configuration
	VectorDimensions int    `yaml:"vector_dimensions,omitempty"`
	SimilarityMetric string `yaml:"similarity_metric,omitempty"`
}

type Document struct {
	ID        string                 `json:"id"`
	Text      string                 `json:"text"`
	Content   string                 `json:"content"`
	Image     string                 `json:"image"`
	ImageData string                 `json:"image_data"`
	URL       string                 `json:"url"`
	Metadata  map[string]interface{} `json:"metadata"`
}

type DocumentOperations interface {
	// CreateDocument creates a new document in the specified collection
	CreateDocument(ctx context.Context, collectionName string, document *Document) error

	// CreateDocuments creates multiple documents in batch
	CreateDocuments(ctx context.Context, collectionName string, documents []*Document) error

	// GetDocument retrieves a document by ID
	GetDocument(ctx context.Context, collectionName, documentID string) (*Document, error)

	// UpdateDocument updates an existing document
	UpdateDocument(ctx context.Context, collectionName string, document *Document) error

	// DeleteDocument deletes a document by ID
	DeleteDocument(ctx context.Context, collectionName, documentID string) error

	// DeleteDocuments deletes multiple documents by IDs
	DeleteDocuments(ctx context.Context, collectionName string, documentIDs []string) error

	// DeleteDocumentsByMetadata deletes documents matching metadata criteria
	DeleteDocumentsByMetadata(ctx context.Context, collectionName string, metadata map[string]interface{}) error

	// ListDocuments returns a list of documents in a collection
	ListDocuments(ctx context.Context, collectionName string, limit int, offset int) ([]*Document, error)
}

type ErrorType string

type FieldDefinition struct {
	Name string
	Type string
}

type QueryOperations interface {
	// SearchSemantic performs semantic search using vector embeddings
	SearchSemantic(ctx context.Context, collectionName, query string, options *QueryOptions) ([]*QueryResult, error)

	// SearchBM25 performs keyword-based search using BM25
	SearchBM25(ctx context.Context, collectionName, query string, options *QueryOptions) ([]*QueryResult, error)

	// SearchHybrid performs hybrid search combining vector and keyword search
	SearchHybrid(ctx context.Context, collectionName, query string, options *QueryOptions) ([]*QueryResult, error)

	// SearchByMetadata searches documents by metadata fields
	SearchByMetadata(ctx context.Context, collectionName string, metadata map[string]interface{}, options *QueryOptions) ([]*QueryResult, error)
}

type QueryOptions struct {
	TopK           int     `json:"top_k"`
	Distance       float64 `json:"distance"`
	SearchMetadata bool    `json:"search_metadata"`
	NoTruncate     bool    `json:"no_truncate"`
	UseBM25        bool    `json:"use_bm25"`
}

type QueryResult struct {
	Document Document `json:"document"`
	Score    float64  `json:"score"`
}

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

type SchemaOperations interface {
	// GetSchema retrieves the schema for a collection
	GetSchema(ctx context.Context, collectionName string) (*CollectionSchema, error)

	// UpdateSchema updates the schema for a collection
	UpdateSchema(ctx context.Context, collectionName string, schema *CollectionSchema) error

	// GetDefaultSchema returns a default schema for the given type
	GetDefaultSchema(schemaType SchemaType, collectionName string) *CollectionSchema

	// ValidateSchema validates a schema definition
	ValidateSchema(schema *CollectionSchema) error
}

type SchemaProperty struct {
	Name             string                 `json:"name" yaml:"name"`
	DataType         []string               `json:"dataType" yaml:"datatype"`
	Description      string                 `json:"description,omitempty" yaml:"description,omitempty"`
	NestedProperties []SchemaProperty       `json:"nestedProperties,omitempty" yaml:"nestedproperties,omitempty"`
	JSONSchema       map[string]interface{} `json:"json_schema,omitempty" yaml:"json_schema,omitempty"`
}

type SchemaType string

type VectorDBClient interface {
	// Health checks the health of the vector database instance
	Health(ctx context.Context) error

	// Collection operations
	CollectionOperations

	// Document operations
	DocumentOperations

	// Query operations
	QueryOperations

	// Schema operations
	SchemaOperations
}

type VectorDBError struct {
	Type       ErrorType
	Message    string
	Cause      error
	Collection string
	Document   string
	Details    map[string]interface{}
}

type VectorDBType string