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

fix: filter

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2025-02-08 01:07:05 -08:00
parent ee0152c1fa
commit 9be8bea6f4
3 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
backend/open_webui/utils/chat.py
View File

@ -199,6 +199,7 @@ async def chat_completed(request: Request, form_data: dict, user: Any):
}, },
"__metadata__": metadata, "__metadata__": metadata,
"__request__": request, "__request__": request,
"__model__": model,
} }
try: try:

14
backend/open_webui/utils/filter.py
View File

@ -61,12 +61,14 @@ async def process_filter_functions(
try: try:
# Prepare parameters # Prepare parameters
sig = inspect.signature(handler) sig = inspect.signature(handler)
params = {"body": form_data} params = {"body": form_data} | {
k: v
# Add extra parameters that exist in the handler's signature for k, v in {
for key in list(extra_params.keys()): **extra_params,
if key in sig.parameters: "__id__": filter_id,
params[key] = extra_params[key] }.items()
if k in sig.parameters
}
# Handle user parameters # Handle user parameters
if "__user__" in sig.parameters: if "__user__" in sig.parameters:

1
backend/open_webui/utils/middleware.py
View File

@ -617,6 +617,7 @@ async def process_chat_payload(request, form_data, metadata, user, model):
}, },
"__metadata__": metadata, "__metadata__": metadata,
"__request__": request, "__request__": request,
"__model__": model,
} }
# Initialize events to store additional event to be sent to the client # Initialize events to store additional event to be sent to the client