diff --git a/agent_v0/server_v1/api_stream.py b/agent_v0/server_v1/api_stream.py index 9f63d14d0..afaeee5a8 100644 --- a/agent_v0/server_v1/api_stream.py +++ b/agent_v0/server_v1/api_stream.py @@ -423,6 +423,7 @@ from .replay_engine import ( _SERVER_SIDE_ACTION_TYPES, _handle_extract_text_action, _handle_extract_table_action, + _handle_extract_dossier_action, _handle_t2a_decision_action, _handle_llm_generate_action, _handle_concat_text_vars_action, @@ -4443,6 +4444,15 @@ async def get_next_action(session_id: str, machine_id: str = "default"): ), timeout=180, ) + elif type_ == "extract_dossier": + await asyncio.wait_for( + loop.run_in_executor( + None, + _handle_extract_dossier_action, + action, owning_replay, session_id, + ), + timeout=180, + ) elif type_ == "t2a_decision": await asyncio.wait_for( loop.run_in_executor(