Improvement/add getting api key from env (#1710)

2024-08-19 22:35:28 +05:30
parent 8476ce5d8e
commit 0e0d0b8fc7
3 changed files with 12 additions and 0 deletions
--- a/mem0/llms/azure_openai.py
+++ b/mem0/llms/azure_openai.py
@@ -1,3 +1,4 @@
+import os
 import json
 from typing import Dict, List, Optional

@@ -15,6 +16,9 @@ class AzureOpenAILLM(LLMBase):
        if not self.config.model:
            self.config.model = "gpt-4o"
        self.client = AzureOpenAI()
+        
+        api_key = os.getenv("AZURE_OPENAI_API_KEY") or self.config.api_key
+        self.client = AzureOpenAI(api_key=api_key)

    def _parse_response(self, response, tools):
        """