signalpilot-ai-internal 0.3.3__py3-none-any.whl → 0.4.0__py3-none-any.whl

signalpilot_ai_internal/_version.py

@@ -1,4 +1,4 @@
 # This file is auto-generated by Hatchling. As such, do not:
 #   - modify
 #   - track in version control e.g. be sure to add to .gitignore
-__version__ = VERSION = '0.3.3'
+__version__ = VERSION = '0.4.0'

signalpilot_ai_internal/cache_service.py

@@ -110,7 +110,7 @@ class RobustFileOperations:
     @staticmethod
     def safe_write_json(file_path: Path, data: Any, max_retries: int = 3) -> bool:
         """Safely write JSON data with atomic operations and backups"""
-        print(f"Attempting to write JSON to: {file_path}")
+        # print(f"Attempting to write JSON to: {file_path}")
         
         if not file_path.parent.exists():
             try:
@@ -138,8 +138,6 @@ class RobustFileOperations:
                     
                     # Keep only the most recent backup that's at least 1 hour old
                     RobustFileOperations._cleanup_backups(file_path)
-                else:
-                    print(f"Skipping backup for {file_path} - recent backup exists")
                 
             except Exception as e:
                 print(f"Warning: Could not create backup for {file_path}: {e}")
@@ -165,7 +163,6 @@ class RobustFileOperations:
                 
                 shutil.move(str(temp_path), str(file_path))
                 
-                print(f"Successfully wrote {file_path}")
                 return True
                 
             except Exception as e:

signalpilot_ai_internal/snowflake_schema_service.py

@@ -4,9 +4,12 @@ Provides REST API handlers for Snowflake database schema retrieval and query exe
 Supports multiple databases within a single Snowflake connection.
 
 Behavior:
-- Picks the smallest RUNNING warehouse.
+- If a warehouse is specified in the config, it will be used directly.
+- Otherwise, picks the smallest RUNNING warehouse.
 - If none running, resumes the smallest SUSPENDED warehouse.
 - If none exist, attempts to CREATE a tiny warehouse (requires privilege).
+- If a database is specified in the config, only that database will be processed.
+- Otherwise, all accessible databases will be processed.
 - Builds a catalog with parallel schema processing for performance.
 - For each table, includes detailed column information: name, type, ordinal position,
   nullable, description, default value, and type-specific attributes.
@@ -321,26 +324,45 @@ class SnowflakeSchemaHandler(APIHandler):
             print(f"Warning: Error processing schema {db}.{schema}: {e}", file=sys.stderr)
             return {"schema": schema, "tables": [], "error": str(e)}
     
-    def _build_catalog(self, connector, conn, max_workers: int = 5) -> Dict:
-        """Build complete catalog with parallel schema processing"""
+    def _build_catalog(self, connector, conn, max_workers: int = 5, specified_database: Optional[str] = None, specified_warehouse: Optional[str] = None) -> Dict:
+        """Build complete catalog with parallel schema processing
+        
+        Args:
+            connector: Snowflake connector module
+            conn: Active Snowflake connection
+            max_workers: Number of parallel workers for schema processing
+            specified_database: If provided, only process this database
+            specified_warehouse: If provided, use this warehouse (don't auto-select)
+        """
         cur = conn.cursor(connector.DictCursor)
         try:
-            # 1) Ensure a small warehouse is available
-            preferred_wh = None
-            # Extract warehouse from conn if available
-            try:
-                cur.execute("SELECT CURRENT_WAREHOUSE()")
-                row = cur.fetchone()
-                if row:
-                    preferred_wh = row[0] if isinstance(row, tuple) else row.get("CURRENT_WAREHOUSE()")
-            except:
-                pass
-            
-            wh = self._ensure_warehouse(cur, preferred_wh)
-            cur.execute(f'USE WAREHOUSE "{wh}"')
+            # 1) Handle warehouse selection
+            if specified_warehouse:
+                # Use the explicitly specified warehouse
+                wh = specified_warehouse
+                cur.execute(f'USE WAREHOUSE "{wh}"')
+            else:
+                # Auto-select a warehouse using existing logic
+                preferred_wh = None
+                # Extract warehouse from conn if available
+                try:
+                    cur.execute("SELECT CURRENT_WAREHOUSE()")
+                    row = cur.fetchone()
+                    if row:
+                        preferred_wh = row[0] if isinstance(row, tuple) else row.get("CURRENT_WAREHOUSE()")
+                except:
+                    pass
+                
+                wh = self._ensure_warehouse(cur, preferred_wh)
+                cur.execute(f'USE WAREHOUSE "{wh}"')
 
-            # 2) Databases (no compute requirement, but we already have compute)
-            dbs = self._list_databases(cur)
+            # 2) Handle database selection
+            if specified_database:
+                # Only process the specified database
+                dbs = [specified_database]
+            else:
+                # List all databases
+                dbs = self._list_databases(cur)
             cur.close()
 
             catalog = []
@@ -423,12 +445,22 @@ class SnowflakeSchemaHandler(APIHandler):
                 conn_params = self._get_connection_params(config)
                 max_workers = int(body.get('max_workers', 5))
                 
-                print(f"[SnowflakeSchemaHandler] Connecting with account={conn_params['account']}, user={conn_params['user']}, warehouse={conn_params.get('warehouse')}, role={conn_params.get('role')}")
+                # Extract database and warehouse from config for filtering
+                specified_database = config.get('database')
+                specified_warehouse = config.get('warehouse')
+                
+                print(f"[SnowflakeSchemaHandler] Connecting with account={conn_params['account']}, user={conn_params['user']}, warehouse={conn_params.get('warehouse')}, database={conn_params.get('database')}, role={conn_params.get('role')}")
                 
                 connection = connector.connect(**conn_params, client_session_keep_alive=False)
                 
                 try:
-                    catalog = self._build_catalog(connector, connection, max_workers=max_workers)
+                    catalog = self._build_catalog(
+                        connector, 
+                        connection, 
+                        max_workers=max_workers,
+                        specified_database=specified_database,
+                        specified_warehouse=specified_warehouse
+                    )
                     result = self._format_catalog_as_json(catalog)
                     self.finish(json.dumps(result, indent=2))
                 finally:

{signalpilot_ai_internal-0.3.3.data → signalpilot_ai_internal-0.4.0.data}/data/share/jupyter/labextensions/signalpilot-ai-internal/package.json

@@ -1,6 +1,6 @@
 {
   "name": "signalpilot-ai-internal",
-  "version": "0.3.3",
+  "version": "0.4.0",
   "description": "SignalPilot Agent - Your Jupyter Notebook Assistant",
   "keywords": [
     "jupyter",
@@ -133,7 +133,7 @@
     "outputDir": "signalpilot_ai_internal/labextension",
     "schemaDir": "schema",
     "_build": {
-      "load": "static/remoteEntry.2e2c6ae0baa591126b0a.js",
+      "load": "static/remoteEntry.a0608479b9eb7d9f9b04.js",
       "extension": "./extension",
       "style": "./style"
     }

{signalpilot_ai_internal-0.3.3.data → signalpilot_ai_internal-0.4.0.data}/data/share/jupyter/labextensions/signalpilot-ai-internal/schemas/signalpilot-ai-internal/package.json.orig

@@ -1,6 +1,6 @@
 {
   "name": "signalpilot-ai-internal",
-  "version": "0.3.3",
+  "version": "0.4.0",
   "description": "SignalPilot Agent - Your Jupyter Notebook Assistant",
   "keywords": [
     "jupyter",