dbagent-cli 0.1.4__tar.gz → 0.1.6__tar.gz

{dbagent_cli-0.1.4 → dbagent_cli-0.1.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dbagent-cli
-Version: 0.1.4
+Version: 0.1.6
 Summary: Universal Database Introspection and Script Generation AI Agent (CLI)
 Author: Santhosh Gupta
 Requires-Python: >=3.9

{dbagent_cli-0.1.4 → dbagent_cli-0.1.6}/dbagent/__init__.py

@@ -2,5 +2,5 @@
 DB-Agent: Universal Database Introspection and Script Generation AI Agent (CLI).
 """
 
-__version__ = "0.1.4"
+__version__ = "0.1.6"
 __author__ = "Santhosh Gupta"

{dbagent_cli-0.1.4 → dbagent_cli-0.1.6}/dbagent/agent/generator.py

@@ -51,9 +51,15 @@ Generate ONLY the raw executable SQL query for the given database. No markdown f
 2. Use the exact table and column names from the schema above.
 3. Use dialect-specific syntax for `{dialect_name}`.
 4. Write clean, optimized SQL with uppercase keywords.
-5. If the user asks for recent/time-based data, use the dialect's native datetime functions.
-6. Always include appropriate WHERE clauses, JOINs, and ORDER BY as needed.
-7. Default to LIMIT 50 unless the user specifies a count.
+5. If the user asks for recent/time-based data, use native datetime functions (e.g. for PostgreSQL: `NOW() - INTERVAL '1 hour'`, for SQLite: `datetime('now', '-1 hour')`, for MySQL: `NOW() - INTERVAL 1 HOUR`).
+6. For queries inspecting PostgreSQL foreign key mappings or table relationships via system catalog, use standard joins:
+   SELECT tc.table_name, kcu.column_name, ccu.table_name AS foreign_table_name, ccu.column_name AS foreign_column_name
+   FROM information_schema.table_constraints tc
+   JOIN information_schema.key_column_usage kcu ON tc.constraint_name = kcu.constraint_name
+   JOIN information_schema.constraint_column_usage ccu ON tc.constraint_name = ccu.constraint_name
+   WHERE tc.constraint_type = 'FOREIGN KEY';
+7. Always include appropriate WHERE clauses, JOINs, and ORDER BY as needed.
+8. Default to LIMIT 50 unless the user specifies a count.
 """
 
 

{dbagent_cli-0.1.4 → dbagent_cli-0.1.6}/dbagent/cli.py

@@ -509,7 +509,7 @@ def chat_command(
     console.print(f"[bold green]Connected:[/bold green] {db_name} ({dialect}) - {len(table_names)} tables")
     console.print(f"[bold cyan]AI:[/bold cyan] {llm.name}")
     console.print("[dim]Type your question in plain English. Read queries auto-execute. Write queries ask for confirmation.[/dim]")
-    console.print("[dim]Commands: [bold]:tables[/bold] | [bold]:table <name>[/bold] | [bold]:explain[/bold] | [bold]:history[/bold] | [bold]:auto[/bold]/[bold]:noauto[/bold] | [bold]:run[/bold] | [bold]:export <file>[/bold] | [bold]:exit[/bold][/dim]\n")
+    console.print("[dim]Commands: [bold]:tables[/bold] | [bold]:table <name>[/bold] | [bold]:fk <name>[/bold] | [bold]:explain[/bold] | [bold]:history[/bold] | [bold]:auto[/bold]/[bold]:noauto[/bold] | [bold]:run[/bold] | [bold]:export <file>[/bold] | [bold]:exit[/bold][/dim]\n")
 
     history_file = config_mgr.config_dir / "chat_history.txt"
     session = PromptSession(history=FileHistory(str(history_file)))
@@ -537,15 +537,47 @@ def chat_command(
 
             if user_input.startswith(":table "):
                 tname = user_input.split(" ", 1)[1].strip()
-                with console.status(f"[bold cyan]Inspecting {tname}...[/bold cyan]"):
-                    on_demand_schema = connector.inspect_schema(table_names=[tname], include_samples=True)
-                tmodel = on_demand_schema.get_table(tname)
+                exact, fuzzy = connector.resolve_tables(tname)
+                target = exact[0] if exact else (fuzzy[0] if fuzzy else tname)
+                with console.status(f"[bold cyan]Inspecting {target}...[/bold cyan]"):
+                    on_demand_schema = connector.inspect_schema(table_names=[target], include_samples=True)
+                tmodel = on_demand_schema.get_table(target)
                 if tmodel:
                     SchemaViewer.display_table_detail(tmodel)
                 else:
                     print_error(f"Table '{tname}' not found.")
                 continue
 
+            if user_input.startswith(":relations ") or user_input.startswith(":fk ") or user_input.startswith(":mapped "):
+                tname = user_input.split(" ", 1)[1].strip()
+                exact, fuzzy = connector.resolve_tables(tname)
+                target = exact[0] if exact else (fuzzy[0] if fuzzy else tname)
+                with console.status(f"[bold cyan]Finding relationships for {target}...[/bold cyan]"):
+                    full_schema = connector.inspect_schema(include_samples=False, max_samples=0)
+                
+                # Find FKs where target references others, and others reference target
+                tmodel = full_schema.get_table(target)
+                outgoing = tmodel.foreign_keys if tmodel else []
+                incoming = []
+                for other in full_schema.tables:
+                    for fk in other.foreign_keys:
+                        if fk.referred_table.lower() == target.lower():
+                            incoming.append((other.name, fk))
+
+                console.print(f"\n[bold cyan]Relationships mapped to table: [bold white]{target}[/bold white][/bold cyan]\n")
+                if outgoing:
+                    console.print("[bold green]Outgoing (This table references):[/bold green]")
+                    for fk in outgoing:
+                        console.print(f"  * ({', '.join(fk.constrained_columns)}) -> [bold]{fk.referred_table}[/bold]({', '.join(fk.referred_columns)})")
+                if incoming:
+                    console.print("\n[bold yellow]Incoming (Other tables reference this table):[/bold yellow]")
+                    for src_table, fk in incoming:
+                        console.print(f"  * [bold]{src_table}[/bold]({', '.join(fk.constrained_columns)}) -> ({', '.join(fk.referred_columns)})")
+                if not outgoing and not incoming:
+                    console.print(f"[dim]No foreign key constraints found directly mapped to '{target}'.[/dim]")
+                console.print("")
+                continue
+
             if user_input.lower() == ":explain":
                 if last_result and last_result.sql:
                     print_code(last_result.sql, "sql", "Last Executed SQL")

{dbagent_cli-0.1.4 → dbagent_cli-0.1.6}/dbagent/connectors/relational.py

@@ -449,7 +449,10 @@ class RelationalConnector(BaseConnector):
         """Safely execute a query with row limit."""
         try:
             with self.engine.connect() as conn:
-                result = conn.execute(text(query))
+                try:
+                    result = conn.exec_driver_sql(query)
+                except Exception:
+                    result = conn.execute(text(query))
                 if not result.returns_rows:
                     # DML: commit and report affected rows
                     try:

{dbagent_cli-0.1.4 → dbagent_cli-0.1.6}/dbagent_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dbagent-cli
-Version: 0.1.4
+Version: 0.1.6
 Summary: Universal Database Introspection and Script Generation AI Agent (CLI)
 Author: Santhosh Gupta
 Requires-Python: >=3.9

{dbagent_cli-0.1.4 → dbagent_cli-0.1.6}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "dbagent-cli"
-version = "0.1.4"
+version = "0.1.6"
 description = "Universal Database Introspection and Script Generation AI Agent (CLI)"
 readme = "README.md"
 requires-python = ">=3.9"