feat(mcp): queue long-running calculations

2026-10-29 18:25:15 +08:00
parent 2865811f1b
commit 63af998045
32 changed files with 1588 additions and 72 deletions
@@ -347,20 +347,25 @@ class McpQueryApplication:
                        )
                    result = QueryResult.model_validate(previous["result"])
                else:
-                    result = (
-                        await self.executor(
-                            scope, binding, tool, args, full_snapshot=True
+                    from app.mcp.calculation_jobs import should_queue, submit_calculation
+                    if should_queue(tool, args):
+                        result = await submit_calculation(scope, binding, tool, args, call_id)
+                    else:
+                        result = (
+                            await self.executor(
+                                scope, binding, tool, args, full_snapshot=True
+                            )
+                            if include_full
+                            else await self.executor(scope, binding, tool, args)
                        )
-                        if include_full
-                        else await self.executor(scope, binding, tool, args)
-                    )
                    # Never serialize an unbounded response into a model message or database receipt.
                    if len(result.model_dump_json().encode()) > 8 * 1024 * 1024:
                        return tool_error(
                            "RESULT_TOO_LARGE",
                            "Narrow the query or request a smaller page",
                        )
-                    if binding.task_id and name not in {
+                    if binding.task_id and not should_queue(tool, args) and name not in {
+                        "get_calculation_job", "get_calculation_result", "cancel_calculation_job", "list_calculation_jobs",
                        "get_task_context",
                        "get_task_evidence",
                    }: