AI/open-webui
1
0
Fork 1
mirror of https://git.mirrors.martin98.com/https://github.com/open-webui/open-webui synced 2025-08-20 01:09:06 +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]
] ]

14
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
@ -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(
@ -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:

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

@ -317,7 +317,8 @@ def get_event_emitter(request_info, update_db=True):
) )
) )
emit_tasks = [sio.emit( emit_tasks = [
sio.emit(
"chat-events", "chat-events",
{ {
"chat_id": request_info.get("chat_id", None), "chat_id": request_info.get("chat_id", None),
@ -326,7 +327,8 @@ def get_event_emitter(request_info, update_db=True):
}, },
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