several_clients

app.py

@@ -73,37 +73,48 @@ 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_openai(inp):
+    response = openai.ChatCompletion.create(
+        model="gpt-4o-mini",
+        messages=[
+            {"role": "user", "content": inp}
+        ]
+    )
+    return response.choices[0].message['content']
 
 
-def generate_gpt_comment(piece_or_text, message, context):
+def generate_gpt_comment_vsegpt(inp):
     client = openai.OpenAI(
         api_key='sk-or-vv-44ca318747a45e810149bf769e9dfbe5f42046695875efd7c9121cca590d6906',
         base_url='https://api.vsegpt.ru/v1'
     )
     response = client.chat.completions.create(
         model="gpt-4o-mini",
-        messages=[{"role": "user", "content": prompt.format(piece_or_text, message, context)}]
+        messages=[{"role": "user", "content": inp}]
     ).choices[0].message.content
 
     return response
 
 
+def generate_gpt_comment(inp, client_name):
+    if client_name == "openai":
+        return generate_gpt_comment_openai(inp)
+    elif client_name == "vsegpt":
+        return generate_gpt_comment_vsegpt(inp)
+    else:
+        raise ValueError(f"Unsupported client: {client_name}")
+    
+
 def check_text(text, tool):
     matches = tool.check(text)
     errors = []
     for match in matches:
+        inp = prompt.format(text[match.offset:match.offset + match.errorLength], 
+                            match.message, match.context)
         error_info = {
             'start': match.offset,
             'end': match.offset + match.errorLength,
-            'message': generate_gpt_comment(text[match.offset:match.offset + match.errorLength], match.message, match.context),
+            'message': generate_gpt_comment(inp, client_name="openai"),
         }
         errors.append(error_info)