Parallel Agent Execution

text_2_sql/autogen/Iteration 5 - Agentic Vector Based Text2SQL.ipynb

@@ -9,6 +9,20 @@
         "Licensed under the MIT License."
       ]
     },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "# This is only needed for this notebook to work\n",
+        "import sys\n",
+        "from pathlib import Path\n",
+        "\n",
+        "# Add the parent directory of `src` to the path\n",
+        "sys.path.append(str(Path.cwd() / \"src\"))"
+      ]
+    },
     {
       "cell_type": "markdown",
       "metadata": {},
@@ -28,20 +42,6 @@
         "`uv add --editable text_2_sql_core`"
       ]
     },
-    {
-      "cell_type": "code",
-      "execution_count": null,
-      "metadata": {},
-      "outputs": [],
-      "source": [
-        "# This is only needed for this notebook to work\n",
-        "import sys\n",
-        "from pathlib import Path\n",
-        "\n",
-        "# Add the parent directory of `src` to the path\n",
-        "sys.path.append(str(Path.cwd() / \"src\"))"
-      ]
-    },
     {
       "cell_type": "code",
       "execution_count": null,
@@ -50,8 +50,7 @@
       "source": [
         "import dotenv\n",
         "import logging\n",
-        "from autogen_agentchat.ui import Console\n",
-        "from autogen_text_2_sql.autogen_text_2_sql import AutoGenText2Sql"
+        "from autogen_text_2_sql import AutoGenText2Sql"
       ]
     },
     {
@@ -101,8 +100,8 @@
       "metadata": {},
       "outputs": [],
       "source": [
-        "result = await agentic_text_2_sql.process_question(task=\"What total number of orders in June 2008?\")\n",
-        "await Console(result)\n"
+        "async for message in agentic_text_2_sql.process_question(question=\"What total number of orders in June 2008?\"):\n",
+        "    logging.info(\"Received %s Message from Text2SQL System\", message)"
       ]
     },
     {

text_2_sql/autogen/pyproject.toml

@@ -5,6 +5,7 @@ description = "AutoGen Based Implementation"
 readme = "README.md"
 requires-python = ">=3.12"
 dependencies = [
+    "aiostream>=0.6.4",
     "autogen-agentchat==0.4.0.dev11",
     "autogen-core==0.4.0.dev11",
     "autogen-ext[azure,openai]==0.4.0.dev11",

text_2_sql/autogen/src/autogen_text_2_sql/__init__.py

@@ -0,0 +1,3 @@
+from autogen_text_2_sql.autogen_text_2_sql import AutoGenText2Sql
+
+__all__ = ["AutoGenText2Sql"]