updated notebook

notebook.ipynb

@@ -33,12 +33,24 @@
    "outputs": [],
    "source": [
     "from smolagents import CodeAgent, Tool, InferenceClientModel\n",
+    "\n",
+    "#print(dir(smolagents.models))\n",
+    "#print(smolagents.__version__)\n",
+    "\n",
+    "\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/Qwen2.5-Coder-7B-Instruct\", provider=\"nscale\")\n",
+    "#model = InferenceClientModel()\n",
     "\n",
-    "from app import BasicAgent\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",
+    "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",
     "\n",
     "agent = CodeAgent(tools=[fetch_file], model=model, additional_authorized_imports=['requests', 'bs4'])\n",
     "\n",
@@ -68,10 +80,13 @@
     "  }\n",
     "\n",
     "question_text = item[\"question\"]\n",
-    "print(f\"Question: {question_text}\")\n",
     "\n",
-    "task_id = item[\"task_id\"]\n",
-    "submitted_answer = agent(question_text)\n"
+    "question = f\"{system_promt} {question_text}\"\n",
+    "\n",
+    "print(f\"Question: {question}\")\n",
+    "\n",
+    "#task_id = item[\"task_id\"]\n",
+    "submitted_answer = agent(question)\n"
    ]
   }
  ],