From f687668e72f12a27d702ed408e28eca463f077c4 Mon Sep 17 00:00:00 2001 From: Stas Moreinis Date: Wed, 24 Dec 2025 13:30:36 -0800 Subject: [PATCH] adding RO session maker --- agentex/src/temporal/run_healthcheck_workflow.py | 4 +++- agentex/src/temporal/run_worker.py | 4 +++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/agentex/src/temporal/run_healthcheck_workflow.py b/agentex/src/temporal/run_healthcheck_workflow.py index 7d0b097..1a6abfc 100644 --- a/agentex/src/temporal/run_healthcheck_workflow.py +++ b/agentex/src/temporal/run_healthcheck_workflow.py @@ -7,6 +7,7 @@ ) from src.config.dependencies import ( GlobalDependencies, + database_async_read_only_session_maker, database_async_read_write_engine, database_async_read_write_session_maker, ) @@ -47,7 +48,8 @@ async def main() -> None: # Initialize repository and list agents engine = database_async_read_write_engine() session_maker = database_async_read_write_session_maker(engine) - agent_repo = AgentRepository(session_maker) + read_only_session_maker = database_async_read_only_session_maker(engine) + agent_repo = AgentRepository(session_maker, read_only_session_maker) agents = await agent_repo.list() adapter = TemporalAdapter(temporal_client=global_dependencies.temporal_client) diff --git a/agentex/src/temporal/run_worker.py b/agentex/src/temporal/run_worker.py index d1a9010..d366509 100644 --- a/agentex/src/temporal/run_worker.py +++ b/agentex/src/temporal/run_worker.py @@ -14,6 +14,7 @@ from src.adapters.temporal.client_factory import TemporalClientFactory from src.config.dependencies import ( + database_async_read_only_session_maker, database_async_read_write_engine, database_async_read_write_session_maker, httpx_client, @@ -166,7 +167,8 @@ async def main() -> None: # Create session maker engine = database_async_read_write_engine() session_maker = database_async_read_write_session_maker(engine) - agent_repo = AgentRepository(session_maker) + read_only_session_maker = database_async_read_only_session_maker(engine) + agent_repo = AgentRepository(session_maker, read_only_session_maker) health_check_worker_task = create_health_check_worker( agent_repo=agent_repo, http_client=httpx_client(),