vsegptv3

app.py

@@ -1,5 +1,5 @@
 import language_tool_python
-from openai import OpenAI, ChatCompletion
+import openai
 import streamlit as st
 import subprocess
 from pathlib import Path
@@ -66,6 +66,7 @@ def install_java():
 
 @st.cache_resource
 def load_assets():
+    openai.api_key = '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'
     install_java()
     tool = language_tool_python.LanguageTool('ru-RU',
                                          language_tool_download_version="6.1")
@@ -73,8 +74,7 @@ def load_assets():
 
 
 def generate_gpt_comment_openai(inp):
-    response = ChatCompletion.create(
-        api_key='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',
+    response = openai.ChatCompletion.create(
         model="gpt-4o-mini",
         messages=[
             {"role": "user", "content": inp}
@@ -84,9 +84,9 @@ def generate_gpt_comment_openai(inp):
 
 
 def generate_gpt_comment_vsegpt(inp):
-    client = OpenAI(
+    client = openai.OpenAI(
         api_key='sk-or-vv-44ca318747a45e810149bf769e9dfbe5f42046695875efd7c9121cca590d6906',
-        base_url='https://api.vsegpt.ru/v1'
+        base_url='https://api.vsegpt.ru/v1',
     )
     response = client.chat.completions.create(
         model="gpt-4o-mini",