Dba 738 index optimization by llm (#64)

* Initial implementation of optimization by LLM

Author: jssmith

pyproject.toml

@@ -11,6 +11,7 @@ dependencies = [
     "pglast==7.2.0",
     "attrs>=25.3.0",
     "psycopg-pool>=3.2.6",
+    "instructor>=1.7.9",
 ]
 license = "mit"
 license-files = ["LICENSE"]

src/postgres_mcp/dta/__init__.py

@@ -9,7 +9,7 @@
 
 from ..artifacts import ErrorResult
 from ..artifacts import ExplainPlanArtifact
-from ..sql import IndexConfig
+from ..sql import IndexDefinition
 from ..sql import SafeSqlDriver
 from ..sql import SqlBindParams
 from ..sql import check_postgres_version_requirement
@@ -114,7 +114,7 @@ async def explain_with_hypothetical_indexes(
 
             # Convert the index definitions to IndexConfig objects
             indexes = frozenset(
-                IndexConfig(
+                IndexDefinition(
                     table=idx["table"],
                     columns=tuple(idx["columns"]),
                     using=idx.get("using", "btree"),
@@ -185,7 +185,7 @@ async def _run_explain_query(self, query: str, analyze: bool = False, generic_pl
     async def generate_explain_plan_with_hypothetical_indexes(
         self,
         query_text: str,
-        indexes: frozenset[IndexConfig],
+        indexes: frozenset[IndexDefinition],
         use_generic_plan: bool = False,
         dta=None,
     ) -> dict[str, Any]: