cellcog 2.1.2__tar.gz → 2.2.0__tar.gz

{cellcog-2.1.2 → cellcog-2.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cellcog
-Version: 2.1.2
+Version: 2.2.0
 Summary: Python SDK for CellCog - Any-to-Any AI for agents. Fire-and-forget pattern with WebSocket notifications.
 Project-URL: Homepage, https://cellcog.ai
 Project-URL: Documentation, https://cellcog.ai/developer/docs

{cellcog-2.1.2 → cellcog-2.2.0}/cellcog/__init__.py

@@ -48,7 +48,7 @@ from .exceptions import (
     SDKUpgradeRequiredError,
 )
 
-__version__ = "2.1.1"
+__version__ = "2.2.0"
 __all__ = [
     "CellCogClient",
     "CellCogError",

{cellcog-2.1.2 → cellcog-2.2.0}/cellcog/files.py

@@ -66,8 +66,10 @@ class FileProcessor:
             attrs = match.group(1)
             file_path = match.group(2).strip()
 
-            # Skip empty paths and URLs — not local files
-            if not file_path or file_path.startswith(("http://", "https://")):
+            # Skip empty paths and URLs — not local files. gs:// is a CellCog
+            # storage reference (already-uploaded file); leave it untouched so it
+            # is not mistaken for a missing local file (which would raise).
+            if not file_path or file_path.startswith(("http://", "https://", "gs://")):
                 return match.group(0)
 
             # Cross-platform local file detection:
@@ -272,17 +274,25 @@ class FileProcessor:
         """
         Generate safe download path under ~/.cellcog/chats/{chat_id}/.
 
-        Handles two blob_name formats:
+        Handles these blob_name formats:
+        - gs://bucket/{chat_id}/path/to/file.ext (canonical explicit GS URL)
         - {chat_id}//home/app/path/to/file.ext  (double slash = from CellCog's /home/app/)
         - {chat_id}/path/to/file.ext             (single slash = relative path)
 
         Args:
-            blob_name: Full blob name from CellCog
+            blob_name: Full blob name (or gs:// URL) from CellCog
             chat_id: Chat ID for organizing downloads
 
         Returns:
             Local path like ~/.cellcog/chats/{chat_id}/{path}
         """
+        # Strip a canonical gs://bucket/ prefix down to the bare blob name
+        # ({chat_id}/path...) so the existing chat_id-prefix logic applies.
+        if blob_name.startswith("gs://"):
+            # gs://bucket/{chat_id}/path → drop scheme + bucket (first 3 segments)
+            parts = blob_name.split("/", 3)
+            blob_name = parts[3] if len(parts) > 3 else parts[-1]
+
         if "/" not in blob_name:
             # Malformed - use blob_name as filename
             return str(self.default_download_dir / chat_id / blob_name)

{cellcog-2.1.2 → cellcog-2.2.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "cellcog"
-version = "2.1.2"
+version = "2.2.0"
 description = "Python SDK for CellCog - Any-to-Any AI for agents. Fire-and-forget pattern with WebSocket notifications."
 readme = "README.md"
 license = {text = "MIT"}

cellcog-2.2.0/tests/test_gs_url_handling.py

@@ -0,0 +1,54 @@
+"""Standalone tests for SDK gs:// SHOW_FILE handling.
+
+Run: python3 tests/test_gs_url_handling.py
+"""
+
+import sys
+from pathlib import Path
+
+sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
+
+from cellcog.config import Config
+from cellcog.files import FileProcessor
+
+CHAT = "687a8aa4fcf12067f5a5b9e0"
+failed = 0
+
+
+def check(label, got, want):
+    global failed
+    ok = got == want
+    if not ok:
+        failed += 1
+        print(f"[FAIL] {label}\n       got : {got}\n       want: {want}")
+    else:
+        print(f"[OK ] {label}")
+
+
+fp = FileProcessor(Config())
+
+# --- download-path derivation strips gs://bucket/ then chat_id ---
+# Real CellCog blobs from /home/app are {chat_id}//home/app/... (double slash:
+# {chat_id}/ + the absolute /home/app/... path). The /home/app/ prefix is stripped.
+p = fp._generate_auto_download_path(f"gs://cellcog_files/{CHAT}//home/app/out.png", CHAT)
+check("gs:// download path strips scheme+bucket+chat+/home/app",
+      p, str(fp.default_download_dir / CHAT / "out.png"))
+
+# A non-/home/app relative path keeps its sub-path after the chat_id.
+p = fp._generate_auto_download_path(f"gs://cellcog_files/{CHAT}/sub/dir/file.pdf", CHAT)
+check("gs:// download path nested",
+      p, str(fp.default_download_dir / CHAT / "sub/dir/file.pdf"))
+
+# --- legacy bare blob still works (back-compat), double-slash /home/app form ---
+p = fp._generate_auto_download_path(f"{CHAT}//home/app/out.png", CHAT)
+check("legacy bare blob still derives correctly",
+      p, str(fp.default_download_dir / CHAT / "out.png"))
+
+# --- outgoing: gs:// is left untouched (not treated as missing local file) ---
+msg = f"Here <SHOW_FILE>gs://cellcog_files/{CHAT}/a.png</SHOW_FILE> done"
+transformed, uploaded = fp.transform_outgoing(msg)
+check("outgoing gs:// unchanged", transformed, msg)
+check("outgoing gs:// nothing uploaded", uploaded, [])
+
+print("\n" + ("ALL PASS" if failed == 0 else f"{failed} FAILURE(S)"))
+sys.exit(1 if failed else 0)