Spaces:

janar
/

retrival_aug_llm

Running

retrival_aug_llm / api /db /embedding.py

add the embedding config file

5fa437b about 3 years ago

394 Bytes

from collections import namedtuple
from langchain.embeddings import OpenAIEmbeddings, ElasticsearchEmbeddings, SentenceTransformerEmbeddings

Embedding = namedtuple("Embedding",["embedding", "dimension", "distance"])

EMBEDDINGS = {
    'SENTENCE': Embedding(SentenceTransformerEmbeddings(),768, "Cosine"),
    'OPEN_AI': Embedding(OpenAIEmbeddings(), 1563, "Cosine"),
}
                       

	from collections import namedtuple
	from langchain.embeddings import OpenAIEmbeddings, ElasticsearchEmbeddings, SentenceTransformerEmbeddings

	Embedding = namedtuple("Embedding",["embedding", "dimension", "distance"])

	EMBEDDINGS = {
	'SENTENCE': Embedding(SentenceTransformerEmbeddings(),768, "Cosine"),
	'OPEN_AI': Embedding(OpenAIEmbeddings(), 1563, "Cosine"),
	}