UniqAI
/
open-webui
mirror of https://github.com/open-webui/open-webui
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

refac: __user__ param

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
if content is not None:
print(content)
result = json.loads(content)
print(result)
@ -223,13 +222,13 @@ async def get_function_call_response(prompt, tool_id, template, task_model_id, u
try:
# Get the signature of the function
sig = inspect.signature(function)
# Check if 'user' is a parameter of the function
if "user" in sig.parameters:
# Call the function with the 'user' parameter included
# Check if '__user__' is a parameter of the function
if "__user__" in sig.parameters:
# Call the function with the '__user__' parameter included
function_result = function(
**{
**result["parameters"],
"user": {
"__user__": {
"id": user.id,
"name": user.name,
"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(
function
).items()
if param_name != "return"
if param_name != "return" and param_name != "__user__"
},
"required": [
name