Skip to content

feat: add AzureOpenAIEmbeddingFunction#83

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

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

Conversation

@kaijchen

@kaijchen kaijchen commented Nov 19, 2025

Copy link
Copy Markdown

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
Copy Markdown
Author

@zc277584121 @codingjaguar PTAL, thanks.

@xiaofan-luan

Copy link
Copy Markdown

/lgtm
/approve

@xiaofan-luan

Copy link
Copy Markdown

@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