cyberdesk 2.0.0__py3-none-any.whl → 2.1.1__py3-none-any.whl

cyberdesk/__init__.py

@@ -27,7 +27,7 @@ from .client import (
     AttachmentType,
 )
 
-__version__ = "2.0.0"
+__version__ = "2.1.1"
 
 __all__ = [
     "CyberdeskClient",

cyberdesk/client.py

@@ -4,6 +4,7 @@ from uuid import UUID
 from pathlib import Path
 import httpx
 from dataclasses import dataclass
+from datetime import datetime, timezone
 
 # Import the generated client
 from openapi_client.cyberdesk_cloud_client import AuthenticatedClient
@@ -14,6 +15,17 @@ from openapi_client.cyberdesk_cloud_client.api.machines import (
     get_machine_v1_machines_machine_id_get,
     update_machine_v1_machines_machine_id_patch,
     delete_machine_v1_machines_machine_id_delete,
+    get_machine_pools_v1_machines_machine_id_pools_get,
+    update_machine_pools_v1_machines_machine_id_pools_put,
+)
+from openapi_client.cyberdesk_cloud_client.api.pools import (
+    list_pools_v1_pools_get,
+    create_pool_v1_pools_post,
+    get_pool_v1_pools_pool_id_get,
+    update_pool_v1_pools_pool_id_patch,
+    delete_pool_v1_pools_pool_id_delete,
+    add_machines_to_pool_v1_pools_pool_id_machines_post,
+    remove_machines_from_pool_v1_pools_pool_id_machines_delete,
 )
 from openapi_client.cyberdesk_cloud_client.api.workflows import (
     list_workflows_v1_workflows_get,
@@ -58,6 +70,12 @@ from openapi_client.cyberdesk_cloud_client.models import (
     MachineUpdate,
     MachineResponse,
     MachineStatus,
+    MachinePoolUpdate,
+    PoolCreate,
+    PoolUpdate,
+    PoolResponse,
+    PoolWithMachines,
+    MachinePoolAssignment,
     WorkflowCreate,
     WorkflowUpdate,
     WorkflowResponse,
@@ -80,6 +98,7 @@ from openapi_client.cyberdesk_cloud_client.models import (
     RunAttachmentDownloadUrlResponse,
     AttachmentType,
     PaginatedResponseMachineResponse,
+    PaginatedResponsePoolResponse,
     PaginatedResponseWorkflowResponse,
     PaginatedResponseRunResponse,
     PaginatedResponseConnectionResponse,
@@ -94,6 +113,12 @@ __all__ = [
     "MachineUpdate",
     "MachineResponse",
     "MachineStatus",
+    "MachinePoolUpdate",
+    "PoolCreate",
+    "PoolUpdate",
+    "PoolResponse",
+    "PoolWithMachines",
+    "MachinePoolAssignment",
     "WorkflowCreate",
     "WorkflowUpdate",
     "WorkflowResponse",
@@ -139,6 +164,26 @@ def _to_unset_or_value(value: Optional[Any]) -> Union[Unset, Any]:
     return UNSET if value is None else value
 
 
+def _to_iso_utc_str(value: Optional[Union[str, datetime]]) -> Optional[str]:
+    """Convert a datetime or string to an ISO8601 UTC string.
+
+    - If value is a string, it is returned unchanged (assumed to be ISO8601)
+    - If value is a timezone-aware datetime, it is converted to UTC ISO string
+    - If value is a naive datetime, it is treated as UTC and suffixed accordingly
+    - If value is None, returns None
+    """
+    if value is None:
+        return None
+    if isinstance(value, str):
+        return value
+    if isinstance(value, datetime):
+        if value.tzinfo is None:
+            return value.replace(tzinfo=timezone.utc).isoformat()
+        return value.astimezone(timezone.utc).isoformat()
+    return None
+
+
+
 class MachinesAPI:
     """Machines API endpoints."""
     
@@ -149,15 +194,28 @@ class MachinesAPI:
         self,
         skip: Optional[int] = None,
         limit: Optional[int] = None,
-        status: Optional[MachineStatus] = None
+        status: Optional[MachineStatus] = None,
+        *,
+        created_at_from: Optional[Union[str, datetime]] = None,
+        created_at_to: Optional[Union[str, datetime]] = None,
     ) -> ApiResponse:
-        """List machines with optional filtering."""
+        """List machines with optional filtering.
+
+        Args:
+            skip: Pagination skip
+            limit: Pagination limit
+            status: Machine status filter
+            created_at_from: Optional start datetime (UTC or ISO string)
+            created_at_to: Optional end datetime (UTC or ISO string)
+        """
         try:
             response = await list_machines_v1_machines_get.asyncio(
                 client=self.client,
                 skip=_to_unset_or_value(skip),
                 limit=_to_unset_or_value(limit),
-                status=status
+                status=status,
+                created_at_from=_to_unset_or_value(_to_iso_utc_str(created_at_from)),
+                created_at_to=_to_unset_or_value(_to_iso_utc_str(created_at_to)),
             )
             return ApiResponse(data=response)
         except Exception as e:
@@ -167,7 +225,10 @@ class MachinesAPI:
         self,
         skip: Optional[int] = None,
         limit: Optional[int] = None,
-        status: Optional[MachineStatus] = None
+        status: Optional[MachineStatus] = None,
+        *,
+        created_at_from: Optional[Union[str, datetime]] = None,
+        created_at_to: Optional[Union[str, datetime]] = None,
     ) -> ApiResponse:
         """List machines with optional filtering (synchronous)."""
         try:
@@ -175,7 +236,9 @@ class MachinesAPI:
                 client=self.client,
                 skip=_to_unset_or_value(skip),
                 limit=_to_unset_or_value(limit),
-                status=status
+                status=status,
+                created_at_from=_to_unset_or_value(_to_iso_utc_str(created_at_from)),
+                created_at_to=_to_unset_or_value(_to_iso_utc_str(created_at_to)),
             )
             return ApiResponse(data=response)
         except Exception as e:
@@ -270,6 +333,299 @@ class MachinesAPI:
             return ApiResponse(data={"success": True})
         except Exception as e:
             return ApiResponse(error=e)
+    
+    async def get_pools(self, machine_id: str) -> ApiResponse:
+        """Get all pools that a machine belongs to."""
+        try:
+            response = await get_machine_pools_v1_machines_machine_id_pools_get.asyncio(
+                client=self.client,
+                machine_id=_to_uuid(machine_id)
+            )
+            return ApiResponse(data=response)
+        except Exception as e:
+            return ApiResponse(error=e)
+    
+    def get_pools_sync(self, machine_id: str) -> ApiResponse:
+        """Get all pools that a machine belongs to (synchronous)."""
+        try:
+            response = get_machine_pools_v1_machines_machine_id_pools_get.sync(
+                client=self.client,
+                machine_id=_to_uuid(machine_id)
+            )
+            return ApiResponse(data=response)
+        except Exception as e:
+            return ApiResponse(error=e)
+    
+    async def update_pools(self, machine_id: str, data: MachinePoolUpdate) -> ApiResponse:
+        """Update a machine's pool assignments.
+        
+        This replaces all existing pool assignments with the new ones.
+        
+        Args:
+            machine_id: The machine ID
+            data: MachinePoolUpdate with pool_ids list
+        
+        Returns:
+            ApiResponse with updated MachineResponse
+        """
+        try:
+            response = await update_machine_pools_v1_machines_machine_id_pools_put.asyncio(
+                client=self.client,
+                machine_id=_to_uuid(machine_id),
+                body=data
+            )
+            return ApiResponse(data=response)
+        except Exception as e:
+            return ApiResponse(error=e)
+    
+    def update_pools_sync(self, machine_id: str, data: MachinePoolUpdate) -> ApiResponse:
+        """Update a machine's pool assignments (synchronous).
+        
+        This replaces all existing pool assignments with the new ones.
+        
+        Args:
+            machine_id: The machine ID
+            data: MachinePoolUpdate with pool_ids list
+        
+        Returns:
+            ApiResponse with updated MachineResponse
+        """
+        try:
+            response = update_machine_pools_v1_machines_machine_id_pools_put.sync(
+                client=self.client,
+                machine_id=_to_uuid(machine_id),
+                body=data
+            )
+            return ApiResponse(data=response)
+        except Exception as e:
+            return ApiResponse(error=e)
+
+
+class PoolsAPI:
+    """Pools API endpoints."""
+    
+    def __init__(self, client: AuthenticatedClient):
+        self.client = client
+    
+    async def list(
+        self,
+        skip: Optional[int] = None,
+        limit: Optional[int] = None
+    ) -> ApiResponse:
+        """List pools for the organization."""
+        try:
+            response = await list_pools_v1_pools_get.asyncio(
+                client=self.client,
+                skip=_to_unset_or_value(skip),
+                limit=_to_unset_or_value(limit)
+            )
+            return ApiResponse(data=response)
+        except Exception as e:
+            return ApiResponse(error=e)
+    
+    def list_sync(
+        self,
+        skip: Optional[int] = None,
+        limit: Optional[int] = None
+    ) -> ApiResponse:
+        """List pools for the organization (synchronous)."""
+        try:
+            response = list_pools_v1_pools_get.sync(
+                client=self.client,
+                skip=_to_unset_or_value(skip),
+                limit=_to_unset_or_value(limit)
+            )
+            return ApiResponse(data=response)
+        except Exception as e:
+            return ApiResponse(error=e)
+    
+    async def create(self, data: PoolCreate) -> ApiResponse:
+        """Create a new pool."""
+        try:
+            response = await create_pool_v1_pools_post.asyncio(
+                client=self.client,
+                body=data
+            )
+            return ApiResponse(data=response)
+        except Exception as e:
+            return ApiResponse(error=e)
+    
+    def create_sync(self, data: PoolCreate) -> ApiResponse:
+        """Create a new pool (synchronous)."""
+        try:
+            response = create_pool_v1_pools_post.sync(
+                client=self.client,
+                body=data
+            )
+            return ApiResponse(data=response)
+        except Exception as e:
+            return ApiResponse(error=e)
+    
+    async def get(self, pool_id: str, include_machines: bool = False) -> ApiResponse:
+        """Get a specific pool by ID.
+        
+        Args:
+            pool_id: The pool ID
+            include_machines: Whether to include full machine details
+        
+        Returns:
+            ApiResponse with PoolResponse or PoolWithMachines
+        """
+        try:
+            response = await get_pool_v1_pools_pool_id_get.asyncio(
+                client=self.client,
+                pool_id=_to_uuid(pool_id),
+                include_machines=include_machines
+            )
+            return ApiResponse(data=response)
+        except Exception as e:
+            return ApiResponse(error=e)
+    
+    def get_sync(self, pool_id: str, include_machines: bool = False) -> ApiResponse:
+        """Get a specific pool by ID (synchronous).
+        
+        Args:
+            pool_id: The pool ID
+            include_machines: Whether to include full machine details
+        
+        Returns:
+            ApiResponse with PoolResponse or PoolWithMachines
+        """
+        try:
+            response = get_pool_v1_pools_pool_id_get.sync(
+                client=self.client,
+                pool_id=_to_uuid(pool_id),
+                include_machines=include_machines
+            )
+            return ApiResponse(data=response)
+        except Exception as e:
+            return ApiResponse(error=e)
+    
+    async def update(self, pool_id: str, data: PoolUpdate) -> ApiResponse:
+        """Update a pool's details."""
+        try:
+            response = await update_pool_v1_pools_pool_id_patch.asyncio(
+                client=self.client,
+                pool_id=_to_uuid(pool_id),
+                body=data
+            )
+            return ApiResponse(data=response)
+        except Exception as e:
+            return ApiResponse(error=e)
+    
+    def update_sync(self, pool_id: str, data: PoolUpdate) -> ApiResponse:
+        """Update a pool's details (synchronous)."""
+        try:
+            response = update_pool_v1_pools_pool_id_patch.sync(
+                client=self.client,
+                pool_id=_to_uuid(pool_id),
+                body=data
+            )
+            return ApiResponse(data=response)
+        except Exception as e:
+            return ApiResponse(error=e)
+    
+    async def delete(self, pool_id: str) -> ApiResponse:
+        """Delete a pool. This will not delete the machines in the pool."""
+        try:
+            await delete_pool_v1_pools_pool_id_delete.asyncio(
+                client=self.client,
+                pool_id=_to_uuid(pool_id)
+            )
+            return ApiResponse(data={"success": True})
+        except Exception as e:
+            return ApiResponse(error=e)
+    
+    def delete_sync(self, pool_id: str) -> ApiResponse:
+        """Delete a pool. This will not delete the machines in the pool (synchronous)."""
+        try:
+            delete_pool_v1_pools_pool_id_delete.sync(
+                client=self.client,
+                pool_id=_to_uuid(pool_id)
+            )
+            return ApiResponse(data={"success": True})
+        except Exception as e:
+            return ApiResponse(error=e)
+    
+    async def add_machines(self, pool_id: str, data: MachinePoolAssignment) -> ApiResponse:
+        """Add machines to a pool.
+        
+        Args:
+            pool_id: The pool ID
+            data: MachinePoolAssignment with machine_ids list
+        
+        Returns:
+            ApiResponse with PoolWithMachines
+        """
+        try:
+            response = await add_machines_to_pool_v1_pools_pool_id_machines_post.asyncio(
+                client=self.client,
+                pool_id=_to_uuid(pool_id),
+                body=data
+            )
+            return ApiResponse(data=response)
+        except Exception as e:
+            return ApiResponse(error=e)
+    
+    def add_machines_sync(self, pool_id: str, data: MachinePoolAssignment) -> ApiResponse:
+        """Add machines to a pool (synchronous).
+        
+        Args:
+            pool_id: The pool ID
+            data: MachinePoolAssignment with machine_ids list
+        
+        Returns:
+            ApiResponse with PoolWithMachines
+        """
+        try:
+            response = add_machines_to_pool_v1_pools_pool_id_machines_post.sync(
+                client=self.client,
+                pool_id=_to_uuid(pool_id),
+                body=data
+            )
+            return ApiResponse(data=response)
+        except Exception as e:
+            return ApiResponse(error=e)
+    
+    async def remove_machines(self, pool_id: str, data: MachinePoolAssignment) -> ApiResponse:
+        """Remove machines from a pool.
+        
+        Args:
+            pool_id: The pool ID
+            data: MachinePoolAssignment with machine_ids list
+        
+        Returns:
+            ApiResponse with success status
+        """
+        try:
+            await remove_machines_from_pool_v1_pools_pool_id_machines_delete.asyncio(
+                client=self.client,
+                pool_id=_to_uuid(pool_id),
+                body=data
+            )
+            return ApiResponse(data={"success": True})
+        except Exception as e:
+            return ApiResponse(error=e)
+    
+    def remove_machines_sync(self, pool_id: str, data: MachinePoolAssignment) -> ApiResponse:
+        """Remove machines from a pool (synchronous).
+        
+        Args:
+            pool_id: The pool ID
+            data: MachinePoolAssignment with machine_ids list
+        
+        Returns:
+            ApiResponse with success status
+        """
+        try:
+            remove_machines_from_pool_v1_pools_pool_id_machines_delete.sync(
+                client=self.client,
+                pool_id=_to_uuid(pool_id),
+                body=data
+            )
+            return ApiResponse(data={"success": True})
+        except Exception as e:
+            return ApiResponse(error=e)
 
 
 class WorkflowsAPI:
@@ -278,25 +634,61 @@ class WorkflowsAPI:
     def __init__(self, client: AuthenticatedClient):
         self.client = client
     
-    async def list(self, skip: Optional[int] = None, limit: Optional[int] = None) -> ApiResponse:
-        """List workflows."""
+    async def list(
+        self,
+        skip: Optional[int] = None,
+        limit: Optional[int] = None,
+        *,
+        created_at_from: Optional[Union[str, datetime]] = None,
+        created_at_to: Optional[Union[str, datetime]] = None,
+        updated_at_from: Optional[Union[str, datetime]] = None,
+        updated_at_to: Optional[Union[str, datetime]] = None,
+    ) -> ApiResponse:
+        """List workflows with optional date-time filtering.
+
+        Args:
+            skip: Pagination skip
+            limit: Pagination limit
+            created_at_from: Start datetime for created_at filter (UTC or ISO)
+            created_at_to: End datetime for created_at filter (UTC or ISO)
+            updated_at_from: Start datetime for updated_at filter (UTC or ISO)
+            updated_at_to: End datetime for updated_at filter (UTC or ISO)
+        """
         try:
             response = await list_workflows_v1_workflows_get.asyncio(
                 client=self.client,
                 skip=_to_unset_or_value(skip),
-                limit=_to_unset_or_value(limit)
+                limit=_to_unset_or_value(limit),
+                created_at_from=_to_unset_or_value(_to_iso_utc_str(created_at_from)),
+                created_at_to=_to_unset_or_value(_to_iso_utc_str(created_at_to)),
+                updated_at_from=_to_unset_or_value(_to_iso_utc_str(updated_at_from)),
+                updated_at_to=_to_unset_or_value(_to_iso_utc_str(updated_at_to)),
+                test=1
             )
             return ApiResponse(data=response)
         except Exception as e:
             return ApiResponse(error=e)
     
-    def list_sync(self, skip: Optional[int] = None, limit: Optional[int] = None) -> ApiResponse:
-        """List workflows (synchronous)."""
+    def list_sync(
+        self,
+        skip: Optional[int] = None,
+        limit: Optional[int] = None,
+        *,
+        created_at_from: Optional[Union[str, datetime]] = None,
+        created_at_to: Optional[Union[str, datetime]] = None,
+        updated_at_from: Optional[Union[str, datetime]] = None,
+        updated_at_to: Optional[Union[str, datetime]] = None,
+    ) -> ApiResponse:
+        """List workflows (synchronous) with optional date-time filtering."""
         try:
             response = list_workflows_v1_workflows_get.sync(
                 client=self.client,
                 skip=_to_unset_or_value(skip),
-                limit=_to_unset_or_value(limit)
+                limit=_to_unset_or_value(limit),
+                created_at_from=_to_unset_or_value(_to_iso_utc_str(created_at_from)),
+                created_at_to=_to_unset_or_value(_to_iso_utc_str(created_at_to)),
+                updated_at_from=_to_unset_or_value(_to_iso_utc_str(updated_at_from)),
+                updated_at_to=_to_unset_or_value(_to_iso_utc_str(updated_at_to)),
             )
             return ApiResponse(data=response)
         except Exception as e:
@@ -405,7 +797,10 @@ class RunsAPI:
         limit: Optional[int] = None,
         status: Optional[RunStatus] = None,
         workflow_id: Optional[str] = None,
-        machine_id: Optional[str] = None
+        machine_id: Optional[str] = None,
+        *,
+        created_at_from: Optional[Union[str, datetime]] = None,
+        created_at_to: Optional[Union[str, datetime]] = None,
     ) -> ApiResponse:
         """List runs with optional filtering."""
         try:
@@ -415,7 +810,9 @@ class RunsAPI:
                 limit=_to_unset_or_value(limit),
                 status=status,
                 workflow_id=_to_uuid(workflow_id) if workflow_id else UNSET,
-                machine_id=_to_uuid(machine_id) if machine_id else UNSET
+                machine_id=_to_uuid(machine_id) if machine_id else UNSET,
+                created_at_from=_to_unset_or_value(_to_iso_utc_str(created_at_from)),
+                created_at_to=_to_unset_or_value(_to_iso_utc_str(created_at_to)),
             )
             return ApiResponse(data=response)
         except Exception as e:
@@ -427,7 +824,10 @@ class RunsAPI:
         limit: Optional[int] = None,
         status: Optional[RunStatus] = None,
         workflow_id: Optional[str] = None,
-        machine_id: Optional[str] = None
+        machine_id: Optional[str] = None,
+        *,
+        created_at_from: Optional[Union[str, datetime]] = None,
+        created_at_to: Optional[Union[str, datetime]] = None,
     ) -> ApiResponse:
         """List runs with optional filtering (synchronous)."""
         try:
@@ -437,14 +837,28 @@ class RunsAPI:
                 limit=_to_unset_or_value(limit),
                 status=status,
                 workflow_id=_to_uuid(workflow_id) if workflow_id else UNSET,
-                machine_id=_to_uuid(machine_id) if machine_id else UNSET
+                machine_id=_to_uuid(machine_id) if machine_id else UNSET,
+                created_at_from=_to_unset_or_value(_to_iso_utc_str(created_at_from)),
+                created_at_to=_to_unset_or_value(_to_iso_utc_str(created_at_to)),
             )
             return ApiResponse(data=response)
         except Exception as e:
             return ApiResponse(error=e)
     
     async def create(self, data: RunCreate) -> ApiResponse:
-        """Create a new run."""
+        """Create a new run.
+        
+        Args:
+            data: RunCreate object with:
+                - workflow_id: The workflow to run
+                - machine_id: Optional specific machine ID
+                - pool_ids: Optional list of pool IDs (machine must be in ALL specified pools)
+                - input_values: Optional input values for workflow variables
+                - file_inputs: Optional files to upload to the machine
+        
+        Returns:
+            ApiResponse with RunResponse
+        """
         try:
             response = await create_run_v1_runs_post.asyncio(
                 client=self.client,
@@ -455,7 +869,19 @@ class RunsAPI:
             return ApiResponse(error=e)
     
     def create_sync(self, data: RunCreate) -> ApiResponse:
-        """Create a new run (synchronous)."""
+        """Create a new run (synchronous).
+        
+        Args:
+            data: RunCreate object with:
+                - workflow_id: The workflow to run
+                - machine_id: Optional specific machine ID
+                - pool_ids: Optional list of pool IDs (machine must be in ALL specified pools)
+                - input_values: Optional input values for workflow variables
+                - file_inputs: Optional files to upload to the machine
+        
+        Returns:
+            ApiResponse with RunResponse
+        """
         try:
             response = create_run_v1_runs_post.sync(
                 client=self.client,
@@ -542,8 +968,13 @@ class RunsAPI:
         - Returns immediately with created run details
         
         Args:
-            data: RunBulkCreate object containing workflow_id, machine_id, 
-                  input_values, file_inputs, and count (max 1000)
+            data: RunBulkCreate object containing:
+                - workflow_id: The workflow to run
+                - machine_id: Optional specific machine ID 
+                - pool_ids: Optional list of pool IDs (machine must be in ALL specified pools)
+                - input_values: Optional input values for workflow variables
+                - file_inputs: Optional files to upload to the machine
+                - count: Number of runs to create (max 1000)
         
         Returns:
             ApiResponse with RunBulkCreateResponse containing:
@@ -569,8 +1000,13 @@ class RunsAPI:
         - Returns immediately with created run details
         
         Args:
-            data: RunBulkCreate object containing workflow_id, machine_id, 
-                  input_values, file_inputs, and count (max 1000)
+            data: RunBulkCreate object containing:
+                - workflow_id: The workflow to run
+                - machine_id: Optional specific machine ID 
+                - pool_ids: Optional list of pool IDs (machine must be in ALL specified pools)
+                - input_values: Optional input values for workflow variables
+                - file_inputs: Optional files to upload to the machine
+                - count: Number of runs to create (max 1000)
         
         Returns:
             ApiResponse with RunBulkCreateResponse containing:
@@ -675,7 +1111,12 @@ class TrajectoriesAPI:
         self,
         skip: Optional[int] = None,
         limit: Optional[int] = None,
-        workflow_id: Optional[str] = None
+        workflow_id: Optional[str] = None,
+        *,
+        created_at_from: Optional[Union[str, datetime]] = None,
+        created_at_to: Optional[Union[str, datetime]] = None,
+        updated_at_from: Optional[Union[str, datetime]] = None,
+        updated_at_to: Optional[Union[str, datetime]] = None,
     ) -> ApiResponse:
         """List trajectories with optional filtering."""
         try:
@@ -683,7 +1124,11 @@ class TrajectoriesAPI:
                 client=self.client,
                 skip=_to_unset_or_value(skip),
                 limit=_to_unset_or_value(limit),
-                workflow_id=_to_uuid(workflow_id) if workflow_id else UNSET
+                workflow_id=_to_uuid(workflow_id) if workflow_id else UNSET,
+                created_at_from=_to_unset_or_value(_to_iso_utc_str(created_at_from)),
+                created_at_to=_to_unset_or_value(_to_iso_utc_str(created_at_to)),
+                updated_at_from=_to_unset_or_value(_to_iso_utc_str(updated_at_from)),
+                updated_at_to=_to_unset_or_value(_to_iso_utc_str(updated_at_to)),
             )
             return ApiResponse(data=response)
         except Exception as e:
@@ -693,7 +1138,12 @@ class TrajectoriesAPI:
         self,
         skip: Optional[int] = None,
         limit: Optional[int] = None,
-        workflow_id: Optional[str] = None
+        workflow_id: Optional[str] = None,
+        *,
+        created_at_from: Optional[Union[str, datetime]] = None,
+        created_at_to: Optional[Union[str, datetime]] = None,
+        updated_at_from: Optional[Union[str, datetime]] = None,
+        updated_at_to: Optional[Union[str, datetime]] = None,
     ) -> ApiResponse:
         """List trajectories with optional filtering (synchronous)."""
         try:
@@ -701,7 +1151,11 @@ class TrajectoriesAPI:
                 client=self.client,
                 skip=_to_unset_or_value(skip),
                 limit=_to_unset_or_value(limit),
-                workflow_id=_to_uuid(workflow_id) if workflow_id else UNSET
+                workflow_id=_to_uuid(workflow_id) if workflow_id else UNSET,
+                created_at_from=_to_unset_or_value(_to_iso_utc_str(created_at_from)),
+                created_at_to=_to_unset_or_value(_to_iso_utc_str(created_at_to)),
+                updated_at_from=_to_unset_or_value(_to_iso_utc_str(updated_at_from)),
+                updated_at_to=_to_unset_or_value(_to_iso_utc_str(updated_at_to)),
             )
             return ApiResponse(data=response)
         except Exception as e:
@@ -1173,6 +1627,7 @@ class CyberdeskClient:
         
         # Initialize API endpoints
         self.machines = MachinesAPI(self._client)
+        self.pools = PoolsAPI(self._client)
         self.workflows = WorkflowsAPI(self._client)
         self.runs = RunsAPI(self._client)
         self.connections = ConnectionsAPI(self._client)