agentex-sdk 0.4.12__tar.gz → 0.4.13__tar.gz

agentex_sdk-0.4.13/.release-please-manifest.json

@@ -0,0 +1,3 @@
+{
+  ".": "0.4.13"
+}

{agentex_sdk-0.4.12 → agentex_sdk-0.4.13}/CHANGELOG.md

@@ -1,5 +1,14 @@
 # Changelog
 
+## 0.4.13 (2025-09-12)
+
+Full Changelog: [v0.4.12...v0.4.13](https://github.com/scaleapi/agentex-python/compare/v0.4.12...v0.4.13)
+
+### Features
+
+* **api:** api update ([0102183](https://github.com/scaleapi/agentex-python/commit/0102183a8f5a23dbdaf905ffbe7ffbcf59bf7b21))
+* **api:** api update ([8a6edb1](https://github.com/scaleapi/agentex-python/commit/8a6edb13046ca24bf6c45fc018e32de498d48869))
+
 ## 0.4.12 (2025-09-08)
 
 Full Changelog: [v0.4.11...v0.4.12](https://github.com/scaleapi/agentex-python/compare/v0.4.11...v0.4.12)

agentex_sdk-0.4.13/CLAUDE.md

@@ -0,0 +1,93 @@
+# CLAUDE.md
+
+This file provides guidance to Claude Code (claude.ai/code) when working with code in this repository.
+
+## Development Commands
+
+### Package Management in the top level repo
+- Use `rye` for dependency management (preferred)
+- Run `./scripts/bootstrap` to set up the environment
+- Or use `rye sync --all-features` directly
+
+Special note: the individual tutorials maintain their own tutorial specific virtualenv using `uv`. So when testing/running tutorials, you `uv run` instead of `rye run`.  Everything else is similar.
+
+#### Testing
+- Run tests: `rye run pytest` or `./scripts/test`
+- Run specific test: `rye run pytest path/to/test_file.py::TestClass::test_method -v`
+- Mock server is automatically started for tests, runs on port 4010
+
+#### Linting and Formatting
+- Format code: `rye run format` or `./scripts/format`
+  * The repository is still in flux, so running format might accidentally change files that aren't part of your scope of changes. So always run `run rye format` with additional arguments to constrain the formatting to the files that you are modifying.
+- Lint code: `rye run lint` or `./scripts/lint`
+- Type check: `rye run typecheck` (runs both pyright and mypy)
+
+### Building and Running
+- Build package: `rye build`
+
+
+
+### CLI Commands
+The package provides the `agentex` CLI with these main commands:
+- `agentex agents` - Get, list, run, build, and deploy agents
+- `agentex tasks` - Get, list, and delete tasks  
+- `agentex secrets` - Sync, get, list, and delete secrets
+- `agentex uv` - UV wrapper with AgentEx-specific enhancements
+- `agentex init` - Initialize new agent projects
+
+### Agent Development
+- Run agents: `agentex agents run --manifest manifest.yaml`
+- Debug agents: `agentex agents run --manifest manifest.yaml --debug-worker`
+- Debug with custom port: `agentex agents run --manifest manifest.yaml --debug-worker --debug-port 5679`
+
+## Architecture Overview
+
+### Code Structure
+- `/src/agentex/` - Core SDK and generated API client code
+- `/src/agentex/lib/` - Custom library code (not modified by code generator)
+  - `/cli/` - Command-line interface implementation
+  - `/core/` - Core services, adapters, and temporal workflows
+  - `/sdk/` - SDK utilities and FastACP implementation
+  - `/types/` - Custom type definitions
+  - `/utils/` - Utility functions
+- `/examples/` - Example implementations and tutorials
+- `/tests/` - Test suites
+
+### Key Components
+
+**SDK Architecture:**
+- **Client Layer**: HTTP client for AgentEx API (`_client.py`, `resources/`)
+- **CLI Layer**: Typer-based command interface (`lib/cli/`)
+- **Core Services**: Temporal workflows, adapters, and services (`lib/core/`)
+- **FastACP**: Fast Agent Communication Protocol implementation (`lib/sdk/fastacp/`)
+- **State Machine**: Workflow state management (`lib/sdk/state_machine/`)
+
+**Temporal Integration:**
+- Workflow definitions in `lib/core/temporal/`
+- Activity definitions for different providers
+- Worker implementations for running temporal workflows
+
+**Agent Framework:**
+- Manifest-driven agent configuration
+- Support for multiple agent types (sync, temporal-based)
+- Debugging support with VS Code integration
+
+### Code Generation
+Most SDK code is auto-generated. Manual changes are preserved in:
+- `src/agentex/lib/` directory
+- `examples/` directory
+- Merge conflicts may occur between manual patches and generator changes
+
+### Key Dependencies
+- `temporalio` - Temporal workflow engine
+- `typer` - CLI framework  
+- `pydantic` - Data validation
+- `httpx` - HTTP client
+- `fastapi` - Web framework
+- `ruff` - Linting and formatting
+- `pytest` - Testing framework
+
+### Environment Requirements
+- Python 3.12+ required
+- Uses Rye for dependency management
+- Supports both sync and async client patterns

{agentex_sdk-0.4.12 → agentex_sdk-0.4.13}/CONTRIBUTING.md

@@ -126,3 +126,33 @@ You can release to package managers by using [the `Publish PyPI` GitHub action](
 
 If you need to manually release a package, you can run the `bin/publish-pypi` script with a `PYPI_TOKEN` set on
 the environment.
+
+## 🤖 **Vibe Coding Setup**
+
+This repository is setup with some pre-canned prompts for [Claude Code](https://docs.anthropic.com/en/docs/claude-code) as well as [Cursor](https://cursor.com/).
+
+### Cursor
+
+Access to Cursor can be acquired by asking for it in #it-help.  Then just loading this repo in the Cursor IDE should enable the prompts.
+
+### Claude Code
+
+### 1. Install Claude Code
+```bash
+npm install -g @anthropic-ai/claude-code
+```
+
+### 2. Request a LiteLLM API Key
+Visit the [LiteLLM User Guide](https://scale.atlassian.net/wiki/spaces/EPD/pages/1490354189/LiteLLM+User+Guide#Requesting-LiteLLM-Key-for-Generic-Usage) to request your API key.
+
+### 3. Set Environment Variables
+```bash
+export ANTHROPIC_AUTH_TOKEN=${LITELLM_PROXY_API_KEY}
+export ANTHROPIC_BASE_URL="https://litellm.ml-serving-internal.scale.com"
+```
+
+### 4. Start Claude Code
+```bash
+claude
+```
+This should be run from inside the main repo directory. If you run the command from a terminal inside VSCode, then Claude will use the VSCode editor to show diffs etc. 

{agentex_sdk-0.4.12 → agentex_sdk-0.4.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agentex-sdk
-Version: 0.4.12
+Version: 0.4.13
 Summary: The official Python library for the agentex API
 Project-URL: Homepage, https://github.com/scaleapi/agentex-python
 Project-URL: Repository, https://github.com/scaleapi/agentex-python

{agentex_sdk-0.4.12 → agentex_sdk-0.4.13}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agentex-sdk"
-version = "0.4.12"
+version = "0.4.13"
 description = "The official Python library for the agentex API"
 dynamic = ["readme"]
 license = "Apache-2.0"

{agentex_sdk-0.4.12 → agentex_sdk-0.4.13}/src/agentex/_constants.py

@@ -6,9 +6,9 @@ RAW_RESPONSE_HEADER = "X-Stainless-Raw-Response"
 OVERRIDE_CAST_TO_HEADER = "____stainless_override_cast_to"
 
 # default timeout is 1 minute
-DEFAULT_TIMEOUT = httpx.Timeout(timeout=60, connect=5.0)
-DEFAULT_MAX_RETRIES = 2
-DEFAULT_CONNECTION_LIMITS = httpx.Limits(max_connections=100, max_keepalive_connections=20)
+DEFAULT_TIMEOUT = httpx.Timeout(timeout=300, connect=5.0)
+DEFAULT_MAX_RETRIES = 0
+DEFAULT_CONNECTION_LIMITS = httpx.Limits(max_connections=1000, max_keepalive_connections=1000)
 
 INITIAL_RETRY_DELAY = 0.5
 MAX_RETRY_DELAY = 8.0

{agentex_sdk-0.4.12 → agentex_sdk-0.4.13}/src/agentex/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "agentex"
-__version__ = "0.4.12"  # x-release-please-version
+__version__ = "0.4.13"  # x-release-please-version

{agentex_sdk-0.4.12 → agentex_sdk-0.4.13}/src/agentex/lib/adk/_modules/acp.py

@@ -59,6 +59,7 @@ class ACPModule:
         start_to_close_timeout: timedelta = timedelta(seconds=5),
         heartbeat_timeout: timedelta = timedelta(seconds=5),
         retry_policy: RetryPolicy = DEFAULT_RETRY_POLICY,
+        request: dict[str, Any] | None = None,
     ) -> Task:
         """
         Create a new task.
@@ -71,6 +72,7 @@ class ACPModule:
             start_to_close_timeout: The start to close timeout for the task.
             heartbeat_timeout: The heartbeat timeout for the task.
             retry_policy: The retry policy for the task.
+            request: Additional request context including headers to forward to the agent.
 
         Returns:
             The task entry.
@@ -85,6 +87,7 @@ class ACPModule:
                     params=params,
                     trace_id=trace_id,
                     parent_span_id=parent_span_id,
+                    request=request,
                 ),
                 response_type=Task,
                 start_to_close_timeout=start_to_close_timeout,
@@ -99,6 +102,7 @@ class ACPModule:
                 params=params,
                 trace_id=trace_id,
                 parent_span_id=parent_span_id,
+                request=request,
             )
 
     async def send_event(
@@ -112,15 +116,22 @@ class ACPModule:
         start_to_close_timeout: timedelta = timedelta(seconds=5),
         heartbeat_timeout: timedelta = timedelta(seconds=5),
         retry_policy: RetryPolicy = DEFAULT_RETRY_POLICY,
+        request: dict[str, Any] | None = None,
     ) -> Event:
         """
         Send an event to a task.
 
         Args:
             task_id: The ID of the task to send the event to.
-            data: The data to send to the event.
+            content: The content to send to the event.
             agent_id: The ID of the agent to send the event to.
             agent_name: The name of the agent to send the event to.
+            trace_id: The trace ID for the event.
+            parent_span_id: The parent span ID for the event.
+            start_to_close_timeout: The start to close timeout for the event.
+            heartbeat_timeout: The heartbeat timeout for the event.
+            retry_policy: The retry policy for the event.
+            request: Additional request context including headers to forward to the agent.
 
         Returns:
             The event entry.
@@ -135,6 +146,7 @@ class ACPModule:
                     content=content,
                     trace_id=trace_id,
                     parent_span_id=parent_span_id,
+                    request=request,
                 ),
                 response_type=None,
                 start_to_close_timeout=start_to_close_timeout,
@@ -149,6 +161,7 @@ class ACPModule:
                 content=content,
                 trace_id=trace_id,
                 parent_span_id=parent_span_id,
+                request=request,
             )
 
     async def send_message(
@@ -162,15 +175,22 @@ class ACPModule:
         start_to_close_timeout: timedelta = timedelta(seconds=5),
         heartbeat_timeout: timedelta = timedelta(seconds=5),
         retry_policy: RetryPolicy = DEFAULT_RETRY_POLICY,
+        request: dict[str, Any] | None = None,
     ) -> List[TaskMessage]:
         """
         Send a message to a task.
 
         Args:
-            task_id: The ID of the task to send the message to.
             content: The task message content to send to the task.
+            task_id: The ID of the task to send the message to.
             agent_id: The ID of the agent to send the message to.
             agent_name: The name of the agent to send the message to.
+            trace_id: The trace ID for the message.
+            parent_span_id: The parent span ID for the message.
+            start_to_close_timeout: The start to close timeout for the message.
+            heartbeat_timeout: The heartbeat timeout for the message.
+            retry_policy: The retry policy for the message.
+            request: Additional request context including headers to forward to the agent.
 
         Returns:
             The message entry.
@@ -185,6 +205,7 @@ class ACPModule:
                     content=content,
                     trace_id=trace_id,
                     parent_span_id=parent_span_id,
+                    request=request,
                 ),
                 response_type=TaskMessage,
                 start_to_close_timeout=start_to_close_timeout,
@@ -199,6 +220,7 @@ class ACPModule:
                 content=content,
                 trace_id=trace_id,
                 parent_span_id=parent_span_id,
+                request=request,
             )
 
     async def cancel_task(
@@ -212,6 +234,7 @@ class ACPModule:
         start_to_close_timeout: timedelta = timedelta(seconds=5),
         heartbeat_timeout: timedelta = timedelta(seconds=5),
         retry_policy: RetryPolicy = DEFAULT_RETRY_POLICY,
+        request: dict[str, Any] | None = None,
     ) -> Task:
         """
         Cancel a task by sending cancel request to the agent that owns the task.
@@ -226,6 +249,7 @@ class ACPModule:
             start_to_close_timeout: The start to close timeout for the task.
             heartbeat_timeout: The heartbeat timeout for the task.
             retry_policy: The retry policy for the task.
+            request: Additional request context including headers to forward to the agent.
 
         Returns:
             The task entry.
@@ -244,6 +268,7 @@ class ACPModule:
                     agent_name=agent_name,
                     trace_id=trace_id,
                     parent_span_id=parent_span_id,
+                    request=request,
                 ),
                 response_type=None,
                 start_to_close_timeout=start_to_close_timeout,
@@ -258,4 +283,5 @@ class ACPModule:
                 agent_name=agent_name,
                 trace_id=trace_id,
                 parent_span_id=parent_span_id,
+                request=request,
             )

{agentex_sdk-0.4.12 → agentex_sdk-0.4.13}/src/agentex/lib/core/services/adk/acp/acp.py

@@ -9,6 +9,10 @@ from agentex.types.task import Task
 from agentex.types.task_message import TaskMessage
 from agentex.types.task_message_content import TaskMessageContent
 from agentex.types.task_message_content_param import TaskMessageContentParam
+from agentex.types.agent_rpc_params import (
+    ParamsCancelTaskRequest as RpcParamsCancelTaskRequest,
+    ParamsSendEventRequest as RpcParamsSendEventRequest,
+)
 
 logger = make_logger(__name__)
 
@@ -30,6 +34,7 @@ class ACPService:
         params: dict[str, Any] | None = None,
         trace_id: str | None = None,
         parent_span_id: str | None = None,
+        request: dict[str, Any] | None = None,
     ) -> Task:
         trace = self._tracer.trace(trace_id=trace_id)
         async with trace.span(
@@ -43,6 +48,10 @@ class ACPService:
             },
         ) as span:
             heartbeat_if_in_workflow("task create")
+            
+            # Extract headers from request; pass-through to agent
+            extra_headers = request.get("headers") if request else None
+            
             if agent_name:
                 json_rpc_response = await self._agentex_client.agents.rpc_by_name(
                     agent_name=agent_name,
@@ -51,6 +60,7 @@ class ACPService:
                         "name": name,
                         "params": params,
                     },
+                    extra_headers=extra_headers,
                 )
             elif agent_id:
                 json_rpc_response = await self._agentex_client.agents.rpc(
@@ -60,6 +70,7 @@ class ACPService:
                         "name": name,
                         "params": params,
                     },
+                    extra_headers=extra_headers,
                 )
             else:
                 raise ValueError("Either agent_name or agent_id must be provided")
@@ -78,6 +89,7 @@ class ACPService:
         task_name: str | None = None,
         trace_id: str | None = None,
         parent_span_id: str | None = None,
+        request: dict[str, Any] | None = None,
     ) -> List[TaskMessage]:
         trace = self._tracer.trace(trace_id=trace_id)
         async with trace.span(
@@ -92,6 +104,10 @@ class ACPService:
             },
         ) as span:
             heartbeat_if_in_workflow("message send")
+            
+            # Extract headers from request; pass-through to agent
+            extra_headers = request.get("headers") if request else None
+            
             if agent_name:
                 json_rpc_response = await self._agentex_client.agents.rpc_by_name(
                     agent_name=agent_name,
@@ -101,6 +117,7 @@ class ACPService:
                         "content": cast(TaskMessageContentParam, content.model_dump()),
                         "stream": False,
                     },
+                    extra_headers=extra_headers,
                 )
             elif agent_id:
                 json_rpc_response = await self._agentex_client.agents.rpc(
@@ -111,12 +128,13 @@ class ACPService:
                         "content": cast(TaskMessageContentParam, content.model_dump()),
                         "stream": False,
                     },
+                    extra_headers=extra_headers,
                 )
             else:
                 raise ValueError("Either agent_name or agent_id must be provided")
 
             task_messages: List[TaskMessage] = []
-            logger.info(f"json_rpc_response: {json_rpc_response}")
+            logger.info("json_rpc_response: %s", json_rpc_response)
             if isinstance(json_rpc_response.result, list):
                 for message in json_rpc_response.result:
                     task_message = TaskMessage.model_validate(message)
@@ -137,6 +155,7 @@ class ACPService:
         task_name: str | None = None,
         trace_id: str | None = None,
         parent_span_id: str | None = None,
+        request: dict[str, Any] | None = None,
     ) -> Event:
         trace = self._tracer.trace(trace_id=trace_id)
         async with trace.span(
@@ -146,27 +165,33 @@ class ACPService:
                 "agent_id": agent_id,
                 "agent_name": agent_name,
                 "task_id": task_id,
+                "task_name": task_name,
                 "content": content,
             },
         ) as span:
             heartbeat_if_in_workflow("event send")
+            
+            # Extract headers from request; pass-through to agent
+            extra_headers = request.get("headers") if request else None
+            
+            rpc_event_params: RpcParamsSendEventRequest = {
+                "task_id": task_id,
+                "task_name": task_name,
+                "content": cast(TaskMessageContentParam, content.model_dump()),
+            }
             if agent_name:
                 json_rpc_response = await self._agentex_client.agents.rpc_by_name(
                     agent_name=agent_name,
                     method="event/send",
-                    params={
-                        "task_id": task_id,
-                        "content": cast(TaskMessageContentParam, content.model_dump()),
-                    },
+                    params=rpc_event_params,
+                    extra_headers=extra_headers,
                 )
             elif agent_id:
                 json_rpc_response = await self._agentex_client.agents.rpc(
                     agent_id=agent_id,
                     method="event/send",
-                    params={
-                        "task_id": task_id,
-                        "content": cast(TaskMessageContentParam, content.model_dump()),
-                    },
+                    params=rpc_event_params,
+                    extra_headers=extra_headers,
                 )
             else:
                 raise ValueError("Either agent_name or agent_id must be provided")
@@ -184,7 +209,27 @@ class ACPService:
         agent_name: str | None = None,
         trace_id: str | None = None,
         parent_span_id: str | None = None,
-    ) -> Task:        
+        request: dict[str, Any] | None = None,
+    ) -> Task:
+        """
+        Cancel a task by sending cancel request to the agent that owns the task.
+        
+        Args:
+            task_id: ID of the task to cancel (passed to agent in params)
+            task_name: Name of the task to cancel (passed to agent in params)  
+            agent_id: ID of the agent that owns the task
+            agent_name: Name of the agent that owns the task
+            trace_id: Trace ID for tracing
+            parent_span_id: Parent span ID for tracing
+            request: Additional request context including headers to forward to the agent
+            
+        Returns:
+            Task entry representing the cancelled task
+            
+        Raises:
+            ValueError: If neither agent_name nor agent_id is provided,
+                       or if neither task_name nor task_id is provided
+        """
         # Require agent identification
         if not agent_name and not agent_id:
             raise ValueError("Either agent_name or agent_id must be provided to identify the agent that owns the task")
@@ -192,7 +237,6 @@ class ACPService:
         # Require task identification
         if not task_name and not task_id:
             raise ValueError("Either task_name or task_id must be provided to identify the task to cancel")
-        
         trace = self._tracer.trace(trace_id=trace_id)
         async with trace.span(
             parent_id=parent_span_id,
@@ -206,8 +250,11 @@ class ACPService:
         ) as span:
             heartbeat_if_in_workflow("task cancel")
             
+            # Extract headers from request; pass-through to agent
+            extra_headers = request.get("headers") if request else None
+            
             # Build params for the agent (task identification)
-            params = {}
+            params: RpcParamsCancelTaskRequest = {}
             if task_id:
                 params["task_id"] = task_id
             if task_name:
@@ -219,12 +266,15 @@ class ACPService:
                     agent_name=agent_name,
                     method="task/cancel",
                     params=params,
+                    extra_headers=extra_headers,
                 )
             else:  # agent_id is provided (validated above)
+                assert agent_id is not None
                 json_rpc_response = await self._agentex_client.agents.rpc(
                     agent_id=agent_id,
                     method="task/cancel",
                     params=params,
+                    extra_headers=extra_headers,
                 )
 
             task_entry = Task.model_validate(json_rpc_response.result)

{agentex_sdk-0.4.12 → agentex_sdk-0.4.13}/src/agentex/lib/core/temporal/activities/adk/acp/acp_activities.py

@@ -26,6 +26,7 @@ class TaskCreateParams(BaseModelWithTraceParams):
     agent_id: str | None = None
     agent_name: str | None = None
     params: dict[str, Any] | None = None
+    request: dict[str, Any] | None = None
 
 
 class MessageSendParams(BaseModelWithTraceParams):
@@ -33,6 +34,7 @@ class MessageSendParams(BaseModelWithTraceParams):
     agent_name: str | None = None
     task_id: str | None = None
     content: TaskMessageContent
+    request: dict[str, Any] | None = None
 
 
 class EventSendParams(BaseModelWithTraceParams):
@@ -40,6 +42,7 @@ class EventSendParams(BaseModelWithTraceParams):
     agent_name: str | None = None
     task_id: str | None = None
     content: TaskMessageContent
+    request: dict[str, Any] | None = None
 
 
 class TaskCancelParams(BaseModelWithTraceParams):
@@ -47,6 +50,7 @@ class TaskCancelParams(BaseModelWithTraceParams):
     task_name: str | None = None
     agent_id: str | None = None
     agent_name: str | None = None
+    request: dict[str, Any] | None = None
 
 
 class ACPActivities:
@@ -60,6 +64,9 @@ class ACPActivities:
             agent_id=params.agent_id,
             agent_name=params.agent_name,
             params=params.params,
+            trace_id=params.trace_id,
+            parent_span_id=params.parent_span_id,
+            request=params.request,
         )
 
     @activity.defn(name=ACPActivityName.MESSAGE_SEND)
@@ -69,6 +76,9 @@ class ACPActivities:
             agent_name=params.agent_name,
             task_id=params.task_id,
             content=params.content,
+            trace_id=params.trace_id,
+            parent_span_id=params.parent_span_id,
+            request=params.request,
         )
 
     @activity.defn(name=ACPActivityName.EVENT_SEND)
@@ -78,6 +88,9 @@ class ACPActivities:
             agent_name=params.agent_name,
             task_id=params.task_id,
             content=params.content,
+            trace_id=params.trace_id,
+            parent_span_id=params.parent_span_id,
+            request=params.request,
         )
 
     @activity.defn(name=ACPActivityName.TASK_CANCEL)
@@ -89,4 +102,5 @@ class ACPActivities:
             agent_name=params.agent_name,
             trace_id=params.trace_id,
             parent_span_id=params.parent_span_id,
+            request=params.request,
         )

{agentex_sdk-0.4.12 → agentex_sdk-0.4.13}/src/agentex/lib/core/temporal/workers/worker.py

@@ -4,7 +4,7 @@ import os
 import uuid
 from collections.abc import Callable
 from concurrent.futures import ThreadPoolExecutor
-from typing import Any
+from typing import Any, overload
 
 from aiohttp import web
 from temporalio.client import Client
@@ -99,10 +99,28 @@ class AgentexWorker:
         self.healthy = False
         self.health_check_port = health_check_port
 
+    @overload
     async def run(
         self,
         activities: list[Callable],
+        *,
         workflow: type,
+    ) -> None: ...
+    
+    @overload
+    async def run(
+        self,
+        activities: list[Callable],
+        *,
+        workflows: list[type],
+    ) -> None: ...
+
+    async def run(
+        self,
+        activities: list[Callable],
+        *,
+        workflow: type | None = None,
+        workflows: list[type] | None = None,
     ):
         await self.start_health_check_server()
         await self._register_agent()
@@ -115,11 +133,14 @@ class AgentexWorker:
         if debug_enabled:
             logger.info("🐛 [WORKER] Temporal debug mode enabled - deadlock detection disabled")
         
+        if workflow is None and workflows is None:
+            raise ValueError("Either workflow or workflows must be provided")
+        
         worker = Worker(
             client=temporal_client,
             task_queue=self.task_queue,
             activity_executor=ThreadPoolExecutor(max_workers=self.max_workers),
-            workflows=[workflow],
+            workflows=[workflow] if workflows is None else workflows,
             activities=activities,
             workflow_runner=UnsandboxedWorkflowRunner(),
             max_concurrent_activities=self.max_concurrent_activities,