Method run_agent

src/google/adk/cli/api_server.py:1460–1523

(req: RunAgentRequest, request: Request)

    @app.post("/run", response_model_exclude_none=True)
    async def run_agent(req: RunAgentRequest, request: Request) -> list[Event]:
      app_name = req.app_name or self.default_app_name
      if not app_name:
        raise HTTPException(
            status_code=400,
            detail="app_name is required when ADK_DEFAULT_APP_NAME is not set",
        )
      req.app_name = app_name
      self.current_app_name_ref.value = req.app_name
      runner = await self.get_runner_async(req.app_name)
      _set_telemetry_context_if_needed(runner)
      run_config = (
          RunConfig(custom_metadata=req.custom_metadata)
          if req.custom_metadata
          else None
      )

      async def worker():
        try:
          async with Aclosing(
              runner.run_async(
                  user_id=req.user_id,
                  session_id=req.session_id,
                  new_message=req.new_message,
                  state_delta=req.state_delta,
                  invocation_id=req.invocation_id,
                  run_config=run_config,
              )
          ) as agen:
            return [event async for event in agen]
        except SessionNotFoundError as e:
          raise HTTPException(status_code=404, detail=str(e)) from e

      worker_task = asyncio.create_task(worker())

      async def monitor():
        try:
          while True:
            message = await request.receive()
            if message.get("type") == "http.disconnect":
              logger.warning(
                  "Client disconnected. Aborting agent run for session %s.",
                  req.session_id,
              )
              worker_task.cancel()
              break
        except asyncio.CancelledError:
          pass
        except Exception as e:  # pylint: disable=broad-exception-caught
          logger.error("Exception in disconnect monitor: %s", e, exc_info=True)

      monitor_task = asyncio.create_task(monitor())

      try:
        events = await worker_task
        logger.info("Generated %s events in agent run", len(events))
        logger.debug("Events generated: %s", events)
        return events