More changes to packaging

Author: BenConstable9

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

@@ -29,7 +29,7 @@
         "import dotenv\n",
         "import logging\n",
         "from autogen_agentchat.task import Console\n",
-        "from agentic_text_2_sql import AgenticText2Sql"
+        "from autogen_text_2_sql import AutoGenText2Sql"
       ]
     },
     {
@@ -63,7 +63,7 @@
       "metadata": {},
       "outputs": [],
       "source": [
-        "agentic_text_2_sql = AgenticText2Sql(target_engine=\"TSQL\", engine_specific_rules=\"Use TOP X to limit the number of rows returned instead of LIMIT X. NEVER USE LIMIT X as it produces a syntax error.\").agentic_flow"
+        "agentic_text_2_sql = AutoGenText2Sql(target_engine=\"TSQL\", engine_specific_rules=\"Use TOP X to limit the number of rows returned instead of LIMIT X. NEVER USE LIMIT X as it produces a syntax error.\").agentic_flow"
       ]
     },
     {

text_2_sql/autogen/src/autogen/autogen_text_2_sql.py

@@ -2,15 +2,15 @@
 # Licensed under the MIT License.
 from autogen_agentchat.task import TextMentionTermination, MaxMessageTermination
 from autogen_agentchat.teams import SelectorGroupChat
-from llm_model_creator import LLMModelCreator
-from llm_agent_creator import LLMAgentCreator
+from creators.llm_model_creator import LLMModelCreator
+from creators.llm_agent_creator import LLMAgentCreator
 import logging
 from custom_agents.sql_query_cache_agent import SqlQueryCacheAgent
 import json
 import os
 
 
-class AgenticText2Sql:
+class AutoGenText2Sql:
     def __init__(self, target_engine: str, engine_specific_rules: str):
         self.use_query_cache = False
         self.pre_run_query_cache = False
@@ -135,7 +135,7 @@ def agentic_flow(self):
             allow_repeated_speaker=False,
             model_client=LLMModelCreator.get_model("4o-mini"),
             termination_condition=self.termination_condition,
-            selector_func=AgenticText2Sql.selector,
+            selector_func=AutoGenText2Sql.selector,
         )
 
         return agentic_flow