Update agent.py

agent.py

@@ -1,5 +1,5 @@
 from typing import Annotated, Sequence, TypedDict
-from langchain_community.llms import HuggingFaceHub
+from langchain_community.llms import HuggingFaceHub,HuggingFaceEndpoint  
 from langchain_core.messages import BaseMessage, HumanMessage, AIMessage
 from langgraph.graph import StateGraph, END
 from langchain_core.agents import AgentAction, AgentFinish
@@ -14,14 +14,19 @@ class AgentState(TypedDict):
 
 def build_graph():
     # 1. 初始化模型 - 使用HuggingFace免费接口
-    llm = HuggingFaceHub(
-        repo_id="mistralai/Mistral-7B-Instruct-v0.2",
-        model_kwargs={"temperature": 0.1, "max_new_tokens": 500}
+    llm = HuggingFaceEndpoint(
+        endpoint_url="https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.2",
+        task="text-generation",
+        model_kwargs={
+            "temperature": 0.1,
+            "max_new_tokens": 500,
+            "repetition_penalty": 1.2
+        }
     )
     
     # 2. 创建ReAct代理
     prompt = hub.pull("hwchase17/react")
-    tools = get_tools()  # 从ai_tools获取工具
+    tools = get_tools()
     agent = create_react_agent(llm, tools, prompt)
     
     # 3. 定义节点行为