@springbrand/chat-client 0.1.3-alpha.3 → 0.1.3-alpha.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@springbrand/chat-client",
-  "version": "0.1.3-alpha.3",
+  "version": "0.1.3-alpha.5",
   "type": "module",
   "files": [
     "src",
@@ -25,7 +25,7 @@
     "@types/react-dom": "^19.2.3",
     "react-dom": "^19.2.7",
     "typescript": "^7.0.2",
-    "@springbrand/agent-runtime": "0.1.3-alpha.4"
+    "@springbrand/agent-runtime": "0.1.3-alpha.6"
   },
   "scripts": {
     "typecheck": "tsc --noEmit"

package/src/use-universal-agent-chat.ts

@@ -8,6 +8,7 @@ import { useAgent, useAgentToolEvents } from "agents/react";
 import {
   type MessageDelivery,
   type MessageDispatchReceipt,
+  type RequestedCapability,
   type RuntimeAssemblyView,
   type RuntimeConfigUpdateResult,
   type RuntimeQueuedSubmission,
@@ -17,6 +18,7 @@ import { createSafeUIMessageAgentConnection } from "./ui-message-stream-guard";
 import { CURRENT_INBOX_AGENT_PATH } from "./chat-sessions";
 
 type ChatMessageMetadata = Record<string, unknown> & {
+  requestedCapabilities?: readonly RequestedCapability[];
   turnStatus?: string;
 };
 type ChatMessage = UIMessage<ChatMessageMetadata>;
@@ -85,9 +87,21 @@ export interface ChatRuntime {
   runtimeLoad: ChatRuntimeLoadState | undefined;
   isStreaming: boolean;
   error: string | Error | undefined;
-  sendText: (text: string, files?: readonly FileUIPart[]) => Promise<boolean>;
-  steerText: (text: string, files?: readonly FileUIPart[]) => Promise<boolean>;
-  enqueueText: (text: string, files?: readonly FileUIPart[]) => Promise<boolean>;
+  sendText: (
+    text: string,
+    files?: readonly FileUIPart[],
+    capabilities?: readonly RequestedCapability[],
+  ) => Promise<boolean>;
+  steerText: (
+    text: string,
+    files?: readonly FileUIPart[],
+    capabilities?: readonly RequestedCapability[],
+  ) => Promise<boolean>;
+  enqueueText: (
+    text: string,
+    files?: readonly FileUIPart[],
+    capabilities?: readonly RequestedCapability[],
+  ) => Promise<boolean>;
   steerQueued: (submissionId: string) => Promise<boolean>;
   cancelQueued: (submissionId: string) => Promise<boolean>;
   stop: () => Promise<void>;
@@ -195,10 +209,18 @@ function markChatTurnPhase(
 function createChatMessage(
   text: string,
   files?: readonly FileUIPart[],
+  capabilities: readonly RequestedCapability[] = [],
 ): ChatMessage | null {
   const normalizedText = text.trim();
   if (!normalizedText && !files?.length) return null;
   const createdAt = Date.now();
+  const seen = new Set<string>();
+  const requestedCapabilities = capabilities.filter((capability) => {
+    const key = `${capability.kind}:${capability.name}`;
+    if (seen.has(key)) return false;
+    seen.add(key);
+    return true;
+  });
   return {
     id: nanoid(),
     role: "user",
@@ -212,6 +234,7 @@ function createChatMessage(
       createdAt,
       authorDisplayName: "You",
       messageSource: "Web",
+      ...(requestedCapabilities.length > 0 ? { requestedCapabilities } : {}),
     },
   };
 }
@@ -572,8 +595,9 @@ export function useUniversalAgentChat(
   const sendText = useCallback(async (
     text: string,
     files?: readonly FileUIPart[],
+    capabilities?: readonly RequestedCapability[],
   ): Promise<boolean> => {
-    const message = createChatMessage(text, files);
+    const message = createChatMessage(text, files, capabilities);
     return message
       ? submitMessage(message, "enqueue", "message")
       : false;
@@ -582,21 +606,28 @@ export function useUniversalAgentChat(
   const dispatchText = useCallback(async (
     text: string,
     files: readonly FileUIPart[] | undefined,
+    capabilities: readonly RequestedCapability[] | undefined,
     delivery: MessageDelivery,
     projection: "message" | "queue",
   ): Promise<boolean> => {
-    const message = createChatMessage(text, files);
+    const message = createChatMessage(text, files, capabilities);
     return message ? submitMessage(message, delivery, projection) : false;
   }, [submitMessage]);
 
   const steerText = useCallback(
-    (text: string, files?: readonly FileUIPart[]) =>
-      dispatchText(text, files, "steer", "message"),
+    (
+      text: string,
+      files?: readonly FileUIPart[],
+      capabilities?: readonly RequestedCapability[],
+    ) => dispatchText(text, files, capabilities, "steer", "message"),
     [dispatchText],
   );
   const enqueueText = useCallback(
-    (text: string, files?: readonly FileUIPart[]) =>
-      dispatchText(text, files, "enqueue", "queue"),
+    (
+      text: string,
+      files?: readonly FileUIPart[],
+      capabilities?: readonly RequestedCapability[],
+    ) => dispatchText(text, files, capabilities, "enqueue", "queue"),
     [dispatchText],
   );
   const steerQueued = useCallback(async (submissionId: string) => {