Update groq_llm.py

groq_llm.py

@@ -1,10 +1,16 @@
-from groq import Groq
+import os
+from langchain_groq import ChatGroq
 
-def get_llm(api_key, streaming=True):
-    client = Groq(api_key=api_key)
-
-    return client.chat.completions.create(
-        model="llama-3.1-8b-instant",   # 👈 THIS LINE
-        messages=[...],
-        stream=streaming
+def get_llm(callbacks=None):
+    """
+    Returns a Groq LLM instance with streaming enabled.
+    callbacks: list of LangChain callback handlers (optional)
+    """
+    return ChatGroq(
+        groq_api_key=os.getenv("GROQ_API_KEY"),
+        model_name="llama-3.1-8b-instant",  # ✅ your chosen model
+        temperature=0.1,
+        max_tokens=1024,
+        streaming=True,
+        callbacks=callbacks
     )