34 lines
954 B
Go
34 lines
954 B
Go
|
package weaviate
|
||
|
|
|
||
|
|
import (
|
||
|
|
"sync"
|
||
|
|
|
||
|
|
"github.com/weaviate/weaviate-go-client/v5/weaviate"
|
||
|
|
)
|
||
|
|
|
||
|
|
type weaviateRepository struct {
|
||
|
|
client *weaviate.Client
|
||
|
|
collectionBaseName string
|
||
|
|
replicationFactor int // 0 = use Weaviate server default
|
||
|
|
desiredShardCount int // 0 = use Weaviate server default
|
||
|
|
// Cache for initialized collections (dimension -> true)
|
||
|
|
initializedCollections sync.Map
|
||
|
|
}
|
||
|
|
|
||
|
|
type WeaviateVectorEmbedding struct {
|
||
|
|
Content string `json:"content"`
|
||
|
|
SourceID string `json:"source_id"`
|
||
|
|
SourceType int `json:"source_type"`
|
||
|
|
ChunkID string `json:"chunk_id"`
|
||
|
|
KnowledgeID string `json:"knowledge_id"`
|
||
|
|
KnowledgeBaseID string `json:"knowledge_base_id"`
|
||
|
|
TagID string `json:"tag_id"`
|
||
|
|
Embedding []float32 `json:"embedding"`
|
||
|
|
IsEnabled bool `json:"is_enabled"`
|
||
|
|
}
|
||
|
|
|
||
|
|
type WeaviateVectorEmbeddingWithScore struct {
|
||
|
|
WeaviateVectorEmbedding
|
||
|
|
Score float64
|
||
|
|
}
|