chuk-artifacts 0.3__py3-none-any.whl → 0.4.1__py3-none-any.whl

chuk_artifacts/providers/ibm_cos_iam.py

@@ -5,6 +5,11 @@ Async wrapper for IBM Cloud Object Storage using IAM API-key (oauth).
 
 ✓ Fits the aioboto3-style interface that ArtifactStore expects:
     • async put_object(...)
+    • async get_object(...)
+    • async head_object(...)
+    • async delete_object(...)
+    • async list_objects_v2(...)
+    • async head_bucket(...)
     • async generate_presigned_url(...)
 ✓ No HMAC keys required - just IBM_COS_APIKEY + IBM_COS_INSTANCE_CRN.
 
@@ -27,6 +32,7 @@ from ibm_botocore.client import Config
 
 # ─────────────────────────────────────────────────────────────────────
 def _sync_client():
+    """Create synchronous IBM COS client with IAM authentication."""
     endpoint = os.getenv(
         "IBM_COS_ENDPOINT",
         "https://s3.us-south.cloud-object-storage.appdomain.cloud",
@@ -49,19 +55,52 @@ def _sync_client():
 
 # ─────────────────────────────────────────────────────────────────────
 class _AsyncIBMClient:
-    """Minimal async façade over synchronous ibm_boto3 S3 client."""
+    """Complete async façade over synchronous ibm_boto3 S3 client."""
+    
     def __init__(self, sync_client):
         self._c = sync_client
 
-    # ---- methods used by ArtifactStore -------------------------------------
-    async def put_object(self, **kw) -> Dict[str, Any]:
-        return await asyncio.to_thread(self._c.put_object, **kw)
+    # ---- Core S3 operations used by ArtifactStore -------------------------
+    async def put_object(self, **kwargs) -> Dict[str, Any]:
+        """Store object in IBM COS."""
+        return await asyncio.to_thread(self._c.put_object, **kwargs)
 
-    async def generate_presigned_url(self, *a, **kw) -> str:
-        return await asyncio.to_thread(self._c.generate_presigned_url, *a, **kw)
+    async def get_object(self, **kwargs) -> Dict[str, Any]:
+        """Retrieve object from IBM COS."""
+        return await asyncio.to_thread(self._c.get_object, **kwargs)
 
-    # ---- cleanup -----------------------------------------------------------
+    async def head_object(self, **kwargs) -> Dict[str, Any]:
+        """Get object metadata from IBM COS."""
+        return await asyncio.to_thread(self._c.head_object, **kwargs)
+
+    async def delete_object(self, **kwargs) -> Dict[str, Any]:
+        """Delete object from IBM COS."""
+        return await asyncio.to_thread(self._c.delete_object, **kwargs)
+
+    async def list_objects_v2(self, **kwargs) -> Dict[str, Any]:
+        """List objects in IBM COS bucket."""
+        return await asyncio.to_thread(self._c.list_objects_v2, **kwargs)
+
+    async def head_bucket(self, **kwargs) -> Dict[str, Any]:
+        """Check if bucket exists in IBM COS."""
+        return await asyncio.to_thread(self._c.head_bucket, **kwargs)
+
+    async def generate_presigned_url(self, *args, **kwargs) -> str:
+        """Generate presigned URL for IBM COS object."""
+        return await asyncio.to_thread(self._c.generate_presigned_url, *args, **kwargs)
+
+    # ---- Additional operations for completeness ---------------------------
+    async def copy_object(self, **kwargs) -> Dict[str, Any]:
+        """Copy object within IBM COS."""
+        return await asyncio.to_thread(self._c.copy_object, **kwargs)
+
+    async def delete_objects(self, **kwargs) -> Dict[str, Any]:
+        """Delete multiple objects from IBM COS."""
+        return await asyncio.to_thread(self._c.delete_objects, **kwargs)
+
+    # ---- Cleanup -----------------------------------------------------------
     async def close(self):
+        """Close the underlying sync client."""
         await asyncio.to_thread(self._c.close)
 
 
@@ -69,6 +108,11 @@ class _AsyncIBMClient:
 def factory() -> Callable[[], AsyncContextManager]:
     """
     Return a zero-arg callable that yields an async-context-manager.
+    
+    Returns
+    -------
+    Callable[[], AsyncContextManager]
+        Factory function that creates IBM COS IAM client context managers
     """
 
     @asynccontextmanager

chuk_artifacts/providers/memory.py

@@ -36,6 +36,7 @@ class _MemoryS3Client:
             If None, creates isolated per-instance storage.
         """
         self._store: Dict[str, Dict[str, Any]] = shared_store if shared_store is not None else {}
+        self._is_shared_store = shared_store is not None
         self._lock = asyncio.Lock()
         self._closed = False
         
@@ -231,8 +232,11 @@ class _MemoryS3Client:
     async def close(self):
         """Clean up resources and mark client as closed."""
         if not self._closed:
-            async with self._lock:
-                self._store.clear()
+            # Only clear the store if it's NOT shared
+            # If it's shared, other clients may still need the data
+            if not self._is_shared_store:
+                async with self._lock:
+                    self._store.clear()
             self._closed = True
 
     # ------------------------------------------------------------
@@ -253,6 +257,10 @@ class _MemoryS3Client:
                 "total_objects": total_objects,
                 "total_bytes": total_bytes,
                 "closed": self._closed,
+                "is_shared_store": self._is_shared_store,
+                "store_id": id(self._store),  # Memory address for debugging
+                "client_id": id(self),        # Client instance ID
+                "store_keys": list(self._store.keys())[:5],  # First 5 keys for debugging
             }
 
     @classmethod
@@ -263,17 +271,32 @@ class _MemoryS3Client:
 
 # ---- public factory -------------------------------------------------------
 
+# Global shared storage for memory provider when used as default
+_default_shared_store: Dict[str, Dict[str, Any]] = {}
+
 def factory(shared_store: Optional[Dict[str, Dict[str, Any]]] = None) -> Callable[[], AsyncContextManager]:
     """
     Return a **zero-arg** factory that yields an async-context client.
     
+    Key behavior for memory provider:
+    - If shared_store is provided, uses that specific storage
+    - If shared_store is None, ALWAYS uses the global shared storage
+    - This ensures all memory clients in the same process share data
+    - Prevents issues where ArtifactStore operations can't see each other's data
+    
     Parameters
     ----------
     shared_store : dict, optional
         If provided, all clients created by this factory will share
-        the same storage dict. Useful for testing scenarios where
-        you need multiple clients to see the same data.
+        the same storage dict. If None, will use a global shared store
+        to ensure consistency across operations within the same process.
     """
+    
+    # CRITICAL: Always use global shared storage when none specified
+    # This prevents the common issue where each ArtifactStore operation
+    # gets a different isolated storage and can't see each other's data
+    if shared_store is None:
+        shared_store = _default_shared_store
 
     @asynccontextmanager
     async def _ctx():
@@ -307,6 +330,11 @@ async def clear_all_memory_stores():
     Emergency cleanup function that clears all active memory stores.
     Useful for test teardown.
     """
+    # Clear the global shared store
+    global _default_shared_store
+    _default_shared_store.clear()
+    
+    # Close all active instances
     instances = list(_MemoryS3Client._instances)
     for instance in instances:
         try:

chuk_artifacts/store.py

@@ -172,21 +172,24 @@ class ArtifactStore:
         filename: Optional[str] = None,
         summary: Optional[str] = None,
         mime: Optional[str] = None,
+        ttl: Optional[int] = None,
     ) -> bool:
         """
         Update an artifact's content, metadata, filename, summary, or mime type.
         All parameters are optional. At least one must be provided.
         """
-        if not any([data, meta, filename, summary, mime]):
+        if not any([data is not None, meta is not None, filename is not None, 
+                   summary is not None, mime is not None, ttl is not None]):
             raise ValueError("At least one update parameter must be provided.")
 
         return await self._core.update_file(
             artifact_id=artifact_id,
             new_data=data,
+            mime=mime,
+            summary=summary,
             meta=meta,
             filename=filename,
-            summary=summary,
-            mime=mime,
+            ttl=ttl,
         )
 
     async def retrieve(self, artifact_id: str) -> bytes:
@@ -633,4 +636,50 @@ class ArtifactStore:
         return self
 
     async def __aexit__(self, exc_type, exc_val, exc_tb):
-        await self.close()
+        await self.close()
+
+    async def get_sandbox_info(self) -> Dict[str, Any]:
+        """
+        Get sandbox information and metadata.
+        
+        Returns
+        -------
+        Dict[str, Any]
+            Dictionary containing sandbox information including:
+            - sandbox_id: The current sandbox identifier
+            - bucket: The storage bucket name
+            - storage_provider: The storage provider type
+            - session_provider: The session provider type
+            - session_ttl_hours: Default session TTL
+            - grid_prefix_pattern: The grid path pattern for this sandbox
+            - created_at: Timestamp of when this info was retrieved
+        """
+        from datetime import datetime
+        
+        # Get session manager stats if available
+        session_stats = {}
+        try:
+            session_stats = self._session_manager.get_cache_stats()
+        except Exception:
+            pass  # Session manager might not have stats
+        
+        # Get storage stats if available
+        storage_stats = {}
+        try:
+            storage_stats = await self._admin.get_stats()
+        except Exception:
+            pass  # Storage might not have stats
+        
+        return {
+            "sandbox_id": self.sandbox_id,
+            "bucket": self.bucket,
+            "storage_provider": self._storage_provider_name,
+            "session_provider": self._session_provider_name,
+            "session_ttl_hours": self.session_ttl_hours,
+            "max_retries": self.max_retries,
+            "grid_prefix_pattern": self.get_session_prefix_pattern(),
+            "created_at": datetime.utcnow().isoformat() + "Z",
+            "session_stats": session_stats,
+            "storage_stats": storage_stats,
+            "closed": self._closed,
+        }