临时优化一下。

This commit is contained in:
liuwei
2025-03-13 17:59:43 +08:00
parent a559dcaaf0
commit 28e99a31d3
2 changed files with 10 additions and 9 deletions

View File

@@ -144,7 +144,8 @@ class DifyChat:
"query": query,
"conversation_id": session_id
}
# 准备请求数据
# 准备请求数据
data = {
"sys.files": [],
"user": user_id,
@@ -166,14 +167,14 @@ class DifyChat:
# 发送请求
url = f"{self.base_url}/workflows/run"
self.LOG.info(f"发送请求到Dify API: {url}")
self.LOG.info(f"请求数据: {json.dumps(data, ensure_ascii=False)}")
try:
# 使用普通请求(非流式)
response = requests.post(url, headers=headers, json=data, proxies=proxies)
if response.status_code != 200:
self.LOG.error(f"Dify API请求失败: {response.status_code} {response.text}")
return f"请求失败,状态码: {response.status_code}"
@@ -181,11 +182,11 @@ class DifyChat:
# 解析响应
response_data = response.json()
self.LOG.info(f"收到Dify API响应: {json.dumps(response_data, ensure_ascii=False)}")
# 提取回答内容
answer = ""
total_tokens = 0
# 获取输出内容
outputs = response_data.get("data", {}).get("outputs", {})
if outputs:
@@ -212,7 +213,7 @@ class DifyChat:
for item_key, item_value in item.items():
if isinstance(item_value, str) and item_value.strip():
answer += item_value
# 获取token使用情况
total_tokens = response_data.get("data", {}).get("total_tokens", 0)