fruitpicker01 commited on
Commit
5dd2fb4
·
verified ·
1 Parent(s): f9ecb94

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -0
app.py CHANGED
@@ -1113,6 +1113,22 @@ def generate_message_gpt4o(prompt):
1113
  except Exception as e:
1114
  return f"Ошибка при обращении к ChatGPT-4o-Latest: {e}"
1115
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1116
  def clean_message(message):
1117
  # Если сообщение не заканчивается на точку, восклицательный знак или вопросительный знак, обрезаем его до последнего такого знака
1118
  if not message.endswith(('.', '!', '?')):
 
1113
  except Exception as e:
1114
  return f"Ошибка при обращении к ChatGPT-4o-Latest: {e}"
1115
 
1116
+ def generate_message_gpt4o_mini(prompt):
1117
+ try:
1118
+ headers = {
1119
+ "Content-Type": "application/json",
1120
+ "Authorization": f"Bearer {openai_api_key}"
1121
+ }
1122
+ data = {
1123
+ "model": "gpt-4o-mini",
1124
+ "messages": [{"role": "system", "content": prompt}]
1125
+ }
1126
+ response = requests.post("https://api.openai.com/v1/chat/completions", json=data, headers=headers)
1127
+ response_data = response.json()
1128
+ return response_data["choices"][0]["message"]["content"].strip()
1129
+ except Exception as e:
1130
+ return f"Ошибка при обращении к GPT-4o-Mini: {e}"
1131
+
1132
  def clean_message(message):
1133
  # Если сообщение не заканчивается на точку, восклицательный знак или вопросительный знак, обрезаем его до последнего такого знака
1134
  if not message.endswith(('.', '!', '?')):