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 {
	GetChunk(ctx context.Context, id string) (SearchResult, error)
	GetChunks(ctx context.Context, ids []string) ([]SearchResult, error)
	GetAll(ctx context.Context, queries []Query, selectFields []string, cursor *string, count int64) ([]SearchResult, *string, error)
	GetNearest(ctx context.Context, vector []float32, queries []Query, selectFields []string, threshold float64, limit int64) ([]SearchResult, error)
	Add(ctx context.Context, id string, embedding []float32, metadata map[string]interface{}) error
	Delete(ctx context.Context, id string) error
	DeleteAll(ctx context.Context, queries []Query) ([]DeleteResult, error)
	Close(ctx context.Context) error
}

type VectorStoreType string

type WeaviateConfig struct {
	// Connection settings
	Scheme string `json:"scheme"` // "http" or "https" - REQUIRED
	Host   string `json:"host"`   // "localhost:8080" - REQUIRED

	// 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)

	// Class name
	ClassName string `json:"class_name,omitempty"`

	// Select fields
	Properties []WeaviateProperties `json:"properties,omitempty"`
}

type WeaviateProperties struct {
	Name     string   `json:"name"`
	DataType []string `json:"data_type"`
}

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