proxy后向兼容，修复部分报错

RockChinQ · Mar 3, 2023 · c23d114 · c23d114
1 parent c57642b
commit c23d114
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/pkg/openai/manager.py b/pkg/openai/manager.py
@@ -39,7 +39,7 @@ def request_completion(self, prompts):
         ai: ModelRequest = create_openai_model_request(
             config.completion_api_params['model'], 
             'user', 
-            config.openai_config["http_proxy"]
+            config.openai_config["http_proxy"] if "http_proxy" in config.openai_config else None
         )
         ai.request(
             prompts,

diff --git a/pkg/openai/modelmgr.py b/pkg/openai/modelmgr.py
@@ -45,8 +45,9 @@ async def __a_request__(self, **kwargs):
 
     def request(self, **kwargs):
         if self.proxy != None: #异步请求
+            loop = asyncio.new_event_loop()
             self.runtime = threading.Thread(
-                target=asyncio.run,
+                target=loop.run_until_complete,
                 args=(self.__a_request__(**kwargs),)
             )
             self.runtime.start()