make code stronger

This commit is contained in:
blade 2025-10-22 18:05:38 +08:00
parent d66110a849
commit dee344dbea
2 changed files with 12 additions and 7 deletions

View File

@ -267,14 +267,19 @@ class TruthSocialRetriever:
search_options={"forced_search": True}, # 强制联网搜索
result_format="message",
)
# 获取response的token
if response.status_code == 200:
response_contents = (
response.get("output", {})
.get("choices", [])[0]
.get("message", {})
.get("content", "")
)
# 获取response的token
token = response.get("usage", {}).get("total_tokens", 0)
else:
response_contents = f"{response.code} {response.message}"
token = 0
return response_contents, token
except Exception as e:
logger.error(f"分析推文失败: {e}")