Update app.py
Browse files
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-
|
| 33 |
-
'gpt-
|
| 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-
|
| 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-
|
| 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 |
|