Spaces:
Runtime error
Runtime error
File size: 330 Bytes
4962437 |
1 2 3 4 5 6 7 8 9 10 11 |
# This file contains the function that embeds the input into a vector
from chromadb import EmbeddingFunction
def openai_embed(self, input, api_key, model_name):
openai = EmbeddingFunction.OpenAIEmbeddingFunction(
api_key=api_key,
model_name=model_name
)
embedding = openai(input)
return embedding |