roddomsatish/VectorEmbeddings

1from sentence_transformers import SentenceTransformer2 3 4# Func.on to generate .tle and Logline from simple chaining5def generate_vector(prompt, api_key):6    #os.environ["OPENAI_API_KEY"] = api_key7    #embeddings = OpenAIEmbeddings()8    #text_embedding = embeddings.embed_query(prompt)9    model = SentenceTransformer('all-MiniLM-L6-v2')10    text_embeddings = model.encode(prompt)11    return text_embeddings