@meetsmore-oss/use-ai-client 1.6.0 → 1.7.0

package/dist/index.d.ts

@@ -1,7 +1,6 @@
 import { z } from 'zod';
 export { z } from 'zod';
-import * as _meetsmore_oss_use_ai_core from '@meetsmore-oss/use-ai-core';
-import { ToolAnnotations, ToolDefinition, WorkflowStatus, FeedbackValue, AgentInfo, McpHeadersMap, MultimodalContent, AGUIEvent, Message as Message$1, UseAIClientMessage } from '@meetsmore-oss/use-ai-core';
+import { ToolAnnotations, ToolDefinition, WorkflowStatus, FeedbackValue, AgentInfo, MultimodalContent, UseAIForwardedProps, AGUIEvent, Message as Message$1, UseAIClientMessage } from '@meetsmore-oss/use-ai-core';
 export { AgentInfo, ToolAnnotations, ToolDefinition } from '@meetsmore-oss/use-ai-core';
 import * as react_jsx_runtime from 'react/jsx-runtime';
 import React$1, { ReactNode } from 'react';
@@ -941,7 +940,6 @@ declare class UseAIClient {
     private _tools;
     private _messages;
     private _state;
-    private mcpHeadersProvider?;
     private _availableAgents;
     private _defaultAgent;
     private _selectedAgent;
@@ -982,20 +980,15 @@ declare class UseAIClient {
      * @param state - The current state object to provide to the AI
      */
     updateState(state: unknown): void;
-    /**
-     * Sets the MCP headers provider.
-     * The provider will be called each time a message is sent to get fresh headers.
-     *
-     * @param provider - Function that returns MCP headers configuration
-     */
-    setMcpHeadersProvider(provider: () => McpHeadersMap | Promise<McpHeadersMap>): void;
     /**
      * Sends a user prompt to the AI.
      *
      * @param prompt - The user's prompt/question (text part)
      * @param multimodalContent - Optional multimodal content (text, images, files)
+     * @param forwardedProps - Optional props to forward to the server (e.g., telemetryMetadata, mcpHeaders).
+     *                         Internally merged with other forwardedProps.
      */
-    sendPrompt(prompt: string, multimodalContent?: MultimodalContent[]): Promise<void>;
+    sendPrompt(prompt: string, multimodalContent?: MultimodalContent[], forwardedProps?: UseAIForwardedProps): Promise<void>;
     /**
      * Sends the result of a tool execution back to the server.
      *
@@ -1162,6 +1155,11 @@ interface SendMessageOptions {
     openChat?: boolean;
     /** Metadata to set on the new chat (only used when newChat: true) */
     metadata?: ChatMetadata;
+    /**
+     * Forwarded props for observability and configuration (e.g., telemetryMetadata, mcpHeaders).
+     * This is merged with provider-level forwardedProps (message-level takes precedence).
+     */
+    forwardedProps?: UseAIForwardedProps;
 }
 interface UseChatManagementOptions {
     /** Chat repository for persistence */
@@ -1173,7 +1171,7 @@ interface UseChatManagementOptions {
     /** Setter for messages state (owned by provider) */
     setMessages: React.Dispatch<React.SetStateAction<Message[]>>;
     /** Callback to send a message (from UseAIProvider) */
-    onSendMessage?: (message: string, attachments?: FileAttachment[]) => Promise<void>;
+    onSendMessage?: (message: string, attachments?: FileAttachment[], forwardedProps?: UseAIForwardedProps) => Promise<void>;
     /** Callback to open/close the chat panel */
     setOpen?: (open: boolean) => void;
     /** Whether the client is connected */
@@ -1407,33 +1405,41 @@ interface UseAIProviderProps extends UseAIConfig {
      */
     chatRepository?: ChatRepository;
     /**
-     * Callback to provide HTTP headers for MCP endpoints.
-     * Called each time AI is invoked by use-ai.
-     * Returns a mapping of MCP endpoint patterns to header configurations.
-     *
-     * Patterns can be:
-     * - Constant strings: `https://api.example.com` - Exact match
-     * - Glob patterns: `https://*.meetsmore.com` - Wildcard matching using picomatch
+     * Provider function for forwarded props (telemetry metadata, MCP headers, etc.).
+     * Called before each message is sent. Can be sync or async.
+     * Props from this provider are merged with message-level forwardedProps,
+     * with message-level taking precedence.
      *
      * @example
-     * ```typescript
-     * mcpHeadersProvider={() => ({
-     *   // Exact match
-     *   'https://api.example.com': {
-     *     headers: { 'Authorization': `Bearer ${userToken}` }
-     *   },
-     *   // Wildcard subdomain
-     *   'https://*.meetsmore.com': {
-     *     headers: { 'X-API-Key': apiKey }
-     *   },
-     *   // Multiple wildcards
-     *   '*://*.example.com': {
-     *     headers: { 'X-Custom': 'value' }
-     *   }
-     * })}
+     * ```tsx
+     * <UseAIProvider
+     *   serverUrl="wss://your-server.com"
+     *   forwardedPropsProvider={() => ({
+     *     mcpHeaders: {
+     *       // Exact match
+     *       'https://api.example.com': {
+     *         headers: { 'Authorization': `Bearer ${userToken}` }
+     *       },
+     *       // Wildcard subdomain
+     *       'https://*.meetsmore.com': {
+     *         headers: { 'X-API-Key': apiKey }
+     *       },
+     *       // Multiple wildcards
+     *       '*://*.example.com': {
+     *         headers: { 'X-Custom': 'value' }
+     *       },
+     *     },
+     *     telemetryMetadata: {
+     *       userId: currentUser.id,
+     *       tenantId: tenant.id,
+     *     },
+     *   })}
+     * >
+     *   <App />
+     * </UseAIProvider>
      * ```
      */
-    mcpHeadersProvider?: () => _meetsmore_oss_use_ai_core.McpHeadersMap | Promise<_meetsmore_oss_use_ai_core.McpHeadersMap>;
+    forwardedPropsProvider?: () => UseAIForwardedProps | Promise<UseAIForwardedProps>;
     /**
      * Configuration for file uploads.
      * File upload is enabled by default with EmbedFileUploadBackend, 10MB max size,
@@ -1547,7 +1553,7 @@ interface UseAIProviderProps extends UseAIConfig {
  * }
  * ```
  */
-declare function UseAIProvider({ serverUrl, children, systemPrompt, CustomButton, CustomChat, chatRepository, mcpHeadersProvider, fileUploadConfig: fileUploadConfigProp, commandRepository, renderChat, theme: customTheme, strings: customStrings, visibleAgentIds, onOpenChange, }: UseAIProviderProps): react_jsx_runtime.JSX.Element;
+declare function UseAIProvider({ serverUrl, children, systemPrompt, CustomButton, CustomChat, chatRepository, forwardedPropsProvider, fileUploadConfig: fileUploadConfigProp, commandRepository, renderChat, theme: customTheme, strings: customStrings, visibleAgentIds, onOpenChange, }: UseAIProviderProps): react_jsx_runtime.JSX.Element;
 /**
  * Hook to access the UseAI context.
  * When used outside a UseAIProvider, returns a no-op context and logs a warning.

package/dist/index.js

@@ -3072,8 +3072,6 @@ var UseAIClient = class {
   _tools = [];
   _messages = [];
   _state = null;
-  // MCP headers provider
-  mcpHeadersProvider;
   // Agent selection
   _availableAgents = [];
   _defaultAgent = null;
@@ -3232,22 +3230,15 @@ var UseAIClient = class {
   updateState(state) {
     this._state = state;
   }
-  /**
-   * Sets the MCP headers provider.
-   * The provider will be called each time a message is sent to get fresh headers.
-   *
-   * @param provider - Function that returns MCP headers configuration
-   */
-  setMcpHeadersProvider(provider) {
-    this.mcpHeadersProvider = provider;
-  }
   /**
    * Sends a user prompt to the AI.
    *
    * @param prompt - The user's prompt/question (text part)
    * @param multimodalContent - Optional multimodal content (text, images, files)
+   * @param forwardedProps - Optional props to forward to the server (e.g., telemetryMetadata, mcpHeaders).
+   *                         Internally merged with other forwardedProps.
    */
-  async sendPrompt(prompt, multimodalContent) {
+  async sendPrompt(prompt, multimodalContent, forwardedProps) {
     let messageContent = prompt;
     if (multimodalContent && multimodalContent.length > 0) {
       messageContent = multimodalContent.map((part) => {
@@ -3277,14 +3268,6 @@ var UseAIClient = class {
       // Type cast needed for Message type compatibility
     };
     this._messages.push(userMessage);
-    let mcpHeaders;
-    if (this.mcpHeadersProvider) {
-      try {
-        mcpHeaders = await this.mcpHeadersProvider();
-      } catch (error) {
-        console.error("[UseAIClient] Failed to get MCP headers:", error);
-      }
-    }
     const runInput = {
       threadId: this.threadId,
       // Use getter to ensure non-null
@@ -3299,8 +3282,8 @@ var UseAIClient = class {
       state: this._state,
       context: [],
       forwardedProps: {
-        ...mcpHeaders ? { mcpHeaders } : {},
-        ...this._selectedAgent ? { agent: this._selectedAgent } : {}
+        ...this._selectedAgent ? { agent: this._selectedAgent } : {},
+        ...forwardedProps || {}
       }
     };
     this.send({
@@ -4104,7 +4087,7 @@ function useChatManagement({
     isProcessingQueueRef.current = true;
     while (pendingMessagesRef.current.length > 0) {
       const { message, options } = pendingMessagesRef.current.shift();
-      const { newChat = false, attachments = [], openChat = true, metadata } = options ?? {};
+      const { newChat = false, attachments = [], openChat = true, metadata, forwardedProps } = options ?? {};
       if (newChat) {
         await createNewChat({ metadata });
       }
@@ -4126,7 +4109,7 @@ function useChatManagement({
           };
         })
       );
-      await onSendMessage(message, fileAttachments.length > 0 ? fileAttachments : void 0);
+      await onSendMessage(message, fileAttachments.length > 0 ? fileAttachments : void 0, forwardedProps);
       if (openChat && setOpen) {
         setOpen(true);
       }
@@ -4794,7 +4777,7 @@ function UseAIProvider({
   CustomButton,
   CustomChat,
   chatRepository,
-  mcpHeadersProvider,
+  forwardedPropsProvider,
   fileUploadConfig: fileUploadConfigProp,
   commandRepository,
   renderChat = true,
@@ -4845,9 +4828,9 @@ function UseAIProvider({
     clientRef,
     connected
   });
-  const stableSendMessage = useCallback11(async (message, attachments) => {
+  const stableSendMessage = useCallback11(async (message, attachments, forwardedProps) => {
     if (handleSendMessageRef.current) {
-      await handleSendMessageRef.current(message, attachments);
+      await handleSendMessageRef.current(message, attachments, forwardedProps);
     }
   }, []);
   const toolExecution = useToolExecution({
@@ -4907,9 +4890,6 @@ function UseAIProvider({
   useEffect10(() => {
     console.log("[UseAIProvider] Initializing client with serverUrl:", serverUrl);
     const client = new UseAIClient(serverUrl);
-    if (mcpHeadersProvider) {
-      client.setMcpHeadersProvider(mcpHeadersProvider);
-    }
     const unsubscribeConnection = client.onConnectionStateChange((isConnected) => {
       console.log("[UseAIProvider] Connection state changed:", isConnected);
       setConnected(isConnected);
@@ -4983,13 +4963,6 @@ function UseAIProvider({
       client.disconnect();
     };
   }, [serverUrl]);
-  useEffect10(() => {
-    const client = clientRef.current;
-    if (!client) return;
-    if (mcpHeadersProvider) {
-      client.setMcpHeadersProvider(mcpHeadersProvider);
-    }
-  }, [mcpHeadersProvider]);
   const lastRegisteredToolsRef = useRef11("");
   useEffect10(() => {
     const client = clientRef.current;
@@ -5009,7 +4982,7 @@ function UseAIProvider({
       console.error("Failed to register tools:", err);
     }
   }, [hasTools, aggregatedTools, connected]);
-  const handleSendMessage = useCallback11(async (message, attachments) => {
+  const handleSendMessage = useCallback11(async (message, attachments, messageForwardedProps) => {
     if (!clientRef.current) return;
     setStreamingText("");
     const activatedChatId = activatePendingChat();
@@ -5063,8 +5036,18 @@ function UseAIProvider({
       }
       setLoading(true);
     }
-    await clientRef.current.sendPrompt(message, multimodalContent);
-  }, [activatePendingChat, currentChatId, saveUserMessage, fileUploadConfig, getCurrentChat]);
+    const providerResult = forwardedPropsProvider ? forwardedPropsProvider() : {};
+    const providerProps = providerResult instanceof Promise ? await providerResult : providerResult;
+    const mergedForwardedProps = {
+      ...providerProps,
+      ...messageForwardedProps
+    };
+    await clientRef.current.sendPrompt(
+      message,
+      multimodalContent,
+      Object.keys(mergedForwardedProps).length > 0 ? mergedForwardedProps : void 0
+    );
+  }, [activatePendingChat, currentChatId, saveUserMessage, fileUploadConfig, getCurrentChat, forwardedPropsProvider]);
   handleSendMessageRef.current = handleSendMessage;
   const value = {
     serverUrl,