simple smolagent with duck duck go search tool and youtube transcription tool

app.py

@@ -10,7 +10,7 @@ from langchain_community.document_loaders import YoutubeLoader
 
 
 #client = genai.Client(api_key=os.getenv("API_KEY"))
-model = LiteLLMModel(model_id="gemini/gemini-2.0-flash-exp",
+model = LiteLLMModel(model_id="gemini/gemini-2.0-flash-lite",
                      api_key=os.getenv("API_KEY"))
 # (Keep Constants as is)
 # --- Constants ---

test.ipynb

@@ -363,6 +363,258 @@
     "What does Teal'c say in response to the question \"\"Isn't that hot?\"\"\")"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": 12,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\u001b[1;31mInit signature:\u001b[0m\n",
+      "\u001b[0mCodeAgent\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m    \u001b[0mtools\u001b[0m\u001b[1;33m:\u001b[0m \u001b[0mList\u001b[0m\u001b[1;33m[\u001b[0m\u001b[0msmolagents\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mtools\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mTool\u001b[0m\u001b[1;33m]\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m    \u001b[0mmodel\u001b[0m\u001b[1;33m:\u001b[0m \u001b[0mCallable\u001b[0m\u001b[1;33m[\u001b[0m\u001b[1;33m[\u001b[0m\u001b[0mList\u001b[0m\u001b[1;33m[\u001b[0m\u001b[0mDict\u001b[0m\u001b[1;33m[\u001b[0m\u001b[0mstr\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mstr\u001b[0m\u001b[1;33m]\u001b[0m\u001b[1;33m]\u001b[0m\u001b[1;33m]\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0msmolagents\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mmodels\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mChatMessage\u001b[0m\u001b[1;33m]\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m    \u001b[0mprompt_templates\u001b[0m\u001b[1;33m:\u001b[0m \u001b[0mOptional\u001b[0m\u001b[1;33m[\u001b[0m\u001b[0msmolagents\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0magents\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mPromptTemplates\u001b[0m\u001b[1;33m]\u001b[0m \u001b[1;33m=\u001b[0m \u001b[1;32mNone\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m    \u001b[0mgrammar\u001b[0m\u001b[1;33m:\u001b[0m \u001b[0mOptional\u001b[0m\u001b[1;33m[\u001b[0m\u001b[0mDict\u001b[0m\u001b[1;33m[\u001b[0m\u001b[0mstr\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mstr\u001b[0m\u001b[1;33m]\u001b[0m\u001b[1;33m]\u001b[0m \u001b[1;33m=\u001b[0m \u001b[1;32mNone\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m    \u001b[0madditional_authorized_imports\u001b[0m\u001b[1;33m:\u001b[0m \u001b[0mOptional\u001b[0m\u001b[1;33m[\u001b[0m\u001b[0mList\u001b[0m\u001b[1;33m[\u001b[0m\u001b[0mstr\u001b[0m\u001b[1;33m]\u001b[0m\u001b[1;33m]\u001b[0m \u001b[1;33m=\u001b[0m \u001b[1;32mNone\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m    \u001b[0mplanning_interval\u001b[0m\u001b[1;33m:\u001b[0m \u001b[0mOptional\u001b[0m\u001b[1;33m[\u001b[0m\u001b[0mint\u001b[0m\u001b[1;33m]\u001b[0m \u001b[1;33m=\u001b[0m \u001b[1;32mNone\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m    \u001b[0mexecutor_type\u001b[0m\u001b[1;33m:\u001b[0m \u001b[0mstr\u001b[0m \u001b[1;33m|\u001b[0m \u001b[1;32mNone\u001b[0m \u001b[1;33m=\u001b[0m \u001b[1;34m'local'\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m    \u001b[0mexecutor_kwargs\u001b[0m\u001b[1;33m:\u001b[0m \u001b[0mOptional\u001b[0m\u001b[1;33m[\u001b[0m\u001b[0mDict\u001b[0m\u001b[1;33m[\u001b[0m\u001b[0mstr\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mAny\u001b[0m\u001b[1;33m]\u001b[0m\u001b[1;33m]\u001b[0m \u001b[1;33m=\u001b[0m \u001b[1;32mNone\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m    \u001b[0mmax_print_outputs_length\u001b[0m\u001b[1;33m:\u001b[0m \u001b[0mOptional\u001b[0m\u001b[1;33m[\u001b[0m\u001b[0mint\u001b[0m\u001b[1;33m]\u001b[0m \u001b[1;33m=\u001b[0m \u001b[1;32mNone\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m    \u001b[1;33m**\u001b[0m\u001b[0mkwargs\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n",
+      "\u001b[1;31mSource:\u001b[0m        \n",
+      "\u001b[1;32mclass\u001b[0m \u001b[0mCodeAgent\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mMultiStepAgent\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m    \u001b[1;34m\"\"\"\n",
+      "    In this agent, the tool calls will be formulated by the LLM in code format, then parsed and executed.\n",
+      "\n",
+      "    Args:\n",
+      "        tools (`list[Tool]`): [`Tool`]s that the agent can use.\n",
+      "        model (`Callable[[list[dict[str, str]]], ChatMessage]`): Model that will generate the agent's actions.\n",
+      "        prompt_templates ([`~agents.PromptTemplates`], *optional*): Prompt templates.\n",
+      "        grammar (`dict[str, str]`, *optional*): Grammar used to parse the LLM output.\n",
+      "        additional_authorized_imports (`list[str]`, *optional*): Additional authorized imports for the agent.\n",
+      "        planning_interval (`int`, *optional*): Interval at which the agent will run a planning step.\n",
+      "        executor_type (`str`, default `\"local\"`): Which executor type to use between `\"local\"`, `\"e2b\"`, or `\"docker\"`.\n",
+      "        executor_kwargs (`dict`, *optional*): Additional arguments to pass to initialize the executor.\n",
+      "        max_print_outputs_length (`int`, *optional*): Maximum length of the print outputs.\n",
+      "        **kwargs: Additional keyword arguments.\n",
+      "\n",
+      "    \"\"\"\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m    \u001b[1;32mdef\u001b[0m \u001b[0m__init__\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mself\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mtools\u001b[0m\u001b[1;33m:\u001b[0m \u001b[0mList\u001b[0m\u001b[1;33m[\u001b[0m\u001b[0mTool\u001b[0m\u001b[1;33m]\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mmodel\u001b[0m\u001b[1;33m:\u001b[0m \u001b[0mCallable\u001b[0m\u001b[1;33m[\u001b[0m\u001b[1;33m[\u001b[0m\u001b[0mList\u001b[0m\u001b[1;33m[\u001b[0m\u001b[0mDict\u001b[0m\u001b[1;33m[\u001b[0m\u001b[0mstr\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mstr\u001b[0m\u001b[1;33m]\u001b[0m\u001b[1;33m]\u001b[0m\u001b[1;33m]\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mChatMessage\u001b[0m\u001b[1;33m]\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mprompt_templates\u001b[0m\u001b[1;33m:\u001b[0m \u001b[0mOptional\u001b[0m\u001b[1;33m[\u001b[0m\u001b[0mPromptTemplates\u001b[0m\u001b[1;33m]\u001b[0m \u001b[1;33m=\u001b[0m \u001b[1;32mNone\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mgrammar\u001b[0m\u001b[1;33m:\u001b[0m \u001b[0mOptional\u001b[0m\u001b[1;33m[\u001b[0m\u001b[0mDict\u001b[0m\u001b[1;33m[\u001b[0m\u001b[0mstr\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mstr\u001b[0m\u001b[1;33m]\u001b[0m\u001b[1;33m]\u001b[0m \u001b[1;33m=\u001b[0m \u001b[1;32mNone\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0madditional_authorized_imports\u001b[0m\u001b[1;33m:\u001b[0m \u001b[0mOptional\u001b[0m\u001b[1;33m[\u001b[0m\u001b[0mList\u001b[0m\u001b[1;33m[\u001b[0m\u001b[0mstr\u001b[0m\u001b[1;33m]\u001b[0m\u001b[1;33m]\u001b[0m \u001b[1;33m=\u001b[0m \u001b[1;32mNone\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mplanning_interval\u001b[0m\u001b[1;33m:\u001b[0m \u001b[0mOptional\u001b[0m\u001b[1;33m[\u001b[0m\u001b[0mint\u001b[0m\u001b[1;33m]\u001b[0m \u001b[1;33m=\u001b[0m \u001b[1;32mNone\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mexecutor_type\u001b[0m\u001b[1;33m:\u001b[0m \u001b[0mstr\u001b[0m \u001b[1;33m|\u001b[0m \u001b[1;32mNone\u001b[0m \u001b[1;33m=\u001b[0m \u001b[1;34m\"local\"\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mexecutor_kwargs\u001b[0m\u001b[1;33m:\u001b[0m \u001b[0mOptional\u001b[0m\u001b[1;33m[\u001b[0m\u001b[0mDict\u001b[0m\u001b[1;33m[\u001b[0m\u001b[0mstr\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mAny\u001b[0m\u001b[1;33m]\u001b[0m\u001b[1;33m]\u001b[0m \u001b[1;33m=\u001b[0m \u001b[1;32mNone\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mmax_print_outputs_length\u001b[0m\u001b[1;33m:\u001b[0m \u001b[0mOptional\u001b[0m\u001b[1;33m[\u001b[0m\u001b[0mint\u001b[0m\u001b[1;33m]\u001b[0m \u001b[1;33m=\u001b[0m \u001b[1;32mNone\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[1;33m**\u001b[0m\u001b[0mkwargs\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m    \u001b[1;33m)\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0madditional_authorized_imports\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0madditional_authorized_imports\u001b[0m \u001b[1;32mif\u001b[0m \u001b[0madditional_authorized_imports\u001b[0m \u001b[1;32melse\u001b[0m \u001b[1;33m[\u001b[0m\u001b[1;33m]\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mauthorized_imports\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0msorted\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mset\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mBASE_BUILTIN_MODULES\u001b[0m\u001b[1;33m)\u001b[0m \u001b[1;33m|\u001b[0m \u001b[0mset\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0madditional_authorized_imports\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mmax_print_outputs_length\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mmax_print_outputs_length\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mprompt_templates\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mprompt_templates\u001b[0m \u001b[1;32mor\u001b[0m \u001b[0myaml\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0msafe_load\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[0mimportlib\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mresources\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mfiles\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;34m\"smolagents.prompts\"\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mjoinpath\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;34m\"code_agent.yaml\"\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mread_text\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0msuper\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0m__init__\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[0mtools\u001b[0m\u001b[1;33m=\u001b[0m\u001b[0mtools\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[0mmodel\u001b[0m\u001b[1;33m=\u001b[0m\u001b[0mmodel\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[0mprompt_templates\u001b[0m\u001b[1;33m=\u001b[0m\u001b[0mprompt_templates\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[0mgrammar\u001b[0m\u001b[1;33m=\u001b[0m\u001b[0mgrammar\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[0mplanning_interval\u001b[0m\u001b[1;33m=\u001b[0m\u001b[0mplanning_interval\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[1;33m**\u001b[0m\u001b[0mkwargs\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[1;32mif\u001b[0m \u001b[1;34m\"*\"\u001b[0m \u001b[1;32min\u001b[0m \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0madditional_authorized_imports\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mlogger\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mlog\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                \u001b[1;34m\"Caution: you set an authorization for all imports, meaning your agent can decide to import any package it deems necessary. This might raise issues if the package is not installed in your environment.\"\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                \u001b[1;36m0\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mexecutor_type\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mexecutor_type\u001b[0m \u001b[1;32mor\u001b[0m \u001b[1;34m\"local\"\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mexecutor_kwargs\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mexecutor_kwargs\u001b[0m \u001b[1;32mor\u001b[0m \u001b[1;33m{\u001b[0m\u001b[1;33m}\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mpython_executor\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mcreate_python_executor\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m    \u001b[1;32mdef\u001b[0m \u001b[0mcreate_python_executor\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mself\u001b[0m\u001b[1;33m)\u001b[0m \u001b[1;33m->\u001b[0m \u001b[0mPythonExecutor\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mmatch\u001b[0m \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mexecutor_type\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[0mcase\u001b[0m \u001b[1;34m\"e2b\"\u001b[0m \u001b[1;33m|\u001b[0m \u001b[1;34m\"docker\"\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                \u001b[1;32mif\u001b[0m \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mmanaged_agents\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                    \u001b[1;32mraise\u001b[0m \u001b[0mException\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;34m\"Managed agents are not yet supported with remote code execution.\"\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                \u001b[1;32mif\u001b[0m \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mexecutor_type\u001b[0m \u001b[1;33m==\u001b[0m \u001b[1;34m\"e2b\"\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                    \u001b[1;32mreturn\u001b[0m \u001b[0mE2BExecutor\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0madditional_authorized_imports\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mlogger\u001b[0m\u001b[1;33m,\u001b[0m \u001b[1;33m**\u001b[0m\u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mexecutor_kwargs\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                \u001b[1;32melse\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                    \u001b[1;32mreturn\u001b[0m \u001b[0mDockerExecutor\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0madditional_authorized_imports\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mlogger\u001b[0m\u001b[1;33m,\u001b[0m \u001b[1;33m**\u001b[0m\u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mexecutor_kwargs\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[0mcase\u001b[0m \u001b[1;34m\"local\"\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                \u001b[1;32mreturn\u001b[0m \u001b[0mLocalPythonExecutor\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                    \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0madditional_authorized_imports\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                    \u001b[0mmax_print_outputs_length\u001b[0m\u001b[1;33m=\u001b[0m\u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mmax_print_outputs_length\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                \u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[0mcase\u001b[0m \u001b[0m_\u001b[0m\u001b[1;33m:\u001b[0m  \u001b[1;31m# if applicable\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                \u001b[1;32mraise\u001b[0m \u001b[0mValueError\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33mf\"\u001b[0m\u001b[1;33mUnsupported executor type: \u001b[0m\u001b[1;33m{\u001b[0m\u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mexecutor_type\u001b[0m\u001b[1;33m}\u001b[0m\u001b[1;33m\"\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m    \u001b[1;32mdef\u001b[0m \u001b[0minitialize_system_prompt\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mself\u001b[0m\u001b[1;33m)\u001b[0m \u001b[1;33m->\u001b[0m \u001b[0mstr\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0msystem_prompt\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mpopulate_template\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mprompt_templates\u001b[0m\u001b[1;33m[\u001b[0m\u001b[1;34m\"system_prompt\"\u001b[0m\u001b[1;33m]\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[0mvariables\u001b[0m\u001b[1;33m=\u001b[0m\u001b[1;33m{\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                \u001b[1;34m\"tools\"\u001b[0m\u001b[1;33m:\u001b[0m \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mtools\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                \u001b[1;34m\"managed_agents\"\u001b[0m\u001b[1;33m:\u001b[0m \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mmanaged_agents\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                \u001b[1;34m\"authorized_imports\"\u001b[0m\u001b[1;33m:\u001b[0m \u001b[1;33m(\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                    \u001b[1;34m\"You can import from any package you want.\"\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                    \u001b[1;32mif\u001b[0m \u001b[1;34m\"*\"\u001b[0m \u001b[1;32min\u001b[0m \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mauthorized_imports\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                    \u001b[1;32melse\u001b[0m \u001b[0mstr\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mauthorized_imports\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                \u001b[1;33m)\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[1;33m}\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[1;32mreturn\u001b[0m \u001b[0msystem_prompt\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m    \u001b[1;32mdef\u001b[0m \u001b[0mstep\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mself\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mmemory_step\u001b[0m\u001b[1;33m:\u001b[0m \u001b[0mActionStep\u001b[0m\u001b[1;33m)\u001b[0m \u001b[1;33m->\u001b[0m \u001b[0mUnion\u001b[0m\u001b[1;33m[\u001b[0m\u001b[1;32mNone\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mAny\u001b[0m\u001b[1;33m]\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[1;34m\"\"\"\n",
+      "        Perform one step in the ReAct framework: the agent thinks, acts, and observes the result.\n",
+      "        Returns None if the step is not final.\n",
+      "        \"\"\"\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mmemory_messages\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mwrite_memory_to_messages\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0minput_messages\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mmemory_messages\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mcopy\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[1;31m# Add new step in logs\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mmemory_step\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mmodel_input_messages\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mmemory_messages\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mcopy\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[1;32mtry\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[0madditional_args\u001b[0m \u001b[1;33m=\u001b[0m \u001b[1;33m{\u001b[0m\u001b[1;34m\"grammar\"\u001b[0m\u001b[1;33m:\u001b[0m \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mgrammar\u001b[0m\u001b[1;33m}\u001b[0m \u001b[1;32mif\u001b[0m \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mgrammar\u001b[0m \u001b[1;32mis\u001b[0m \u001b[1;32mnot\u001b[0m \u001b[1;32mNone\u001b[0m \u001b[1;32melse\u001b[0m \u001b[1;33m{\u001b[0m\u001b[1;33m}\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[0mchat_message\u001b[0m\u001b[1;33m:\u001b[0m \u001b[0mChatMessage\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mmodel\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0minput_messages\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                \u001b[0mstop_sequences\u001b[0m\u001b[1;33m=\u001b[0m\u001b[1;33m[\u001b[0m\u001b[1;34m\"<end_code>\"\u001b[0m\u001b[1;33m,\u001b[0m \u001b[1;34m\"Observation:\"\u001b[0m\u001b[1;33m,\u001b[0m \u001b[1;34m\"Calling tools:\"\u001b[0m\u001b[1;33m]\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                \u001b[1;33m**\u001b[0m\u001b[0madditional_args\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[0mmemory_step\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mmodel_output_message\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mchat_message\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[0mmodel_output\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mchat_message\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mcontent\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[1;31m# This adds <end_code> sequence to the history.\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[1;31m# This will nudge ulterior LLM calls to finish with <end_code>, thus efficiently stopping generation.\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[1;32mif\u001b[0m \u001b[0mmodel_output\u001b[0m \u001b[1;32mand\u001b[0m \u001b[0mmodel_output\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mstrip\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mendswith\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;34m\"```\"\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                \u001b[0mmodel_output\u001b[0m \u001b[1;33m+=\u001b[0m \u001b[1;34m\"<end_code>\"\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                \u001b[0mmemory_step\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mmodel_output_message\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mcontent\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mmodel_output\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[0mmemory_step\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mmodel_output\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mmodel_output\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[1;32mexcept\u001b[0m \u001b[0mException\u001b[0m \u001b[1;32mas\u001b[0m \u001b[0me\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[1;32mraise\u001b[0m \u001b[0mAgentGenerationError\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33mf\"\u001b[0m\u001b[1;33mError in generating model output:\\n\u001b[0m\u001b[1;33m{\u001b[0m\u001b[0me\u001b[0m\u001b[1;33m}\u001b[0m\u001b[1;33m\"\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mlogger\u001b[0m\u001b[1;33m)\u001b[0m \u001b[1;32mfrom\u001b[0m \u001b[0me\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mlogger\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mlog_markdown\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[0mcontent\u001b[0m\u001b[1;33m=\u001b[0m\u001b[0mmodel_output\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[0mtitle\u001b[0m\u001b[1;33m=\u001b[0m\u001b[1;34m\"Output message of the LLM:\"\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[0mlevel\u001b[0m\u001b[1;33m=\u001b[0m\u001b[0mLogLevel\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mDEBUG\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[1;31m# Parse\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[1;32mtry\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[0mcode_action\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mfix_final_answer_code\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mparse_code_blobs\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mmodel_output\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[1;32mexcept\u001b[0m \u001b[0mException\u001b[0m \u001b[1;32mas\u001b[0m \u001b[0me\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[0merror_msg\u001b[0m \u001b[1;33m=\u001b[0m \u001b[1;33mf\"\u001b[0m\u001b[1;33mError in code parsing:\\n\u001b[0m\u001b[1;33m{\u001b[0m\u001b[0me\u001b[0m\u001b[1;33m}\u001b[0m\u001b[1;33m\\nMake sure to provide correct code blobs.\u001b[0m\u001b[1;33m\"\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[1;32mraise\u001b[0m \u001b[0mAgentParsingError\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0merror_msg\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mlogger\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mmemory_step\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mtool_calls\u001b[0m \u001b[1;33m=\u001b[0m \u001b[1;33m[\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[0mToolCall\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                \u001b[0mname\u001b[0m\u001b[1;33m=\u001b[0m\u001b[1;34m\"python_interpreter\"\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                \u001b[0marguments\u001b[0m\u001b[1;33m=\u001b[0m\u001b[0mcode_action\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                \u001b[0mid\u001b[0m\u001b[1;33m=\u001b[0m\u001b[1;33mf\"\u001b[0m\u001b[1;33mcall_\u001b[0m\u001b[1;33m{\u001b[0m\u001b[0mlen\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mmemory\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0msteps\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m}\u001b[0m\u001b[1;33m\"\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[1;33m]\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[1;31m# Execute\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mlogger\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mlog_code\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mtitle\u001b[0m\u001b[1;33m=\u001b[0m\u001b[1;34m\"Executing parsed code:\"\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mcontent\u001b[0m\u001b[1;33m=\u001b[0m\u001b[0mcode_action\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mlevel\u001b[0m\u001b[1;33m=\u001b[0m\u001b[0mLogLevel\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mINFO\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mis_final_answer\u001b[0m \u001b[1;33m=\u001b[0m \u001b[1;32mFalse\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[1;32mtry\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[0moutput\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mexecution_logs\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mis_final_answer\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mpython_executor\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mcode_action\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[0mexecution_outputs_console\u001b[0m \u001b[1;33m=\u001b[0m \u001b[1;33m[\u001b[0m\u001b[1;33m]\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[1;32mif\u001b[0m \u001b[0mlen\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mexecution_logs\u001b[0m\u001b[1;33m)\u001b[0m \u001b[1;33m>\u001b[0m \u001b[1;36m0\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                \u001b[0mexecution_outputs_console\u001b[0m \u001b[1;33m+=\u001b[0m \u001b[1;33m[\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                    \u001b[0mText\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;34m\"Execution logs:\"\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mstyle\u001b[0m\u001b[1;33m=\u001b[0m\u001b[1;34m\"bold\"\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                    \u001b[0mText\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mexecution_logs\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                \u001b[1;33m]\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[0mobservation\u001b[0m \u001b[1;33m=\u001b[0m \u001b[1;34m\"Execution logs:\\n\"\u001b[0m \u001b[1;33m+\u001b[0m \u001b[0mexecution_logs\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[1;32mexcept\u001b[0m \u001b[0mException\u001b[0m \u001b[1;32mas\u001b[0m \u001b[0me\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[1;32mif\u001b[0m \u001b[0mhasattr\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mpython_executor\u001b[0m\u001b[1;33m,\u001b[0m \u001b[1;34m\"state\"\u001b[0m\u001b[1;33m)\u001b[0m \u001b[1;32mand\u001b[0m \u001b[1;34m\"_print_outputs\"\u001b[0m \u001b[1;32min\u001b[0m \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mpython_executor\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mstate\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                \u001b[0mexecution_logs\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mstr\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mpython_executor\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mstate\u001b[0m\u001b[1;33m[\u001b[0m\u001b[1;34m\"_print_outputs\"\u001b[0m\u001b[1;33m]\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                \u001b[1;32mif\u001b[0m \u001b[0mlen\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mexecution_logs\u001b[0m\u001b[1;33m)\u001b[0m \u001b[1;33m>\u001b[0m \u001b[1;36m0\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                    \u001b[0mexecution_outputs_console\u001b[0m \u001b[1;33m=\u001b[0m \u001b[1;33m[\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                        \u001b[0mText\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;34m\"Execution logs:\"\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mstyle\u001b[0m\u001b[1;33m=\u001b[0m\u001b[1;34m\"bold\"\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                        \u001b[0mText\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mexecution_logs\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                    \u001b[1;33m]\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                    \u001b[0mmemory_step\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mobservations\u001b[0m \u001b[1;33m=\u001b[0m \u001b[1;34m\"Execution logs:\\n\"\u001b[0m \u001b[1;33m+\u001b[0m \u001b[0mexecution_logs\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                    \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mlogger\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mlog\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mGroup\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33m*\u001b[0m\u001b[0mexecution_outputs_console\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mlevel\u001b[0m\u001b[1;33m=\u001b[0m\u001b[0mLogLevel\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mINFO\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[0merror_msg\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mstr\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0me\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[1;32mif\u001b[0m \u001b[1;34m\"Import of \"\u001b[0m \u001b[1;32min\u001b[0m \u001b[0merror_msg\u001b[0m \u001b[1;32mand\u001b[0m \u001b[1;34m\" is not allowed\"\u001b[0m \u001b[1;32min\u001b[0m \u001b[0merror_msg\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mlogger\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mlog\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                    \u001b[1;34m\"[bold red]Warning to user: Code execution failed due to an unauthorized import - Consider passing said import under `additional_authorized_imports` when initializing your CodeAgent.\"\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                    \u001b[0mlevel\u001b[0m\u001b[1;33m=\u001b[0m\u001b[0mLogLevel\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mINFO\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                \u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[1;32mraise\u001b[0m \u001b[0mAgentExecutionError\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0merror_msg\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mlogger\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mtruncated_output\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mtruncate_content\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mstr\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0moutput\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mobservation\u001b[0m \u001b[1;33m+=\u001b[0m \u001b[1;34m\"Last output from code snippet:\\n\"\u001b[0m \u001b[1;33m+\u001b[0m \u001b[0mtruncated_output\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mmemory_step\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mobservations\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mobservation\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mexecution_outputs_console\u001b[0m \u001b[1;33m+=\u001b[0m \u001b[1;33m[\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[0mText\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                \u001b[1;33mf\"\u001b[0m\u001b[1;33m{\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;34m'Out - Final answer'\u001b[0m \u001b[1;32mif\u001b[0m \u001b[0mis_final_answer\u001b[0m \u001b[1;32melse\u001b[0m \u001b[1;34m'Out'\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m}\u001b[0m\u001b[1;33m: \u001b[0m\u001b[1;33m{\u001b[0m\u001b[0mtruncated_output\u001b[0m\u001b[1;33m}\u001b[0m\u001b[1;33m\"\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m                \u001b[0mstyle\u001b[0m\u001b[1;33m=\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33mf\"\u001b[0m\u001b[1;33mbold \u001b[0m\u001b[1;33m{\u001b[0m\u001b[0mYELLOW_HEX\u001b[0m\u001b[1;33m}\u001b[0m\u001b[1;33m\"\u001b[0m \u001b[1;32mif\u001b[0m \u001b[0mis_final_answer\u001b[0m \u001b[1;32melse\u001b[0m \u001b[1;34m\"\"\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[1;33m)\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[1;33m]\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mlogger\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mlog\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mGroup\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33m*\u001b[0m\u001b[0mexecution_outputs_console\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mlevel\u001b[0m\u001b[1;33m=\u001b[0m\u001b[0mLogLevel\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mINFO\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mmemory_step\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0maction_output\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0moutput\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[1;32mreturn\u001b[0m \u001b[0moutput\u001b[0m \u001b[1;32mif\u001b[0m \u001b[0mis_final_answer\u001b[0m \u001b[1;32melse\u001b[0m \u001b[1;32mNone\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m    \u001b[1;32mdef\u001b[0m \u001b[0mto_dict\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mself\u001b[0m\u001b[1;33m)\u001b[0m \u001b[1;33m->\u001b[0m \u001b[0mdict\u001b[0m\u001b[1;33m[\u001b[0m\u001b[0mstr\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mAny\u001b[0m\u001b[1;33m]\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[1;34m\"\"\"Convert the agent to a dictionary representation.\n",
+      "\n",
+      "        Returns:\n",
+      "            `dict`: Dictionary representation of the agent.\n",
+      "        \"\"\"\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0magent_dict\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0msuper\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mto_dict\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0magent_dict\u001b[0m\u001b[1;33m[\u001b[0m\u001b[1;34m\"authorized_imports\"\u001b[0m\u001b[1;33m]\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mauthorized_imports\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0magent_dict\u001b[0m\u001b[1;33m[\u001b[0m\u001b[1;34m\"executor_type\"\u001b[0m\u001b[1;33m]\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mexecutor_type\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0magent_dict\u001b[0m\u001b[1;33m[\u001b[0m\u001b[1;34m\"executor_kwargs\"\u001b[0m\u001b[1;33m]\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mexecutor_kwargs\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0magent_dict\u001b[0m\u001b[1;33m[\u001b[0m\u001b[1;34m\"max_print_outputs_length\"\u001b[0m\u001b[1;33m]\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mself\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mmax_print_outputs_length\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[1;32mreturn\u001b[0m \u001b[0magent_dict\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m    \u001b[1;33m@\u001b[0m\u001b[0mclassmethod\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m    \u001b[1;32mdef\u001b[0m \u001b[0mfrom_dict\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mcls\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0magent_dict\u001b[0m\u001b[1;33m:\u001b[0m \u001b[0mdict\u001b[0m\u001b[1;33m[\u001b[0m\u001b[0mstr\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mAny\u001b[0m\u001b[1;33m]\u001b[0m\u001b[1;33m,\u001b[0m \u001b[1;33m**\u001b[0m\u001b[0mkwargs\u001b[0m\u001b[1;33m)\u001b[0m \u001b[1;33m->\u001b[0m \u001b[1;34m\"CodeAgent\"\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[1;34m\"\"\"Create CodeAgent from a dictionary representation.\n",
+      "\n",
+      "        Args:\n",
+      "            agent_dict (`dict[str, Any]`): Dictionary representation of the agent.\n",
+      "            **kwargs: Additional keyword arguments that will override agent_dict values.\n",
+      "\n",
+      "        Returns:\n",
+      "            `CodeAgent`: Instance of the CodeAgent class.\n",
+      "        \"\"\"\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[1;31m# Add CodeAgent-specific parameters to kwargs\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mcode_agent_kwargs\u001b[0m \u001b[1;33m=\u001b[0m \u001b[1;33m{\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[1;34m\"additional_authorized_imports\"\u001b[0m\u001b[1;33m:\u001b[0m \u001b[0magent_dict\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mget\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;34m\"authorized_imports\"\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[1;34m\"executor_type\"\u001b[0m\u001b[1;33m:\u001b[0m \u001b[0magent_dict\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mget\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;34m\"executor_type\"\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[1;34m\"executor_kwargs\"\u001b[0m\u001b[1;33m:\u001b[0m \u001b[0magent_dict\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mget\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;34m\"executor_kwargs\"\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m            \u001b[1;34m\"max_print_outputs_length\"\u001b[0m\u001b[1;33m:\u001b[0m \u001b[0magent_dict\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mget\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;34m\"max_print_outputs_length\"\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[1;33m}\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[1;31m# Filter out None values\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mcode_agent_kwargs\u001b[0m \u001b[1;33m=\u001b[0m \u001b[1;33m{\u001b[0m\u001b[0mk\u001b[0m\u001b[1;33m:\u001b[0m \u001b[0mv\u001b[0m \u001b[1;32mfor\u001b[0m \u001b[0mk\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mv\u001b[0m \u001b[1;32min\u001b[0m \u001b[0mcode_agent_kwargs\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mitems\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33m)\u001b[0m \u001b[1;32mif\u001b[0m \u001b[0mv\u001b[0m \u001b[1;32mis\u001b[0m \u001b[1;32mnot\u001b[0m \u001b[1;32mNone\u001b[0m\u001b[1;33m}\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[1;31m# Update with any additional kwargs\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[0mcode_agent_kwargs\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mupdate\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mkwargs\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[1;31m# Call the parent class's from_dict method\u001b[0m\u001b[1;33m\n",
+      "\u001b[0m        \u001b[1;32mreturn\u001b[0m \u001b[0msuper\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mfrom_dict\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0magent_dict\u001b[0m\u001b[1;33m,\u001b[0m \u001b[1;33m**\u001b[0m\u001b[0mcode_agent_kwargs\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n",
+      "\u001b[1;31mFile:\u001b[0m           c:\\users\\pulkitme\\appdata\\local\\miniconda3\\lib\\site-packages\\smolagents\\agents.py\n",
+      "\u001b[1;31mType:\u001b[0m           ABCMeta\n",
+      "\u001b[1;31mSubclasses:\u001b[0m     "
+     ]
+    }
+   ],
+   "source": [
+    "CodeAgent??"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,