Create llm_clients.py

llm_clients.py

@@ -0,0 +1,21 @@
+import os
+from groq import Groq
+import google.generativeai as genai
+
+# Load Keys
+groq_client = Groq(api_key=os.getenv("GROQ_API_KEY"))
+
+genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
+gemini_model = genai.GenerativeModel("gemini-1.5-flash")
+
+def call_llama(prompt):
+    response = groq_client.chat.completions.create(
+        model="llama-3.1-8b-instant",
+        messages=[{"role": "user", "content": prompt}]
+    )
+    return response.choices[0].message.content
+
+def call_gemini(prompt):
+    response = gemini_model.generate_content(prompt)
+    return response.text
+