accept workflows list

Author: jatong

src/agentex/lib/core/temporal/workers/worker.py

@@ -4,7 +4,7 @@
 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,25 @@ def __init__(
         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()
@@ -119,7 +134,7 @@ async def run(
             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,