总结加入，tokens显示

base/func_doubao.py

@@ -89,13 +89,35 @@ class Doubao():
         return False
 
 
-# 解析JSON
 def extract_content(data_string):
     try:
         data = json.loads(data_string)
+
         # 提取content字段
         content = data["choices"][0]["message"].get("content", "")
-        return content
+
+        # 提取tokens相关内容，加入容错处理
+        tokens_usage = data.get("usage", {})
+
+        # 确保tokens_usage是字典类型
+        if isinstance(tokens_usage, dict):
+            prompt_tokens = tokens_usage.get("prompt_tokens", 0)
+            completion_tokens = tokens_usage.get("completion_tokens", 0)
+            total_tokens = tokens_usage.get("total_tokens", 0)
+        else:
+            prompt_tokens = completion_tokens = total_tokens = 0
+
+        # 如果tokens信息为空，提供默认值或提示
+        if prompt_tokens == 0 and completion_tokens == 0 and total_tokens == 0:
+            tokens_info = "\n\n【tokens】暂无数据"
+        else:
+            tokens_info = (f"\n\n【tokens】输入: {prompt_tokens} 生成: {completion_tokens} 总: {total_tokens}")
+
+        # 将tokens信息添加到content后面，返回为字符串
+        content_with_tokens = content + tokens_info
+
+        return content_with_tokens
+
     except json.JSONDecodeError:
         print("Invalid JSON")
         return None