Upload agent.py

agent.py

@@ -153,7 +153,7 @@ tools = [
 ]
 
 # Build graph function
-def build_graph(provider: str = "huggingface"):
+def build_graph(provider: str = "groq"):
     """Build the graph"""
     # Load environment variables from .env file
     if provider == "google":
@@ -161,7 +161,7 @@ def build_graph(provider: str = "huggingface"):
         llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
     elif provider == "groq":
         # Groq https://console.groq.com/docs/models
-        llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
+        llm = ChatGroq(model="qwen-qwq-32b", temperature=0, api_key=os.getenv("GROQ_API_KEY")) # optional : qwen-qwq-32b gemma2-9b-it
     elif provider == "huggingface":
         # TODO: Add huggingface endpoint
         llm = ChatHuggingFace(