spice-mcp 0.1.5__py3-none-any.whl → 0.1.7__py3-none-any.whl

spice_mcp/adapters/spellbook/explorer.py

@@ -349,13 +349,17 @@ class SpellbookExplorer(CatalogExplorer):
             return {}
 
     def _parse_sql_columns(self, sql_file: Path) -> list[TableColumn]:
-        """Parse SQL file to extract column names from SELECT statements."""
+        """
+        Parse SQL file to extract column names from SELECT statements.
+        
+        Note: This is a best-effort heuristic and may not be perfect for complex SQL.
+        For accurate column information, use Dune's DESCRIBE TABLE or query the actual table.
+        """
         try:
             with open(sql_file, encoding="utf-8") as f:
                 sql = f.read()
             
-            # Look for SELECT ... FROM patterns
-            # Match: SELECT col1, col2, col3 FROM ...
+            # Look for SELECT ... FROM patterns (simple heuristic)
             select_match = re.search(
                 r"SELECT\s+(.+?)\s+FROM",
                 sql,
@@ -364,27 +368,20 @@ class SpellbookExplorer(CatalogExplorer):
             
             if select_match:
                 cols_str = select_match.group(1)
-                # Split by comma, but handle function calls and aliases
+                # Simple split - may not handle all nested cases perfectly
+                # This is OK since column info is optional and best-effort
                 cols = []
                 for col in cols_str.split(","):
                     col = col.strip()
-                    # Extract column name (handle aliases: col AS alias -> col)
-                    if " AS " in col.upper():
-                        col = col.split(" AS ", 1)[0].strip()
-                    elif " " in col and not col.startswith("("):
-                        # Might be alias without AS
-                        parts = col.split()
-                        col = parts[0].strip()
-                    
-                    # Clean up function calls: function(col) -> col
-                    col = re.sub(r"^\w+\((.+)\)", r"\1", col)
+                    # Basic cleanup - remove obvious SQL noise
+                    col = col.split()[-1] if col else ""
                     col = col.strip().strip('"').strip("'")
                     
-                    if col and col not in ["*", "DISTINCT"]:
+                    if col and col not in ["*", "DISTINCT", "FROM"]:
                         cols.append(
                             TableColumn(
                                 name=col,
-                                dune_type="VARCHAR",  # Default, can't infer from SQL
+                                dune_type="VARCHAR",
                                 polars_dtype="Utf8",
                             )
                         )

spice_mcp/mcp/server.py

@@ -347,7 +347,7 @@ def _unified_discover_impl(
     schema: str | None = None,
     limit: int = 50,
     source: Literal["dune", "spellbook", "both"] = "both",
-    include_columns: bool = True,
+    include_columns: bool = False,
 ) -> dict[str, Any]:
     """
     Unified discovery implementation that can search Dune API, Spellbook repo, or both.
@@ -496,7 +496,7 @@ def dune_discover(
     schema: str | None = None,
     limit: int = 50,
     source: Literal["dune", "spellbook", "both"] = "both",
-    include_columns: bool = True,
+    include_columns: bool = False,
 ) -> dict[str, Any]:
     """
     PRIMARY discovery tool for finding tables in Dune.
@@ -520,7 +520,9 @@ def dune_discover(
         limit: Maximum number of tables to return
         source: Where to search - "dune" (Dune API only), "spellbook" (GitHub repo only),
                 or "both" (default: searches both and merges results)
-        include_columns: Whether to include column details for Spellbook models (default: True)
+        include_columns: Whether to include column details (default: False).
+                        Note: Column info from Spellbook SQL is unreliable.
+                        Use dune_describe_table on the actual Dune table for accurate columns.
     
     Returns:
         Dictionary with:
@@ -534,9 +536,10 @@ def dune_discover(
           - dune_alias: Actual Dune table alias (for Spellbook models)
           - dune_table: Verified, queryable Dune table name (e.g., "sui_walrus.base_table")
           - verified: True (all returned tables are verified to exist)
-          - columns: Column details (for Spellbook models, if include_columns=True)
         - 'source': The source parameter used
         - 'message': Helpful message if no tables found
+        
+    Note: To get accurate column information, use dune_describe_table on the dune_table value.
     
     Examples:
         # Search both sources for walrus - returns verified tables only
@@ -611,7 +614,7 @@ def _spellbook_find_models_impl(
     keyword: str | list[str] | None = None,
     schema: str | None = None,
     limit: int = 50,
-    include_columns: bool = True,
+    include_columns: bool = False,
 ) -> dict[str, Any]:
     """
     Implementation for spellbook model discovery.

{spice_mcp-0.1.5.dist-info → spice_mcp-0.1.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: spice-mcp
-Version: 0.1.5
+Version: 0.1.7
 Summary: mcp server built ontop of dune api endpoint
 Author-email: Evan-Kim2028 <ekcopersonal@gmail.com>
 License-File: LICENSE

{spice_mcp-0.1.5.dist-info → spice_mcp-0.1.7.dist-info}/RECORD

@@ -17,14 +17,14 @@ spice_mcp/adapters/dune/typing_utils.py,sha256=EpWneGDn-eQdo6lkLuESR09KXkDj9OqGz
 spice_mcp/adapters/dune/urls.py,sha256=bcuPERkFQduRTT2BrgzVhoFrMn-Lkvw9NmktcBZYEig,3902
 spice_mcp/adapters/dune/user_agent.py,sha256=c6Kt4zczbuT9mapDoh8-3sgm268MUtvyIRxDF9yJwXQ,218
 spice_mcp/adapters/spellbook/__init__.py,sha256=D2cdVtSUbmAJdbPRvAyKxYS4-wUQ3unXyX4ZFYxenuk,150
-spice_mcp/adapters/spellbook/explorer.py,sha256=nfecYztzxfBXp9b9IKcP4dVICCnPzhHvPfNJKFhaKxA,14856
+spice_mcp/adapters/spellbook/explorer.py,sha256=PQCUJjXnaf8YSziINVNSqIXsns_y6l62JgjBId4b7jo,14654
 spice_mcp/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 spice_mcp/core/errors.py,sha256=jlfTuyRaAaA_oU07KUk-1pDAAa43KG0BbZc5CINXtoE,3256
 spice_mcp/core/models.py,sha256=i0C_-UE16OWyyZo_liooEJeYvbChE5lpK80aN2OF4lk,1795
 spice_mcp/core/ports.py,sha256=nEdeA3UH7v0kB_hbguMrpDljb9EhSxUAO0SdhjpoijQ,1618
 spice_mcp/logging/query_history.py,sha256=doE9lod64uzJxlA2XzHH2-VAmC6WstYAkQ0taEAxiIM,4315
 spice_mcp/mcp/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-spice_mcp/mcp/server.py,sha256=oi0RRaSithCgUt0Qt6Tb3gks32LN0dOOwN7kX-BrN7s,32263
+spice_mcp/mcp/server.py,sha256=M6IPxQD--eecdNryTIL8JlkWPAXrj256P0UCEktdOt0,32438
 spice_mcp/mcp/tools/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 spice_mcp/mcp/tools/base.py,sha256=zJkVxLgXR48iZcJeng8cZ2rXvbyicagoGlMN7BK7Img,1041
 spice_mcp/mcp/tools/execute_query.py,sha256=K1YpuQGwvVM20A4_h9zNlkeG37J7jbY7BPzLm6vPAsY,16033
@@ -35,8 +35,8 @@ spice_mcp/service_layer/discovery_service.py,sha256=202O0SzCZGQukd9kb2JYfarLygZH
 spice_mcp/service_layer/query_admin_service.py,sha256=4q1NAAuTui7cm83Aq2rFDLIzKTHX17yzbSoSJyCmLbI,1356
 spice_mcp/service_layer/query_service.py,sha256=q0eAVW5I3sUxm29DgzPN_cH3rZEzmKwmdE3Xj4qP9lI,3878
 spice_mcp/service_layer/verification_service.py,sha256=dPA88p9zKqg62bNjN_4c5QFEUBHCWjZph8pn2a5zrUI,6057
-spice_mcp-0.1.5.dist-info/METADATA,sha256=ag4hjEMz-qxvImxJPxpsQN_0F9BOUdbk6TF8zAXW5I4,6093
-spice_mcp-0.1.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-spice_mcp-0.1.5.dist-info/entry_points.txt,sha256=4XiXX13Vy-oiUJwlcO_82OltBaxFnEnkJ-76sZGm5os,56
-spice_mcp-0.1.5.dist-info/licenses/LICENSE,sha256=r0GNDnDY1RSkVQp7kEEf6MQU21OrNGJkxUHIsv6eyLk,1079
-spice_mcp-0.1.5.dist-info/RECORD,,
+spice_mcp-0.1.7.dist-info/METADATA,sha256=1nVDHRV6aJzq2ANEmfaCO5E2phc_m93kQh0P0KUBrkE,6093
+spice_mcp-0.1.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+spice_mcp-0.1.7.dist-info/entry_points.txt,sha256=4XiXX13Vy-oiUJwlcO_82OltBaxFnEnkJ-76sZGm5os,56
+spice_mcp-0.1.7.dist-info/licenses/LICENSE,sha256=r0GNDnDY1RSkVQp7kEEf6MQU21OrNGJkxUHIsv6eyLk,1079
+spice_mcp-0.1.7.dist-info/RECORD,,