Update agent.py

agent.py

@@ -148,22 +148,9 @@ tools = [
 ]
 
 # Build graph function
-def build_graph(provider: str = "groq"):
+def build_graph():
     """Build the graph"""
-    # Load environment variables from .env file
-    if provider == "groq":
-        # Groq https://console.groq.com/docs/models
-        llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
-    elif provider == "huggingface":
-        # TODO: Add huggingface endpoint
-        llm = ChatHuggingFace(
-            llm=HuggingFaceEndpoint(
-                url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
-                temperature=0,
-            ),
-        )
-    else:
-        raise ValueError("Invalid provider. Choose, 'groq' or 'huggingface'.")
+    llm = ChatGroq(model="qwen-qwq-32b", temperature=0)
     # Bind tools to LLM
     llm_with_tools = llm.bind_tools(tools)