klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

AzureOpenai access from behind company proxies. (#1459)

Browse Source
This commit is contained in:
Pranav Puranik
2024-08-01 13:53:38 -05:00
committed by GitHub
parent 563a130141
commit c197a5fe93
8 changed files with 173 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
embedchain/embedchain/embedder/azure_openai.py
View File

@@ -1,6 +1,6 @@
from typing import Optional
from langchain_community.embeddings import AzureOpenAIEmbeddings
from langchain_openai import AzureOpenAIEmbeddings
from embedchain.config import BaseEmbedderConfig
from embedchain.embedder.base import BaseEmbedder
@@ -14,7 +14,11 @@ class AzureOpenAIEmbedder(BaseEmbedder):
if self.config.model is None:
self.config.model = "text-embedding-ada-002"
embeddings = AzureOpenAIEmbeddings(deployment=self.config.deployment_name)
embeddings = AzureOpenAIEmbeddings(
deployment=self.config.deployment_name,
http_client=self.config.http_client,
http_async_client=self.config.http_async_client,
)
embedding_fn = BaseEmbedder._langchain_default_concept(embeddings)
self.set_embedding_fn(embedding_fn=embedding_fn)