embedding_store

package

v1.6.2 Latest Latest Go to latest Published: Nov 13, 2024 License: Apache-2.0 Imports: 5 Imported by: 0

Details

This section is empty.

This section is empty.

func NewEmbeddingStoreModel() contracts.IEmbeddingStore

NewEmbeddingStoreModel initializes a new CodeEmbeddingStoreModel.

type EmbeddingStore struct {
	EmbeddingsStore map[string][]float64
	CodeStore       map[string]string
	// contains filtered or unexported fields
}

EmbeddingStore holds the embeddings and their corresponding code chunks.

func (store *EmbeddingStore) CosineSimilarity(vec1, vec2 []float64) float64

func (store *EmbeddingStore) FindRelevantChunks(queryEmbedding []float64, topN int, embeddingModel string, threshold float64) []string

FindRelevantChunks retrieves the relevant code chunks from the embedding store based on a similarity threshold.

func (store *EmbeddingStore) FindThresholdByModel(modelName string) float64

func (store *EmbeddingStore) Save(key string, code string, embeddings []float64)

Save stores the embeddings and corresponding code in the store. If the existing code is different, it updates the entry.

Path	Synopsis
contracts