feat(agent-runner): support run steering

2026-07-25 13:56:08 +00:00 · 2026-06-11 23:03:44 +08:00
parent f92bd95cc8
commit 54a2f7060d
9 changed files with 299 additions and 6 deletions
@@ -84,15 +84,19 @@ class ChatMessageHandler(handler.MessageHandler):
                    query.user_message.content = [event_ctx.event.user_message_alter]

            text_length = 0
-            try:
-                is_stream = await query.adapter.is_stream_output_supported()
-            except AttributeError:
-                is_stream = False
-
            try:
                # Mark start time for telemetry
                start_ts = time.time()

+                if await self.ap.agent_run_orchestrator.try_claim_steering_from_query(query):
+                    yield entities.StageProcessResult(result_type=entities.ResultType.INTERRUPT, new_query=query)
+                    return
+
+                try:
+                    is_stream = await query.adapter.is_stream_output_supported()
+                except AttributeError:
+                    is_stream = False
+
                # Create a single resp_message_id for the entire streaming response
                resp_message_id = uuid.uuid4()
                chunk_count = 0