def get_embedding()

in src/chat_request.py [0:0]

• 11 lines of code
• 1 McCabe index (conditional complexity)

def get_embedding(question: str):
    connection = AzureOpenAIConnection(        
                    azure_deployment=os.getenv("AZURE_OPENAI_EMBEDDING_DEPLOYMENT", ""),
                    api_version=os.getenv("AZURE_OPENAI_API_VERSION", ""),
                    api_base=os.getenv("AZURE_OPENAI_ENDPOINT", "")
                    )
                
    client = init_azure_openai_client(connection)

    return client.embeddings.create(
            input=question,
            model=os.getenv("AZURE_OPENAI_EMBEDDING_MODEL", ""),
        ).data[0].embedding