update suggest_menu function tool prompt for better output consistency. getting outputs like 'formal dinner', etc. without additional prompt

unit2/smolagents/code_agents.ipynb

@@ -39,12 +39,12 @@
     },
     {
       "cell_type": "markdown",
-      "source": [
-        "Let's also login to the Hugging Face Hub to have access to the Inference API."
-      ],
       "metadata": {
         "id": "hcfn1TDmVqqg"
-      }
+      },
+      "source": [
+        "Let's also login to the Hugging Face Hub to have access to the Inference API."
+      ]
     },
     {
       "cell_type": "code",
@@ -725,6 +725,9 @@
     },
     {
       "cell_type": "markdown",
+      "metadata": {
+        "id": "waE-prPUVzrh"
+      },
       "source": [
         "When you run this example, the output will **display a trace of the workflow steps being executed**. It will also print the corresponding Python code with the message:\n",
         "\n",
@@ -736,10 +739,7 @@
         "```\n",
         "\n",
         "After a few steps, you'll see the generated playlist that Alfred can use for the party! 🎵"
-      ],
-      "metadata": {
-        "id": "waE-prPUVzrh"
-      }
+      ]
     },
     {
       "cell_type": "markdown",
@@ -1021,7 +1021,7 @@
         "    \"\"\"\n",
         "    Suggests a menu based on the occasion.\n",
         "    Args:\n",
-        "        occasion: The type of occasion for the party.\n",
+        "        occasion: The type of occasion for the party. enums: ['casual', 'formal', 'superhero', 'others']\n",
         "    \"\"\"\n",
         "    if occasion == \"casual\":\n",
         "        return \"Pizza, snacks, and drinks.\"\n",
@@ -1039,14 +1039,14 @@
     },
     {
       "cell_type": "markdown",
+      "metadata": {
+        "id": "QZmuYtGPWKfO"
+      },
       "source": [
         "The agent will run for a few steps until finding the answer.\n",
         "\n",
         "The menu is ready! 🥗"
-      ],
-      "metadata": {
-        "id": "QZmuYtGPWKfO"
-      }
+      ]
     },
     {
       "cell_type": "markdown",
@@ -1590,15 +1590,15 @@
     },
     {
       "cell_type": "markdown",
+      "metadata": {
+        "id": "T7pLiWCcWSsU"
+      },
       "source": [
         "These examples are just the beginning of what you can do with code agents, and we're already starting to see their utility for preparing the party.\n",
         "You can learn more about how to build code agents in the [smolagents documentation](https://huggingface.co/docs/smolagents).\n",
         "\n",
         "`smolagents` specializes in agents that write and execute Python code snippets, offering sandboxed execution for security. It supports both open-source and proprietary language models, making it adaptable to various development environments."
-      ],
-      "metadata": {
-        "id": "T7pLiWCcWSsU"
-      }
+      ]
     },
     {
       "cell_type": "markdown",
@@ -3555,12 +3555,12 @@
     },
     {
       "cell_type": "markdown",
-      "source": [
-        "To download the agent again, use the code below:"
-      ],
       "metadata": {
         "id": "3iPLK1IQWguH"
-      }
+      },
+      "source": [
+        "To download the agent again, use the code below:"
+      ]
     },
     {
       "cell_type": "code",
@@ -4539,14 +4539,14 @@
     },
     {
       "cell_type": "markdown",
+      "metadata": {
+        "id": "qz-vwxu9Wobn"
+      },
       "source": [
         "What's also exciting is that shared agents are directly available as Hugging Face Spaces, allowing you to interact with them in real-time. You can explore other agents [here](https://huggingface.co/spaces/davidberenstein1957/smolagents-and-tools).\n",
         "\n",
         "For example, the _AlfredAgent_ is available [here](https://huggingface.co/spaces/sergiopaniego/AlfredAgent)."
-      ],
-      "metadata": {
-        "id": "qz-vwxu9Wobn"
-      }
+      ]
     },
     {
       "cell_type": "markdown",
@@ -4580,14 +4580,14 @@
     },
     {
       "cell_type": "markdown",
+      "metadata": {
+        "id": "IVBwrNgbXPS-"
+      },
       "source": [
         "Next, Alfred has already created an account on Langfuse and has his API keys ready. If you haven’t done so yet, you can sign up for Langfuse Cloud [here](https://cloud.langfuse.com/) or explore [alternatives](https://huggingface.co/docs/smolagents/tutorials/inspect_runs).  \n",
         "\n",
         "Once you have your API keys, they need to be properly configured as follows:"
-      ],
-      "metadata": {
-        "id": "IVBwrNgbXPS-"
-      }
+      ]
     },
     {
       "cell_type": "code",
@@ -4612,12 +4612,12 @@
     },
     {
       "cell_type": "markdown",
-      "source": [
-        "Finally, Alfred is ready to initialize the `SmolagentsInstrumentor` and start tracking his agent's performance.  "
-      ],
       "metadata": {
         "id": "ZIRRj-PJXRGH"
-      }
+      },
+      "source": [
+        "Finally, Alfred is ready to initialize the `SmolagentsInstrumentor` and start tracking his agent's performance.  "
+      ]
     },
     {
       "cell_type": "code",
@@ -4641,12 +4641,12 @@
     },
     {
       "cell_type": "markdown",
-      "source": [
-        "Alfred is now connected 🔌! The runs from `smolagents` are being logged in Langfuse, giving him full visibility into the agent's behavior. With this setup, he's ready to revisit previous runs and refine his Party Preparator Agent even further.  "
-      ],
       "metadata": {
         "id": "B0oLBr_qXSjA"
-      }
+      },
+      "source": [
+        "Alfred is now connected 🔌! The runs from `smolagents` are being logged in Langfuse, giving him full visibility into the agent's behavior. With this setup, he's ready to revisit previous runs and refine his Party Preparator Agent even further.  "
+      ]
     },
     {
       "cell_type": "code",
@@ -5212,14 +5212,14 @@
     },
     {
       "cell_type": "markdown",
+      "metadata": {
+        "id": "KBpqTBK1XVLz"
+      },
       "source": [
         "Alfred can now access this logs [here](https://cloud.langfuse.com/project/cm7bq0abj025rad078ak3luwi/traces/995fc019255528e4f48cf6770b0ce27b?timestamp=2025-02-19T10%3A28%3A36.929Z) to review and analyze them.  \n",
         "\n",
         "Meanwhile, the [suggested playlist](https://open.spotify.com/playlist/0gZMMHjuxMrrybQ7wTMTpw) sets the perfect vibe for the party preparations. Cool, right? 🎶\n"
-      ],
-      "metadata": {
-        "id": "KBpqTBK1XVLz"
-      }
+      ]
     }
   ],
   "metadata": {
@@ -10631,4 +10631,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 0
-}
+}