Added who_am_i tool to get the user's fullname via HF API.

app.py

@@ -20,6 +20,19 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     """
     return "What magic will you build ?"
 
+@tool
+def who_am_i()-> str:
+    """A tool that return the full name of the user given by Hugging Face API
+
+    Returns:
+        str: The user's full name
+    """
+    response = requests.get("https://huggingface.co/api/whoami-v2")
+    if response.status_code == 200:
+        return response.json().get("fullname", "Error fetching user's full name")
+    else:
+        return "Error fetching user's full name"
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -46,8 +59,7 @@ visit_webpage = VisitWebpageTool()
 model = HfApiModel(
 max_tokens=2096,
 temperature=0.5,
-# model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
-model_id='Qwen/QwQ-32B',
+model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
 custom_role_conversions=None,
 )
 
@@ -60,7 +72,14 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[get_current_time_in_timezone, web_search, visit_webpage, final_answer], ## add your tools here (don't remove final answer)
+    tools=[
+        get_current_time_in_timezone,
+        web_search,
+        visit_webpage,
+        image_generation_tool,
+        who_am_i,
+        final_answer
+    ], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,

prompts.yaml

@@ -28,6 +28,23 @@
   final_answer(image)
   ```<end_code>
 
+  ---
+  Task: "Get the full name of the user that asks."
+
+  Thought: I will proceed step by step and use the following tools: `who_am_i` to find the full name of the user.
+  Code:
+  ```py
+  full_name = who_am_i()
+  print(full_name)
+  ```<end_code>
+  Observation: "Lucas Didone"
+
+  Thought: I will show the user's name.
+  Code:
+  ```py
+  final_answer(full_name)
+  ```<end_code>
+
   ---
   Task: "What is the result of the following operation: 5 + 3 + 1294.678?"