Merge S2: region markers include imports

Author: samjulien

showcase/integrations/ag2/src/agents/agent.py

@@ -5,6 +5,7 @@
 the agent via the AG-UI protocol.
 """
 
+# @region[weather-tool-backend]
 from __future__ import annotations
 
 import json
@@ -34,7 +35,6 @@
 # =====
 # Tools
 # =====
-# @region[weather-tool-backend]
 async def get_weather(
     location: Annotated[str, "City name to get weather for"],
 ) -> dict[str, str | float]:

showcase/integrations/ag2/src/agents/subagents.py

@@ -17,6 +17,8 @@
 shared-state channel.
 """
 
+# @region[supervisor-delegation-tools]
+# @region[subagent-setup]
 import asyncio
 import logging
 import uuid
@@ -61,7 +63,6 @@ class SubagentsSnapshot(BaseModel):
 # prompt. They don't share memory or tools with the supervisor — the
 # supervisor only sees what each sub-agent's final reply produces.
 
-# @region[subagent-setup]
 _SUB_LLM_CONFIG = LLMConfig({"model": "gpt-4o-mini", "stream": False})
 
 _research_agent = ConversableAgent(
@@ -217,7 +218,6 @@ async def _run_delegation(
 # ---------------------------------------------------------------------------
 
 
-# @region[supervisor-delegation-tools]
 # Each @tool wraps a sub-agent invocation. The supervisor LLM "calls"
 # these tools to delegate work; each call asynchronously runs the
 # matching sub-agent, records the delegation into shared state via

showcase/integrations/ag2/src/app/api/copilotkit-voice/[[...slug]]/route.ts

@@ -12,6 +12,8 @@
 // `transcriptionService` option. V2 URL-routes on `/info`, `/agent/:id/run`,
 // `/transcribe`, etc., so the route lives at `[[...slug]]/route.ts`.
 
+// @region[voice-runtime]
+// @region[transcription-service-guard]
 import type { NextRequest } from "next/server";
 import {
   CopilotRuntime,
@@ -27,7 +29,6 @@ const AGENT_URL = process.env.AGENT_URL || "http://localhost:8000";
 
 const voiceDemoAgent = new HttpAgent({ url: `${AGENT_URL}/` });
 
-// @region[transcription-service-guard]
 class GuardedOpenAITranscriptionService extends TranscriptionService {
   private delegate: TranscriptionServiceOpenAI | null;
 
@@ -55,7 +56,6 @@ let cachedHandler: ((req: Request) => Promise<Response>) | null = null;
 function getHandler(): (req: Request) => Promise<Response> {
   if (cachedHandler) return cachedHandler;
 
-  // @region[voice-runtime]
   const runtime = new CopilotRuntime({
     // @ts-ignore -- see main route.ts; published agents type generic mismatch
     agents: {

showcase/integrations/ag2/src/app/demos/a2ui-fixed-schema/a2ui-backend.snippet.py

@@ -9,6 +9,8 @@
 #
 # Mirrors the convention from `tool-rendering/render-flight-tool.snippet.tsx`.
 
+# @region[backend-render-operations]
+# @region[backend-schema-json-load]
 from pathlib import Path
 import json
 
@@ -38,7 +40,6 @@ def render(*args, **kwargs): ...
 CATALOG_ID = "flight-catalog"
 
 
-# @region[backend-schema-json-load]
 # Schemas are JSON so they can be authored and reviewed independently of
 # the backend code. `a2ui.load_schema` is just a thin `json.load` wrapper
 # that resolves the path against the schemas directory.
@@ -47,7 +48,6 @@ def render(*args, **kwargs): ...
 
 
 def emit_render_operations(origin: str, destination: str, airline: str, price: float):
-    # @region[backend-render-operations]
     # The a2ui middleware detects the `a2ui_operations` container in this
     # tool result and forwards the ops to the frontend renderer. The
     # frontend catalog resolves component names to local React components.

showcase/integrations/ag2/src/app/demos/a2ui-fixed-schema/a2ui/catalog.ts

@@ -8,14 +8,14 @@
  * the agent's fixed schema (src/agents/a2ui_schemas/flight_schema.json) can
  * compose custom and basic components interchangeably.
  */
+// @region[catalog-creation]
 import { createCatalog } from "@copilotkit/a2ui-renderer";
 
 import { flightDefinitions } from "./definitions";
 import { flightRenderers } from "./renderers";
 
 export const CATALOG_ID = "copilotkit://flight-fixed-catalog";
 
-// @region[catalog-creation]
 export const fixedCatalog = createCatalog(flightDefinitions, flightRenderers, {
   catalogId: CATALOG_ID,
   includeBasicCatalog: true,

showcase/integrations/ag2/src/app/demos/a2ui-fixed-schema/a2ui/definitions.ts

@@ -16,6 +16,7 @@
  * This matches the canonical catalog's `DynString` helper:
  *   examples/integrations/langgraph-python/src/app/declarative-generative-ui/definitions.ts
  */
+// @region[definitions-types]
 import { z } from "zod";
 import type { CatalogDefinitions } from "@copilotkit/a2ui-renderer";
 
@@ -25,7 +26,6 @@ import type { CatalogDefinitions } from "@copilotkit/a2ui-renderer";
  */
 const DynString = z.union([z.string(), z.object({ path: z.string() })]);
 
-// @region[definitions-types]
 export const flightDefinitions = {
   Title: {
     description: "A prominent heading for the flight card.",

showcase/integrations/ag2/src/app/demos/agentic-chat-reasoning/page.tsx

@@ -17,6 +17,7 @@
 // labeled "Agent reasoning"). That is the "per-message conditional rendering
 // via slots" path — the public, stable way to customize reasoning output.
 
+  // @region[reasoning-block-render]
 import React from "react";
 import {
   CopilotKit,
@@ -44,7 +45,6 @@ export default function AgenticChatReasoningDemo() {
 // Inner — wires a custom `reasoningMessage` slot that makes the thinking
 // chain visually prominent, then renders the chat.
 function Chat() {
-  // @region[reasoning-block-render]
   return (
     <CopilotChat
       agentId="agentic-chat-reasoning"

showcase/integrations/ag2/src/app/demos/chat-slots/page.tsx

@@ -1,5 +1,8 @@
 "use client";
 
+// @region[register-assistant-message-slot]
+// @region[register-disclaimer-slot]
+// @region[register-welcome-slot]
 import React from "react";
 import {
   CopilotKit,
@@ -32,13 +35,10 @@ function Chat() {
     available: "always",
   });
 
-  // @region[register-welcome-slot]
   const welcomeScreen = CustomWelcomeScreen;
   // @endregion[register-welcome-slot]
-  // @region[register-disclaimer-slot]
   const input = { disclaimer: CustomDisclaimer };
   // @endregion[register-disclaimer-slot]
-  // @region[register-assistant-message-slot]
   const messageView = {
     assistantMessage:
       CustomAssistantMessage as unknown as typeof CopilotChatAssistantMessage,

showcase/integrations/ag2/src/app/demos/declarative-gen-ui/a2ui/catalog.ts

@@ -11,12 +11,12 @@
  * Reference:
  *   https://docs.copilotkit.ai/integrations/langgraph/generative-ui/a2ui
  */
+// @region[create-catalog]
 import { createCatalog } from "@copilotkit/a2ui-renderer";
 
 import { myDefinitions } from "./definitions";
 import { myRenderers } from "./renderers";
 
-// @region[create-catalog]
 export const myCatalog = createCatalog(myDefinitions, myRenderers, {
   catalogId: "declarative-gen-ui-catalog",
   includeBasicCatalog: true,

showcase/integrations/ag2/src/app/demos/declarative-gen-ui/a2ui/definitions.ts

@@ -11,10 +11,10 @@
  * `includeBasicCatalog: true` so the built-in A2UI primitives (Text, Row,
  * Column, Image, Card, Button, …) come along for free.
  */
+// @region[definitions-zod]
 import { z } from "zod";
 import type { CatalogDefinitions } from "@copilotkit/a2ui-renderer";
 
-// @region[definitions-zod]
 export const myDefinitions = {
   Card: {
     description: