临时优化一下。

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)

View File

@@ -73,10 +73,10 @@ class MessageStorage:
if time_diff < timedelta(hours=3):
# 小于 3 小时,取 8 小时前
last_summary_time = (current_time - timedelta(hours=8)).strftime('%Y-%m-%d %H:%M:%S')
last_summary_time = (current_time - timedelta(hours=3)).strftime('%Y-%m-%d %H:%M:%S')
elif time_diff > timedelta(days=1):
# 大于 24 小时,取 10 小时前
last_summary_time = (current_time - timedelta(hours=10)).strftime('%Y-%m-%d %H:%M:%S')
last_summary_time = (current_time - timedelta(hours=6)).strftime('%Y-%m-%d %H:%M:%S')
# 更新 Redis 存储的当前时间
self._get_redis_connection().set(key, current_date)