Update tools/sql_tool.py

tools/sql_tool.py

@@ -63,10 +63,10 @@ def run_duckdb_query(query: str) -> str:
             conn.close()
 
 @tool
-def get_table_schema(table_name: str = "positions") -> str:
+def get_table_schema(table_name: str = "my_db.main.masterdataset_v") -> str:
     """
     Returns the schema (column names and data types) for the specified table in the MotherDuck database.
-    Defaults to the 'positions' table.
+    Defaults to the 'my_db.main.masterdataset_v' table.
     """
     conn = None
     try:
@@ -78,7 +78,9 @@ def get_table_schema(table_name: str = "positions") -> str:
         schema_df = conn.execute(query).fetchdf()
         
         if schema_df.empty:
-            return f"Error: Table '{table_name}' not found in the MotherDuck database. Available tables: {conn.execute('SHOW TABLES;').fetchnames()}"
+            # Also fetch available table names for better error reporting
+            available_tables = conn.execute('SHOW TABLES;').fetchnames()
+            return f"Error: Table '{table_name}' not found. Available tables: {available_tables}"
 
         # Format the schema into a simple string: name TYPE, name TYPE, ...
         schema_parts = [f"{row['name']} {row['type']}" for index, row in schema_df.iterrows()]
@@ -90,4 +92,4 @@ def get_table_schema(table_name: str = "positions") -> str:
         return f"DuckDB Schema Error: {e}"
     finally:
         if conn:
-            conn.close()
+            conn.close()