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

package/dist/bundled.js

@@ -20817,8 +20817,6 @@ var UseAIClient = class {
   _tools = [];
   _messages = [];
   _state = null;
-  // MCP headers provider
-  mcpHeadersProvider;
   // Agent selection
   _availableAgents = [];
   _defaultAgent = null;
@@ -20977,22 +20975,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) => {
@@ -21022,14 +21013,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 (error48) {
-        console.error("[UseAIClient] Failed to get MCP headers:", error48);
-      }
-    }
     const runInput = {
       threadId: this.threadId,
       // Use getter to ensure non-null
@@ -21044,8 +21027,8 @@ var UseAIClient = class {
       state: this._state,
       context: [],
       forwardedProps: {
-        ...mcpHeaders ? { mcpHeaders } : {},
-        ...this._selectedAgent ? { agent: this._selectedAgent } : {}
+        ...this._selectedAgent ? { agent: this._selectedAgent } : {},
+        ...forwardedProps || {}
       }
     };
     this.send({
@@ -35616,7 +35599,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 });
       }
@@ -35638,7 +35621,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);
       }
@@ -36306,7 +36289,7 @@ function UseAIProvider({
   CustomButton,
   CustomChat,
   chatRepository,
-  mcpHeadersProvider,
+  forwardedPropsProvider,
   fileUploadConfig: fileUploadConfigProp,
   commandRepository,
   renderChat = true,
@@ -36357,9 +36340,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({
@@ -36419,9 +36402,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);
@@ -36495,13 +36475,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;
@@ -36521,7 +36494,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();
@@ -36575,8 +36548,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 value2 = {
     serverUrl,