mistpe commited on
Commit
338d1f0
·
verified ·
1 Parent(s): 4319607

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -29,8 +29,8 @@ client = OpenAI(api_key=API_KEY, base_url=BASE_URL)
29
  # 定义可用的模型列表
30
  AVAILABLE_MODELS = {
31
  'gpt-3.5-turbo': 'GPT-3.5 Turbo',
32
- 'gpt-4': 'GPT-4',
33
- 'gpt-4-turbo': 'GPT-4 Turbo',
34
  }
35
 
36
  # 存储用户会话信息
@@ -174,7 +174,7 @@ def generate_response_xml(from_user_name, to_user_name, output_content):
174
  response.content_type = 'application/xml'
175
  return response
176
 
177
- def get_openai_response(messages, model="gpt-3.5-turbo", functions=None, function_call=None):
178
  try:
179
  response = client.chat.completions.create(
180
  model=model,
@@ -232,7 +232,7 @@ def wechatai():
232
  user_message_content, from_user_name, to_user_name = getUserMessageContentFromXML(xml_str)
233
 
234
  if from_user_name not in user_sessions:
235
- user_sessions[from_user_name] = {'model': 'gpt-3.5-turbo', 'messages': [], 'pending_response': []}
236
 
237
  session = user_sessions[from_user_name]
238
 
 
29
  # 定义可用的模型列表
30
  AVAILABLE_MODELS = {
31
  'gpt-3.5-turbo': 'GPT-3.5 Turbo',
32
+ 'gpt-4o': 'GPT-4o',
33
+ 'gpt-4o-mini': 'GPT-4o-mini',
34
  }
35
 
36
  # 存储用户会话信息
 
174
  response.content_type = 'application/xml'
175
  return response
176
 
177
+ def get_openai_response(messages, model="gpt-4o-mini", functions=None, function_call=None):
178
  try:
179
  response = client.chat.completions.create(
180
  model=model,
 
232
  user_message_content, from_user_name, to_user_name = getUserMessageContentFromXML(xml_str)
233
 
234
  if from_user_name not in user_sessions:
235
+ user_sessions[from_user_name] = {'model': 'gpt-4o-mini', 'messages': [], 'pending_response': []}
236
 
237
  session = user_sessions[from_user_name]
238