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

chuk_artifacts/__init__.py

@@ -12,6 +12,7 @@ from dotenv import load_dotenv
 
 # Core classes
 from .store import ArtifactStore
+from .models import ArtifactEnvelope
 
 # Exception classes
 from .exceptions import (
@@ -42,6 +43,9 @@ __all__ = [
     # Main class
     "ArtifactStore",
     
+    # Models
+    "ArtifactEnvelope",
+    
     # Exceptions
     "ArtifactStoreError", 
     "ArtifactNotFoundError",
@@ -146,4 +150,4 @@ def configure_logging(level: str = "INFO"):
             "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
         )
         handler.setFormatter(formatter)
-        logger.addHandler(handler)
+        logger.addHandler(handler)

chuk_artifacts/core.py

@@ -19,7 +19,7 @@ from typing import Any, Dict, Optional, TYPE_CHECKING
 if TYPE_CHECKING:
     from .store import ArtifactStore
 
-from .exceptions import ArtifactStoreError, ProviderError, SessionError
+from .exceptions import ArtifactStoreError, ProviderError, SessionError, ArtifactNotFoundError
 
 logger = logging.getLogger(__name__)
 
@@ -102,44 +102,55 @@ class CoreStorageOperations:
                 raise ProviderError(f"Storage failed: {e}") from e
 
     async def update_file(
-            self,
-            artifact_id: str,
-            new_data: bytes,
-            *,
-            mime: Optional[str] = None,
-            summary: Optional[str] = None,
-            meta: Optional[Dict[str, Any]] = None,
-            filename: Optional[str] = None,
-            ttl: Optional[int] = None,
-        ) -> None:
-            """
-            Update the contents of an existing artifact.
+        self,
+        artifact_id: str,
+        new_data: Optional[bytes] = None,
+        *,
+        mime: Optional[str] = None,
+        summary: Optional[str] = None,
+        meta: Optional[Dict[str, Any]] = None,
+        filename: Optional[str] = None,
+        ttl: Optional[int] = None,
+    ) -> bool:
+        """
+        Update an artifact's content, metadata, filename, summary, or mime type.
+        
+        Parameters
+        ----------
+        artifact_id : str
+            ID of the artifact to update
+        new_data : bytes, optional
+            New data to overwrite the existing artifact
+        mime : str, optional
+            New MIME type
+        summary : str, optional
+            New summary
+        meta : dict, optional
+            Updated metadata
+        filename : str, optional
+            New filename
+        ttl : int, optional
+            New TTL
+            
+        Returns
+        -------
+        bool
+            True if update was successful
+        """
+        if self.artifact_store._closed:
+            raise ArtifactStoreError("Store is closed")
 
-            Parameters
-            ----------
-            artifact_id : str
-                ID of the artifact to update.
-            new_data : bytes
-                New data to overwrite the existing artifact.
-            mime : Optional[str]
-                Optional new MIME type.
-            summary : Optional[str]
-                Optional new summary.
-            meta : Optional[Dict[str, Any]]
-                Optional updated metadata.
-            filename : Optional[str]
-                Optional new filename.
-            ttl : Optional[int]
-                Optional TTL to reset (if provided).
-            """
-            if self.artifact_store._closed:
-                raise ArtifactStoreError("Store is closed")
+        if not any([new_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.")
 
-            try:
-                record = await self._get_record(artifact_id)
-                key = record["key"]
-                session_id = record["session_id"]
+        try:
+            record = await self._get_record(artifact_id)
+            key = record["key"]
+            session_id = record["session_id"]
 
+            # Update data if provided
+            if new_data is not None:
                 # Overwrite in object storage
                 await self._store_with_retry(
                     new_data,
@@ -148,31 +159,37 @@ class CoreStorageOperations:
                     filename or record.get("filename"),
                     session_id,
                 )
+                
+                # Update size and hash in metadata
+                record["bytes"] = len(new_data)
+                record["sha256"] = hashlib.sha256(new_data).hexdigest()
 
-                # Update metadata
-                record.update({
-                    "mime": mime or record["mime"],
-                    "summary": summary or record["summary"],
-                    "meta": meta or record["meta"],
-                    "filename": filename or record.get("filename"),
-                    "bytes": len(new_data),
-                    "sha256": hashlib.sha256(new_data).hexdigest(),
-                    "updated_at": datetime.utcnow().isoformat() + "Z",
-                     "ttl": ttl or record["ttl"],
-                })
+            # Update metadata fields
+            if mime is not None:
+                record["mime"] = mime
+            if summary is not None:
+                record["summary"] = summary
+            if filename is not None:
+                record["filename"] = filename
+            if meta is not None:
+                record["meta"] = meta
+            if ttl is not None:
+                record["ttl"] = ttl
 
-                if ttl is not None:
-                    record["ttl"] = ttl
+            # Add update timestamp
+            record["updated_at"] = datetime.utcnow().isoformat() + "Z"
 
-                session_ctx_mgr = self.artifact_store._session_factory()
-                async with session_ctx_mgr as session:
-                    await session.setex(artifact_id, record["ttl"], json.dumps(record))
+            # Store updated metadata
+            session_ctx_mgr = self.artifact_store._session_factory()
+            async with session_ctx_mgr as session:
+                await session.setex(artifact_id, record["ttl"], json.dumps(record))
 
-                logger.info("Artifact updated successfully", extra={"artifact_id": artifact_id})
+            logger.info("Artifact updated successfully", extra={"artifact_id": artifact_id})
+            return True
 
-            except Exception as e:
-                logger.error(f"Update failed for artifact {artifact_id}: {e}")
-                raise ProviderError(f"Artifact update failed: {e}") from e
+        except Exception as e:
+            logger.error(f"Update failed for artifact {artifact_id}: {e}")
+            raise ProviderError(f"Artifact update failed: {e}") from e
             
     async def retrieve(self, artifact_id: str) -> bytes:
         """Retrieve artifact data."""
@@ -246,7 +263,7 @@ class CoreStorageOperations:
         raise last_exception
 
     async def _get_record(self, artifact_id: str) -> Dict[str, Any]:
-        """Get artifact metadata."""
+        """Get artifact metadata record from session provider."""
         try:
             session_ctx_mgr = self.artifact_store._session_factory()
             async with session_ctx_mgr as session:
@@ -255,7 +272,7 @@ class CoreStorageOperations:
             raise SessionError(f"Session error for {artifact_id}: {e}") from e
         
         if raw is None:
-            raise ProviderError(f"Artifact {artifact_id} not found")
+            raise ArtifactNotFoundError(f"Artifact {artifact_id} not found")
         
         try:
             return json.loads(raw)

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:

{chuk_artifacts-0.4.dist-info → chuk_artifacts-0.4.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chuk-artifacts
-Version: 0.4
+Version: 0.4.1
 Summary: Chuk Artifacts provides a production-ready, modular artifact storage system that works seamlessly across multiple storage backends (memory, filesystem, AWS S3, IBM Cloud Object Storage) with Redis or memory-based metadata caching and strict session-based security.
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{chuk_artifacts-0.4.dist-info → chuk_artifacts-0.4.1.dist-info}/RECORD

@@ -1,23 +1,23 @@
-chuk_artifacts/__init__.py,sha256=-4S9FWKVcQSa2ZD3GVbmbpGZPcl0cTQN_TFZLSqV7lQ,3605
+chuk_artifacts/__init__.py,sha256=9HIUqLd14M4CGcQvTsa4XEd8SBCjYRIqoKfTh_GEG4w,3683
 chuk_artifacts/admin.py,sha256=lUgmKBPxJh-0FYrGWjkACXQOl8lbVEDPJaeGVsWZmC4,6071
 chuk_artifacts/base.py,sha256=BtuVnC9M8QI1znyTdBxjZ6knIKP_k0yUfLfh7inGJUc,2559
 chuk_artifacts/batch.py,sha256=x8ARrWJ24I9fAXXodzvh31uMxYrvwZCGGJhUCM4vMJ4,5099
 chuk_artifacts/config.py,sha256=MaUzHzKPoBUyERviEpv8JVvPybMzSksgLyj0b7AO3Sc,7664
-chuk_artifacts/core.py,sha256=u1w0A-DZUoTvnnTrtYNMNxqLdKqvRHJZBaXHsjIHQ8E,9465
+chuk_artifacts/core.py,sha256=K23Z9WtbHXCaacWV5mVRPYYCF13qfB0cnG-x3wxXyDs,9893
 chuk_artifacts/exceptions.py,sha256=f-s7Mg7c8vMXsbgqO2B6lMHdXcJQNvsESAY4GhJaV4g,814
 chuk_artifacts/grid.py,sha256=PqDQGPSyxzDkQp7X-JKJvUmVmrmjdVv1wbzQ6Ysj6gg,4328
 chuk_artifacts/metadata.py,sha256=KSWZxaIxpOP_SdxUVXLttQ65aXJGx9ylqkW4n3997fI,7845
 chuk_artifacts/models.py,sha256=XlI-PATNO67vlGVEOnphCx78mfmofcBRAiVU79Ohs9o,881
 chuk_artifacts/presigned.py,sha256=-GE8r0CfUZuPNA_jnSGTfX7kuws6kYCPe7C4y6FItdo,11491
 chuk_artifacts/provider_factory.py,sha256=T0IXx1C8gygJzp417oB44_DxEaZoZR7jcdwQy8FghRE,3398
-chuk_artifacts/store.py,sha256=86PmXrjg4so-Dy492HPLTasGA5wT8T2AyGOUa21o2UQ,27380
+chuk_artifacts/store.py,sha256=IAXRQbtvR5jw4GVZKOT_P9_g1B5Kr912NLlESQXUUQI,27533
 chuk_artifacts/providers/__init__.py,sha256=3lN1lAy1ETT1mQslJo1f22PPR1W4CyxmsqJBclzH4NE,317
 chuk_artifacts/providers/filesystem.py,sha256=fj8p2SVq8V8GQCnqId77Fkwn_PgOquIj6UEaGTiqUe0,15949
 chuk_artifacts/providers/ibm_cos.py,sha256=Kf4ZqxP7aHeYKm6gM03aponMzx-lIyWn2_TL8_X4A_U,3677
 chuk_artifacts/providers/ibm_cos_iam.py,sha256=7mDs0VvVh1OOUmSc-fgghbTa_euongazwZ0QG3wjdJo,5068
 chuk_artifacts/providers/memory.py,sha256=QpJFxJuZGf548ulctmNGtUpHlRzyx1_PW5F1XNVIKrE,12002
 chuk_artifacts/providers/s3.py,sha256=QSSiYjNENQfwxJ2-Yw-CJrO8lNZcIrsfdn2Vo_DD-b4,2931
-chuk_artifacts-0.4.dist-info/METADATA,sha256=BiPcDJkHk8XcsY-mUVUc8G1ywbmfiT2rWmqjHYcGjdQ,18706
-chuk_artifacts-0.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-chuk_artifacts-0.4.dist-info/top_level.txt,sha256=1_PVMtWXR0A-ZmeH6apF9mPaMtU0i23JE6wmN4GBRDI,15
-chuk_artifacts-0.4.dist-info/RECORD,,
+chuk_artifacts-0.4.1.dist-info/METADATA,sha256=sDw8-dWWFBngeUNzXYD3s7t7D-5_Pa5szsh5clfucPA,18708
+chuk_artifacts-0.4.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+chuk_artifacts-0.4.1.dist-info/top_level.txt,sha256=1_PVMtWXR0A-ZmeH6apF9mPaMtU0i23JE6wmN4GBRDI,15
+chuk_artifacts-0.4.1.dist-info/RECORD,,