Adding proxy server settings to azure openai (#1753)

2024-08-29 04:48:50 -05:00
parent deeb4f2250
commit fee3c27af3
7 changed files with 46 additions and 6 deletions
--- a/mem0/embeddings/azure_openai.py
+++ b/mem0/embeddings/azure_openai.py
@@ -17,7 +17,7 @@ class AzureOpenAIEmbedding(EmbeddingBase):
            self.config.embedding_dims = 1536

        api_key = os.getenv("AZURE_OPENAI_API_KEY") or self.config.api_key
-        self.client = AzureOpenAI(api_key=api_key)
+        self.client = AzureOpenAI(api_key=api_key, http_client=self.config.http_client)

    def embed(self, text):
        """