add tool choices for agent. (#1126)

2025-01-13 14:42:31 +08:00
parent fe24decd72
commit 3a7ccb0a75
5 changed files with 81 additions and 34 deletions
--- a/tests/agent/test.py
+++ b/tests/agent/test.py
@@ -10,7 +10,7 @@ import requests
 def generate_answer_agent_api(url, prompt):
    proxies = {"http": ""}
    payload = {
-        "query": prompt,
+        "messages": prompt,
    }
    response = requests.post(url, json=payload, proxies=proxies)
    answer = response.json()["text"]
@@ -21,7 +21,7 @@ def process_request(url, query, is_stream=False):
    proxies = {"http": ""}

    payload = {
-        "query": query,
+        "messages": query,
    }

    try: