From 0e0d0b8fc7c8da190c73eb7030e9c623c1b02a6b Mon Sep 17 00:00:00 2001
From: dbcontributions <navyaalapati13@gmail.com>
Date: Mon, 19 Aug 2024 22:35:28 +0530
Subject: [PATCH] Improvement/add getting api key from env (#1710)

---
 mem0/llms/azure_openai.py | 4 ++++
 mem0/llms/groq.py         | 4 ++++
 mem0/llms/together.py     | 4 ++++
 3 files changed, 12 insertions(+)

diff --git a/mem0/llms/azure_openai.py b/mem0/llms/azure_openai.py
index 0998a03a..3de9cedc 100644
--- 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):
         """
diff --git a/mem0/llms/groq.py b/mem0/llms/groq.py
index d66d92a9..e5b9015e 100644
--- a/mem0/llms/groq.py
+++ b/mem0/llms/groq.py
@@ -1,3 +1,4 @@
+import os
 import json
 from typing import Dict, List, Optional
 
@@ -20,6 +21,9 @@ class GroqLLM(LLMBase):
             self.config.model = "llama3-70b-8192"
         self.client = Groq()
 
+        api_key = os.getenv("GROQ_API_KEY") or self.config.api_key
+        self.client = Groq(api_key=api_key)
+
     def _parse_response(self, response, tools):
         """
         Process the response based on whether tools are used or not.
diff --git a/mem0/llms/together.py b/mem0/llms/together.py
index 4dbc9ae6..1d827d3e 100644
--- a/mem0/llms/together.py
+++ b/mem0/llms/together.py
@@ -1,3 +1,4 @@
+import os
 import json
 from typing import Dict, List, Optional
 
@@ -20,6 +21,9 @@ class TogetherLLM(LLMBase):
             self.config.model = "mistralai/Mixtral-8x7B-Instruct-v0.1"
         self.client = Together()
 
+        api_key = os.getenv("TOGETHER_API_KEY") or self.config.api_key
+        self.client = Together(api_key=api_key)
+    
     def _parse_response(self, response, tools):
         """
         Process the response based on whether tools are used or not.