feature:全球政治经济新闻

utils/ai/dify_news_analyze.py

@@ -0,0 +1,73 @@
+#
+# curl -X POST 'http://192.168.2.240/v1/chat-messages' \
+# --header 'Authorization: Bearer {api_key}' \
+# --header 'Content-Type: application/json' \
+# --data-raw '{
+#     "inputs": {},
+#     "query": "What are the specs of the iPhone 13 Pro Max?",
+#     "response_mode": "streaming",
+#     "conversation_id": "",
+#     "user": "abc-123",
+#     "files": [
+#       {
+#         "type": "image",
+#         "transfer_method": "remote_url",
+#         "url": "https://cloud.dify.ai/logo/logo-site.png"
+#       }
+#     ]
+# }'
+import json
+
+import requests
+
+
+def dify_news_title_analyze(content):
+    # 设置Authorization和URL
+    authorization = "Bearer app-rhhKkbvHd2IAQoGX7xTzXZJj"  # 请替换为真实的Authorization token
+    url = 'http://192.168.2.240/v1/chat-messages'
+
+    data = {
+        "response_mode": "blocking",
+        "conversation_id": "",
+        "inputs": {},
+        "query": content,
+        "user": "a-bot"
+    }
+
+    # 设置请求头
+    headers = {
+        "Content-Type": "application/json; charset=utf-8",
+        "Authorization": authorization
+    }
+
+    # 发送POST请求
+    response = requests.post(url, headers=headers, data=json.dumps(data), )
+    response.encoding = 'utf-8'
+
+    # 输出响应内容
+    print(response.status_code)
+    print(response.json())
+    return extract_content(response.json())
+
+
+def extract_content(data):
+    """解析API响应内容
+    Args:
+        data: API返回的响应数据，可以是字典或字符串
+    Returns:
+        str: 提取的answer内容
+    """
+    try:
+        # 如果是字符串，尝试解析为字典
+        if isinstance(data, str):
+            data = json.dumps(data)
+        # 如果是字典，直接获取answer
+        if isinstance(data, dict):
+            answer = data.get('answer', '')
+            if answer:
+                return answer
+                
+        return None
+    except Exception as e:
+        print(f"解析响应失败: {str(e)}")
+        return None