Apex123 commited on
Commit
9510bbe
·
verified ·
1 Parent(s): 85cd929

Update server.py

Browse files
Files changed (1) hide show
  1. server.py +31 -1
server.py CHANGED
@@ -18,4 +18,34 @@ def chat():
18
  user_message = ""
19
  for m in messages[::-1]:
20
  if m.get("role") == "user":
21
- user_message = m.get("content"_
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
18
  user_message = ""
19
  for m in messages[::-1]:
20
  if m.get("role") == "user":
21
+ user_message = m.get("content", "")
22
+ break
23
+
24
+ # 模擬回覆
25
+ reply = f"你剛剛說的是:「{user_message}」,這是來自 Flask 模型的回覆 ✅"
26
+
27
+ # 回傳格式符合 OpenAI Chat API 格式
28
+ return jsonify({
29
+ "id": "chatcmpl-xyz",
30
+ "object": "chat.completion",
31
+ "created": int(time.time()),
32
+ "model": "local-flask-test-001",
33
+ "choices": [
34
+ {
35
+ "index": 0,
36
+ "message": {
37
+ "role": "assistant",
38
+ "content": reply
39
+ },
40
+ "finish_reason": "stop"
41
+ }
42
+ ],
43
+ "usage": {
44
+ "prompt_tokens": 10,
45
+ "completion_tokens": 10,
46
+ "total_tokens": 20
47
+ }
48
+ })
49
+
50
+ if __name__ == '__main__':
51
+ app.run(host='0.0.0.0', port=7860)