Update agent.py

agent.py

@@ -16,12 +16,11 @@ def build_graph():
     # 1. 初始化模型 - 使用HuggingFace免费接口
     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
-        }
+        max_new_tokens=500,          # 直接指定
+        temperature=0.1,             # 直接指定
+        repetition_penalty=1.2,      # 直接指定
+        top_p=0.9,                   # 可选参数
+        # 其他参数也可以直接在这里指定
     )
     
     # 2. 创建ReAct代理