Update app.py

app.py

@@ -1,23 +1,33 @@
 import os
 from flask import Flask, request, make_response
-import xml.etree.ElementTree as ET
 from wechatpy import parse_message, create_reply
 from wechatpy.utils import check_signature
 from wechatpy.exceptions import InvalidSignatureException
 from openai import OpenAI
+from dotenv import load_dotenv
+
+# 加载环境变量
+load_dotenv()
 
 app = Flask(__name__)
 
 # 配置
-TOKEN = os.getenv("TOKEN")
+TOKEN = os.getenv('TOKEN')
 API_KEY = os.getenv("OPENAI_API_KEY")
 BASE_URL = os.getenv("OPENAI_BASE_URL")
 client = OpenAI(api_key=API_KEY, base_url=BASE_URL)
 
+# 定义可用的模型列表
+AVAILABLE_MODELS = {
+    'gpt-3.5-turbo': 'GPT-3.5 Turbo',
+    'gpt-4': 'GPT-4',
+    'gpt-4-turbo': 'GPT-4 Turbo',
+}
+
 # 存储用户会话信息
 user_sessions = {}
 
-def get_openai_response(messages, model="gpt-4o-mini"):
+def get_openai_response(messages, model="gpt-3.5-turbo"):
     try:
         response = client.chat.completions.create(
             model=model,
@@ -31,6 +41,9 @@ def get_openai_response(messages, model="gpt-4o-mini"):
 def split_message(message, max_length=500):
     return [message[i:i+max_length] for i in range(0, len(message), max_length)]
 
+def list_available_models():
+    return "\n".join([f"{key}: {value}" for key, value in AVAILABLE_MODELS.items()])
+
 @app.route('/', methods=['GET', 'POST'])
 def wechat():
     if request.method == 'GET':
@@ -56,14 +69,19 @@ def wechat():
             user_id = msg.source
             content = msg.content
 
-            if content.startswith('/model'):
-                # 切换模型
+            if content.lower() == '/models':
+                return create_reply(f"可用的模型列表：\n{list_available_models()}\n\n使用 /model 模型名称 来切换模型", msg).render()
+
+            if content.lower().startswith('/model'):
                 model = content.split(' ')[1]
-                user_sessions[user_id] = {'model': model, 'messages': []}
-                return create_reply(f'模型已切换为 {model}', msg).render()
+                if model in AVAILABLE_MODELS:
+                    user_sessions[user_id] = {'model': model, 'messages': []}
+                    return create_reply(f'模型已切换为 {AVAILABLE_MODELS[model]}', msg).render()
+                else:
+                    return create_reply(f'无效的模型名称。可用的模型有：\n{list_available_models()}', msg).render()
 
             if user_id not in user_sessions:
-                user_sessions[user_id] = {'model': 'gpt-4o-mini', 'messages': []}
+                user_sessions[user_id] = {'model': 'gpt-3.5-turbo', 'messages': []}
 
             session = user_sessions[user_id]
             session['messages'].append({"role": "user", "content": content})
@@ -71,7 +89,6 @@ def wechat():
             response = get_openai_response(session['messages'], session['model'])
             session['messages'].append({"role": "assistant", "content": response})
 
-            # 分割长消息
             response_parts = split_message(response)
             for part in response_parts[:-1]:
                 create_reply(part, msg).render()