Specifying endpoints/deployments for Azure OpenAI embeddings

langchain-ai · Apr 11, 2024 · 70d7cc6 · 70d7cc6
1 parent 892af90
commit 70d7cc6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/.env.example b/.env.example
@@ -6,6 +6,7 @@ AZURE_OPENAI_DEPLOYMENT_NAME=placeholder
 AZURE_OPENAI_API_KEY=placeholder
 AZURE_OPENAI_API_BASE=placeholder
 AZURE_OPENAI_API_VERSION=placeholder
+AZURE_OPENAI_EMBEDDINGS_DEPLOYMENT_NAME=placeholder
 CONNERY_RUNNER_URL=https://your-personal-connery-runner-url
 CONNERY_RUNNER_API_KEY=placeholder
 PROXY_URL=your_proxy_url

diff --git a/backend/app/upload.py b/backend/app/upload.py
@@ -63,7 +63,11 @@ def _determine_azure_or_openai_embeddings() -> PGVector:
     if os.environ.get("AZURE_OPENAI_API_KEY"):
         return PGVector(
             connection_string=PG_CONNECTION_STRING,
-            embedding_function=AzureOpenAIEmbeddings(),
+            embedding_function=AzureOpenAIEmbeddings(
+                azure_endpoint=os.environ.get("AZURE_OPENAI_API_BASE"),
+                azure_deployment=os.environ.get("AZURE_OPENAI_EMBEDDINGS_DEPLOYMENT_NAME"),
+                openai_api_version=os.environ.get("AZURE_OPENAI_API_VERSION")
+            ),
             use_jsonb=True,
         )
     raise ValueError(