Specifying endpoints/deployments for Azure OpenAI embeddings

Reformatting for linting
This commit is contained in:
Samuel Pierre
2024-04-11 16:53:39 +02:00
parent 892af90489
commit 34941aae39
2 changed files with 8 additions and 1 deletions
+1
View File
@@ -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
+7 -1
View File
@@ -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(