Feat: First tool, character's quotes

I'm using Star Trek theme for my first smolagent

app.py

@@ -4,19 +4,41 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+import random
 
 from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
+def get_star_trek_quote(character: str)-> str: #it's import to specify the return type
     #Keep this format for the description / args / args description but feel free to modify the tool
-    """A tool that does nothing yet 
+    """
+    Fetches a quote from a Star Trek character.
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        character: The name of the character whose quote is requested (e.g., "Seven of Nine", "Spock", "Picard", "Data").
+
+    Returns: 
+        A quote from the specified Star Trek character.
     """
-    return "What magic will you build ?"
+    quotes = {
+        "Seven of Nine": [
+            "It is unsettling. You say that I am a human being and yet, I am also Borg... Part of me not unlike your replicator... Not unlike the Doctor. Will you one day choose to abandon me as well?",
+            "Fun will now commence."
+        ],
+        "Spock": [
+            "Live and prosper.",
+            "The needs of the many outweigh the needs of the few."
+        ],
+        "Picard": [
+            "Make it so",
+            "Earth was once a violent planet too. At times, the chaos threatened the very fabric of life. But like you, we evolved. We found better ways to handle our conflicts. But I think noone can deny that the seed of violence remains within each of us, We must recognise that, because that violence is capable of consuming each of us."
+        ],
+        "Data": [
+            "It is interesting that people try to find meaningful patterns in things that are essentially random. I have noticed that the images they percieve sometimes suggest what they are thinking about at that particular moment. Besides, it is clearly a bunny rabbit.",
+            "One is my name. The other is not."
+        ]
+    }
+    return quotes.get(character, ["Sorry, no quotes available for this character"])[0]
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -55,7 +77,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_star_trek_quote], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,