def generate_embeddings()

in Autogen_v0.4/rag_agent/openai_helper.py [0:0]

• 13 lines of code
• 2 McCabe index (conditional complexity)

def generate_embeddings(text, model=model):
    try:
        openai_client = AzureOpenAI(
            azure_endpoint=endpoint,
            api_key=api_key,
            api_version="2023-05-15",
        )
        embedding = openai_client.embeddings.create(input = [text], model=model).data[0].embedding
        return embedding
    except Exception as e:
        print("Error calling OpenAI:" + openai_client.base_url)
        print(e)
        raise