fleet-python 0.2.112__tar.gz → 0.2.113__tar.gz

{fleet_python-0.2.112/fleet_python.egg-info → fleet_python-0.2.113}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fleet-python
-Version: 0.2.112
+Version: 0.2.113
 Summary: Python SDK for Fleet environments
 Author-email: Fleet AI <nic@fleet.so>
 License: Apache-2.0

{fleet_python-0.2.112 → fleet_python-0.2.113}/fleet/__init__.py

@@ -76,7 +76,7 @@ from . import env
 from . import global_client as _global_client
 from ._async import global_client as _async_global_client
 
-__version__ = "0.2.112"
+__version__ = "0.2.113"
 
 __all__ = [
     # Core classes

{fleet_python-0.2.112 → fleet_python-0.2.113}/fleet/_async/__init__.py

@@ -44,7 +44,7 @@ from ..types import VerifierFunction
 from .. import env
 from . import global_client as _async_global_client
 
-__version__ = "0.2.112"
+__version__ = "0.2.113"
 
 __all__ = [
     # Core classes

{fleet_python-0.2.112 → fleet_python-0.2.113}/fleet/_async/base.py

@@ -26,7 +26,7 @@ from .exceptions import (
 try:
     from .. import __version__
 except ImportError:
-    __version__ = "0.2.112"
+    __version__ = "0.2.113"
 
 logger = logging.getLogger(__name__)
 

{fleet_python-0.2.112 → fleet_python-0.2.113}/fleet/_async/judge.py

@@ -12,6 +12,8 @@ from ..judge import (
     JudgeResult,
     Rubric,
     _build_grade_request,
+    _collect_image_from_env_async,
+    _guess_media_type,
     _parse_grade_response,
 )
 
@@ -75,6 +77,29 @@ class AsyncJudge:
             collect: File patterns for orchestrator to collect (agentic mode).
             task_id: Optional task ID for tracking.
         """
+        # Resolve Image.from_env images asynchronously before building request
+        resolved_images = images
+        if images and not agentic:
+            resolved_images = {}
+            for label, img in images.items():
+                if img.source == "env" and img._env is not None:
+                    b64 = await _collect_image_from_env_async(img._env, img.filename)
+                    if b64 is not None:
+                        resolved_images[label] = Image.from_base64(
+                            b64,
+                            img.filename or "image.png",
+                            _guess_media_type(img.filename or "image.png"),
+                        )
+                    else:
+                        # Async collection failed — use collect source directly
+                        # (don't keep the env image or serialize() will retry sync)
+                        resolved_images[label] = Image(
+                            source="collect",
+                            filename=img.filename,
+                        )
+                else:
+                    resolved_images[label] = img
+
         body = _build_grade_request(
             self._instance_id,
             rubric,
@@ -84,7 +109,7 @@ class AsyncJudge:
             context=context,
             reference_claims=reference_claims,
             conversation=conversation,
-            images=images,
+            images=resolved_images,
             model=model,
             provider=provider,
             agentic=agentic,

{fleet_python-0.2.112 → fleet_python-0.2.113}/fleet/base.py

@@ -27,7 +27,7 @@ from .exceptions import (
 try:
     from . import __version__
 except ImportError:
-    __version__ = "0.2.112"
+    __version__ = "0.2.113"
 
 logger = logging.getLogger(__name__)
 

{fleet_python-0.2.112 → fleet_python-0.2.113}/fleet/judge.py

@@ -176,6 +176,8 @@ class Image:
                 "data": self.data,
                 "media_type": self.media_type or _guess_media_type(self.filename or "image.png"),
             }
+        elif self.source == "collect":
+            d = {"source": "collect", "selector": self.filename}
         elif self.source == "env":
             if agentic:
                 d = {"source": "collect", "selector": self.filename}
@@ -330,6 +332,86 @@ def _collect_image_from_env(env: Any, filename: str) -> Optional[str]:
     return None
 
 
+async def _collect_image_from_env_async(env: Any, filename: str) -> Optional[str]:
+    """Async version of _collect_image_from_env.
+
+    Collects an image from an AsyncEnv using DB -> notebook -> filesystem strategies.
+    Returns base64-encoded image data, or None if not found.
+    """
+    # Strategy 1: DB files table
+    try:
+        current = env.db("current")
+        where = f"path = '{filename}' OR path LIKE '%/{filename}'"
+        rows = _extract_query_rows(
+            await current.query(f"SELECT path, hex(content) AS content_hex FROM files WHERE {where}")
+        )
+        candidates = {}
+        for row in rows:
+            path, chex = row.get("path", ""), row.get("content_hex", "")
+            if path and chex:
+                try:
+                    candidates[path] = bytes.fromhex(chex)
+                except Exception:
+                    pass
+        # Prefer non-dataroom paths
+        non_dr = [p for p in candidates if not p.startswith("dataroom/")]
+        best = sorted(non_dr or list(candidates.keys()), key=len)
+        if best:
+            logger.debug("Loaded image from DB (async): %s", best[0])
+            return base64.b64encode(candidates[best[0]]).decode()
+    except Exception as e:
+        logger.debug("DB image query failed (async): %s", e)
+
+    # Strategy 2: Notebook cell outputs
+    try:
+        current = env.db("current")
+        nb_rows = _extract_query_rows(
+            await current.query(
+                "SELECT path, hex(content) AS content_hex FROM files "
+                "WHERE path LIKE 'notebooks/%.ipynb'"
+            )
+        )
+        for nb_row in nb_rows:
+            chex = nb_row.get("content_hex", "")
+            if not chex:
+                continue
+            try:
+                nb_bytes = bytes.fromhex(chex)
+                nb = json.loads(nb_bytes.decode("utf-8"))
+                for cell in reversed(nb.get("cells", [])):
+                    for output in cell.get("outputs", []):
+                        if output.get("output_type") in ("display_data", "execute_result"):
+                            img_data = output.get("data", {}).get("image/png")
+                            if img_data:
+                                if isinstance(img_data, list):
+                                    img_data = "".join(img_data)
+                                img_data = img_data.strip()
+                                if img_data:
+                                    logger.debug("Loaded image from notebook (async): %s", nb_row.get("path"))
+                                    return img_data
+            except Exception:
+                pass
+    except Exception as e:
+        logger.debug("Notebook image query failed (async): %s", e)
+
+    # Strategy 3: Filesystem fallback
+    search_paths = [
+        filename,
+        f"/app/workspace/{filename}",
+        f"/workspace/{filename}",
+    ]
+    for fp in search_paths:
+        try:
+            if os.path.exists(fp):
+                with open(fp, "rb") as f:
+                    logger.debug("Loaded image from filesystem (async): %s", fp)
+                    return base64.b64encode(f.read()).decode()
+        except Exception:
+            pass
+
+    return None
+
+
 # ---------------------------------------------------------------------------
 # Accumulator printing (verifier protocol)
 # ---------------------------------------------------------------------------

{fleet_python-0.2.112 → fleet_python-0.2.113/fleet_python.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fleet-python
-Version: 0.2.112
+Version: 0.2.113
 Summary: Python SDK for Fleet environments
 Author-email: Fleet AI <nic@fleet.so>
 License: Apache-2.0

{fleet_python-0.2.112 → fleet_python-0.2.113}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 [project]
 name = "fleet-python"
 
-version = "0.2.112"
+version = "0.2.113"
 description = "Python SDK for Fleet environments"
 authors = [
     {name = "Fleet AI", email = "nic@fleet.so"},