polishing

microsoft · Jan 7, 2024 · 16c79a1 · 16c79a1
1 parent 9119d98
commit 16c79a1
Show file tree

Hide file tree

Showing 6 changed files with 26 additions and 17 deletions.
diff --git a/notebook/agentchat_MathChat.ipynb b/notebook/agentchat_MathChat.ipynb
@@ -52,7 +52,6 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# The wolfram alpha app id is required for this example (the assistant may choose to query Wolfram Alpha).\n",
     "import os\n",
     "\n",
     "import autogen\n",
@@ -294,6 +293,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "# The wolfram alpha app id is required for this example (the assistant may choose to query Wolfram Alpha).\n",
     "if \"WOLFRAM_ALPHA_APPID\" not in os.environ:\n",
     "    os.environ[\"WOLFRAM_ALPHA_APPID\"] = open(\"wolfram.txt\").read().strip()\n",
     "\n",

diff --git a/notebook/agentchat_function_call.ipynb b/notebook/agentchat_function_call.ipynb
@@ -57,7 +57,6 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# define functions according to the function description\n",
     "from IPython import get_ipython\n",
     "from typing_extensions import Annotated\n",
     "\n",
@@ -216,6 +215,9 @@
     ")\n",
     "\n",
     "\n",
+    "# define functions according to the function description\n",
+    "\n",
+    "\n",
     "@user_proxy.register_for_execution()\n",
     "@chatbot.register_for_llm(name=\"python\", description=\"run cell in ipython and return the execution result.\")\n",
     "def exec_python(cell: Annotated[str, \"Valid Python cell to execute.\"]) -> str:\n",

diff --git a/notebook/agentchat_function_call_async.ipynb b/notebook/agentchat_function_call_async.ipynb
@@ -66,7 +66,6 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# define functions according to the function description\n",
     "import time\n",
     "\n",
     "from typing_extensions import Annotated\n",
@@ -196,6 +195,7 @@
     "    code_execution_config={\"work_dir\": \"coding\"},\n",
     ")\n",
     "\n",
+    "# define functions according to the function description\n",
     "\n",
     "# An example async function\n",
     "\n",

diff --git a/notebook/agentchat_graph_modelling_language_using_select_speaker.ipynb b/notebook/agentchat_graph_modelling_language_using_select_speaker.ipynb
@@ -33,26 +33,32 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
     "%%capture --no-stderr\n",
     "# %pip install \"pyautogen>=0.2.3\"\n",
-    "import random\n",
-    "from typing import Dict, List\n",
-    "\n",
-    "import matplotlib.pyplot as plt\n",
-    "import networkx as nx\n",
-    "\n",
-    "import autogen\n",
-    "from autogen.agentchat.assistant_agent import AssistantAgent\n",
-    "from autogen.agentchat.groupchat import GroupChat\n",
-    "\n",
     "%pip install networkX~=3.2.1\n",
     "%pip install matplotlib~=3.8.1"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import random  # noqa E402\n",
+    "\n",
+    "import matplotlib.pyplot as plt  # noqa E402\n",
+    "import networkx as nx  # noqa E402\n",
+    "\n",
+    "import autogen  # noqa E402\n",
+    "from autogen.agentchat.assistant_agent import AssistantAgent  # noqa E402\n",
+    "from autogen.agentchat.groupchat import GroupChat  # noqa E402"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 3,

diff --git a/notebook/agentchat_microsoft_fabric.ipynb b/notebook/agentchat_microsoft_fabric.ipynb
@@ -176,11 +176,11 @@
     }
    ],
    "source": [
-    "# Choose different models\n",
     "from synapse.ml.mlflow import get_mlflow_env_config\n",
     "\n",
     "import autogen\n",
     "\n",
+    "# Choose different models\n",
     "config_list = [\n",
     "    {\n",
     "        \"model\": \"gpt-4-turbo\",\n",

diff --git a/notebook/agenteval_cq_math.ipynb b/notebook/agenteval_cq_math.ipynb
@@ -83,8 +83,6 @@
    "source": [
     "import json\n",
     "\n",
-    "# You can set your own log path - we also limited the number of samples to avoid additional costs.\n",
-    "# By removing the condition about limitations on the number of samples per category, you can run it on all 120 problems\n",
     "import os\n",
     "from pathlib import Path\n",
     "\n",
@@ -744,6 +742,9 @@
     }
    ],
    "source": [
+    "# You can set your own log path - we also limited the number of samples to avoid additional costs.\n",
+    "# By removing the condition about limitations on the number of samples per category, you can run it on all 120 problems\n",
+    "\n",
     "log_path = \"../test/test_files/agenteval-in-out/agentchat_results/\"\n",
     "\n",
     "# The file is no longer in the repo, we can download it from an older commit\n",