Update agent.py

agent.py

@@ -46,8 +46,12 @@ def build_agent():
     #     huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN")
     # )
     
-    llm = HuggingFaceHub(repo_id="distilbert/distilgpt2",
-    task="text-generation")
+    llm = HuggingFaceHub(repo_id="distilbert/distilgpt2", task="text-generation",
+                    model_kwargs={
+        "temperature": 0.7,         # Controls randomness (0.0 = deterministic)
+        "max_new_tokens": 100       # Optional: max output length
+    }, 
+    verbose=True )
     
     tools = [use_wikipedia_tool, use_search_tool]