vectorstore

type Config struct {
	Enabled bool            `json:"enabled"`
	Type    VectorStoreType `json:"type"`
	Config  any             `json:"config"`
}

type DeleteResult struct {
	ID     string
	Status DeleteStatus
	Error  string
}

type DeleteStatus string

type Query struct {
	Field    string
	Operator QueryOperator
	Value    interface{}
}

type QueryOperator string

type SearchResult struct {
	ID         string
	Score      *float64
	Properties map[string]interface{}
}

type VectorStore interface {
	CreateNamespace(ctx context.Context, namespace string, properties map[string]VectorStoreProperties) error
	DeleteNamespace(ctx context.Context, namespace string) error
	GetChunk(ctx context.Context, namespace string, id string) (SearchResult, error)
	GetChunks(ctx context.Context, namespace string, ids []string) ([]SearchResult, error)
	GetAll(ctx context.Context, namespace string, queries []Query, selectFields []string, cursor *string, limit int64) ([]SearchResult, *string, error)
	GetNearest(ctx context.Context, namespace string, vector []float32, queries []Query, selectFields []string, threshold float64, limit int64) ([]SearchResult, error)
	Add(ctx context.Context, namespace string, id string, embedding []float32, metadata map[string]interface{}) error
	Delete(ctx context.Context, namespace string, id string) error
	DeleteAll(ctx context.Context, namespace string, queries []Query) ([]DeleteResult, error)
	Close(ctx context.Context, namespace string) error
}

type VectorStoreProperties struct {
	DataType    VectorStorePropertyType `json:"data_type"`
	Description string                  `json:"description"`
}

type VectorStorePropertyType string

type VectorStoreType string

type WeaviateConfig struct {
	// Connection settings
	Scheme     string              `json:"scheme"`                // "http" or "https" - REQUIRED
	Host       string              `json:"host"`                  // "localhost:8080" - REQUIRED
	GrpcConfig *WeaviateGrpcConfig `json:"grpc_config,omitempty"` // grpc config for weaviate (optional)

	// Authentication settings (optional)
	ApiKey  string            `json:"api_key,omitempty"` // API key for authentication
	Headers map[string]string `json:"headers,omitempty"` // Additional headers

	// Connection settings
	Timeout time.Duration `json:"timeout,omitempty"` // Request timeout (optional)
}

type WeaviateGrpcConfig struct {
	// Host is the host of the weaviate server (host:port).
	// If host is without a port number then the 80 port for insecured and 443 port for secured connections will be used.
	Host string `json:"host"`
	// Secured is a boolean flag indicating if the connection is secured
	Secured bool `json:"secured"`
}

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