Merge pull request #608 from RockChinQ/fix/reverse-proxy-invalid

Fix: 反向代理设置无效
RockChinQ · Nov 21, 2023 · 7890eac · 7890eac
2 parents eebd6a6 + e15f359
commit 7890eac
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/main.py b/main.py
@@ -241,6 +241,7 @@ def start(first_time_init=False):
 
             # 配置openai api_base
             if "reverse_proxy" in config.openai_config and config.openai_config["reverse_proxy"] is not None:
+                logging.debug("设置反向代理: "+config.openai_config['reverse_proxy'])
                 openai.base_url = config.openai_config["reverse_proxy"]
 
             # 主启动流程

diff --git a/pkg/openai/manager.py b/pkg/openai/manager.py
@@ -33,7 +33,8 @@ def __init__(self, api_key: str):
         # logging.info("文字总使用量：%d", self.audit_mgr.get_total_text_length())
 
         self.client = openai.Client(
-            api_key=self.key_mgr.get_using_key()
+            api_key=self.key_mgr.get_using_key(),
+            base_url=openai.base_url
         )
 
         context.set_openai_manager(self)