AI/open-webui
1
0
Fork 1
mirror of https://git.mirrors.martin98.com/https://github.com/open-webui/open-webui synced 2025-08-19 22:39:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

chore: format

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2025-05-03 23:48:24 +04:00
parent aefd5d9557
commit 4cfb99248d
4 changed files with 32 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
backend/open_webui/retrieval/web/yacy.py
View File

@ -53,7 +53,7 @@ def search_yacy(
# Check if provided a json API URL # Check if provided a json API URL
if not query_url.endswith("yacysearch.json"): if not query_url.endswith("yacysearch.json"):
# Strip all query parameters from the URL # Strip all query parameters from the URL
query_url = query_url.rstrip('/') + "/yacysearch.json" query_url = query_url.rstrip("/") + "/yacysearch.json"
log.debug(f"searching {query_url}") log.debug(f"searching {query_url}")
@ -79,7 +79,9 @@ def search_yacy(
sorted_results = get_filtered_results(sorted_results, filter_list) sorted_results = get_filtered_results(sorted_results, filter_list)
return [ return [
SearchResult( SearchResult(
link=result["link"], title=result.get("title"), snippet=result.get("description") link=result["link"],
title=result.get("title"),
snippet=result.get("description"),
) )
for result in sorted_results[:count] for result in sorted_results[:count]
] ]

18
backend/open_webui/routers/audio.py
View File

@ -33,7 +33,7 @@ from open_webui.config import (
WHISPER_MODEL_AUTO_UPDATE, WHISPER_MODEL_AUTO_UPDATE,
WHISPER_MODEL_DIR, WHISPER_MODEL_DIR,
CACHE_DIR, CACHE_DIR,
WHISPER_LANGUAGE WHISPER_LANGUAGE,
) )
from open_webui.constants import ERROR_MESSAGES from open_webui.constants import ERROR_MESSAGES
@ -154,6 +154,7 @@ class STTConfigForm(BaseModel):
AZURE_BASE_URL: str AZURE_BASE_URL: str
AZURE_MAX_SPEAKERS: str AZURE_MAX_SPEAKERS: str
class AudioConfigUpdateForm(BaseModel): class AudioConfigUpdateForm(BaseModel):
tts: TTSConfigForm tts: TTSConfigForm
stt: STTConfigForm stt: STTConfigForm
@ -184,7 +185,7 @@ async def get_audio_config(request: Request, user=Depends(get_admin_user)):
"AZURE_REGION": request.app.state.config.AUDIO_STT_AZURE_REGION, "AZURE_REGION": request.app.state.config.AUDIO_STT_AZURE_REGION,
"AZURE_LOCALES": request.app.state.config.AUDIO_STT_AZURE_LOCALES, "AZURE_LOCALES": request.app.state.config.AUDIO_STT_AZURE_LOCALES,
"AZURE_BASE_URL": request.app.state.config.AUDIO_STT_AZURE_BASE_URL, "AZURE_BASE_URL": request.app.state.config.AUDIO_STT_AZURE_BASE_URL,
"AZURE_MAX_SPEAKERS": request.app.state.config.AUDIO_STT_AZURE_MAX_SPEAKERS, "AZURE_MAX_SPEAKERS": request.app.state.config.AUDIO_STT_AZURE_MAX_SPEAKERS,
}, },
} }
@ -215,7 +216,9 @@ async def update_audio_config(
request.app.state.config.AUDIO_STT_AZURE_REGION = form_data.stt.AZURE_REGION request.app.state.config.AUDIO_STT_AZURE_REGION = form_data.stt.AZURE_REGION
request.app.state.config.AUDIO_STT_AZURE_LOCALES = form_data.stt.AZURE_LOCALES request.app.state.config.AUDIO_STT_AZURE_LOCALES = form_data.stt.AZURE_LOCALES
request.app.state.config.AUDIO_STT_AZURE_BASE_URL = form_data.stt.AZURE_BASE_URL request.app.state.config.AUDIO_STT_AZURE_BASE_URL = form_data.stt.AZURE_BASE_URL
request.app.state.config.AUDIO_STT_AZURE_MAX_SPEAKERS = form_data.stt.AZURE_MAX_SPEAKERS request.app.state.config.AUDIO_STT_AZURE_MAX_SPEAKERS = (
form_data.stt.AZURE_MAX_SPEAKERS
)
if request.app.state.config.STT_ENGINE == "": if request.app.state.config.STT_ENGINE == "":
request.app.state.faster_whisper_model = set_faster_whisper_model( request.app.state.faster_whisper_model = set_faster_whisper_model(
@ -245,7 +248,7 @@ async def update_audio_config(
"AZURE_REGION": request.app.state.config.AUDIO_STT_AZURE_REGION, "AZURE_REGION": request.app.state.config.AUDIO_STT_AZURE_REGION,
"AZURE_LOCALES": request.app.state.config.AUDIO_STT_AZURE_LOCALES, "AZURE_LOCALES": request.app.state.config.AUDIO_STT_AZURE_LOCALES,
"AZURE_BASE_URL": request.app.state.config.AUDIO_STT_AZURE_BASE_URL, "AZURE_BASE_URL": request.app.state.config.AUDIO_STT_AZURE_BASE_URL,
"AZURE_MAX_SPEAKERS": request.app.state.config.AUDIO_STT_AZURE_MAX_SPEAKERS, "AZURE_MAX_SPEAKERS": request.app.state.config.AUDIO_STT_AZURE_MAX_SPEAKERS,
}, },
} }
@ -509,7 +512,7 @@ def transcribe(request: Request, file_path):
file_path, file_path,
beam_size=5, beam_size=5,
vad_filter=request.app.state.config.WHISPER_VAD_FILTER, vad_filter=request.app.state.config.WHISPER_VAD_FILTER,
language=WHISPER_LANGUAGE language=WHISPER_LANGUAGE,
) )
log.info( log.info(
"Detected language '%s' with probability %f" "Detected language '%s' with probability %f"
@ -698,7 +701,10 @@ def transcribe(request: Request, file_path):
) )
} }
url = base_url or f"https://{region}.api.cognitive.microsoft.com/speechtotext/transcriptions:transcribe?api-version=2024-11-15" url = (
base_url
or f"https://{region}.api.cognitive.microsoft.com/speechtotext/transcriptions:transcribe?api-version=2024-11-15"
)
# Use context manager to ensure file is properly closed # Use context manager to ensure file is properly closed
with open(file_path, "rb") as audio_file: with open(file_path, "rb") as audio_file:

24
backend/open_webui/socket/main.py
View File

@ -193,7 +193,7 @@ async def connect(sid, environ, auth):
await sio.emit("user-list", {"user_ids": list(USER_POOL.keys())}) await sio.emit("user-list", {"user_ids": list(USER_POOL.keys())})
await sio.emit("usage", {"models": get_models_in_use()}) await sio.emit("usage", {"models": get_models_in_use()})
return True return True
return False return False
@ -317,16 +317,18 @@ def get_event_emitter(request_info, update_db=True):
) )
) )
emit_tasks = [sio.emit( emit_tasks = [
"chat-events", sio.emit(
{ "chat-events",
"chat_id": request_info.get("chat_id", None), {
"message_id": request_info.get("message_id", None), "chat_id": request_info.get("chat_id", None),
"data": event_data, "message_id": request_info.get("message_id", None),
}, "data": event_data,
to=session_id, },
) to=session_id,
for session_id in session_ids] )
for session_id in session_ids
]
await asyncio.gather(*emit_tasks) await asyncio.gather(*emit_tasks)

4
backend/open_webui/utils/tools.py
View File

@ -291,7 +291,9 @@ def convert_function_to_pydantic_model(func: Callable) -> type[BaseModel]:
param_description = function_param_descriptions.get(name, None) param_description = function_param_descriptions.get(name, None)
if param_description: if param_description:
field_defs[name] = type_hint, Field(default_value, description=param_description) field_defs[name] = type_hint, Field(
default_value, description=param_description
)
else: else:
field_defs[name] = type_hint, default_value field_defs[name] = type_hint, default_value