vector

type Config struct {
	Type       string                 `json:"type"`              // Type of vector store (e.g., "pinecone", "weaviate", "local")
	Dimension  int                    `json:"dimension"`         // Dimension of vectors
	MetricType string                 `json:"metric_type"`       // Distance metric type (e.g., "cosine", "euclidean", "dot_product")
	Options    map[string]interface{} `json:"options,omitempty"` // Additional store-specific options
}

type Document struct {
	ID       string                 `json:"id"`
	Content  string                 `json:"content"`
	Vector   Vector                 `json:"vector"`
	Metadata map[string]interface{} `json:"metadata,omitempty"`
}

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

type MetricType string

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

type SearchResult struct {
	Document Document `json:"document"`
	Score    float32  `json:"score"`
}

type Store interface {
	// Insert adds documents to the store
	Insert(ctx context.Context, docs ...Document) error

	// Search finds similar documents based on query vector
	Search(ctx context.Context, queryVector Vector, limit int) ([]SearchResult, error)

	// Delete removes documents from the store
	Delete(ctx context.Context, ids ...string) error

	// Update updates existing documents
	Update(ctx context.Context, docs ...Document) error

	// Get retrieves documents by their IDs
	Get(ctx context.Context, ids ...string) ([]Document, error)

	// Clear removes all documents from the store
	Clear(ctx context.Context) error

	// Close closes the store connection
	Close() error
}

type StoreType string

type Vector []float32

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