Update agent.py

agent.py

@@ -40,10 +40,15 @@ def use_wikipedia_tool(query: str) -> str:
 
 def build_agent():
     # llm = ChatOllama(model="llama3.1")
-    llm = HuggingFaceEndpoint(
-        endpoint_url="https://api-inference.huggingface.co/models/deepseek-ai/DeepSeek-Prover-V2-671B",
-        huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN")
-    )
+    # llm = HuggingFaceEndpoint(
+    #     endpoint_url="https://api-inference.huggingface.co/models/deepseek-ai/DeepSeek-Prover-V2-671B",
+    #     huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN")
+    # )
+    
+    llm = HuggingFaceHub(
+    repo_id="deepseek-ai/deepseek-coder-6.7b-instruct",
+    model_kwargs={"temperature": 0.2, "max_new_tokens": 256})
+    
     tools = [use_wikipedia_tool, use_search_tool]
 
     system_template = (