edison-client 0.10.1.dev352__tar.gz → 0.11.1.dev194__tar.gz

{edison_client-0.10.1.dev352 → edison_client-0.11.1.dev194}/AGENTS.md

@@ -138,8 +138,10 @@ async def main():
 
 - `clients/rest_client.py`: Main EdisonClient implementation with sync/async methods
 - `clients/job_client.py`: Job-specific client functionality
+- `clients/chat_methods.py`: Chat session SDK methods
 - `clients/data_storage_methods.py`: Data storage integration
 - `models/app.py`: TaskRequest, TaskResponse, RuntimeConfig models
+- `models/chat.py`: Chat session request/response models
 - `utils/auth.py`: Authentication utilities
 
 **Documentation:**

{edison_client-0.10.1.dev352/src/edison_client.egg-info → edison_client-0.11.1.dev194}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: edison-client
-Version: 0.10.1.dev352
+Version: 0.11.1.dev194
 Summary: A client for interacting with endpoints of the Edison Scientific service.
 Author-email: Edison Scientific technical staff <hello@futurehouse.org>
 License:                                  Apache License
@@ -221,7 +221,7 @@ Requires-Dist: google-resumable-media
 Requires-Dist: httpx
 Requires-Dist: httpx-aiohttp
 Requires-Dist: ldp>=0.22.0
-Requires-Dist: litellm
+Requires-Dist: litellm<=1.82.6
 Requires-Dist: openai>=2.7
 Requires-Dist: pydantic
 Requires-Dist: python-dotenv

{edison_client-0.10.1.dev352 → edison_client-0.11.1.dev194}/pyproject.toml

@@ -25,7 +25,7 @@ dependencies = [
     "httpx",
     "httpx-aiohttp",
     "ldp>=0.22.0",
-    "litellm",
+    "litellm<=1.82.6",  # Pinned: supply chain attack in 1.82.7 and 1.82.8
     "openai>=2.7",  # Pin to keep recent
     "pydantic",
     "python-dotenv",

{edison_client-0.10.1.dev352 → edison_client-0.11.1.dev194}/src/edison_client/clients/chat_methods.py

@@ -17,6 +17,8 @@ from edison_client.models.chat import (
     ChatRecoverPayload,
     ChatRecoverResponse,
     ConversationDetail,
+    QueueMessagePayload,
+    QueueMessageResponse,
     StoreMessagePayload,
 )
 from edison_client.utils.general import (
@@ -501,3 +503,103 @@ class ChatMethods:
             if isinstance(e, _BASE_CONNECTION_ERRORS):
                 raise
             raise ChatError(f"An unexpected error occurred: {e!r}") from e
+
+    # ── queue_chat_message ─────────────────────────────────────────────
+
+    @retry(
+        stop=stop_after_attempt(3),
+        wait=wait_exponential(multiplier=1, max=10),
+        retry=retry_if_connection_error,
+        before_sleep=before_sleep_log(logger, logging.WARNING),
+    )
+    def queue_chat_message(
+        self,
+        session_id: UUID,
+        project_id: UUID,
+        message: str,
+    ) -> QueueMessageResponse:
+        """Queue a message for injection into a running chat session.
+
+        The message is placed in the pending queue and picked up by the
+        rollout's ``before_transition`` hook on its next step.
+
+        Args:
+            session_id: Target chat session ID.
+            project_id: Project the session belongs to.
+            message: User message content.
+
+        Returns:
+            QueueMessageResponse with the pending message ID and target ID.
+
+        Raises:
+            ChatError: If there's an error queueing the message.
+        """
+        try:
+            payload = QueueMessagePayload(
+                target_id=session_id,
+                project_id=project_id,
+                message=message,
+            )
+            response = self.client.post(
+                "/v0.1/conversations/queue",
+                json=payload.model_dump(mode="json"),
+            )
+            response.raise_for_status()
+            return QueueMessageResponse.model_validate(response.json())
+        except HTTPStatusError as e:
+            self._handle_chat_http_errors(e, "queueing chat message")
+        except ChatError:
+            raise
+        except Exception as e:
+            if isinstance(e, _BASE_CONNECTION_ERRORS):
+                raise
+            raise ChatError(f"An unexpected error occurred: {e!r}") from e
+
+    @retry(
+        stop=stop_after_attempt(3),
+        wait=wait_exponential(multiplier=1, max=10),
+        retry=retry_if_connection_error,
+        before_sleep=before_sleep_log(logger, logging.WARNING),
+    )
+    async def aqueue_chat_message(
+        self,
+        session_id: UUID,
+        project_id: UUID,
+        message: str,
+    ) -> QueueMessageResponse:
+        """Queue a message for injection into a running chat session (async).
+
+        The message is placed in the pending queue and picked up by the
+        rollout's ``before_transition`` hook on its next step.
+
+        Args:
+            session_id: Target chat session ID.
+            project_id: Project the session belongs to.
+            message: User message content.
+
+        Returns:
+            QueueMessageResponse with the pending message ID and target ID.
+
+        Raises:
+            ChatError: If there's an error queueing the message.
+        """
+        try:
+            payload = QueueMessagePayload(
+                target_id=session_id,
+                project_id=project_id,
+                message=message,
+            )
+            response = await self.async_client.post(
+                "/v0.1/conversations/queue",
+                json=payload.model_dump(mode="json"),
+            )
+            response.raise_for_status()
+            return QueueMessageResponse.model_validate(response.json())
+        except HTTPStatusError as e:
+            self._handle_chat_http_errors(e, "queueing chat message")
+        except ChatError:
+            raise
+        except Exception as e:
+            if isinstance(e, _BASE_CONNECTION_ERRORS):
+                raise
+            raise ChatError(f"An unexpected error occurred: {e!r}") from e

{edison_client-0.10.1.dev352 → edison_client-0.11.1.dev194}/src/edison_client/clients/data_storage_methods.py

@@ -4,6 +4,7 @@ import json
 import logging
 import shutil
 import tempfile
+import time
 import zipfile
 from os import PathLike
 from pathlib import Path
@@ -643,59 +644,103 @@ class DataStorageMethods:  # noqa: PLR0904
         except Exception as e:
             raise DataStorageError(f"Failed to download from GCS: {e}") from e
 
-    def _download_from_gcs(self, signed_url: str, file_name: str | None = None) -> Path:
+    def _download_from_gcs(
+        self,
+        signed_url: str,
+        file_name: str | None = None,
+        max_retries: int = 5,
+        retry_wait_base: int = 2,
+    ) -> Path:
         """Download file from GCS using signed URL and handle unzipping if needed (sync version).
 
+        Resumes interrupted downloads using HTTP Range headers rather than restarting
+        from byte 0. Retries up to max_retries times with exponential backoff.
+
         Args:
             signed_url: The signed URL to download from
             file_name: The name of the file to download
+            max_retries: Maximum number of resume attempts on connection failure
+            retry_wait_base: Base seconds for exponential backoff between retries
         Returns:
             Path to the downloaded file (or unzipped directory if it was a zip)
         """
         file_name = file_name or "downloaded_file"
-
+        logger.info("Downloading file from GCS with signed URL: %s", signed_url)
         try:
             with tempfile.TemporaryDirectory() as temp_dir_str:
                 temp_dir = Path(temp_dir_str)
                 temp_file = temp_dir / file_name
+                offset = 0
+
+                for attempt in range(max_retries + 1):
+                    headers = {}
+                    if offset > 0:
+                        headers["Range"] = f"bytes={offset}-"
+
+                    try:
+                        with requests_lib.get(
+                            signed_url, stream=True, timeout=30, headers=headers
+                        ) as response:
+                            response.raise_for_status()
+
+                            if offset == 0:
+                                content_disposition = response.headers.get(
+                                    "content-disposition", ""
+                                )
+                                filename = file_name
+                                if "filename=" in content_disposition:
+                                    filename = content_disposition.split("filename=")[
+                                        -1
+                                    ].strip('"')
+                                if filename != file_name:
+                                    temp_file = temp_dir / filename
+
+                            mode = "ab" if offset > 0 else "wb"
+                            with open(temp_file, mode) as f:
+                                for chunk in response.iter_content(chunk_size=8192):
+                                    if chunk:
+                                        f.write(chunk)
+                                        offset += len(chunk)
+
+                        break  # download complete
+
+                    except (
+                        requests_lib.exceptions.ChunkedEncodingError,
+                        requests_lib.exceptions.ConnectionError,
+                        requests_lib.exceptions.Timeout,
+                    ) as e:
+                        if attempt == max_retries:
+                            raise
+                        wait = retry_wait_base**attempt
+                        logger.warning(
+                            f"Download interrupted at {offset:,} bytes "
+                            f"(attempt {attempt + 1}/{max_retries}), "
+                            f"resuming in {wait}s: {e}"
+                        )
+                        time.sleep(wait)
 
-                with requests_lib.get(signed_url, stream=True, timeout=30) as response:
-                    response.raise_for_status()
-
-                    content_disposition = response.headers.get(
-                        "content-disposition", ""
-                    )
-                    filename = file_name
-                    if "filename=" in content_disposition:
-                        filename = content_disposition.split("filename=")[-1].strip('"')
-
-                    if filename != file_name:
-                        temp_file = temp_dir / filename
-
-                    with open(temp_file, "wb") as f:
-                        f.writelines(response.iter_content(chunk_size=8192))
+                logger.debug(
+                    f"Downloaded file to {temp_file} (size: {temp_file.stat().st_size:,} bytes)"
+                )
 
-                    logger.debug(
-                        f"Downloaded file to {temp_file} (size: {temp_file.stat().st_size:,} bytes)"
-                    )
+                if self._is_zip_file(temp_file):
+                    logger.debug(f"File {temp_file} is a zip file, extracting...")
+                    extracted_path = self._extract_zip_file(temp_file, temp_dir)
 
-                    if self._is_zip_file(temp_file):
-                        logger.debug(f"File {temp_file} is a zip file, extracting...")
-                        extracted_path = self._extract_zip_file(temp_file, temp_dir)
+                    final_temp_dir = Path(tempfile.mkdtemp())
+                    final_path = final_temp_dir / extracted_path.name
 
-                        final_temp_dir = Path(tempfile.mkdtemp())
-                        final_path = final_temp_dir / extracted_path.name
+                    if extracted_path.is_dir():
+                        shutil.copytree(extracted_path, final_path)
+                    else:
+                        shutil.copy2(extracted_path, final_path)
 
-                        if extracted_path.is_dir():
-                            shutil.copytree(extracted_path, final_path)
-                        else:
-                            shutil.copy2(extracted_path, final_path)
+                    return final_path
 
-                        return final_path
-                    final_temp_dir = Path(tempfile.mkdtemp())
-                    final_file = final_temp_dir / temp_file.name
-                    shutil.copy2(temp_file, final_file)
-                    return final_file
+                final_temp_dir = Path(tempfile.mkdtemp())
+                final_file = final_temp_dir / temp_file.name
+                shutil.copy2(temp_file, final_file)
+                return final_file
 
         except Exception as e:
             raise DataStorageError(f"Failed to download from GCS: {e}") from e

{edison_client-0.10.1.dev352 → edison_client-0.11.1.dev194}/src/edison_client/clients/job_client.py

@@ -1,3 +1,6 @@
+import asyncio
+import gzip
+import json
 import logging
 from typing import ClassVar
 from uuid import UUID, uuid4
@@ -27,6 +30,7 @@ logger = logging.getLogger(__name__)
 
 class JobClient:
     REQUEST_TIMEOUT: ClassVar[float] = 30.0  # sec
+    LARGE_PAYLOAD_TIMEOUT: ClassVar[float] = 120.0  # sec, for streaming uploads
     MAX_RETRY_ATTEMPTS: ClassVar[int] = 3
     RETRY_MULTIPLIER: ClassVar[int] = 1
     MAX_RETRY_WAIT: ClassVar[int] = 10
@@ -141,13 +145,25 @@ class JobClient:
             trajectory_timestep=self.current_timestep,
         )
 
+        def _serialize_and_compress() -> tuple[bytes, int]:
+            json_bytes = json.dumps(data.model_dump(mode="json")).encode("utf-8")
+            return gzip.compress(json_bytes), len(json_bytes)
+
+        compressed, raw_size = await asyncio.to_thread(_serialize_and_compress)
+        logger.debug(
+            f"Compressed agent state payload: {raw_size} -> {len(compressed)} bytes "
+            f"({len(compressed) / raw_size:.1%})",
+        )
+
         try:
             async with httpx_aiohttp.HttpxAiohttpClient(
-                timeout=self.REQUEST_TIMEOUT
+                timeout=self.LARGE_PAYLOAD_TIMEOUT
             ) as client:
                 url = f"{self.base_uri}/v0.1/trajectories/{self.trajectory_id}/agent-state"
                 headers = {
                     "Authorization": f"Bearer {self.oauth_jwt}",
+                    "Content-Type": "application/json",
+                    "Content-Encoding": "gzip",
                     "x-trajectory-id": self.trajectory_id,
                 }
 
@@ -168,7 +184,7 @@ class JobClient:
 
                     response = await client.post(
                         url=url,
-                        json=data.model_dump(mode="json"),
+                        content=compressed,
                         headers=headers,
                     )
                 response.raise_for_status()
@@ -182,7 +198,7 @@ class JobClient:
             )
         except httpx.TimeoutException:
             logger.exception(
-                f"Timeout while storing agent state after {self.REQUEST_TIMEOUT}s",
+                f"Timeout while storing agent state after {self.LARGE_PAYLOAD_TIMEOUT}s",
             )
             raise
         except httpx.NetworkError:

{edison_client-0.10.1.dev352 → edison_client-0.11.1.dev194}/src/edison_client/models/app.py

@@ -63,7 +63,9 @@ class JobNames(StrEnum):
         """Get human-readable descriptions for each job type."""
         return {
             cls.LITERATURE: "Paper analysis and literature review",
-            cls.LITERATURE_HIGH: "High-reasoning mode paper analysis and literature review",
+            cls.LITERATURE_HIGH: (
+                "High-reasoning mode paper analysis and literature review"
+            ),
             cls.CROW: "Paper analysis and literature review",
             cls.FALCON: "Paper analysis and literature review",
             cls.ANALYSIS: "Data analysis with Python/notebooks",
@@ -413,6 +415,10 @@ class DockerContainerConfiguration(BaseModel):
         default="10Gi",
         description="Disk space for the agent workspace volume (e.g., '10Gi')",
     )
+    tmp_size: str = Field(
+        default="5Gi",
+        description="Size limit for the /tmp emptyDir volume (e.g., '5Gi')",
+    )
     gpu_count: int | None = Field(
         default=None,
         description="Number of NVIDIA GPUs to allocate. Requires CELERY backend.",
@@ -444,6 +450,19 @@ class DockerContainerConfiguration(BaseModel):
             raise ValueError("Disk space must not exceed 100Gi")
         return v
 
+    @field_validator("tmp_size")
+    @classmethod
+    def validate_tmp_size(cls, v: str) -> str:
+        match = re.match(r"^(\d+)Gi$", v)
+        if not match:
+            raise ValueError("tmp_size must be in Gi format (e.g., '5Gi')")
+        value = int(match.group(1))
+        if value < 1:
+            raise ValueError("tmp_size must be at least 1Gi")
+        if value > 50:  # noqa: PLR2004
+            raise ValueError("tmp_size must not exceed 50Gi")
+        return v
+
     @field_validator("memory")
     @classmethod
     def validate_memory(cls, v: str) -> str:
@@ -977,6 +996,14 @@ class TaskRequest(BaseModel):
         default=None,
         description="Project ID for the /conversations/queue completion notification",
     )
+    tags: list[str] | None = Field(
+        default=None,
+        description="Grouping association tags for a task.",
+    )
+    workspace_id: str | None = Field(
+        default=None,
+        description="JuiceFS workspace identifier for persistent storage across sandbox pods.",
+    )
 
     @model_validator(mode="after")
     def validate_caller_fields(self) -> Self:

{edison_client-0.10.1.dev352 → edison_client-0.11.1.dev194}/src/edison_client/models/chat.py

@@ -24,7 +24,9 @@ class ChatMessagePayload(BaseModel):
     message: str = Field(description="User message content")
     ttl_seconds: int | None = Field(
         default=None,
-        description="Optional sandbox TTL override in seconds (60-86400)",
+        ge=60,
+        le=86400,
+        description="Optional sandbox TTL override in seconds",
     )
     job_name: str | None = Field(
         default=None, description="Job name identifying which agent to chat with"
@@ -53,7 +55,7 @@ class ChatRecoverPayload(BaseModel):
     project_id: UUID = Field(description="Project ID for recovery")
     job_name: str | None = Field(
         default=None,
-        description="Job name for the agent (used to resolve the sandbox template on recovery)",
+        description="Job name for the agent (used to resolve the sandbox template on recovery) -- optional will infer recovery template from the default service template (dummy-env)",
     )
 
 
@@ -93,6 +95,7 @@ class ChatConversation(BaseModel):
     created_at: datetime = Field(
         description="Timestamp when the conversation was created"
     )
+    # stolen from the service side code, but we should consider a more robust schema
     messages: list[Any] = Field(description="The messages in this conversation")
 
 
@@ -100,7 +103,7 @@ class ChatConversationList(BaseModel):
     """Response listing conversations."""
 
     conversations: list[ChatConversation] = Field(
-        description="List of conversations grouped by session_id"
+        description="List of conversations associated with the provided session_id if any"
     )
 
 
@@ -122,3 +125,18 @@ class StoreMessagePayload(BaseModel):
     metadata: dict | None = Field(
         default=None, description="Optional metadata to attach to the message"
     )
+
+
+class QueueMessagePayload(BaseModel):
+    """Payload for queueing a message to a running rollout's pending queue."""
+
+    target_id: UUID = Field(description="Session ID (chat) or trajectory ID (run)")
+    project_id: UUID = Field(description="Project the target belongs to")
+    message: str = Field(description="User message content")
+
+
+class QueueMessageResponse(BaseModel):
+    """Response from the queue endpoint."""
+
+    id: UUID = Field(description="Pending message ID")
+    target_id: UUID = Field(description="Session or trajectory ID")

edison_client-0.11.1.dev194/src/edison_client/version.py

@@ -0,0 +1,24 @@
+# file generated by vcs-versioning
+# don't change, don't track in version control
+from __future__ import annotations
+
+__all__ = [
+    "__version__",
+    "__version_tuple__",
+    "version",
+    "version_tuple",
+    "__commit_id__",
+    "commit_id",
+]
+
+version: str
+__version__: str
+__version_tuple__: tuple[int | str, ...]
+version_tuple: tuple[int | str, ...]
+commit_id: str | None
+__commit_id__: str | None
+
+__version__ = version = '0.11.1.dev194'
+__version_tuple__ = version_tuple = (0, 11, 1, 'dev194')
+
+__commit_id__ = commit_id = 'gb74b18d01'

{edison_client-0.10.1.dev352 → edison_client-0.11.1.dev194/src/edison_client.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: edison-client
-Version: 0.10.1.dev352
+Version: 0.11.1.dev194
 Summary: A client for interacting with endpoints of the Edison Scientific service.
 Author-email: Edison Scientific technical staff <hello@futurehouse.org>
 License:                                  Apache License
@@ -221,7 +221,7 @@ Requires-Dist: google-resumable-media
 Requires-Dist: httpx
 Requires-Dist: httpx-aiohttp
 Requires-Dist: ldp>=0.22.0
-Requires-Dist: litellm
+Requires-Dist: litellm<=1.82.6
 Requires-Dist: openai>=2.7
 Requires-Dist: pydantic
 Requires-Dist: python-dotenv

{edison_client-0.10.1.dev352 → edison_client-0.11.1.dev194}/src/edison_client.egg-info/SOURCES.txt

@@ -41,6 +41,7 @@ tests/test_client.py
 tests/test_client_close.py
 tests/test_data_storage_e2e.py
 tests/test_data_storage_methods.py
+tests/test_docker_container_config.py
 tests/test_dss_restclient_api_e2e.py
 tests/test_min_replicas_validation.py
 tests/test_rest.py

{edison_client-0.10.1.dev352 → edison_client-0.11.1.dev194}/src/edison_client.egg-info/requires.txt

@@ -5,7 +5,7 @@ google-resumable-media
 httpx
 httpx-aiohttp
 ldp>=0.22.0
-litellm
+litellm<=1.82.6
 openai>=2.7
 pydantic
 python-dotenv

{edison_client-0.10.1.dev352 → edison_client-0.11.1.dev194}/tests/client_e2e_uploaded_file_trajectory_test.py

@@ -12,7 +12,7 @@ from edison_client.models.data_storage_methods import RawFetchResponse
 pytestmark = [pytest.mark.live, pytest.mark.live_agent]
 
 
-@pytest.mark.timeout(660)
+@pytest.mark.timeout(180)
 def test_upload_and_run_analysis_task(admin_client: RestClient):
     """Test uploading a file with prompt, running analysis task, and downloading outputs."""
     prompt_content = "draw a blue square"
@@ -40,7 +40,7 @@ def test_upload_and_run_analysis_task(admin_client: RestClient):
             name=JobNames.ANALYSIS,  # Use the data analysis crow
             query="Open the attached file and follow the instructions in it. Create an image as specified.",
             runtime_config=RuntimeConfig(
-                timeout=200,
+                timeout=180,
             ),
         )
 
@@ -49,7 +49,7 @@ def test_upload_and_run_analysis_task(admin_client: RestClient):
             files=[data_entry_uri],  # Simple file attachment!
             verbose=True,
             progress_bar=True,
-            timeout=600,
+            timeout=180,
         )
 
         assert len(results) == 1

{edison_client-0.10.1.dev352 → edison_client-0.11.1.dev194}/tests/test_chat_methods.py

@@ -1,5 +1,6 @@
 import asyncio
 import time
+from collections.abc import Generator
 from uuid import UUID, uuid4
 
 import pytest
@@ -10,6 +11,7 @@ from edison_client.models.chat import (
     ChatConversationList,
     ChatMessageResponse,
     ConversationDetail,
+    QueueMessageResponse,
 )
 
 CHAT_JOB_NAME = "job-futurehouse-dummy-env"
@@ -20,9 +22,11 @@ RETRY_ATTEMPTS = 3
 
 
 @pytest.fixture(name="chat_project_id")
-def fixture_chat_project_id(admin_client: RestClient) -> UUID:
-    """Create a fresh project for chat tests."""
-    return admin_client.create_project(f"chat-e2e-{uuid4()}")
+def fixture_chat_project_id(admin_client: RestClient) -> Generator[UUID, None, None]:
+    """Create a fresh project for chat tests, clean up afterwards."""
+    project_id = admin_client.create_project(f"chat-e2e-{uuid4()}")
+    yield project_id
+    admin_client.delete_project(project_id)
 
 
 class TestSyncChatLifecycle:
@@ -133,6 +137,37 @@ class TestSyncChatLifecycle:
             )
         assert "not found" in str(exc_info.value).lower()
 
+    def test_queue_chat_message_bad_project(self, admin_client: RestClient):
+        """Queueing a message with a non-existent project_id returns 404."""
+        with pytest.raises(ChatError, match="404") as exc_info:
+            admin_client.queue_chat_message(
+                session_id=uuid4(),
+                project_id=uuid4(),
+                message="Should fail",
+            )
+        assert "not found" in str(exc_info.value).lower()
+
+    @pytest.mark.live_agent
+    @pytest.mark.timeout(TEST_TIMEOUT)
+    @pytest.mark.flaky(reruns=RETRY_ATTEMPTS)
+    def test_queue_chat_message(self, admin_client: RestClient, chat_project_id: UUID):
+        """Queue a message into a running chat session (sync)."""
+        send_result = admin_client.send_chat_message(
+            project_id=chat_project_id,
+            message="Setting up for queue test",
+            job_name=CHAT_JOB_NAME,
+        )
+        session_id = send_result.session_id
+
+        result = admin_client.queue_chat_message(
+            session_id=session_id,
+            project_id=chat_project_id,
+            message="Injected via queue",
+        )
+        assert isinstance(result, QueueMessageResponse)
+        assert result.id is not None
+        assert result.target_id == session_id
+
 
 class TestAsyncChatLifecycle:
     """Async e2e: asend_chat_message → aget_conversations → aget_conversation → astore_conversation_message."""
@@ -246,3 +281,38 @@ class TestAsyncChatLifecycle:
                 message={"role": "user", "content": "orphan"},
             )
         assert "not found" in str(exc_info.value).lower()
+
+    @pytest.mark.asyncio
+    async def test_aqueue_chat_message_bad_project(self, admin_client: RestClient):
+        """Queueing a message with a non-existent project_id returns 404 (async)."""
+        with pytest.raises(ChatError, match="404") as exc_info:
+            await admin_client.aqueue_chat_message(
+                session_id=uuid4(),
+                project_id=uuid4(),
+                message="Should fail",
+            )
+        assert "not found" in str(exc_info.value).lower()
+
+    @pytest.mark.live_agent
+    @pytest.mark.timeout(TEST_TIMEOUT)
+    @pytest.mark.flaky(reruns=RETRY_ATTEMPTS)
+    @pytest.mark.asyncio
+    async def test_aqueue_chat_message(
+        self, admin_client: RestClient, chat_project_id: UUID
+    ):
+        """Queue a message into a running chat session (async)."""
+        send_result = await admin_client.asend_chat_message(
+            project_id=chat_project_id,
+            message="Setting up for async queue test",
+            job_name=CHAT_JOB_NAME,
+        )
+        session_id = send_result.session_id
+
+        result = await admin_client.aqueue_chat_message(
+            session_id=session_id,
+            project_id=chat_project_id,
+            message="Injected via async queue",
+        )
+        assert isinstance(result, QueueMessageResponse)
+        assert result.id is not None
+        assert result.target_id == session_id

{edison_client-0.10.1.dev352 → edison_client-0.11.1.dev194}/tests/test_data_storage_e2e.py

@@ -626,7 +626,7 @@ class TestDataStorageSearch:
         dataset_id_test_criteria = SearchCriterion(
             field="dataset_id",
             operator=SearchOperator.EQUALS,
-            value="8bb27d71-8ec6-4845-a09e-4eceefb38cba",
+            value="328375e2-591e-4cc9-b17b-cccf040c343a",
         )
 
         description_test_criteria = SearchCriterion(

edison_client-0.11.1.dev194/tests/test_docker_container_config.py

@@ -0,0 +1,28 @@
+import pytest
+from pydantic import ValidationError
+
+from edison_client.models.app import DockerContainerConfiguration
+
+
+def test_tmp_size_default():
+    config = DockerContainerConfiguration(cpu="1", memory="2Gi")
+    assert config.tmp_size == "5Gi"
+
+
+def test_tmp_size_custom():
+    config = DockerContainerConfiguration(cpu="4", memory="8Gi", tmp_size="20Gi")
+    assert config.tmp_size == "20Gi"
+
+
+@pytest.mark.parametrize(
+    ("tmp_size", "error_match"),
+    [
+        ("0Gi", "tmp_size must be at least 1Gi"),
+        ("51Gi", "tmp_size must not exceed 50Gi"),
+        ("5MB", "tmp_size must be in Gi format"),
+        ("abc", "tmp_size must be in Gi format"),
+    ],
+)
+def test_tmp_size_validation(tmp_size, error_match):
+    with pytest.raises(ValidationError, match=error_match):
+        DockerContainerConfiguration(cpu="4", memory="8Gi", tmp_size=tmp_size)

{edison_client-0.10.1.dev352 → edison_client-0.11.1.dev194}/tests/test_rest.py

@@ -95,7 +95,7 @@ def fixture_running_trajectory_id(
 
 
 @pytest.mark.live_agent
-@pytest.mark.timeout(300)
+@pytest.mark.timeout(120)
 @pytest.mark.flaky(reruns=3)
 def test_futurehouse_dummy_env_crow(admin_client: RestClient, task_req: TaskRequest):
     task_id = admin_client.create_task(task_req)
@@ -240,7 +240,7 @@ def test_deployment_config_hidden_from_non_fh_admin(pub_client: RestClient):
 
 
 @pytest.mark.live_agent
-@pytest.mark.timeout(300)
+@pytest.mark.timeout(240)
 @pytest.mark.flaky(reruns=3)
 def test_run_until_done_futurehouse_dummy_env_crow(
     admin_client: RestClient, task_req: TaskRequest
@@ -254,7 +254,7 @@ def test_run_until_done_futurehouse_dummy_env_crow(
 
 
 @pytest.mark.live_agent
-@pytest.mark.timeout(300)
+@pytest.mark.timeout(240)
 @pytest.mark.flaky(reruns=3)
 def test_run_until_done_returns_task_response(
     admin_client: RestClient, task_req: TaskRequest
@@ -274,7 +274,7 @@ def test_run_until_done_returns_task_response(
 
 
 @pytest.mark.live_agent
-@pytest.mark.timeout(300)
+@pytest.mark.timeout(240)
 @pytest.mark.flaky(reruns=3)
 @pytest.mark.asyncio
 async def test_arun_until_done_futurehouse_dummy_env_crow(
@@ -289,7 +289,7 @@ async def test_arun_until_done_futurehouse_dummy_env_crow(
 
 
 @pytest.mark.live_agent
-@pytest.mark.timeout(300)
+@pytest.mark.timeout(240)
 @pytest.mark.flaky(reruns=3)
 @pytest.mark.asyncio
 async def test_arun_until_done_returns_task_response(
@@ -310,13 +310,13 @@ async def test_arun_until_done_returns_task_response(
 
 
 @pytest.mark.live_agent
-@pytest.mark.timeout(500)
+@pytest.mark.timeout(180)
 def test_continuation_task(admin_client: RestClient):
     """Test running a continuation task against data-analysis-crow-high."""
     initial_task = TaskRequest(
         name=JobNames.ANALYSIS,
         query="What is 2 + 2? Just give me the number.",
-        runtime_config=RuntimeConfig(timeout=200),
+        runtime_config=RuntimeConfig(timeout=180),
     )
     print(initial_task)
 
@@ -324,7 +324,7 @@ def test_continuation_task(admin_client: RestClient):
         task_data=initial_task,
         verbose=True,
         progress_bar=True,
-        timeout=220,
+        timeout=180,
     )
 
     assert len(initial_results) == 1
@@ -338,7 +338,7 @@ def test_continuation_task(admin_client: RestClient):
         name=JobNames.ANALYSIS,
         query="Now multiply that result by 3. Just give me the number.",
         runtime_config=RuntimeConfig(
-            timeout=200,
+            timeout=180,
             continued_job_id=initial_result.task_id,
         ),
     )
@@ -347,7 +347,7 @@ def test_continuation_task(admin_client: RestClient):
         task_data=continuation_task,
         verbose=True,
         progress_bar=True,
-        timeout=220,
+        timeout=180,
     )
 
     assert len(continuation_results) == 1
@@ -359,7 +359,7 @@ def test_continuation_task(admin_client: RestClient):
 
 
 @pytest.mark.live_agent
-@pytest.mark.timeout(300)
+@pytest.mark.timeout(60)
 @pytest.mark.flaky(reruns=3)
 @pytest.mark.asyncio
 async def test_timeout_run_until_done_futurehouse_dummy_env_crow(
@@ -622,7 +622,7 @@ class TestAsyncProjectOperations:
 
 
 @pytest.mark.live_agent
-@pytest.mark.timeout(300)
+@pytest.mark.timeout(120)
 @pytest.mark.flaky(reruns=3)
 def test_get_tasks_with_project_filter(admin_client: RestClient, task_req: TaskRequest):
     """Test retrieving trajectories filtered by project_id using real API calls."""
@@ -645,7 +645,7 @@ def test_get_tasks_with_project_filter(admin_client: RestClient, task_req: TaskR
 
 
 @pytest.mark.live_agent
-@pytest.mark.timeout(300)
+@pytest.mark.timeout(120)
 @pytest.mark.flaky(reruns=3)
 @pytest.mark.asyncio
 async def test_aget_tasks_with_project_filter(

edison_client-0.10.1.dev352/src/edison_client/version.py

@@ -1,34 +0,0 @@
-# file generated by setuptools-scm
-# don't change, don't track in version control
-
-__all__ = [
-    "__version__",
-    "__version_tuple__",
-    "version",
-    "version_tuple",
-    "__commit_id__",
-    "commit_id",
-]
-
-TYPE_CHECKING = False
-if TYPE_CHECKING:
-    from typing import Tuple
-    from typing import Union
-
-    VERSION_TUPLE = Tuple[Union[int, str], ...]
-    COMMIT_ID = Union[str, None]
-else:
-    VERSION_TUPLE = object
-    COMMIT_ID = object
-
-version: str
-__version__: str
-__version_tuple__: VERSION_TUPLE
-version_tuple: VERSION_TUPLE
-commit_id: COMMIT_ID
-__commit_id__: COMMIT_ID
-
-__version__ = version = '0.10.1.dev352'
-__version_tuple__ = version_tuple = (0, 10, 1, 'dev352')
-
-__commit_id__ = commit_id = 'gbbd65e731'

{edison_client-0.10.1.dev352 → edison_client-0.11.1.dev194}/src/edison_client/utils/module_utils.py

@@ -75,14 +75,14 @@ def fetch_environment_function_docstring(
         directory (Path): The base directory containing the environment files.
         function_name (str): The name of the function to retrieve the docstring for.
 
+    Returns:
+        str | None: The docstring of the specified function, or None if not found.
+
     Raises:
         ValueError: If multiple classes with the same name are found in different files
                     (making the intended class ambiguous), an error is raised requiring
                     disambiguation.
 
-    Returns:
-        str | None: The docstring of the specified function, or None if not found.
-
     """
     parts = environment_name.split(".")
     class_name = parts[-1]