Update tools.py

tools.py

@@ -15,6 +15,25 @@ from bs4 import BeautifulSoup
 from io import BytesIO
 from PyPDF2 import PdfReader
 
+
+class GroqKeyManager:
+    def __init__(self):
+        self.api_keys = [
+            os.getenv("GROQ_API_KEY"),
+            os.getenv("GROQ_API_KEY_1"),
+            os.getenv("GROQ_API_KEY_2"),
+        ]
+        self.key_index = 0
+        
+    def get_next_key(self):
+        api_key = self.api_keys[self.key_index]
+        self.key_index = (self.key_index + 1) % len(self.api_keys)
+        return api_key
+
+# Create a global instance of the key manager
+groq_key_manager = GroqKeyManager()
+
+
 @tool
 def add(a: float, b: float) -> float:
     """ Adds two numbers.
@@ -109,13 +128,20 @@ def transcribe_audio(audio_file: str, file_extension: str) -> str:
         with open(filename, 'wb') as file:  # opens a new file for writing with a name like, e.g. tmp.mp3
             file.write(response.content)    # write(w) the binary(b) contents (audio file) to disk
         
-        # transcribe audio with OpenAI Whisper
-        client = OpenAI()
+        # Get the next Groq API key
+        api_key = groq_key_manager.get_next_key()
+        
+        # Initialize the client with Groq's base URL and API key
+        client = OpenAI(
+            base_url="https://api.groq.com/openai/v1",
+            api_key=api_key
+        )
         
-        # read(r) the audio file from disk in binary(b) mode "rb"; the "with" block ensures the file is automatically closed afterward
+        # Note: Check if Groq supports audio transcription
+        # If not, you might need to keep using OpenAI's Whisper for this specific function
         with open(filename, "rb") as audio_content:
             transcription = client.audio.transcriptions.create(
-                model="whisper-1",
+                model="whisper-large-v3-turbo",  # Check if Groq has an equivalent model
                 file=audio_content
             )
         return transcription.text
@@ -158,15 +184,19 @@ def query_image(query: str, image_url: str) -> str:
         image_url (str): the image's URL
     """
     try:
-        client = OpenAI()
+        api_key = groq_key_manager.get_next_key()
+        client = OpenAI(
+            base_url="https://api.groq.com/openai/v1",
+            api_key=api_key
+        )
         response = client.responses.create(
-            model="gpt-4o-mini",
+            model="meta-llama/llama-4-scout-17b-16e-instruct",
             input=[
                 {
                     "role": "user",
                     "content": [
                         {"type": "input_text", "text": query},
-                        {"type": "input_image","image_url": image_url},
+                        {"type": "input_image", "image_url": image_url},
                     ],
                 }
             ],
@@ -222,4 +252,8 @@ def read_excel(file_url: str) -> str:
         return df.to_csv(index=False)  # convert dataframe to CSV string for easy processing
 
     except Exception as e:
-        return f"read_excel failed: {str(e)}"
+        return f"read_excel failed: {str(e)}"
+
+
+
+