Public/kotaemon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use new Langchain's dedicated Azure OpenAI embedding class (#76)

Browse Source 
* Use new Langchain's dedicated Azure OpenAI embedding class

* Update test
This commit is contained in:
Nguyen Trung Duc (john)
2023-11-15 14:46:32 +07:00
committed by GitHub
parent b159897ac6
commit b52f312d8e
2 changed files with 6 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_embedding_models.py
View File

@@ -21,7 +21,7 @@ def test_azureopenai_embeddings_raw(openai_embedding_call):
model = AzureOpenAIEmbeddings(
model="text-embedding-ada-002",
deployment="embedding-deployment",
openai_api_base="https://test.openai.azure.com/",
azure_endpoint="https://test.openai.azure.com/",
openai_api_key="some-key",
)
output = model("Hello world")
@@ -39,7 +39,7 @@ def test_azureopenai_embeddings_batch_raw(openai_embedding_call):
model = AzureOpenAIEmbeddings(
model="text-embedding-ada-002",
deployment="embedding-deployment",
openai_api_base="https://test.openai.azure.com/",
azure_endpoint="https://test.openai.azure.com/",
openai_api_key="some-key",
)
output = model(["Hello world", "Goodbye world"])