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

This commit is contained in:
LINSTCL
2023-03-03 15:20:42 +08:00
parent c57642bd4e
commit c23d114094
2 changed files with 3 additions and 2 deletions

View File

@@ -39,7 +39,7 @@ class OpenAIInteract:
ai: ModelRequest = create_openai_model_request( ai: ModelRequest = create_openai_model_request(
config.completion_api_params['model'], config.completion_api_params['model'],
'user', 'user',
config.openai_config["http_proxy"] config.openai_config["http_proxy"] if "http_proxy" in config.openai_config else None
) )
ai.request( ai.request(
prompts, prompts,

View File

@@ -45,8 +45,9 @@ class ModelRequest():
def request(self, **kwargs): def request(self, **kwargs):
if self.proxy != None: #异步请求 if self.proxy != None: #异步请求
loop = asyncio.new_event_loop()
self.runtime = threading.Thread( self.runtime = threading.Thread(
target=asyncio.run, target=loop.run_until_complete,
args=(self.__a_request__(**kwargs),) args=(self.__a_request__(**kwargs),)
) )
self.runtime.start() self.runtime.start()