Merge pull request #299 from ihmily/main

Fix:修复OpenAI未开代理报错bug以及设置gpt模型
wzpan · Jan 2, 2024 · 8607167 · 8607167
2 parents abe2681 + 44c85e6
commit 8607167
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/robot/AI.py b/robot/AI.py
@@ -207,6 +207,8 @@ def __init__(
             if proxy:
                 logger.info(f"{self.SLUG} 使用代理：{proxy}")
                 self.openai.proxy = proxy
+            else:
+                self.openai.proxy = None
 
         except Exception:
             logger.critical("OpenAI 初始化失败，请升级 Python 版本至 > 3.6")
@@ -243,8 +245,8 @@ def stream_chat(self, texts):
             "Authorization": "Bearer " + self.openai.api_key,
         }
 
-        data = {"model": "gpt-3.5-turbo", "messages": self.context, "stream": True}
-        logger.info("开始流式请求")
+        data = {"model": self.model, "messages": self.context, "stream": True}
+        logger.info(f"使用模型：{self.model}，开始流式请求")
         url = self.api_base + "/completions"
         # 请求接收流式数据
         try: