Skip to content

feat: add AzureOpenAIEmbeddingFunction#83

Open
kaijchen wants to merge 1 commit intomilvus-io:mainfrom
kaijchen:azure
Open

feat: add AzureOpenAIEmbeddingFunction#83
kaijchen wants to merge 1 commit intomilvus-io:mainfrom
kaijchen:azure

Conversation

@kaijchen
Copy link

@kaijchen kaijchen commented Nov 19, 2025

Description

This PR adds support for Azure OpenAI Service by introducing a new AzureOpenAIEmbeddingFunction class.

Changes

  • Added AzureOpenAIEmbeddingFunction to pymilvus/model/dense/azure_openai.py.
  • Implemented the wrapper using openai.AzureOpenAI from the OpenAI SDK.
  • Exposed api_version and azure_endpoint in the __init__ method.

Verification

I have manually tested this implementation against a live Azure OpenAI instance:

  • Verified successful client initialization with api_key, azure_endpoint, and api_version.
  • Verified successful embedding generation.

Related Issue

Fixes #82

Signed-off-by: Kaijie Chen <chenkaijie.jay@bytedance.com>
@kaijchen
Copy link
Author

@zc277584121 @codingjaguar PTAL, thanks.

@xiaofan-luan
Copy link

/lgtm
/approve

@xiaofan-luan
Copy link

@codingjaguar

shall we merge this and update the document?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Add support for Azure OpenAI Embedding Function

2 participants