UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-16 21:42:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

refac: __user__ param

Browse Source
This commit is contained in:
Timothy J. Baek 2024-06-11 10:35:13 -07:00
parent 8a86f32700
commit 67cd9b76ae
2 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
backend/main.py
View File

@ -205,7 +205,6 @@ async def get_function_call_response(prompt, tool_id, template, task_model_id, u
# Parse the function response # Parse the function response
if content is not None: if content is not None:
print(content) print(content)
result = json.loads(content) result = json.loads(content)
print(result) print(result)
@ -223,13 +222,13 @@ async def get_function_call_response(prompt, tool_id, template, task_model_id, u
try: try:
# Get the signature of the function # Get the signature of the function
sig = inspect.signature(function) sig = inspect.signature(function)
# Check if 'user' is a parameter of the function # Check if '__user__' is a parameter of the function
if "user" in sig.parameters: if "__user__" in sig.parameters:
# Call the function with the 'user' parameter included # Call the function with the '__user__' parameter included
function_result = function( function_result = function(
**{ **{
**result["parameters"], **result["parameters"],
"user": { "__user__": {
"id": user.id, "id": user.id,
"name": user.name, "name": user.name,
"role": user.role, "role": user.role,

2
backend/utils/tools.py
View File

@ -57,7 +57,7 @@ def get_tools_specs(tools) -> List[dict]:
for param_name, param_annotation in get_type_hints( for param_name, param_annotation in get_type_hints(
function function
).items() ).items()
if param_name != "return" if param_name != "return" and param_name != "__user__"
}, },
"required": [ "required": [
name name