UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-04-24 00:06:11 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

black formatting

Browse Source
This commit is contained in:
Didier FOURNOUT 2025-02-13 16:12:46 +00:00
parent 6d62e71c34
commit 06062568c7
3 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
backend/open_webui/routers/ollama.py
View File

@ -115,7 +115,7 @@ async def send_post_request(
stream: bool = True, stream: bool = True,
key: Optional[str] = None, key: Optional[str] = None,
content_type: Optional[str] = None, content_type: Optional[str] = None,
user: UserModel = None user: UserModel = None,
): ):
r = None r = None
@ -317,7 +317,9 @@ async def get_all_models(request: Request, user: UserModel=None):
key = api_config.get("key", None) key = api_config.get("key", None)
if enable: if enable:
request_tasks.append(send_get_request(f"{url}/api/tags", key, user=user)) request_tasks.append(
send_get_request(f"{url}/api/tags", key, user=user)
)
else: else:
request_tasks.append(asyncio.ensure_future(asyncio.sleep(0, None))) request_tasks.append(asyncio.ensure_future(asyncio.sleep(0, None)))
@ -531,7 +533,7 @@ async def get_ollama_loaded_models(request: Request, user=Depends(get_verified_u
url, {} url, {}
), # Legacy support ), # Legacy support
).get("key", None), ).get("key", None),
user=user user=user,
) )
for idx, url in enumerate(request.app.state.config.OLLAMA_BASE_URLS) for idx, url in enumerate(request.app.state.config.OLLAMA_BASE_URLS)
] ]

2
backend/open_webui/routers/openai.py
View File

@ -70,7 +70,7 @@ async def send_get_request(url, key=None, user: UserModel=None):
if ENABLE_FORWARD_USER_INFO_HEADERS if ENABLE_FORWARD_USER_INFO_HEADERS
else {} else {}
), ),
} },
) as response: ) as response:
return await response.json() return await response.json()
except Exception as e: except Exception as e: