kailash 0.9.18__py3-none-any.whl → 0.9.19__py3-none-any.whl

kailash/mcp_server/subscriptions.py

@@ -1075,9 +1075,9 @@ class ResourceSubscriptionManager:
         This method should be overridden or configured to fetch actual resource data.
         For now, it returns basic resource information from the monitored state.
         """
-        async with self._resource_monitor._lock:
-            if uri in self._resource_monitor._resource_states:
-                state = self._resource_monitor._resource_states[uri]
+        async with self.resource_monitor._lock:
+            if uri in self.resource_monitor._resource_states:
+                state = self.resource_monitor._resource_states[uri]
                 return {
                     "uri": uri,
                     "content": state.get("content", {}),

kailash/nodes/data/async_sql.py

@@ -1530,15 +1530,28 @@ class SQLiteAdapter(DatabaseAdapter):
 
             if fetch_mode == FetchMode.ONE:
                 row = await cursor.fetchone()
-                return self._convert_row(dict(row)) if row else None
+                result = self._convert_row(dict(row)) if row else None
             elif fetch_mode == FetchMode.ALL:
                 rows = await cursor.fetchall()
-                return [self._convert_row(dict(row)) for row in rows]
+                result = [self._convert_row(dict(row)) for row in rows]
             elif fetch_mode == FetchMode.MANY:
                 if not fetch_size:
                     raise ValueError("fetch_size required for MANY mode")
                 rows = await cursor.fetchmany(fetch_size)
-                return [self._convert_row(dict(row)) for row in rows]
+                result = [self._convert_row(dict(row)) for row in rows]
+            else:
+                result = []
+
+            # Check if this was an INSERT and capture lastrowid for SQLite
+            if query.strip().upper().startswith("INSERT") and (
+                not result or result == [] or result is None
+            ):
+                # For INSERT without RETURNING, capture lastrowid
+                lastrowid = cursor.lastrowid if hasattr(cursor, "lastrowid") else None
+                if lastrowid is not None:
+                    return {"lastrowid": lastrowid}
+
+            return result
         else:
             # Create new connection for non-transactional queries
             if self._is_memory_db:
@@ -1557,6 +1570,19 @@ class SQLiteAdapter(DatabaseAdapter):
                         raise ValueError("fetch_size required for MANY mode")
                     rows = await cursor.fetchmany(fetch_size)
                     result = [self._convert_row(dict(row)) for row in rows]
+                else:
+                    result = []
+
+                # Check if this was an INSERT and capture lastrowid for SQLite
+                if query.strip().upper().startswith("INSERT") and (
+                    not result or result == []
+                ):
+                    # For INSERT without RETURNING, capture lastrowid
+                    lastrowid = (
+                        cursor.lastrowid if hasattr(cursor, "lastrowid") else None
+                    )
+                    if lastrowid is not None:
+                        result = {"lastrowid": lastrowid}
 
                 # Commit for memory databases (needed for INSERT/UPDATE/DELETE)
                 await db.commit()
@@ -1577,9 +1603,24 @@ class SQLiteAdapter(DatabaseAdapter):
                         if not fetch_size:
                             raise ValueError("fetch_size required for MANY mode")
                         rows = await cursor.fetchmany(fetch_size)
-                        return [self._convert_row(dict(row)) for row in rows]
+                        result = [self._convert_row(dict(row)) for row in rows]
+                    else:
+                        result = []
 
-                await db.commit()
+                    # Check if this was an INSERT and capture lastrowid for SQLite
+                    if query.strip().upper().startswith("INSERT") and (
+                        not result or result == []
+                    ):
+                        # For INSERT without RETURNING, capture lastrowid
+                        lastrowid = (
+                            cursor.lastrowid if hasattr(cursor, "lastrowid") else None
+                        )
+                        if lastrowid is not None:
+                            await db.commit()  # Commit before returning
+                            return {"lastrowid": lastrowid}
+
+                    await db.commit()
+                    return result
 
     async def execute_many(
         self,
@@ -3421,28 +3462,37 @@ class AsyncSQLDatabaseNode(AsyncNode):
                 parameter_types=parameter_types,
             )
 
-            # Ensure all data is JSON-serializable (safety net for adapter inconsistencies)
-            result = self._ensure_serializable(result)
+            # Check for special SQLite lastrowid result
+            if isinstance(result, dict) and "lastrowid" in result:
+                # This is a special SQLite INSERT result
+                formatted_data = result  # Keep as-is
+                row_count = 1  # One row was inserted
+            else:
+                # Ensure all data is JSON-serializable (safety net for adapter inconsistencies)
+                result = self._ensure_serializable(result)
 
-            # Format results based on requested format
-            formatted_data = self._format_results(result, result_format)
+                # Format results based on requested format
+                formatted_data = self._format_results(result, result_format)
+                row_count = None  # Will be calculated below
 
             # For DataFrame, we need special handling for row count
-            row_count = 0
-            if result_format == "dataframe":
-                try:
-                    row_count = len(formatted_data)
-                except:
-                    # If pandas isn't available, formatted_data is still a list
+            if row_count is None:  # Only calculate if not already set
+                if result_format == "dataframe":
+                    try:
+                        row_count = len(formatted_data)
+                    except:
+                        # If pandas isn't available, formatted_data is still a list
+                        row_count = (
+                            len(result)
+                            if isinstance(result, list)
+                            else (1 if result else 0)
+                        )
+                else:
                     row_count = (
                         len(result)
                         if isinstance(result, list)
                         else (1 if result else 0)
                     )
-            else:
-                row_count = (
-                    len(result) if isinstance(result, list) else (1 if result else 0)
-                )
 
             # Extract column names if available
             columns = []
@@ -4677,13 +4727,30 @@ class AsyncSQLDatabaseNode(AsyncNode):
             self._adapter = None
 
     def __del__(self):
-        """Ensure connections are closed."""
+        """Ensure connections are closed safely."""
         if self._adapter and self._connected:
-            # Schedule cleanup in the event loop if it exists
+            # Try to schedule cleanup, but be resilient to event loop issues
             try:
-                loop = asyncio.get_event_loop()
-                if not loop.is_closed():
-                    loop.create_task(self.cleanup())
-            except RuntimeError:
-                # No event loop, can't clean up async resources
+                import asyncio
+
+                # Check if there's a running event loop that's not closed
+                try:
+                    loop = asyncio.get_running_loop()
+                    if loop and not loop.is_closed():
+                        # Create cleanup task only if loop is healthy
+                        try:
+                            loop.create_task(self.cleanup())
+                        except RuntimeError as e:
+                            # Loop might be closing, ignore gracefully
+                            logger.debug(f"Could not schedule cleanup task: {e}")
+                    else:
+                        logger.debug("Event loop is closed, skipping async cleanup")
+                except RuntimeError:
+                    # No running event loop - this is normal during shutdown
+                    logger.debug(
+                        "No running event loop for cleanup, connections will be cleaned by GC"
+                    )
+            except Exception as e:
+                # Complete fallback - any unexpected error should not crash __del__
+                logger.debug(f"Error during connection cleanup: {e}")
                 pass

kailash/runtime/local.py

@@ -2333,7 +2333,10 @@ class LocalRuntime:
         else:
             # Standard node execution (backward compatibility)
             try:
-                if hasattr(node, "async_run"):
+                if hasattr(node, "execute_async"):
+                    # For AsyncNode and its subclasses, use execute_async which handles event loop properly
+                    node_result = await node.execute_async(**inputs)
+                elif hasattr(node, "async_run"):
                     node_result = await node.async_run(**inputs)
                 else:
                     node_result = node.execute(**inputs)

{kailash-0.9.18.dist-info → kailash-0.9.19.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kailash
-Version: 0.9.18
+Version: 0.9.19
 Summary: Python SDK for the Kailash container-node architecture
 Home-page: https://github.com/integrum/kailash-python-sdk
 Author: Integrum

{kailash-0.9.18.dist-info → kailash-0.9.19.dist-info}/RECORD

@@ -93,7 +93,7 @@ kailash/mcp_server/oauth.py,sha256=GFC2O2ueiTTI6V-91Huevhc3K8CxrHe22knuHfuCTqY,5
 kailash/mcp_server/protocol.py,sha256=NIdEwJT21JT9ItajXniPNvCbZtTbpqyOC_ZezqsguGE,35694
 kailash/mcp_server/registry_integration.py,sha256=B8CSLq_O1ea3cXrbVjC3bB_OFgHIP-KS9dk77mNM02I,19791
 kailash/mcp_server/server.py,sha256=yFp1F4QQl6gkTY_9JJWmiMiwfT-zACLJLubz-NR5sCw,108675
-kailash/mcp_server/subscriptions.py,sha256=J0FXg6_8lOffQ5SSwm2_DmqIS2pOii6ikxORANLCy1o,58589
+kailash/mcp_server/subscriptions.py,sha256=UK0Ssjab-lHJ16DFPi6lmvLh5xMYNRZia0DgYb6aQ60,58586
 kailash/mcp_server/transports.py,sha256=fBa7CTVYTDb0ZbBQTsZ2d8rKvcVuqBIteczq8eqarr4,49919
 kailash/mcp_server/servers/ai_registry.py,sha256=IdF_keUuJlMsvjLjSAykxxbm46K4qA7eCj7T-lYSrzk,10007
 kailash/mcp_server/utils/__init__.py,sha256=R20N-iiKXUPxc9MOh6vPO1vIfkPmwhEQ5KNFgGd4xSs,771
@@ -220,7 +220,7 @@ kailash/nodes/compliance/data_retention.py,sha256=90bH_eGwlcDzUdklAJeXQM-RcuLUGQ
 kailash/nodes/compliance/gdpr.py,sha256=ZMoHZjAo4QtGwtFCzGMrAUBFV3TbZOnJ5DZGZS87Bas,70548
 kailash/nodes/data/__init__.py,sha256=f0h4ysvXxlyFcNJLvDyXrgJ0ixwDF1cS0pJ2QNPakhg,5213
 kailash/nodes/data/async_connection.py,sha256=wfArHs9svU48bxGZIiixSV2YVn9cukNgEjagwTRu6J4,17250
-kailash/nodes/data/async_sql.py,sha256=dhDBn5Ont0XBLnZz0_gG8s_8dossj50J0upuvanU7fw,185523
+kailash/nodes/data/async_sql.py,sha256=9C-XRTDrzpVwFRrI13ym539UajT0Qgh9jmIjisDPi28,188864
 kailash/nodes/data/async_vector.py,sha256=HtwQLO25IXu8Vq80qzU8rMkUAKPQ2qM0x8YxjXHlygU,21005
 kailash/nodes/data/bulk_operations.py,sha256=WVopmosVkIlweFxVt3boLdCPc93EqpYyQ1Ez9mCIt0c,34453
 kailash/nodes/data/directory.py,sha256=fbfLqD_ijRubk-4xew3604QntPsyDxqaF4k6TpfyjDg,9923
@@ -341,7 +341,7 @@ kailash/runtime/async_local.py,sha256=sYNggSU0R-oo8cCvU5ayodDBqASzUhxu994ZvZxDSC
 kailash/runtime/compatibility_reporter.py,sha256=TOQD0ODnJdsxEPyNSYOV_zQxu60X_yvHeu26seFOMEA,19807
 kailash/runtime/docker.py,sha256=sZknVl1PCGfAZeyc0-exTuKlllSyjYlFIgJoiB3CRNs,23500
 kailash/runtime/hierarchical_switch_executor.py,sha256=k6aPGbpf6z2m6dTbHrEyuDR8ZCvOqUanBGYp70arQn0,20782
-kailash/runtime/local.py,sha256=6c2RtT3YisYPnBAOho6lbd4fYyMiJG8EyUBOGmcei_U,201159
+kailash/runtime/local.py,sha256=nIQRWUwSHVg2Daafq_JggBLf-zTDBaGMcwObBzVI0po,201389
 kailash/runtime/parallel.py,sha256=-M9VVG36RxnrrmdbcBe9IjQWb58tAEEo76RQQ2uIXaE,21084
 kailash/runtime/parallel_cyclic.py,sha256=yANZHnePjhCPuCFbq3lFQA1K6jbCv5Of5-vIKbCsmZk,19863
 kailash/runtime/parameter_injection.py,sha256=kG4GhmarsRr5t3VDFbc2G1HSbsZJg6UmienHCE2Ru7o,14852
@@ -424,10 +424,10 @@ kailash/workflow/templates.py,sha256=XQMAKZXC2dlxgMMQhSEOWAF3hIbe9JJt9j_THchhAm8
 kailash/workflow/type_inference.py,sha256=i1F7Yd_Z3elTXrthsLpqGbOnQBIVVVEjhRpI0HrIjd0,24492
 kailash/workflow/validation.py,sha256=LdbIPQSokCqSLfWTBhJR82pa_0va44pcVu9dpEM4rvY,45177
 kailash/workflow/visualization.py,sha256=nHBW-Ai8QBMZtn2Nf3EE1_aiMGi9S6Ui_BfpA5KbJPU,23187
-kailash-0.9.18.dist-info/licenses/LICENSE,sha256=9GYZHXVUmx6FdFRNzOeE_w7a_aEGeYbqTVmFtJlrbGk,13438
-kailash-0.9.18.dist-info/licenses/NOTICE,sha256=9ssIK4LcHSTFqriXGdteMpBPTS1rSLlYtjppZ_bsjZ0,723
-kailash-0.9.18.dist-info/METADATA,sha256=7kNOPQ-Zpyh4bZVQ9khKjjrzDDvzBHrsBHaex36vXZY,23528
-kailash-0.9.18.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-kailash-0.9.18.dist-info/entry_points.txt,sha256=M_q3b8PG5W4XbhSgESzIJjh3_4OBKtZFYFsOdkr2vO4,45
-kailash-0.9.18.dist-info/top_level.txt,sha256=z7GzH2mxl66498pVf5HKwo5wwfPtt9Aq95uZUpH6JV0,8
-kailash-0.9.18.dist-info/RECORD,,
+kailash-0.9.19.dist-info/licenses/LICENSE,sha256=9GYZHXVUmx6FdFRNzOeE_w7a_aEGeYbqTVmFtJlrbGk,13438
+kailash-0.9.19.dist-info/licenses/NOTICE,sha256=9ssIK4LcHSTFqriXGdteMpBPTS1rSLlYtjppZ_bsjZ0,723
+kailash-0.9.19.dist-info/METADATA,sha256=RY2liVVkhKdErnyayfo4_vH2OyMVKgvtvimhAT7JWvA,23528
+kailash-0.9.19.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+kailash-0.9.19.dist-info/entry_points.txt,sha256=M_q3b8PG5W4XbhSgESzIJjh3_4OBKtZFYFsOdkr2vO4,45
+kailash-0.9.19.dist-info/top_level.txt,sha256=z7GzH2mxl66498pVf5HKwo5wwfPtt9Aq95uZUpH6JV0,8
+kailash-0.9.19.dist-info/RECORD,,