Apex123 commited on
Commit
fd060cc
·
verified ·
1 Parent(s): 822fb48

Update server.py

Browse files
Files changed (1) hide show
  1. server.py +16 -13
server.py CHANGED
@@ -1,9 +1,12 @@
1
  from flask import Flask, request, jsonify
2
  from flask_cors import CORS
3
  import time
 
 
4
 
5
  app = Flask(__name__)
6
  CORS(app) # 讓 Open WebUI 可跨域存取
 
7
 
8
  @app.route('/')
9
  def hello():
@@ -11,21 +14,21 @@ def hello():
11
 
12
  @app.route('/v1/chat/completions', methods=['POST'])
13
  def chat():
14
- print("💬 收到請求")
15
-
16
- data = request.get_json()
17
- print("資料內容:", data)
18
 
19
- # 取得使用者輸入
20
  messages = data.get("messages", [])
21
- user_message = ""
22
- for m in messages[::-1]:
23
- if m.get("role") == "user":
24
- user_message = m.get("content", "")
25
- break
26
-
27
- # 模擬回覆
28
- reply = f"你剛剛說的是:「{user_message}」,這是來自 Flask 模型的回覆 ✅"
 
 
 
29
 
30
  # 回傳格式符合 OpenAI Chat API 格式
31
  return jsonify({
 
1
  from flask import Flask, request, jsonify
2
  from flask_cors import CORS
3
  import time
4
+ import openai
5
+
6
 
7
  app = Flask(__name__)
8
  CORS(app) # 讓 Open WebUI 可跨域存取
9
+ openai.api_key = os.getenv("openaikey") # 或從 .env 載入
10
 
11
  @app.route('/')
12
  def hello():
 
14
 
15
  @app.route('/v1/chat/completions', methods=['POST'])
16
  def chat():
17
+
18
+ data = request.get_json(force=True)
 
 
19
 
 
20
  messages = data.get("messages", [])
21
+ model = data.get("model", "gpt-3.5-turbo")
22
+
23
+ try:
24
+ response = openai.ChatCompletion.create(
25
+ model=model,
26
+ messages=messages
27
+ )
28
+ reply = response.choices[0].message.content
29
+ except Exception as e:
30
+ reply = f"⚠️ OpenAI API 錯誤:{str(e)}"
31
+ print(reply)
32
 
33
  # 回傳格式符合 OpenAI Chat API 格式
34
  return jsonify({