Update links

Author: BenConstable9

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

@@ -10,10 +10,12 @@ dependencies = [
     "autogen-ext[azure,openai]==0.4.0.dev7",
     "grpcio>=1.68.1",
     "pyyaml>=6.0.2",
+    "text-2-sql-core",
 ]
 
 [dependency-groups]
 dev = [
+    "autogen-text-2-sql",
     "black>=24.10.0",
     "ipykernel>=6.29.5",
     "jupyter>=1.1.1",
@@ -24,4 +26,5 @@ dev = [
 ]
 
 [tool.uv.sources]
-text_2_sql_core = { path = "text_2_sql/text_2_sql_core" }
+autogen-text-2-sql = { workspace = true }
+text-2-sql-core = { workspace = true }

text_2_sql/autogen/pyproject.toml

@@ -2,10 +2,10 @@
 # Licensed under the MIT License.
 from autogen_agentchat.task import TextMentionTermination, MaxMessageTermination
 from autogen_agentchat.teams import SelectorGroupChat
-from creators.llm_model_creator import LLMModelCreator
-from creators.llm_agent_creator import LLMAgentCreator
+from autogen_text_2_sql.creators.llm_model_creator import LLMModelCreator
+from autogen_text_2_sql.creators.llm_agent_creator import LLMAgentCreator
 import logging
-from custom_agents.sql_query_cache_agent import SqlQueryCacheAgent
+from autogen_text_2_sql.custom_agents.sql_query_cache_agent import SqlQueryCacheAgent
 import json
 import os
 

text_2_sql/autogen/__init__.py …togen/src/autogen_text_2_sql/__init__.pytext_2_sql/autogen/__init__.py renamed to text_2_sql/autogen/src/autogen_text_2_sql/__init__.py text_2_sql/autogen/__init__.py renamed to text_2_sql/autogen/src/autogen_text_2_sql/__init__.py

@@ -3,7 +3,7 @@
 import yaml
 from autogen_core.components.tools import FunctionTool
 from autogen_agentchat.agents import AssistantAgent
-from ..text_2_sql_core.connectors.sql import SqlConnector
+from text_2_sql_core.connectors.sql import SqlConnector
 from llm_model_creator import LLMModelCreator
 from jinja2 import Template
 from datetime import datetime

…utogen/src/autogen/autogen_text_2_sql.py …autogen_text_2_sql/autogen_text_2_sql.pytext_2_sql/autogen/src/autogen/autogen_text_2_sql.py renamed to text_2_sql/autogen/src/autogen_text_2_sql/autogen_text_2_sql.py text_2_sql/autogen/src/autogen/autogen_text_2_sql.py renamed to text_2_sql/autogen/src/autogen_text_2_sql/autogen_text_2_sql.py

@@ -1,7 +1,7 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
 from autogen_ext.models import AzureOpenAIChatCompletionClient
-from ..text_2_sql_core.utils.environment import IdentityType, get_identity_type
+from text_2_sql_core.utils.environment import IdentityType, get_identity_type
 
 from azure.identity import DefaultAzureCredential, get_bearer_token_provider
 import os

…rc/autogen/creators/llm_agent_creator.py …text_2_sql/creators/llm_agent_creator.pytext_2_sql/autogen/src/autogen/creators/llm_agent_creator.py renamed to text_2_sql/autogen/src/autogen_text_2_sql/creators/llm_agent_creator.py text_2_sql/autogen/src/autogen/creators/llm_agent_creator.py renamed to text_2_sql/autogen/src/autogen_text_2_sql/creators/llm_agent_creator.py

@@ -17,7 +17,29 @@
         "\n",
         "This notebook demonstrates how the SQL plugin can be integrated with Semantic Kernel and Azure OpenAI to answer questions from the database based on the schemas provided. \n",
         "\n",
-        "A multi-shot approach is used for SQL generation for more reliable results and reduced token usage. More details can be found in the README.md."
+        "A multi-shot approach is used for SQL generation for more reliable results and reduced token usage. More details can be found in the README.md.\n",
+        "\n",
+        "### Dependencies\n",
+        "\n",
+        "To install dependencies:\n",
+        "\n",
+        "`uv sync --package semantic_kernel_text_2_sql`\n",
+        "\n",
+        "`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\"))"
       ]
     },
     {

…rc/autogen/creators/llm_model_creator.py …text_2_sql/creators/llm_model_creator.pytext_2_sql/autogen/src/autogen/creators/llm_model_creator.py renamed to text_2_sql/autogen/src/autogen_text_2_sql/creators/llm_model_creator.py text_2_sql/autogen/src/autogen/creators/llm_model_creator.py renamed to text_2_sql/autogen/src/autogen_text_2_sql/creators/llm_model_creator.py

@@ -17,7 +17,29 @@
         "\n",
         "This notebook demonstrates how the SQL plugin can be integrated with Semantic Kernel and Azure OpenAI to answer questions from the database based on the schemas provided. \n",
         "\n",
-        "A multi-shot approach is used for SQL generation for more reliable results and reduced token usage. More details can be found in the README.md."
+        "A multi-shot approach is used for SQL generation for more reliable results and reduced token usage. More details can be found in the README.md.\n",
+        "\n",
+        "### Dependencies\n",
+        "\n",
+        "To install dependencies:\n",
+        "\n",
+        "`uv sync --package semantic_kernel_text_2_sql`\n",
+        "\n",
+        "`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\"))"
       ]
     },
     {