UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-02-21 12:29:29 +00:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

feat: user valves support

Browse Source
This commit is contained in:
Timothy J. Baek 2024-06-22 12:14:12 -07:00
parent d362fd027e
commit 6ccb5e8f67
1 changed files with 19 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
backend/main.py
View File

@ -278,6 +278,9 @@ async def get_function_call_response(
"email": user.email, "email": user.email,
"name": user.name, "name": user.name,
"role": user.role, "role": user.role,
"valves": Tools.get_user_valves_by_id_and_user_id(
tool_id, user.id
),
}, },
} }
@ -401,6 +404,9 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware):
"email": user.email, "email": user.email,
"name": user.name, "name": user.name,
"role": user.role, "role": user.role,
"valves": Functions.get_user_valves_by_id_and_user_id(
filter_id, user.id
),
}, },
} }
@ -871,10 +877,14 @@ async def generate_chat_completions(form_data: dict, user=Depends(get_verified_u
"email": user.email, "email": user.email,
"name": user.name, "name": user.name,
"role": user.role, "role": user.role,
"valves": Functions.get_user_valves_by_id_and_user_id(
pipe_id, user.id
),
}, },
} }
if form_data["stream"]: if form_data["stream"]:
async def stream_content(): async def stream_content():
if inspect.iscoroutinefunction(pipe): if inspect.iscoroutinefunction(pipe):
res = await pipe(**param) res = await pipe(**param)
@ -1010,7 +1020,12 @@ async def chat_completed(form_data: dict, user=Depends(get_verified_user)):
f"{url}/{filter['id']}/filter/outlet", f"{url}/{filter['id']}/filter/outlet",
headers=headers, headers=headers,
json={ json={
"user": {"id": user.id, "name": user.name, "role": user.role}, "user": {
"id": user.id,
"name": user.name,
"email": user.email,
"role": user.role,
},
"body": data, "body": data,
}, },
) )
@ -1064,6 +1079,9 @@ async def chat_completed(form_data: dict, user=Depends(get_verified_user)):
"email": user.email, "email": user.email,
"name": user.name, "name": user.name,
"role": user.role, "role": user.role,
"valves": Functions.get_user_valves_by_id_and_user_id(
filter_id, user.id
),
}, },
} }