diff --git a/workers/file_processing/structure_tool_task.py b/workers/file_processing/structure_tool_task.py index 3fcf1999c4..9b954cadcb 100644 --- a/workers/file_processing/structure_tool_task.py +++ b/workers/file_processing/structure_tool_task.py @@ -435,7 +435,7 @@ def _execute_structure_tool_impl(params: dict) -> dict: json_structure = at_settings.get("json_structure") if isinstance(json_structure, dict): json_structure = json.dumps(json_structure) - agentic_params = { + agentic_params: dict[str, Any] = { "llm_adapter_instance_id": at_output["llm"], "lite_llm_adapter_instance_id": at_settings.get( "lite_llm_adapter_instance_id", "" @@ -452,6 +452,9 @@ def _execute_structure_tool_impl(params: dict) -> dict: "execution_id": execution_id, "PLATFORM_SERVICE_API_KEY": platform_service_api_key, } + group_key = at_settings.get("group_key", "") + if group_key: + agentic_params["group_key"] = group_key at_ctx = ExecutionContext( executor_name="agentic_table", operation="table_extract",