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

chuk_artifacts/core.py

@@ -101,6 +101,79 @@ class CoreStorageOperations:
             else:
                 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.
+
+            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")
+
+            try:
+                record = await self._get_record(artifact_id)
+                key = record["key"]
+                session_id = record["session_id"]
+
+                # Overwrite in object storage
+                await self._store_with_retry(
+                    new_data,
+                    key,
+                    mime or record["mime"],
+                    filename or record.get("filename"),
+                    session_id,
+                )
+
+                # 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"],
+                })
+
+                if ttl is not None:
+                    record["ttl"] = ttl
+
+                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})
+
+            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."""
         if self.artifact_store._closed:

chuk_artifacts/metadata.py

@@ -9,7 +9,7 @@ from __future__ import annotations
 
 import json
 import logging
-from typing import Any, Dict, List, Optional, TYPE_CHECKING
+from typing import Any, Dict, List, TYPE_CHECKING
 
 if TYPE_CHECKING:
     from .store import ArtifactStore
@@ -68,7 +68,7 @@ class MetadataOperations:
         try:
             artifacts = []
             # Use the session manager's canonical prefix instead of building our own
-            prefix = self.artifact_store._session_manager.get_canonical_prefix(session_id)
+            prefix = self.artifact_store.get_canonical_prefix(session_id)
             
             storage_ctx_mgr = self.artifact_store._s3_factory()
             async with storage_ctx_mgr as s3:
@@ -82,7 +82,7 @@ class MetadataOperations:
                     for obj in response.get('Contents', []):
                         key = obj['Key']
                         # Parse the grid key using chuk_sessions
-                        parsed = self.artifact_store._session_manager.parse_grid_key(key)
+                        parsed = self.artifact_store.parse_grid_key(key)
                         if parsed and parsed.get('artifact_id'):
                             artifact_id = parsed['artifact_id']
                             try:
@@ -93,7 +93,7 @@ class MetadataOperations:
                     
                     return artifacts[:limit]
             
-            logger.warning(f"Storage provider doesn't support listing")
+            logger.warning("Storage provider doesn't support listing")
             return []
             
         except Exception as e:

chuk_artifacts/store.py

@@ -12,7 +12,9 @@ Grid Architecture:
 
 from __future__ import annotations
 
-import os, logging, uuid
+import os
+import logging
+import uuid
 from datetime import datetime
 from typing import Any, Dict, List, Callable, AsyncContextManager, Optional, Union
 from chuk_sessions.session_manager import SessionManager
@@ -37,7 +39,6 @@ except ImportError:
 from .exceptions import ArtifactStoreError, ProviderError
 
 # Import chuk_sessions instead of local session manager
-from chuk_sessions.session_manager import SessionManager
 
 # Configure structured logging
 logger = logging.getLogger(__name__)
@@ -161,6 +162,32 @@ class ArtifactStore:
             session_id=session_id,
             ttl=ttl,
         )
+    
+    async def update_file(
+        self,
+        artifact_id: str,
+        *,
+        data: Optional[bytes] = None,
+        meta: Optional[Dict[str, Any]] = None,
+        filename: Optional[str] = None,
+        summary: Optional[str] = None,
+        mime: Optional[str] = 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]):
+            raise ValueError("At least one update parameter must be provided.")
+
+        return await self._core.update_file(
+            artifact_id=artifact_id,
+            new_data=data,
+            meta=meta,
+            filename=filename,
+            summary=summary,
+            mime=mime,
+        )
 
     async def retrieve(self, artifact_id: str) -> bytes:
         """Retrieve artifact data."""

{chuk_artifacts-0.2.2.dist-info → chuk_artifacts-0.3.dist-info}/METADATA

@@ -1,11 +1,9 @@
 Metadata-Version: 2.4
 Name: chuk-artifacts
-Version: 0.2.2
+Version: 0.3
 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.
-License: MIT
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
-License-File: LICENSE
 Requires-Dist: chuk-tool-processor>=0.4
 Requires-Dist: pydantic>=2.10.6
 Requires-Dist: pyyaml>=6.0.2
@@ -19,7 +17,6 @@ Provides-Extra: dev
 Requires-Dist: pytest>=8.3.5; extra == "dev"
 Requires-Dist: pytest-asyncio>=0.26.0; extra == "dev"
 Requires-Dist: ruff>=0.4.6; extra == "dev"
-Dynamic: license-file
 
 # Chuk Artifacts
 
@@ -593,6 +590,9 @@ Move/rename file within same session only (cross-session blocked).
 #### `list_files(session_id, prefix="", limit=100)`
 List files with optional prefix filtering.
 
+#### `update_file(artifact_id, *, data=None, meta=None, filename=None, summary=None, mime=None)`
+Update artifact content, metadata, filename, summary, or MIME type. At least one field must be specified for update.
+
 ### Metadata Operations
 
 #### `update_metadata(artifact_id, *, summary=None, meta=None, merge=True, **kwargs)`

{chuk_artifacts-0.2.2.dist-info → chuk_artifacts-0.3.dist-info}/RECORD

@@ -3,22 +3,21 @@ 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=hokH7cgGE2ZaEwlV8XMKOov3EMvcLS2HufdApLS6l3M,6699
+chuk_artifacts/core.py,sha256=u1w0A-DZUoTvnnTrtYNMNxqLdKqvRHJZBaXHsjIHQ8E,9465
 chuk_artifacts/exceptions.py,sha256=f-s7Mg7c8vMXsbgqO2B6lMHdXcJQNvsESAY4GhJaV4g,814
 chuk_artifacts/grid.py,sha256=IkvAgTLaN-ahRGVC6sysMkP7ALUUleJHaYfr5TRH7ZY,796
-chuk_artifacts/metadata.py,sha256=McyKLviBInpqh2eF621xhqb3Ix7QUj_nVc1gg_tUlqY,7830
+chuk_artifacts/metadata.py,sha256=Dr2IFgYn4ISHYzoSdnOToVO3CmskDqkV_ejnJFAsZos,7785
 chuk_artifacts/models.py,sha256=_foXlkr0DprqgztDw5WtlDc-s1OouLgYNp4XM1Ghp-g,837
 chuk_artifacts/presigned.py,sha256=-GE8r0CfUZuPNA_jnSGTfX7kuws6kYCPe7C4y6FItdo,11491
 chuk_artifacts/provider_factory.py,sha256=T0IXx1C8gygJzp417oB44_DxEaZoZR7jcdwQy8FghRE,3398
-chuk_artifacts/store.py,sha256=HDX2xbWXEXP6B_xylBh1L3ynCxvs_41gZy4drm1bLWc,24814
+chuk_artifacts/store.py,sha256=zotx7jJosFLKBdqfRtLRRUIsRWu7eDOB2TpoldEjYck,25605
 chuk_artifacts/providers/__init__.py,sha256=3lN1lAy1ETT1mQslJo1f22PPR1W4CyxmsqJBclzH4NE,317
 chuk_artifacts/providers/filesystem.py,sha256=F4EjE-_ItPg0RWe7CqameVpOMjU-b7AigEBkm_ZoNrc,15280
 chuk_artifacts/providers/ibm_cos.py,sha256=K1-VAX4UVV9tA161MOeDXOKloQ0hB77jdw1-p46FwmU,4445
 chuk_artifacts/providers/ibm_cos_iam.py,sha256=VtwvCi9rMMcZx6i9l21ob6wM8jXseqvjzgCnAA82RkY,3186
 chuk_artifacts/providers/memory.py,sha256=9sA8xDKh_IMjuHtqR5gTtpzpanZfdBeQLcTk-4-G9Dk,10582
 chuk_artifacts/providers/s3.py,sha256=QSSiYjNENQfwxJ2-Yw-CJrO8lNZcIrsfdn2Vo_DD-b4,2931
-chuk_artifacts-0.2.2.dist-info/licenses/LICENSE,sha256=SG9BmgtPBagPV0d-Fep-msdAGl-E1CeoBL7-EDRH2qA,1066
-chuk_artifacts-0.2.2.dist-info/METADATA,sha256=hpHktbI6xqEx9o8CuGK936IW5FTJQsSFoPIBvAkU23I,21184
-chuk_artifacts-0.2.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-chuk_artifacts-0.2.2.dist-info/top_level.txt,sha256=1_PVMtWXR0A-ZmeH6apF9mPaMtU0i23JE6wmN4GBRDI,15
-chuk_artifacts-0.2.2.dist-info/RECORD,,
+chuk_artifacts-0.3.dist-info/METADATA,sha256=rWEYiBcR92SduU9VzxnQMU1gqmGArE_XiSdWavdav4c,21340
+chuk_artifacts-0.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+chuk_artifacts-0.3.dist-info/top_level.txt,sha256=1_PVMtWXR0A-ZmeH6apF9mPaMtU0i23JE6wmN4GBRDI,15
+chuk_artifacts-0.3.dist-info/RECORD,,

chuk_artifacts-0.2.2.dist-info/licenses/LICENSE

@@ -1,21 +0,0 @@
-MIT License
-
-Copyright (c) 2025 Chris Hay
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in all
-copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.