Update agent.py

agent.py

@@ -173,13 +173,7 @@ tools = [
 def build_graph(provider: str = "huggingface"):
     """Build the graph"""
     # Load environment variables from .env file
-    if provider == "google":
-        # Google Gemini
-        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
-    elif provider == "huggingface":
+    if provider == "huggingface":
         # TODO: Add huggingface endpoint
         """
         llm = ChatHuggingFace(
@@ -191,16 +185,21 @@ def build_graph(provider: str = "huggingface"):
             ),
         )
         """
-        llm_id=HuggingFaceEndpoint(
-            repo_id="Qwen/Qwen2.5-Coder-32B.Instruct",
-            task="text-generation",
-            temperature = 0,
+        llm = ChatHuggingFace(
+            llm=HuggingFaceEndpoint(
+                repo_id="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
+                task="text-generation",  # for chat‐style use “text-generation”
+                max_new_tokens=1024,
+                do_sample=False,
+                repetition_penalty=1.03,
+                temperature=0,
+            ),
+            verbose=True,
         )
-
-        llm=ChatHuggingFace(llm=llm_id)
         
     else:
-        raise ValueError("Invalid provider. Choose 'google', 'groq' or 'huggingface'.")
+        raise ValueError("Invalid provider. Choose 'huggingface'.")
+        
     # Bind tools to LLM
     llm_with_tools = llm.bind_tools(tools)