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