XFE Git
XFE Studio Git
Git 首页 全局搜索
XFE 主站 文档 NuGet
公开
关注 0 Fork 0 Star 0
返回提交历史

XFEstudio/gpt4free

Support timeout in backend_api

6f9b9aeb
hlohaus <983577+hlohaus@users.noreply.github.com>
提交于

代码差异

1 个文件 +5 -4
Modified g4f/Provider/needs_auth/Video.py +5 -4
@@ -32,10 +32,10 @@ class RequestConfig:
32 32 @classmethod
33 33 async def get_response(cls, prompt: str) -> VideoResponse | None:
34 34 if prompt in cls.urls and cls.urls[prompt]:
35 cls.urls[prompt] = list(set(cls.urls[prompt]))
36 return VideoResponse(cls.urls[prompt], prompt, {
35 unique_list = list(set(cls.urls[prompt]))
36 return VideoResponse(unique_list, prompt, {
37 37 "headers": {"authorization": cls.headers.get("authorization")} if cls.headers.get("authorization") else {},
38 "preview": [url.replace("md.mp4", "thumb.webp") for url in cls.urls[prompt]]
38 "preview": [url.replace("md.mp4", "thumb.webp") for url in unique_list]
39 39 })
40 40 async with ClientSession() as session:
41 41 found_urls = []
@@ -131,7 +131,8 @@ class Video(AsyncGeneratorProvider, ProviderModelMixin):
131 131 await page.send(nodriver.cdp.network.enable())
132 132 page.add_handler(nodriver.cdp.network.RequestWillBeSent, on_request)
133 133 if model == "search":
134 asyncio.sleep(5)
134 for _ in range(5):
135 await page.scroll_down(50)
135 136 response = await RequestConfig.get_response(prompt)
136 137 if response:
137 138 yield Reasoning(label="Found", status="")