extends notebook

notebook.ipynb

@@ -32,27 +32,40 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from smolagents import CodeAgent, Tool, InferenceClientModel\n",
-    "\n",
-    "#print(dir(smolagents.models))\n",
-    "#print(smolagents.__version__)\n",
-    "\n",
+    "import yaml\n",
     "\n",
+    "from smolagents import CodeAgent, Tool, InferenceClientModel, DuckDuckGoSearchTool, WikipediaSearchTool\n",
+    "from smolagents import OpenAIServerModel\n",
     "from tools import fetch_file\n",
     "\n",
-    "model = InferenceClientModel(model_id=\"Qwen/Qwen2.5-Coder-7B-Instruct\", provider=\"nscale\")\n",
-    "#model = InferenceClientModel()\n",
+    "#model = InferenceClientModel(model_id=\"Qwen/Qwen3-32B\", provider=\"nscale\")\n",
+    "model = InferenceClientModel(model_id=\"deepseek-ai/DeepSeek-R1-Distill-Llama-70B\", provider=\"nscale\")\n",
+    "\n",
+    "# Set your Gemini API key in the environment variable GEMINI_API_KEY_1\n",
+    "#model = OpenAIServerModel(\n",
+    "#    model_id=\"gemini-2.5-flash\", \n",
+    "#    api_base=\"https://generativelanguage.googleapis.com/v1beta\", \n",
+    "#    api_key=os.getenv(\"GEMINI_API_KEY_1\")\n",
+    "#    ) \n",
+    "#print('API key was found:', os.getenv(\"GEMINI_API_KEY_1\") is not None)\n",
     "\n",
     "#from app import BasicAgent\n",
     "\n",
-    "system_promt = \"\"\"You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template: \n",
+    "system_prompt = \"\"\"You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template: \n",
     "FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.\n",
     "If you are asked for a number, don’t use comma to write your number neither use units such as $ or percent sign unless specified otherwise.\n",
     "If you are asked for a string, don’t use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise.\n",
     "If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string. \n",
+    "\n",
+    "Here is the question: {question}\n",
     "\"\"\"\n",
     "\n",
-    "agent = CodeAgent(tools=[fetch_file], model=model, additional_authorized_imports=['requests', 'bs4'])\n",
+    "agent = CodeAgent(\n",
+    "    model=model, \n",
+    "    tools=[fetch_file, DuckDuckGoSearchTool(), WikipediaSearchTool()], \n",
+    "    add_base_tools=True,\n",
+    "    additional_authorized_imports=['requests', 'bs4'] \n",
+    "    )\n",
     "\n",
     "\n"
    ]
@@ -80,13 +93,12 @@
     "  }\n",
     "\n",
     "question_text = item[\"question\"]\n",
+    "prompt_with_question = system_prompt.format(question=question_text)\n",
     "\n",
-    "question = f\"{system_promt} {question_text}\"\n",
-    "\n",
-    "print(f\"Question: {question}\")\n",
+    "print(f\"Question: {prompt_with_question}\")\n",
     "\n",
     "#task_id = item[\"task_id\"]\n",
-    "submitted_answer = agent(question)\n"
+    "submitted_answer = agent(prompt_with_question)\n"
    ]
   }
  ],

prompts.yaml

@@ -0,0 +1,19 @@
+"system_prompt": |-
+    You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template: 
+    FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
+    If you are asked for a number, don’t use comma to write your number neither use units such as $ or percent sign unless specified otherwise.
+    If you are asked for a string, don’t use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise.
+    If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string. 
+"planning" :   
+  "initial_facts": ""
+  "initial_plan":  ""
+  "update_facts_pre_messages":  ""
+  "update_facts_post_messages":  ""
+  "update_plan_pre_messages":  ""
+  "update_plan_post_messages":  ""
+"managed_agent":
+  "task":  ""
+  "report":  ""
+"final_answer":
+  "pre_messages": ""
+  "post_messages": ""