feat: tool __user__ email

This commit is contained in:
Timothy J. Baek 2024-06-11 10:54:11 -07:00
parent 37ed31b0e2
commit e4fe1fff97
2 changed files with 5 additions and 2 deletions

View File

@ -230,6 +230,7 @@ async def get_function_call_response(prompt, tool_id, template, task_model_id, u
**result["parameters"],
"__user__": {
"id": user.id,
"email": user.email,
"name": user.name,
"role": user.role,
},

View File

@ -20,9 +20,9 @@ class Tools:
# Use Sphinx-style docstrings to document your tools, they will be used for generating tools specifications
# Please refer to function_calling_filter_pipeline.py file from pipelines project for an example
def get_user_name_and_id(self, __user__: dict = {}) -> str:
def get_user_name_and_email_and_id(self, __user__: dict = {}) -> str:
"""
Get the user name and ID from the user object.
Get the user name, Email and ID from the user object.
"""
# Do not include :param for __user__ in the docstring as it should not be shown in the tool's documentation
@ -35,6 +35,8 @@ class Tools:
result += f"User: {__user__['name']}"
if "id" in __user__:
result += f" (ID: {__user__['id']})"
if "email" in __user__:
result += f" (Email: {__user__['email']})"
if result == "":
result = "User: Unknown"