mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-02 03:55:55 +00:00
52 lines
1.0 KiB
Python
52 lines
1.0 KiB
Python
# 定义不同接口请求的模型
|
|
import threading
|
|
import asyncio
|
|
import logging
|
|
|
|
import openai
|
|
|
|
|
|
class RequestBase:
|
|
|
|
req_func: callable
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
raise NotImplementedError
|
|
|
|
def _req(self, **kwargs):
|
|
"""处理代理问题"""
|
|
|
|
ret: dict = {}
|
|
exception: Exception = None
|
|
|
|
async def awrapper(**kwargs):
|
|
nonlocal ret, exception
|
|
|
|
try:
|
|
ret = await self.req_func(**kwargs)
|
|
logging.debug("接口请求返回：%s", str(ret))
|
|
return ret
|
|
except Exception as e:
|
|
exception = e
|
|
|
|
loop = asyncio.new_event_loop()
|
|
|
|
thr = threading.Thread(
|
|
target=loop.run_until_complete,
|
|
args=(awrapper(**kwargs),)
|
|
)
|
|
|
|
thr.start()
|
|
thr.join()
|
|
|
|
if exception is not None:
|
|
raise exception
|
|
|
|
return ret
|
|
|
|
def __iter__(self):
|
|
raise self
|
|
|
|
def __next__(self):
|
|
raise NotImplementedError
|