vsegpt

app.py

@@ -73,14 +73,27 @@ def load_assets():
     return tool
 
 
+# def generate_gpt_comment(piece_or_text, message, context):
+#     response = openai.ChatCompletion.create(
+#         model="gpt-4o-mini",
+#         messages=[
+#             {"role": "user", "content": prompt.format(piece_or_text, message, context)}
+#         ]
+#     )
+#     return response.choices[0].message['content']
+
+
 def generate_gpt_comment(piece_or_text, message, context):
-    response = openai.ChatCompletion.create(
-        model="gpt-4o-mini",
-        messages=[
-            {"role": "user", "content": prompt.format(piece_or_text, message, context)}
-        ]
+    client = openai.OpenAI(
+        api_key='sk-or-vv-44ca318747a45e810149bf769e9dfbe5f42046695875efd7c9121cca590d6906',
+        base_url='https://api.vsegpt.ru/v1'
     )
-    return response.choices[0].message['content']
+    response = client.chat.completions.create(
+        model="gpt-4o-mini",
+        messages=[{"role": "user", "content": prompt.format(piece_or_text, message, context)}]
+    ).choices[0].message.content
+
+    return response
 
 
 def check_text(text, tool):