Merge pull request #292 from samuelp-mw/feature/improve-usage-for-azure

Specifying endpoints/deployments for Azure OpenAI embeddings
langchain-ai · Apr 12, 2024 · 028ed27 · 028ed27
2 parents 8cc32c5 + 34941aa
commit 028ed27
Show file tree

Hide file tree

Showing 2 changed files with 8 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,13 @@ 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(