update sth. at 2024-11-15 19:31:13

app.py

@@ -36,49 +36,28 @@ def chat_with_model(messages):
         return "聊天生成失败，请稍后再试。"
 
 # ---------- chatgpt-4o-mini 模块 ----------
-def chatgpt_4o_mini(messages):
-    """
-    调用 gpt-4o-mini API 进行对话，并解析流式响应。
-    """
-    try:
-        # 构建请求数据
-        data = {
-            "model": "gpt-4o-mini",
-            "messages": [{"role": "system", "content": "你是一个辅助机器人"}] + messages,
-            "stream": True
-        }
-
-        # 发送请求
-        url = 'https://sanbo1200-duck2api.hf.space/completions'
-        headers = {'Content-Type': 'application/json'}
-        
-        # 使用 requests 发送 POST 请求
-        response = requests.post(url, headers=headers, json=data, stream=True)
-        
-        # 检查响应状态码
-        if response.status_code != 200:
-            return f"请求失败，状态码：{response.status_code}"
-
-        # 处理流式响应
-        full_response = ""
-        for line in response.iter_lines():
-            if line:
-                # 解析每一行的内容
-                try:
-                    chunk = json.loads(line.decode('utf-8'))
-                    # 解析返回的每个 chunk
-                    if 'choices' in chunk and len(chunk['choices']) > 0:
-                        content = chunk['choices'][0]['delta'].get('content', '')
-                        full_response += content
-                except Exception as e:
-                    print(f"Error parsing response chunk: {e}")
-        
-        # 返回拼接后的完整对话内容
-        return full_response.strip()
 
-    except Exception as e:
-        print(f"Error during gpt-4o-mini request: {e}")
-        return "gpt-4o-mini 请求失败，请稍后再试。"
+def chatgpt_4o_mini(Query):
+    url = 'https://sanbo1200-duck2api.hf.space/completions'
+    headers = {'Content-Type': 'application/json'}
+    data = {
+        "model": "gpt-4o-mini",
+        "messages": [
+            {"role": "system", "content": "你是一个辅助机器人"},
+            {"role": "user", "content": Query}
+        ],
+        "stream": False
+    }
+    
+    # 发起 HTTP 请求
+    response = requests.post(url, json=data, headers=headers, stream=True)
+    response.encoding = 'utf-8'
+    if response.status_code!= 200:
+        return "请求失败"
+    else:
+        json_data = response.json()
+        return json_data['choices'][0]['message']['content']
+
 
 # ---------- 图像生成模块 ----------
 def image_gen(prompt):

req.py

@@ -0,0 +1,29 @@
+import requests
+import json
+
+def chatgpt_4o_mini(Query):
+    url = 'https://sanbo1200-duck2api.hf.space/completions'
+    headers = {'Content-Type': 'application/json'}
+    data = {
+        "model": "gpt-4o-mini",
+        "messages": [
+            {"role": "system", "content": "你是一个辅助机器人"},
+            {"role": "user", "content": Query}
+        ],
+        "stream": False
+    }
+    
+    # 发起 HTTP 请求
+    response = requests.post(url, json=data, headers=headers, stream=True)
+    response.encoding = 'utf-8'
+    if response.status_code!= 200:
+        return "请求失败"
+    else:
+        json_data = response.json()
+        return json_data['choices'][0]['message']['content']
+
+# 获取回答
+response = chatgpt_4o_mini('你的知识库最后什么日期')
+
+# 输出回答
+print(response)