Use request-scoped project agent config in hosted chat

src/agent/hosted/chat-request-parser.ts

@@ -10,6 +10,7 @@ import {
   hostedChatRequestSchema,
 } from "./chat-request.ts";
 import { RuntimeAgentRunInvocationSchema } from "../runtime/agent-invocation-contract.ts";
+import type { RuntimeAgentMarkdownDefinition } from "../runtime/agent-definition.ts";
 
 /** Public API contract for hosted chat request principal. */
 export type HostedChatRequestPrincipal = {
@@ -49,6 +50,7 @@ export type ParsedHostedChatRequest = {
   runtimeOverrides: ChatRuntimeOverrides | undefined;
   durableRootRun: DurableRootRunDescriptor | undefined;
   persistLatestUserMessageBeforeDurableRun: boolean;
+  agentConfig?: RuntimeAgentMarkdownDefinition;
 };
 
 /** Options accepted by parse hosted chat request. */
@@ -108,6 +110,7 @@ async function verifyHostedChatProjectAccess(input: {
 export async function buildParsedHostedChatRequest(input: {
   chatRequest: HostedChatRequest;
   agentId?: string;
+  agentConfig?: RuntimeAgentMarkdownDefinition;
   authToken: string;
   userId: string;
   verifyProjectAccess?: ParseHostedChatRequestOptions["verifyProjectAccess"];
@@ -125,6 +128,13 @@ export async function buildParsedHostedChatRequest(input: {
   const projectSlug = chatContext.projectSlug;
   const conversationId = chatContext.conversationId;
 
+  if (input.agentConfig && input.agentId && input.agentConfig.id !== input.agentId) {
+    return createValidationErrorResponse({
+      messagePrefix: "Invalid runtime agent invocation",
+      validationMessage: "agentConfig.id must match the requested agent id",
+    });
+  }
+
   const accessError = await verifyHostedChatProjectAccess({
     projectId,
     authToken: input.authToken,
@@ -153,6 +163,7 @@ export async function buildParsedHostedChatRequest(input: {
     runtimeOverrides,
     durableRootRun,
     persistLatestUserMessageBeforeDurableRun: false,
+    ...(input.agentConfig ? { agentConfig: input.agentConfig } : {}),
   };
 }
 
@@ -215,6 +226,7 @@ export async function parseRuntimeAgentRunInvocationHostedChatRequestFromRequest
     userId: invocation.data.run.requestedByUserId,
     chatRequest: chatRequest.data,
     agentId: invocation.data.run.agentId,
+    agentConfig: invocation.data.agentConfig,
     verifyProjectAccess: options.verifyProjectAccess,
   });
 }

src/agent/hosted/chat-request.test.ts

@@ -269,6 +269,36 @@ describe("agent/hosted-chat-request", () => {
     assertEquals(parsed.validatedContext.projectSlug, "demo-project");
   });
 
+  it("preserves request-scoped project agent config from runtime invocations", async () => {
+    const parsed = await parseRuntimeAgentRunInvocationHostedChatRequestFromRequest(
+      new Request("https://agent.example.com/api/runs", {
+        method: "POST",
+        body: JSON.stringify({
+          ...createRuntimeInvocation(),
+          agentConfig: {
+            id: "builder",
+            name: "Builder",
+            description: "Builds with project skills.",
+            instructions: "Use project skills.",
+            skills: ["support-triage"],
+            tools: ["search_knowledge", "get_file"],
+          },
+        }),
+      }),
+      {
+        authenticate: () => Promise.resolve({ userId, authToken: "token_1" }),
+        verifyProjectAccess: () => Promise.resolve({ success: true }),
+      },
+    );
+
+    if (parsed instanceof Response) {
+      throw new Error("Expected parsed request");
+    }
+
+    assertEquals(parsed.agentConfig?.skills, ["support-triage"]);
+    assertEquals(parsed.agentConfig?.tools, ["search_knowledge", "get_file"]);
+  });
+
   it("returns hosted chat project-access errors as stable JSON responses", async () => {
     const response = await parseHostedChatRequestFromRequest(
       new Request("https://agent.example.com/api/runs", {

src/agent/hosted/veryfront-cloud-agent-service.ts

@@ -915,7 +915,8 @@ async function prepareChatExecution(
 
   setPrepareChatExecutionStartAttributes(context, { projectId, userId });
 
-  const agentConfig = await resolveAgentConfig(context, req.agentId ?? getDefaultAgentId(context));
+  const requestedAgentId = req.agentId ?? getDefaultAgentId(context);
+  const agentConfig = req.agentConfig ?? await resolveAgentConfig(context, requestedAgentId);
   const abortController = new AbortController();
   const {
     effectiveMessages,

src/agent/runtime/agent-invocation-contract.test.ts

@@ -240,6 +240,30 @@ describe("agent/runtime-agent-invocation-contract", () => {
     });
   });
 
+  it("preserves the selected project agent config on control-plane stream requests", () => {
+    const parsed = RuntimeAgentRunInvocationSchema.parse(createInvocation({
+      agentConfig: {
+        id: "builder",
+        name: "Builder",
+        description: "Builds with project skills.",
+        instructions: "Use project skills.",
+        skills: ["support-triage"],
+        tools: ["search_knowledge", "get_file"],
+      },
+    }));
+
+    const request = buildRuntimeAgentControlPlaneStreamRequestFromInvocation(parsed);
+
+    assertEquals(request.agentConfig, {
+      id: "builder",
+      name: "Builder",
+      description: "Builds with project skills.",
+      instructions: "Use project skills.",
+      skills: ["support-triage"],
+      tools: ["search_knowledge", "get_file"],
+    });
+  });
+
   it("parses runtime agent invocation request bodies through the public helper", async () => {
     const parsed = await parseRuntimeAgentRunInvocation(
       new Request("http://localhost/api/control-plane/runs/run_1/stream", {

src/agent/runtime/agent-invocation-contract.ts

@@ -2,6 +2,7 @@ import { defineSchema, lazySchema } from "#veryfront/schemas/index.ts";
 import type { InferSchema, RefinementCtx } from "#veryfront/extensions/schema/index.ts";
 import { ensureBuiltinSchemaValidator } from "#veryfront/extensions/builtin-extensions.ts";
 import { parseAgUiJsonRequestOrError } from "../ag-ui/request-shared.ts";
+import { getRuntimeAgentMarkdownDefinitionSchema } from "./agent-definition.ts";
 
 ensureBuiltinSchemaValidator();
 
@@ -313,6 +314,7 @@ export const getRuntimeAgentRunInvocationSchema = defineSchema((v) =>
       { message: "context must be less than 64 KB total" },
     ),
     agentSource: getRuntimeAgentSourceContextSchema().optional(),
+    agentConfig: getRuntimeAgentMarkdownDefinitionSchema().optional(),
     credentials: getRuntimeAgentCredentialsSchema().optional(),
     forwardedProps: v.record(v.string(), v.unknown()).optional().refine(
       (value) => value === undefined || isWithinJsonSizeLimit(value, MAX_FORWARDED_PROPS_BYTES),
@@ -368,6 +370,7 @@ export type RuntimeAgentControlPlaneStreamRequest = {
   context: RuntimeAgentRunInvocation["context"];
   credentials?: RuntimeAgentRunInvocation["credentials"];
   agentSource?: RuntimeAgentRunInvocation["agentSource"];
+  agentConfig?: RuntimeAgentRunInvocation["agentConfig"];
   forwardedProps?: RuntimeAgentRunInvocation["forwardedProps"];
 };
 
@@ -385,6 +388,7 @@ export function buildRuntimeAgentControlPlaneStreamRequestFromInvocation(
     context: input.context,
     ...(input.credentials ? { credentials: input.credentials } : {}),
     ...(input.agentSource ? { agentSource: input.agentSource } : {}),
+    ...(input.agentConfig ? { agentConfig: input.agentConfig } : {}),
     ...(input.forwardedProps ? { forwardedProps: input.forwardedProps } : {}),
   };
 }