diff --git a/core/media/__pycache__/truth_social_retriever.cpython-312.pyc b/core/media/__pycache__/truth_social_retriever.cpython-312.pyc index d25757a..a48cd42 100644 Binary files a/core/media/__pycache__/truth_social_retriever.cpython-312.pyc and b/core/media/__pycache__/truth_social_retriever.cpython-312.pyc differ diff --git a/core/media/truth_social_retriever.py b/core/media/truth_social_retriever.py index cb8affa..f4e7b67 100644 --- a/core/media/truth_social_retriever.py +++ b/core/media/truth_social_retriever.py @@ -267,14 +267,19 @@ class TruthSocialRetriever: search_options={"forced_search": True}, # 强制联网搜索 result_format="message", ) - response_contents = ( - response.get("output", {}) - .get("choices", [])[0] - .get("message", {}) - .get("content", "") - ) + # 获取response的token - token = response.get("usage", {}).get("total_tokens", 0) + if response.status_code == 200: + response_contents = ( + response.get("output", {}) + .get("choices", [])[0] + .get("message", {}) + .get("content", "") + ) + 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}")