全openai或全google

Dockerfile

@@ -8,4 +8,4 @@ RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
 COPY . .
 
-CMD ["gunicorn","-b", "0.0.0.0:7860", "test:app"]
+CMD ["gunicorn","-b", "0.0.0.0:7860", "main:app"]

main.py

@@ -30,6 +30,7 @@ model = genai.GenerativeModel("gemini-2.0-flash")
 # === 初始化OpenAI模型 ===
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 client = OpenAI(api_key=OPENAI_API_KEY)
+text_system_prompt = "你是一個中文的AI助手，請用繁體中文回答"
 
 
 # === 初始設定 ===
@@ -53,8 +54,11 @@ handler = WebhookHandler(channel_secret)
 
 # === AI Query 包裝 ===
 def query(payload):
-    response = model.generate_content(payload)
-    return response.text
+    response = client.responses.create(
+        model="gpt-4.1",
+        input=f"{text_system_prompt}：{payload}",
+    )
+    return response.output_text
 
 
 # === 靜態圖檔路由 ===