@tambo-ai/react 0.58.0 → 0.59.0

package/dist/hooks/use-tambo-stream-status.js

@@ -203,7 +203,7 @@ function useTamboStreamStatus() {
             : undefined;
         const hasComponent = !!message?.component;
         return deriveGlobalStreamStatus(generationStage, propStatus, hasComponent, generationError);
-    }, [generationStage, propStatus, message?.error, message?.component]);
+    }, [generationStage, propStatus, message]);
     return {
         streamStatus,
         propStatus,

package/dist/hooks/use-tambo-stream-status.js.map

@@ -1 +1 @@
-{"version":3,"file":"use-tambo-stream-status.js","sourceRoot":"","sources":["../../src/hooks/use-tambo-stream-status.tsx"],"names":[],"mappings":";AAAA,YAAY,CAAC;;AA+Tb,oDAwCC;AAtWD,iCAAqD;AACrD,sFAAuE;AACvE,8EAA6E;AAC7E,+DAA+D;AAqE/D;;;;GAIG;AACH,SAAS,gBAAgB;IACvB,IAAI,OAAO,MAAM,KAAK,WAAW,EAAE,CAAC;QAClC,MAAM,IAAI,KAAK,CACb,6DAA6D;YAC3D,4CAA4C;YAC5C,6DAA6D,CAChE,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;;;;;;;;GASG;AACH,SAAS,uBAAuB,CAC9B,KAAwB,EACxB,eAAgC,EAChC,SAAiB;IAEjB,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,IAAA,gBAAQ,EAU9C,EAAE,CAAC,CAAC;IAEN,mDAAmD;IACnD,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,eAAe,CAAC,CAAC,IAAI,EAAE,EAAE;YACvB,0DAA0D;YAC1D,MAAM,iBAAiB,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAChD,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,SAAS,KAAK,SAAS,CAC5D,CAAC;YACF,OAAO,iBAAiB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;IAEhB,oFAAoF;IACpF,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,IAAI,CAAC,KAAK;YAAE,OAAO;QAEnB,eAAe,CAAC,CAAC,IAAI,EAAE,EAAE;YACvB,MAAM,OAAO,GAAG,EAAE,GAAG,IAAI,EAAE,CAAC;YAC5B,IAAI,UAAU,GAAG,KAAK,CAAC;YAEvB,oDAAoD;YACpD,MAAM,gBAAgB,GAAa,EAAE,CAAC;YACtC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;gBAC7C,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI;oBAC3B,UAAU,EAAE,KAAK;oBACjB,UAAU,EAAE,KAAK;iBAClB,CAAC;gBAEF,+EAA+E;gBAC/E,MAAM,UAAU,GACd,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,EAAE,CAAC;gBACxD,MAAM,WAAW,GAAG,UAAU,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;gBAEtD,IAAI,WAAW,EAAE,CAAC;oBAChB,gBAAgB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBAC7B,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,mEAAmE;YACnE,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;gBAC7C,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI;oBAC3B,UAAU,EAAE,KAAK;oBACjB,UAAU,EAAE,KAAK;iBAClB,CAAC;gBAEF,+EAA+E;gBAC/E,MAAM,UAAU,GACd,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,EAAE,CAAC;gBACxD,MAAM,WAAW,GAAG,UAAU,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;gBAEtD;;;;mBAIG;gBACH,MAAM,uBAAuB,GAAG,gBAAgB,CAAC,IAAI,CACnD,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,KAAK,GAAG,CACrC,CAAC;gBACF,MAAM,oBAAoB,GACxB,eAAe,KAAK,6CAAe,CAAC,QAAQ,CAAC;gBAC/C,MAAM,UAAU,GACd,OAAO,CAAC,UAAU;oBAClB,CAAC,uBAAuB,IAAI,oBAAoB,CAAC;oBACjD,CAAC,OAAO,CAAC,UAAU,CAAC;gBAEtB,8DAA8D;gBAC9D,IAAI,OAAO,CAAC,UAAU,IAAI,OAAO,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;oBAC1D,2CAA2C;oBAC3C,OAAO;gBACT,CAAC;gBAED,IAAI,WAAW,IAAI,UAAU,EAAE,CAAC;oBAC9B,OAAO,CAAC,GAAG,CAAC,GAAG;wBACb,GAAG,OAAO;wBACV,UAAU,EAAE,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU;wBACnD,UAAU,EAAE,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU;wBAClD,SAAS;qBACV,CAAC;oBACF,UAAU,GAAG,IAAI,CAAC;gBACpB,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,OAAO,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC;QACrC,CAAC,CAAC,CAAC;IACL,CAAC,EAAE,CAAC,KAAK,EAAE,eAAe,EAAE,SAAS,CAAC,CAAC,CAAC;IAExC,mDAAmD;IACnD,OAAO,IAAA,eAAO,EAAC,GAAG,EAAE;QAClB,IAAI,CAAC,KAAK;YAAE,OAAO,EAAqC,CAAC;QAEzD,MAAM,MAAM,GAAG,EAAqC,CAAC;QAErD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YACjC,MAAM,QAAQ,GAAG,YAAY,CAAC,GAAG,CAAC,IAAI;gBACpC,UAAU,EAAE,KAAK;gBACjB,UAAU,EAAE,KAAK;gBACjB,SAAS,EAAE,EAAE;aACd,CAAC;YAEF,+DAA+D;YAC/D,MAAM,wBAAwB,GAC5B,QAAQ,CAAC,UAAU,IAAI,QAAQ,CAAC,SAAS,KAAK,SAAS,CAAC;YAE1D,sFAAsF;YACtF,MAAM,qBAAqB,GACzB,CAAC,wBAAwB;gBACzB,eAAe,KAAK,6CAAe,CAAC,kBAAkB,CAAC;YAEzD,MAAM,CAAC,GAAkB,CAAC,GAAG;gBAC3B,SAAS,EAAE,CAAC,QAAQ,CAAC,UAAU,IAAI,CAAC,wBAAwB;gBAC5D,WAAW,EACT,QAAQ,CAAC,UAAU;oBACnB,CAAC,wBAAwB;oBACzB,qBAAqB;gBACvB,SAAS,EAAE,wBAAwB;gBACnC,KAAK,EAAE,QAAQ,CAAC,KAAK;aACtB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,OAAO,MAAM,CAAC;IAChB,CAAC,EAAE,CAAC,KAAK,EAAE,YAAY,EAAE,eAAe,EAAE,SAAS,CAAC,CAAC,CAAC;AACxD,CAAC;AAED;;;;;;;;;GASG;AACH,SAAS,wBAAwB,CAC/B,eAAgC,EAChC,UAA2C,EAC3C,YAAqB,EACrB,eAAuB;IAEvB,MAAM,YAAY,GAAG,MAAM,CAAC,MAAM,CAAC,UAAU,CAAiB,CAAC;IAE/D,gGAAgG;IAChG,MAAM,kBAAkB,GACtB,YAAY,CAAC,MAAM,GAAG,CAAC,IAAI,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;IAEpE,+DAA+D;IAC/D,MAAM,qBAAqB,GACzB,CAAC,kBAAkB;QACnB,eAAe,KAAK,6CAAe,CAAC,kBAAkB,CAAC;IACzD,MAAM,iBAAiB,GAAG,eAAe,KAAK,6CAAe,CAAC,KAAK,CAAC;IAEpE,mDAAmD;IACnD,MAAM,UAAU,GACd,eAAe,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,KAAK,CAAC;IAE9D,OAAO;QACL,+EAA+E;QAC/E,SAAS,EACP,CAAC,YAAY;YACb,CAAC,CAAC,qBAAqB;gBACrB,CAAC,kBAAkB;gBACnB,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QAE3C,iGAAiG;QACjG,WAAW,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,CAAC;QAEpD,oFAAoF;QACpF,SAAS,EAAE,kBAAkB;QAE7B,kDAAkD;QAClD,OAAO,EACL,iBAAiB;YACjB,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC;YACjC,CAAC,CAAC,eAAe;QAEnB,WAAW,EAAE,UAAU;KACxB,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;GAyBG;AACH,SAAgB,oBAAoB;IAMlC,oDAAoD;IACpD,gBAAgB,EAAE,CAAC;IAEnB,MAAM,EAAE,eAAe,EAAE,GAAG,IAAA,+CAAuB,GAAE,CAAC;IACtD,MAAM,OAAO,GAAG,IAAA,4CAAsB,GAAE,CAAC;IAEzC,uDAAuD;IACvD,MAAM,cAAc,GAAI,OAAO,EAAE,SAAS,EAAE,KAAe,IAAK,EAAY,CAAC;IAE7E,sCAAsC;IACtC,MAAM,UAAU,GAAG,uBAAuB,CACxC,cAAc,EACd,eAAe,EACf,OAAO,EAAE,EAAE,IAAI,EAAE,CAClB,CAAC;IAEF,0EAA0E;IAC1E,MAAM,YAAY,GAAG,IAAA,eAAO,EAAC,GAAG,EAAE;QAChC,MAAM,eAAe,GAAG,OAAO,EAAE,KAAK;YACpC,CAAC,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;YAC1B,CAAC,CAAC,SAAS,CAAC;QACd,MAAM,YAAY,GAAG,CAAC,CAAC,OAAO,EAAE,SAAS,CAAC;QAC1C,OAAO,wBAAwB,CAC7B,eAAe,EACf,UAAU,EACV,YAAY,EACZ,eAAe,CAChB,CAAC;IACJ,CAAC,EAAE,CAAC,eAAe,EAAE,UAAU,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC,CAAC;IAEtE,OAAO;QACL,YAAY;QACZ,UAAU;KACX,CAAC;AACJ,CAAC","sourcesContent":["\"use client\";\nimport { useEffect, useMemo, useState } from \"react\";\nimport { GenerationStage } from \"../model/generate-component-response\";\nimport { useTamboGenerationStage } from \"../providers/tambo-thread-provider\";\nimport { useTamboCurrentMessage } from \"./use-current-message\";\n\n/**\n * Global stream status flags for a specific component in a message.\n * Represents the aggregate state across all props for this component only.\n * Once a component completes, its status remains stable regardless of other generations.\n */\nexport interface StreamStatus {\n  /**\n   * Indicates no tokens have been received for any prop and generation is not active.\n   * Useful for showing initial loading states before any data arrives.\n   */\n  isPending: boolean;\n\n  /**\n   * Indicates active streaming - either generation is streaming OR at least one prop is still streaming.\n   * Use this to show loading animations or skeleton states during data transmission.\n   */\n  isStreaming: boolean;\n\n  /**\n   * Indicates successful completion - generation is complete AND every prop finished without error.\n   * Safe to render the final component when this is true.\n   */\n  isSuccess: boolean;\n\n  /**\n   * Indicates a fatal error occurred in any prop or the stream itself.\n   * Check streamError for details about what went wrong.\n   */\n  isError: boolean;\n\n  /**\n   * The first fatal error encountered during streaming (if any).\n   * Will be undefined if no errors occurred.\n   */\n  streamError?: Error;\n}\n\n/**\n * Streaming status flags for individual component props.\n * Tracks the state of each prop as it streams from the LLM.\n */\nexport interface PropStatus {\n  /**\n   * Indicates no tokens have been received for this specific prop yet.\n   * The prop value is still undefined, null, or empty string.\n   */\n  isPending: boolean;\n\n  /**\n   * Indicates at least one token has been received but streaming is not complete.\n   * The prop has partial content that may still be updating.\n   */\n  isStreaming: boolean;\n\n  /**\n   * Indicates this prop has finished streaming successfully.\n   * The prop value is complete and stable.\n   */\n  isSuccess: boolean;\n\n  /**\n   * The error that occurred during streaming (if any).\n   * Will be undefined if no error occurred for this prop.\n   */\n  error?: Error;\n}\n\n/**\n * SSR Guard - throws during server-side rendering.\n * Ensures the hook is only used in browser contexts.\n * @throws {Error} When called during server-side rendering\n */\nfunction assertClientSide() {\n  if (typeof window === \"undefined\") {\n    throw new Error(\n      \"useTamboStreamStatus can only be used in browser contexts. \" +\n        \"This hook is not compatible with SSR/SSG. \" +\n        \"Consider wrapping it in useEffect or using dynamic imports.\",\n    );\n  }\n}\n\n/**\n * Track streaming status for individual props by monitoring their values.\n * Monitors when props receive their first token and when they complete streaming.\n * Maintains stable state per message - once props complete for a message, they stay complete.\n * @template Props - The type of the component props being tracked\n * @param props - The current component props object\n * @param generationStage - The current generation stage from the LLM\n * @param messageId - The ID of the current message to track component-specific state\n * @returns A record mapping each prop key to its PropStatus\n */\nfunction usePropsStreamingStatus<Props extends Record<string, any>>(\n  props: Props | undefined,\n  generationStage: GenerationStage,\n  messageId: string,\n): Record<keyof Props, PropStatus> {\n  const [propTracking, setPropTracking] = useState<\n    Record<\n      string,\n      {\n        hasStarted: boolean;\n        isComplete: boolean;\n        error?: Error;\n        messageId: string;\n      }\n    >\n  >({});\n\n  /** Reset tracking only when the message changes */\n  useEffect(() => {\n    setPropTracking((prev) => {\n      // If we have tracking data for a different message, reset\n      const hasOldMessageData = Object.values(prev).some(\n        (track) => track.messageId && track.messageId !== messageId,\n      );\n      return hasOldMessageData ? {} : prev;\n    });\n  }, [messageId]);\n\n  /** Track when props start streaming (receive first token) and when they complete */\n  useEffect(() => {\n    if (!props) return;\n\n    setPropTracking((prev) => {\n      const updated = { ...prev };\n      let hasChanges = false;\n\n      // First pass: identify which props are starting now\n      const propsStartingNow: string[] = [];\n      Object.entries(props).forEach(([key, value]) => {\n        const current = prev[key] || {\n          hasStarted: false,\n          isComplete: false,\n        };\n\n        /** A prop starts streaming when it has a non-empty value for the first time */\n        const hasContent =\n          value !== undefined && value !== null && value !== \"\";\n        const justStarted = hasContent && !current.hasStarted;\n\n        if (justStarted) {\n          propsStartingNow.push(key);\n        }\n      });\n\n      // Second pass: update tracking and mark previous props as complete\n      Object.entries(props).forEach(([key, value]) => {\n        const current = prev[key] || {\n          hasStarted: false,\n          isComplete: false,\n        };\n\n        /** A prop starts streaming when it has a non-empty value for the first time */\n        const hasContent =\n          value !== undefined && value !== null && value !== \"\";\n        const justStarted = hasContent && !current.hasStarted;\n\n        /**\n         * A prop is complete when it has started and either:\n         * 1. A following prop has started, OR\n         * 2. Generation is complete (for the final prop)\n         */\n        const hasFollowingPropStarted = propsStartingNow.some(\n          (startingKey) => startingKey !== key,\n        );\n        const isGenerationComplete =\n          generationStage === GenerationStage.COMPLETE;\n        const isComplete =\n          current.hasStarted &&\n          (hasFollowingPropStarted || isGenerationComplete) &&\n          !current.isComplete;\n\n        // Once a prop is complete for this message, it stays complete\n        if (current.isComplete && current.messageId === messageId) {\n          // Skip - already complete for this message\n          return;\n        }\n\n        if (justStarted || isComplete) {\n          updated[key] = {\n            ...current,\n            hasStarted: justStarted ? true : current.hasStarted,\n            isComplete: isComplete ? true : current.isComplete,\n            messageId,\n          };\n          hasChanges = true;\n        }\n      });\n\n      return hasChanges ? updated : prev;\n    });\n  }, [props, generationStage, messageId]);\n\n  /** Convert tracking state to PropStatus objects */\n  return useMemo(() => {\n    if (!props) return {} as Record<keyof Props, PropStatus>;\n\n    const result = {} as Record<keyof Props, PropStatus>;\n\n    Object.keys(props).forEach((key) => {\n      const tracking = propTracking[key] || {\n        hasStarted: false,\n        isComplete: false,\n        messageId: \"\",\n      };\n\n      // If this prop is complete for this message, it stays complete\n      const isCompleteForThisMessage =\n        tracking.isComplete && tracking.messageId === messageId;\n\n      // Only consider generation stage if this prop isn't already complete for this message\n      const isGenerationStreaming =\n        !isCompleteForThisMessage &&\n        generationStage === GenerationStage.STREAMING_RESPONSE;\n\n      result[key as keyof Props] = {\n        isPending: !tracking.hasStarted && !isCompleteForThisMessage,\n        isStreaming:\n          tracking.hasStarted &&\n          !isCompleteForThisMessage &&\n          isGenerationStreaming,\n        isSuccess: isCompleteForThisMessage,\n        error: tracking.error,\n      };\n    });\n\n    return result;\n  }, [props, propTracking, generationStage, messageId]);\n}\n\n/**\n * Derives global StreamStatus from generation stage and individual prop statuses.\n * Aggregates individual prop states into a unified stream status.\n * @template Props - The type of the component props\n * @param generationStage - The current generation stage from the LLM\n * @param propStatus - Status record for each individual prop\n * @param hasComponent - Whether a component exists in the current message\n * @param generationError - Any error from the generation process itself\n * @returns The aggregated StreamStatus for the entire component\n */\nfunction deriveGlobalStreamStatus<Props extends Record<string, any>>(\n  generationStage: GenerationStage,\n  propStatus: Record<keyof Props, PropStatus>,\n  hasComponent: boolean,\n  generationError?: Error,\n): StreamStatus {\n  const propStatuses = Object.values(propStatus) as PropStatus[];\n\n  // If all props are already successful, the component is complete regardless of generation stage\n  const allPropsSuccessful =\n    propStatuses.length > 0 && propStatuses.every((p) => p.isSuccess);\n\n  // Only consider generation stage if not all props are complete\n  const isGenerationStreaming =\n    !allPropsSuccessful &&\n    generationStage === GenerationStage.STREAMING_RESPONSE;\n  const isGenerationError = generationStage === GenerationStage.ERROR;\n\n  /** Find first error from generation or any prop */\n  const firstError =\n    generationError ?? propStatuses.find((p) => p.error)?.error;\n\n  return {\n    /** isPending: no component yet OR (has component but no props have started) */\n    isPending:\n      !hasComponent ||\n      (!isGenerationStreaming &&\n        !allPropsSuccessful &&\n        propStatuses.every((p) => p.isPending)),\n\n    /** isStreaming: any prop is streaming (generation stage doesn't matter if props are complete) */\n    isStreaming: propStatuses.some((p) => p.isStreaming),\n\n    /** isSuccess: all props successful (component is stable once all props complete) */\n    isSuccess: allPropsSuccessful,\n\n    /** isError: generation error OR any prop error */\n    isError:\n      isGenerationError ||\n      propStatuses.some((p) => p.error) ||\n      !!generationError,\n\n    streamError: firstError,\n  };\n}\n\n/**\n * Hook that exposes per-prop and global streaming status for tambo-ai components.\n * Provides streaming readiness flags so consumers can show loaders, skeletons,\n * or errors while LLM-generated props stream in.\n * This hook tracks status for the specific component in the current message only.\n * Once a component's props complete streaming, they remain stable regardless of\n * other components being generated in the thread.\n * @template Props - The type of the component props being tracked (defaults to Record<string, any>)\n * @returns An object containing both global streamStatus and per-prop propStatus\n * @throws {Error} When used during SSR/SSG\n * @example\n * ```tsx\n * // Wait for entire stream\n * const { streamStatus } = useTamboStreamStatus();\n * if (!streamStatus.isSuccess) return <Spinner />;\n * return <Card {...props} />;\n * ```\n * @example\n * ```tsx\n * // Highlight in-flight props\n * const { propStatus } = useTamboStreamStatus<Props>();\n * <h2 className={propStatus.title.isStreaming ? \"animate-pulse\" : \"\"}>\n *   {title}\n * </h2>\n * ```\n */\nexport function useTamboStreamStatus<\n  Props extends Record<string, any> = Record<string, any>,\n>(): {\n  streamStatus: StreamStatus;\n  propStatus: Record<keyof Props, PropStatus>;\n} {\n  /** SSR Guard - ensure client-side only execution */\n  assertClientSide();\n\n  const { generationStage } = useTamboGenerationStage();\n  const message = useTamboCurrentMessage();\n\n  /** Get the current component props from the message */\n  const componentProps = (message?.component?.props as Props) || ({} as Props);\n\n  /** Track per-prop streaming status */\n  const propStatus = usePropsStreamingStatus(\n    componentProps,\n    generationStage,\n    message?.id ?? \"\",\n  );\n\n  /** Derive global stream status from prop statuses and generation stage */\n  const streamStatus = useMemo(() => {\n    const generationError = message?.error\n      ? new Error(message.error)\n      : undefined;\n    const hasComponent = !!message?.component;\n    return deriveGlobalStreamStatus(\n      generationStage,\n      propStatus,\n      hasComponent,\n      generationError,\n    );\n  }, [generationStage, propStatus, message?.error, message?.component]);\n\n  return {\n    streamStatus,\n    propStatus,\n  };\n}\n"]}
+{"version":3,"file":"use-tambo-stream-status.js","sourceRoot":"","sources":["../../src/hooks/use-tambo-stream-status.tsx"],"names":[],"mappings":";AAAA,YAAY,CAAC;;AA+Tb,oDAwCC;AAtWD,iCAAqD;AACrD,sFAAuE;AACvE,8EAA6E;AAC7E,+DAA+D;AAqE/D;;;;GAIG;AACH,SAAS,gBAAgB;IACvB,IAAI,OAAO,MAAM,KAAK,WAAW,EAAE,CAAC;QAClC,MAAM,IAAI,KAAK,CACb,6DAA6D;YAC3D,4CAA4C;YAC5C,6DAA6D,CAChE,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;;;;;;;;GASG;AACH,SAAS,uBAAuB,CAC9B,KAAwB,EACxB,eAAgC,EAChC,SAAiB;IAEjB,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,IAAA,gBAAQ,EAU9C,EAAE,CAAC,CAAC;IAEN,mDAAmD;IACnD,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,eAAe,CAAC,CAAC,IAAI,EAAE,EAAE;YACvB,0DAA0D;YAC1D,MAAM,iBAAiB,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAChD,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,SAAS,KAAK,SAAS,CAC5D,CAAC;YACF,OAAO,iBAAiB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;IAEhB,oFAAoF;IACpF,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,IAAI,CAAC,KAAK;YAAE,OAAO;QAEnB,eAAe,CAAC,CAAC,IAAI,EAAE,EAAE;YACvB,MAAM,OAAO,GAAG,EAAE,GAAG,IAAI,EAAE,CAAC;YAC5B,IAAI,UAAU,GAAG,KAAK,CAAC;YAEvB,oDAAoD;YACpD,MAAM,gBAAgB,GAAa,EAAE,CAAC;YACtC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;gBAC7C,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI;oBAC3B,UAAU,EAAE,KAAK;oBACjB,UAAU,EAAE,KAAK;iBAClB,CAAC;gBAEF,+EAA+E;gBAC/E,MAAM,UAAU,GACd,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,EAAE,CAAC;gBACxD,MAAM,WAAW,GAAG,UAAU,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;gBAEtD,IAAI,WAAW,EAAE,CAAC;oBAChB,gBAAgB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBAC7B,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,mEAAmE;YACnE,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;gBAC7C,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI;oBAC3B,UAAU,EAAE,KAAK;oBACjB,UAAU,EAAE,KAAK;iBAClB,CAAC;gBAEF,+EAA+E;gBAC/E,MAAM,UAAU,GACd,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,EAAE,CAAC;gBACxD,MAAM,WAAW,GAAG,UAAU,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;gBAEtD;;;;mBAIG;gBACH,MAAM,uBAAuB,GAAG,gBAAgB,CAAC,IAAI,CACnD,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,KAAK,GAAG,CACrC,CAAC;gBACF,MAAM,oBAAoB,GACxB,eAAe,KAAK,6CAAe,CAAC,QAAQ,CAAC;gBAC/C,MAAM,UAAU,GACd,OAAO,CAAC,UAAU;oBAClB,CAAC,uBAAuB,IAAI,oBAAoB,CAAC;oBACjD,CAAC,OAAO,CAAC,UAAU,CAAC;gBAEtB,8DAA8D;gBAC9D,IAAI,OAAO,CAAC,UAAU,IAAI,OAAO,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;oBAC1D,2CAA2C;oBAC3C,OAAO;gBACT,CAAC;gBAED,IAAI,WAAW,IAAI,UAAU,EAAE,CAAC;oBAC9B,OAAO,CAAC,GAAG,CAAC,GAAG;wBACb,GAAG,OAAO;wBACV,UAAU,EAAE,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU;wBACnD,UAAU,EAAE,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU;wBAClD,SAAS;qBACV,CAAC;oBACF,UAAU,GAAG,IAAI,CAAC;gBACpB,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,OAAO,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC;QACrC,CAAC,CAAC,CAAC;IACL,CAAC,EAAE,CAAC,KAAK,EAAE,eAAe,EAAE,SAAS,CAAC,CAAC,CAAC;IAExC,mDAAmD;IACnD,OAAO,IAAA,eAAO,EAAC,GAAG,EAAE;QAClB,IAAI,CAAC,KAAK;YAAE,OAAO,EAAqC,CAAC;QAEzD,MAAM,MAAM,GAAG,EAAqC,CAAC;QAErD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YACjC,MAAM,QAAQ,GAAG,YAAY,CAAC,GAAG,CAAC,IAAI;gBACpC,UAAU,EAAE,KAAK;gBACjB,UAAU,EAAE,KAAK;gBACjB,SAAS,EAAE,EAAE;aACd,CAAC;YAEF,+DAA+D;YAC/D,MAAM,wBAAwB,GAC5B,QAAQ,CAAC,UAAU,IAAI,QAAQ,CAAC,SAAS,KAAK,SAAS,CAAC;YAE1D,sFAAsF;YACtF,MAAM,qBAAqB,GACzB,CAAC,wBAAwB;gBACzB,eAAe,KAAK,6CAAe,CAAC,kBAAkB,CAAC;YAEzD,MAAM,CAAC,GAAkB,CAAC,GAAG;gBAC3B,SAAS,EAAE,CAAC,QAAQ,CAAC,UAAU,IAAI,CAAC,wBAAwB;gBAC5D,WAAW,EACT,QAAQ,CAAC,UAAU;oBACnB,CAAC,wBAAwB;oBACzB,qBAAqB;gBACvB,SAAS,EAAE,wBAAwB;gBACnC,KAAK,EAAE,QAAQ,CAAC,KAAK;aACtB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,OAAO,MAAM,CAAC;IAChB,CAAC,EAAE,CAAC,KAAK,EAAE,YAAY,EAAE,eAAe,EAAE,SAAS,CAAC,CAAC,CAAC;AACxD,CAAC;AAED;;;;;;;;;GASG;AACH,SAAS,wBAAwB,CAC/B,eAAgC,EAChC,UAA2C,EAC3C,YAAqB,EACrB,eAAuB;IAEvB,MAAM,YAAY,GAAG,MAAM,CAAC,MAAM,CAAC,UAAU,CAAiB,CAAC;IAE/D,gGAAgG;IAChG,MAAM,kBAAkB,GACtB,YAAY,CAAC,MAAM,GAAG,CAAC,IAAI,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;IAEpE,+DAA+D;IAC/D,MAAM,qBAAqB,GACzB,CAAC,kBAAkB;QACnB,eAAe,KAAK,6CAAe,CAAC,kBAAkB,CAAC;IACzD,MAAM,iBAAiB,GAAG,eAAe,KAAK,6CAAe,CAAC,KAAK,CAAC;IAEpE,mDAAmD;IACnD,MAAM,UAAU,GACd,eAAe,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,KAAK,CAAC;IAE9D,OAAO;QACL,+EAA+E;QAC/E,SAAS,EACP,CAAC,YAAY;YACb,CAAC,CAAC,qBAAqB;gBACrB,CAAC,kBAAkB;gBACnB,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QAE3C,iGAAiG;QACjG,WAAW,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,CAAC;QAEpD,oFAAoF;QACpF,SAAS,EAAE,kBAAkB;QAE7B,kDAAkD;QAClD,OAAO,EACL,iBAAiB;YACjB,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC;YACjC,CAAC,CAAC,eAAe;QAEnB,WAAW,EAAE,UAAU;KACxB,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;GAyBG;AACH,SAAgB,oBAAoB;IAMlC,oDAAoD;IACpD,gBAAgB,EAAE,CAAC;IAEnB,MAAM,EAAE,eAAe,EAAE,GAAG,IAAA,+CAAuB,GAAE,CAAC;IACtD,MAAM,OAAO,GAAG,IAAA,4CAAsB,GAAE,CAAC;IAEzC,uDAAuD;IACvD,MAAM,cAAc,GAAI,OAAO,EAAE,SAAS,EAAE,KAAe,IAAK,EAAY,CAAC;IAE7E,sCAAsC;IACtC,MAAM,UAAU,GAAG,uBAAuB,CACxC,cAAc,EACd,eAAe,EACf,OAAO,EAAE,EAAE,IAAI,EAAE,CAClB,CAAC;IAEF,0EAA0E;IAC1E,MAAM,YAAY,GAAG,IAAA,eAAO,EAAC,GAAG,EAAE;QAChC,MAAM,eAAe,GAAG,OAAO,EAAE,KAAK;YACpC,CAAC,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;YAC1B,CAAC,CAAC,SAAS,CAAC;QACd,MAAM,YAAY,GAAG,CAAC,CAAC,OAAO,EAAE,SAAS,CAAC;QAC1C,OAAO,wBAAwB,CAC7B,eAAe,EACf,UAAU,EACV,YAAY,EACZ,eAAe,CAChB,CAAC;IACJ,CAAC,EAAE,CAAC,eAAe,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC,CAAC;IAE3C,OAAO;QACL,YAAY;QACZ,UAAU;KACX,CAAC;AACJ,CAAC","sourcesContent":["\"use client\";\nimport { useEffect, useMemo, useState } from \"react\";\nimport { GenerationStage } from \"../model/generate-component-response\";\nimport { useTamboGenerationStage } from \"../providers/tambo-thread-provider\";\nimport { useTamboCurrentMessage } from \"./use-current-message\";\n\n/**\n * Global stream status flags for a specific component in a message.\n * Represents the aggregate state across all props for this component only.\n * Once a component completes, its status remains stable regardless of other generations.\n */\nexport interface StreamStatus {\n  /**\n   * Indicates no tokens have been received for any prop and generation is not active.\n   * Useful for showing initial loading states before any data arrives.\n   */\n  isPending: boolean;\n\n  /**\n   * Indicates active streaming - either generation is streaming OR at least one prop is still streaming.\n   * Use this to show loading animations or skeleton states during data transmission.\n   */\n  isStreaming: boolean;\n\n  /**\n   * Indicates successful completion - generation is complete AND every prop finished without error.\n   * Safe to render the final component when this is true.\n   */\n  isSuccess: boolean;\n\n  /**\n   * Indicates a fatal error occurred in any prop or the stream itself.\n   * Check streamError for details about what went wrong.\n   */\n  isError: boolean;\n\n  /**\n   * The first fatal error encountered during streaming (if any).\n   * Will be undefined if no errors occurred.\n   */\n  streamError?: Error;\n}\n\n/**\n * Streaming status flags for individual component props.\n * Tracks the state of each prop as it streams from the LLM.\n */\nexport interface PropStatus {\n  /**\n   * Indicates no tokens have been received for this specific prop yet.\n   * The prop value is still undefined, null, or empty string.\n   */\n  isPending: boolean;\n\n  /**\n   * Indicates at least one token has been received but streaming is not complete.\n   * The prop has partial content that may still be updating.\n   */\n  isStreaming: boolean;\n\n  /**\n   * Indicates this prop has finished streaming successfully.\n   * The prop value is complete and stable.\n   */\n  isSuccess: boolean;\n\n  /**\n   * The error that occurred during streaming (if any).\n   * Will be undefined if no error occurred for this prop.\n   */\n  error?: Error;\n}\n\n/**\n * SSR Guard - throws during server-side rendering.\n * Ensures the hook is only used in browser contexts.\n * @throws {Error} When called during server-side rendering\n */\nfunction assertClientSide() {\n  if (typeof window === \"undefined\") {\n    throw new Error(\n      \"useTamboStreamStatus can only be used in browser contexts. \" +\n        \"This hook is not compatible with SSR/SSG. \" +\n        \"Consider wrapping it in useEffect or using dynamic imports.\",\n    );\n  }\n}\n\n/**\n * Track streaming status for individual props by monitoring their values.\n * Monitors when props receive their first token and when they complete streaming.\n * Maintains stable state per message - once props complete for a message, they stay complete.\n * @template Props - The type of the component props being tracked\n * @param props - The current component props object\n * @param generationStage - The current generation stage from the LLM\n * @param messageId - The ID of the current message to track component-specific state\n * @returns A record mapping each prop key to its PropStatus\n */\nfunction usePropsStreamingStatus<Props extends Record<string, any>>(\n  props: Props | undefined,\n  generationStage: GenerationStage,\n  messageId: string,\n): Record<keyof Props, PropStatus> {\n  const [propTracking, setPropTracking] = useState<\n    Record<\n      string,\n      {\n        hasStarted: boolean;\n        isComplete: boolean;\n        error?: Error;\n        messageId: string;\n      }\n    >\n  >({});\n\n  /** Reset tracking only when the message changes */\n  useEffect(() => {\n    setPropTracking((prev) => {\n      // If we have tracking data for a different message, reset\n      const hasOldMessageData = Object.values(prev).some(\n        (track) => track.messageId && track.messageId !== messageId,\n      );\n      return hasOldMessageData ? {} : prev;\n    });\n  }, [messageId]);\n\n  /** Track when props start streaming (receive first token) and when they complete */\n  useEffect(() => {\n    if (!props) return;\n\n    setPropTracking((prev) => {\n      const updated = { ...prev };\n      let hasChanges = false;\n\n      // First pass: identify which props are starting now\n      const propsStartingNow: string[] = [];\n      Object.entries(props).forEach(([key, value]) => {\n        const current = prev[key] || {\n          hasStarted: false,\n          isComplete: false,\n        };\n\n        /** A prop starts streaming when it has a non-empty value for the first time */\n        const hasContent =\n          value !== undefined && value !== null && value !== \"\";\n        const justStarted = hasContent && !current.hasStarted;\n\n        if (justStarted) {\n          propsStartingNow.push(key);\n        }\n      });\n\n      // Second pass: update tracking and mark previous props as complete\n      Object.entries(props).forEach(([key, value]) => {\n        const current = prev[key] || {\n          hasStarted: false,\n          isComplete: false,\n        };\n\n        /** A prop starts streaming when it has a non-empty value for the first time */\n        const hasContent =\n          value !== undefined && value !== null && value !== \"\";\n        const justStarted = hasContent && !current.hasStarted;\n\n        /**\n         * A prop is complete when it has started and either:\n         * 1. A following prop has started, OR\n         * 2. Generation is complete (for the final prop)\n         */\n        const hasFollowingPropStarted = propsStartingNow.some(\n          (startingKey) => startingKey !== key,\n        );\n        const isGenerationComplete =\n          generationStage === GenerationStage.COMPLETE;\n        const isComplete =\n          current.hasStarted &&\n          (hasFollowingPropStarted || isGenerationComplete) &&\n          !current.isComplete;\n\n        // Once a prop is complete for this message, it stays complete\n        if (current.isComplete && current.messageId === messageId) {\n          // Skip - already complete for this message\n          return;\n        }\n\n        if (justStarted || isComplete) {\n          updated[key] = {\n            ...current,\n            hasStarted: justStarted ? true : current.hasStarted,\n            isComplete: isComplete ? true : current.isComplete,\n            messageId,\n          };\n          hasChanges = true;\n        }\n      });\n\n      return hasChanges ? updated : prev;\n    });\n  }, [props, generationStage, messageId]);\n\n  /** Convert tracking state to PropStatus objects */\n  return useMemo(() => {\n    if (!props) return {} as Record<keyof Props, PropStatus>;\n\n    const result = {} as Record<keyof Props, PropStatus>;\n\n    Object.keys(props).forEach((key) => {\n      const tracking = propTracking[key] || {\n        hasStarted: false,\n        isComplete: false,\n        messageId: \"\",\n      };\n\n      // If this prop is complete for this message, it stays complete\n      const isCompleteForThisMessage =\n        tracking.isComplete && tracking.messageId === messageId;\n\n      // Only consider generation stage if this prop isn't already complete for this message\n      const isGenerationStreaming =\n        !isCompleteForThisMessage &&\n        generationStage === GenerationStage.STREAMING_RESPONSE;\n\n      result[key as keyof Props] = {\n        isPending: !tracking.hasStarted && !isCompleteForThisMessage,\n        isStreaming:\n          tracking.hasStarted &&\n          !isCompleteForThisMessage &&\n          isGenerationStreaming,\n        isSuccess: isCompleteForThisMessage,\n        error: tracking.error,\n      };\n    });\n\n    return result;\n  }, [props, propTracking, generationStage, messageId]);\n}\n\n/**\n * Derives global StreamStatus from generation stage and individual prop statuses.\n * Aggregates individual prop states into a unified stream status.\n * @template Props - The type of the component props\n * @param generationStage - The current generation stage from the LLM\n * @param propStatus - Status record for each individual prop\n * @param hasComponent - Whether a component exists in the current message\n * @param generationError - Any error from the generation process itself\n * @returns The aggregated StreamStatus for the entire component\n */\nfunction deriveGlobalStreamStatus<Props extends Record<string, any>>(\n  generationStage: GenerationStage,\n  propStatus: Record<keyof Props, PropStatus>,\n  hasComponent: boolean,\n  generationError?: Error,\n): StreamStatus {\n  const propStatuses = Object.values(propStatus) as PropStatus[];\n\n  // If all props are already successful, the component is complete regardless of generation stage\n  const allPropsSuccessful =\n    propStatuses.length > 0 && propStatuses.every((p) => p.isSuccess);\n\n  // Only consider generation stage if not all props are complete\n  const isGenerationStreaming =\n    !allPropsSuccessful &&\n    generationStage === GenerationStage.STREAMING_RESPONSE;\n  const isGenerationError = generationStage === GenerationStage.ERROR;\n\n  /** Find first error from generation or any prop */\n  const firstError =\n    generationError ?? propStatuses.find((p) => p.error)?.error;\n\n  return {\n    /** isPending: no component yet OR (has component but no props have started) */\n    isPending:\n      !hasComponent ||\n      (!isGenerationStreaming &&\n        !allPropsSuccessful &&\n        propStatuses.every((p) => p.isPending)),\n\n    /** isStreaming: any prop is streaming (generation stage doesn't matter if props are complete) */\n    isStreaming: propStatuses.some((p) => p.isStreaming),\n\n    /** isSuccess: all props successful (component is stable once all props complete) */\n    isSuccess: allPropsSuccessful,\n\n    /** isError: generation error OR any prop error */\n    isError:\n      isGenerationError ||\n      propStatuses.some((p) => p.error) ||\n      !!generationError,\n\n    streamError: firstError,\n  };\n}\n\n/**\n * Hook that exposes per-prop and global streaming status for tambo-ai components.\n * Provides streaming readiness flags so consumers can show loaders, skeletons,\n * or errors while LLM-generated props stream in.\n * This hook tracks status for the specific component in the current message only.\n * Once a component's props complete streaming, they remain stable regardless of\n * other components being generated in the thread.\n * @template Props - The type of the component props being tracked (defaults to Record<string, any>)\n * @returns An object containing both global streamStatus and per-prop propStatus\n * @throws {Error} When used during SSR/SSG\n * @example\n * ```tsx\n * // Wait for entire stream\n * const { streamStatus } = useTamboStreamStatus();\n * if (!streamStatus.isSuccess) return <Spinner />;\n * return <Card {...props} />;\n * ```\n * @example\n * ```tsx\n * // Highlight in-flight props\n * const { propStatus } = useTamboStreamStatus<Props>();\n * <h2 className={propStatus.title.isStreaming ? \"animate-pulse\" : \"\"}>\n *   {title}\n * </h2>\n * ```\n */\nexport function useTamboStreamStatus<\n  Props extends Record<string, any> = Record<string, any>,\n>(): {\n  streamStatus: StreamStatus;\n  propStatus: Record<keyof Props, PropStatus>;\n} {\n  /** SSR Guard - ensure client-side only execution */\n  assertClientSide();\n\n  const { generationStage } = useTamboGenerationStage();\n  const message = useTamboCurrentMessage();\n\n  /** Get the current component props from the message */\n  const componentProps = (message?.component?.props as Props) || ({} as Props);\n\n  /** Track per-prop streaming status */\n  const propStatus = usePropsStreamingStatus(\n    componentProps,\n    generationStage,\n    message?.id ?? \"\",\n  );\n\n  /** Derive global stream status from prop statuses and generation stage */\n  const streamStatus = useMemo(() => {\n    const generationError = message?.error\n      ? new Error(message.error)\n      : undefined;\n    const hasComponent = !!message?.component;\n    return deriveGlobalStreamStatus(\n      generationStage,\n      propStatus,\n      hasComponent,\n      generationError,\n    );\n  }, [generationStage, propStatus, message]);\n\n  return {\n    streamStatus,\n    propStatus,\n  };\n}\n"]}

package/dist/mcp/__tests__/mcp-client.test.js

@@ -10,7 +10,6 @@ jest.mock("@modelcontextprotocol/sdk/client/index.js", () => ({
         callTool: jest.fn(),
         setRequestHandler: jest.fn(),
         removeRequestHandler: jest.fn(),
-        onclose: null,
     })),
 }));
 jest.mock("@modelcontextprotocol/sdk/client/sse.js", () => ({
@@ -45,7 +44,6 @@ describe("MCPClient", () => {
             callTool: jest.fn(),
             setRequestHandler: jest.fn(),
             removeRequestHandler: jest.fn(),
-            onclose: null,
         };
         mockTransportInstance = {
             sessionId: "test-session-id",
@@ -83,264 +81,6 @@ describe("MCPClient", () => {
             expect(MockedStreamableHTTPClientTransport).toHaveBeenCalledWith(new URL(endpoint), { sessionId: undefined, requestInit: { headers: {} } });
         });
     });
-    describe("reconnect", () => {
-        it("should create new transport and client instances and call connect when reconnect() is called (default behavior)", async () => {
-            const endpoint = "https://api.example.com/mcp";
-            const client = await mcp_client_1.MCPClient.create(endpoint, mcp_client_1.MCPTransport.HTTP, undefined, undefined, undefined);
-            // Clear previous calls to focus on reconnect behavior
-            jest.clearAllMocks();
-            // Create new mock instances to verify new instances are created
-            const newMockClientInstance = {
-                connect: jest.fn().mockResolvedValue(undefined),
-                close: jest.fn().mockResolvedValue(undefined),
-                listTools: jest.fn(),
-                callTool: jest.fn(),
-                onclose: null,
-            };
-            const newMockTransportInstance = {
-                sessionId: "new-session-id",
-            };
-            // Mock the constructors to return new instances
-            MockedClient.mockImplementation(() => newMockClientInstance);
-            MockedStreamableHTTPClientTransport.mockImplementation(() => newMockTransportInstance);
-            await client.reconnect(); // Uses default parameters
-            // Verify old client was closed
-            expect(mockClientInstance.close).toHaveBeenCalled();
-            // Verify new transport was created with preserved session ID (default behavior)
-            expect(MockedStreamableHTTPClientTransport).toHaveBeenCalledWith(new URL(endpoint), { sessionId: "test-session-id", requestInit: { headers: {} } });
-            // Verify new client was created
-            expect(MockedClient).toHaveBeenCalledWith({
-                name: "tambo-mcp-client",
-                version: "1.0.0",
-            }, { capabilities: {} });
-            // Verify new client's connect was called with new transport
-            expect(newMockClientInstance.connect).toHaveBeenCalledWith(newMockTransportInstance);
-        });
-        it("should reconnect without session ID for SSE transport", async () => {
-            const endpoint = "https://api.example.com/mcp";
-            const client = await mcp_client_1.MCPClient.create(endpoint, mcp_client_1.MCPTransport.SSE, undefined, undefined, undefined);
-            // Clear previous calls
-            jest.clearAllMocks();
-            await client.reconnect();
-            expect(mockClientInstance.close).toHaveBeenCalled();
-            expect(MockedSSEClientTransport).toHaveBeenCalledWith(new URL(endpoint), {
-                requestInit: { headers: {} },
-            });
-            expect(mockClientInstance.connect).toHaveBeenCalledWith({});
-        });
-        it("should handle close errors when reportErrorOnClose is true", async () => {
-            const endpoint = "https://api.example.com/mcp";
-            const client = await mcp_client_1.MCPClient.create(endpoint, mcp_client_1.MCPTransport.HTTP, undefined, undefined, undefined);
-            const consoleSpy = jest.spyOn(console, "error").mockImplementation();
-            // Make close throw an error
-            mockClientInstance.close.mockRejectedValue(new Error("Close failed"));
-            await client.reconnect(false, true);
-            expect(consoleSpy).toHaveBeenCalledWith("Error closing Tambo MCP Client:", expect.any(Error));
-            consoleSpy.mockRestore();
-        });
-        it("should not log close errors when reportErrorOnClose is false", async () => {
-            const endpoint = "https://api.example.com/mcp";
-            const client = await mcp_client_1.MCPClient.create(endpoint, mcp_client_1.MCPTransport.HTTP, undefined, undefined, undefined);
-            const consoleSpy = jest.spyOn(console, "error").mockImplementation();
-            // Make close throw an error
-            mockClientInstance.close.mockRejectedValue(new Error("Close failed"));
-            await client.reconnect(false, false);
-            expect(consoleSpy).not.toHaveBeenCalled();
-            consoleSpy.mockRestore();
-        });
-        it("should create new session when newSession is true", async () => {
-            const endpoint = "https://api.example.com/mcp";
-            const client = await mcp_client_1.MCPClient.create(endpoint, mcp_client_1.MCPTransport.HTTP, undefined, undefined, undefined);
-            // Clear previous calls to focus on reconnect behavior
-            jest.clearAllMocks();
-            // Create new mock instances to verify new instances are created
-            const newMockClientInstance = {
-                connect: jest.fn().mockResolvedValue(undefined),
-                close: jest.fn().mockResolvedValue(undefined),
-                listTools: jest.fn(),
-                callTool: jest.fn(),
-                onclose: null,
-            };
-            const newMockTransportInstance = {
-                sessionId: "new-session-id",
-            };
-            // Mock the constructors to return new instances
-            MockedClient.mockImplementation(() => newMockClientInstance);
-            MockedStreamableHTTPClientTransport.mockImplementation(() => newMockTransportInstance);
-            await client.reconnect(true, true); // newSession = true, reportErrorOnClose = true
-            // Verify old client was closed
-            expect(mockClientInstance.close).toHaveBeenCalled();
-            // Verify new transport was created with undefined session ID (new session)
-            expect(MockedStreamableHTTPClientTransport).toHaveBeenCalledWith(new URL(endpoint), { sessionId: undefined, requestInit: { headers: {} } });
-            // Verify new client was created
-            expect(MockedClient).toHaveBeenCalledWith({
-                name: "tambo-mcp-client",
-                version: "1.0.0",
-            }, { capabilities: {} });
-            // Verify new client's connect was called with new transport
-            expect(newMockClientInstance.connect).toHaveBeenCalledWith(newMockTransportInstance);
-        });
-        it("should reuse existing session when newSession is false (default)", async () => {
-            const endpoint = "https://api.example.com/mcp";
-            const client = await mcp_client_1.MCPClient.create(endpoint, mcp_client_1.MCPTransport.HTTP, undefined, undefined, undefined);
-            // Clear previous calls to focus on reconnect behavior
-            jest.clearAllMocks();
-            // Create new mock instances to verify new instances are created
-            const newMockClientInstance = {
-                connect: jest.fn().mockResolvedValue(undefined),
-                close: jest.fn().mockResolvedValue(undefined),
-                listTools: jest.fn(),
-                callTool: jest.fn(),
-                onclose: null,
-            };
-            const newMockTransportInstance = {
-                sessionId: "reused-session-id",
-            };
-            // Mock the constructors to return new instances
-            MockedClient.mockImplementation(() => newMockClientInstance);
-            MockedStreamableHTTPClientTransport.mockImplementation(() => newMockTransportInstance);
-            await client.reconnect(false, true); // newSession = false, reportErrorOnClose = true
-            // Verify old client was closed
-            expect(mockClientInstance.close).toHaveBeenCalled();
-            // Verify new transport was created with preserved session ID
-            expect(MockedStreamableHTTPClientTransport).toHaveBeenCalledWith(new URL(endpoint), { sessionId: "test-session-id", requestInit: { headers: {} } });
-            // Verify new client was created
-            expect(MockedClient).toHaveBeenCalledWith({
-                name: "tambo-mcp-client",
-                version: "1.0.0",
-            }, { capabilities: {} });
-            // Verify new client's connect was called with new transport
-            expect(newMockClientInstance.connect).toHaveBeenCalledWith(newMockTransportInstance);
-        });
-    });
-    describe("onclose", () => {
-        it("should reconnect MCPClient when client is closed by external means (no backoff on manual preemption)", async () => {
-            const endpoint = "https://api.example.com/mcp";
-            const client = await mcp_client_1.MCPClient.create(endpoint, mcp_client_1.MCPTransport.HTTP, undefined, undefined, undefined);
-            jest.useFakeTimers();
-            const consoleSpy = jest.spyOn(console, "warn").mockImplementation();
-            // Create new mock instances to verify reconnection creates new instances
-            const newMockClientInstance = {
-                connect: jest.fn().mockResolvedValue(undefined),
-                close: jest.fn().mockResolvedValue(undefined),
-                listTools: jest.fn(),
-                callTool: jest.fn(),
-                onclose: null,
-            };
-            const newMockTransportInstance = {
-                sessionId: "reconnected-session-id",
-            };
-            // Mock the constructors to return new instances for reconnection
-            MockedClient.mockImplementation(() => newMockClientInstance);
-            MockedStreamableHTTPClientTransport.mockImplementation(() => newMockTransportInstance);
-            // Reset counts after initial creation
-            jest.clearAllMocks();
-            // Trigger automatic onclose (schedules a delayed reconnect)
-            client.onclose();
-            // Manual reconnect should preempt the scheduled automatic attempt
-            const reconnectPromise = client.reconnect();
-            // No timers should be pending after manual preemption
-            await reconnectPromise;
-            // Verify warning message is logged
-            expect(consoleSpy).toHaveBeenCalled();
-            // Verify old client was closed
-            expect(mockClientInstance.close).toHaveBeenCalled();
-            // Verify new transport was created with preserved session ID
-            expect(MockedStreamableHTTPClientTransport).toHaveBeenCalledWith(new URL(endpoint), { sessionId: "test-session-id", requestInit: { headers: {} } });
-            // Verify new client was created
-            expect(MockedClient).toHaveBeenCalledWith({
-                name: "tambo-mcp-client",
-                version: "1.0.0",
-            }, { capabilities: {} });
-            // Verify new client's connect was called with new transport
-            expect(newMockClientInstance.connect).toHaveBeenCalledWith(newMockTransportInstance);
-            // Ensure only a single reconnect attempt occurred
-            expect(MockedClient).toHaveBeenCalledTimes(1);
-            expect(newMockClientInstance.connect).toHaveBeenCalledTimes(1);
-            consoleSpy.mockRestore();
-            jest.useRealTimers();
-        });
-    });
-    describe("reconnect re-entrancy and single-flight", () => {
-        it("prevents re-entrant onclose during deliberate close and coalesces concurrent calls", async () => {
-            const endpoint = "https://api.example.com/mcp";
-            const client = await mcp_client_1.MCPClient.create(endpoint, mcp_client_1.MCPTransport.HTTP, undefined, undefined, undefined);
-            // Simulate an implementation where closing the client would call its own onclose handler
-            const closeImpl = jest.fn(async () => {
-                if (typeof mockClientInstance.onclose === "function") {
-                    // would cause recursion if not detached
-                    mockClientInstance.onclose();
-                }
-                return;
-            });
-            mockClientInstance.close = closeImpl;
-            // Prepare new instances for the reconnect
-            const newMockClientInstance = {
-                connect: jest.fn().mockResolvedValue(undefined),
-                close: jest.fn().mockResolvedValue(undefined),
-                listTools: jest.fn(),
-                callTool: jest.fn(),
-                onclose: null,
-            };
-            MockedClient.mockImplementation(() => newMockClientInstance);
-            // Reset counts after initial creation
-            jest.clearAllMocks();
-            // Trigger auto onclose and manual reconnect nearly simultaneously
-            client.onclose();
-            await client.reconnect();
-            // Should have detached onclose before calling close, avoiding recursion
-            expect(closeImpl).toHaveBeenCalledTimes(1);
-            expect(mockClientInstance.onclose).toBeUndefined();
-            // Single-flight: only one new client/connect
-            expect(MockedClient).toHaveBeenCalledTimes(1);
-            expect(newMockClientInstance.connect).toHaveBeenCalledTimes(1);
-        });
-    });
-    describe("backoff + jitter (automatic reconnect)", () => {
-        it("applies jitter and resets to initial delay after a successful reconnect (manual preempt)", async () => {
-            jest.useFakeTimers();
-            const base = mcp_client_1.MCPClient.BACKOFF_INITIAL_MS;
-            const ratio = mcp_client_1.MCPClient.BACKOFF_JITTER_RATIO;
-            const min = Math.round(base * (1 - ratio));
-            const max = Math.round(base * (1 + ratio));
-            const setTimeoutSpy = jest.spyOn(global, "setTimeout");
-            const randSpy = jest.spyOn(Math, "random").mockReturnValue(0.0); // extreme low jitter
-            const endpoint = "https://api.example.com/mcp";
-            const client = await mcp_client_1.MCPClient.create(endpoint, mcp_client_1.MCPTransport.HTTP, undefined, undefined, undefined);
-            // Prepare one attempt that will succeed to avoid rescheduling
-            MockedClient.mockImplementation(() => ({
-                connect: jest.fn().mockResolvedValue(undefined),
-                close: jest.fn().mockResolvedValue(undefined),
-                listTools: jest.fn(),
-                callTool: jest.fn(),
-                onclose: null,
-            }));
-            // Trigger once to capture the delay
-            client.onclose();
-            expect(setTimeoutSpy).toHaveBeenCalled();
-            const numericDelays = setTimeoutSpy.mock.calls
-                .map((c) => c[1])
-                .filter((v) => typeof v === "number");
-            expect(numericDelays.length).toBeGreaterThan(0);
-            const d = numericDelays[0];
-            expect(d).toBeGreaterThanOrEqual(min);
-            expect(d).toBeLessThanOrEqual(max);
-            // Manual reconnect succeeds and should reset backoff attempts
-            await client.reconnect();
-            // Trigger onclose again and ensure we start from initial range again
-            client.onclose();
-            const numericDelays2 = setTimeoutSpy.mock.calls
-                .map((c) => c[1])
-                .filter((v) => typeof v === "number");
-            expect(numericDelays2.length).toBeGreaterThanOrEqual(2);
-            const dAgain = numericDelays2[1];
-            expect(dAgain).toBeGreaterThanOrEqual(min);
-            expect(dAgain).toBeLessThanOrEqual(max);
-            jest.useRealTimers();
-            setTimeoutSpy.mockRestore();
-            randSpy.mockRestore();
-        });
-    });
     describe("listTools", () => {
         it("should list all tools with pagination", async () => {
             const endpoint = "https://api.example.com/mcp";
@@ -495,12 +235,6 @@ describe("MCPClient", () => {
                 version: "1.0.0",
             }, { capabilities: {} });
         });
-        it("should set onclose handler", async () => {
-            const endpoint = "https://api.example.com/mcp";
-            const _client = await mcp_client_1.MCPClient.create(endpoint, mcp_client_1.MCPTransport.HTTP, undefined, undefined, undefined);
-            expect(mockClientInstance.onclose).toBeDefined();
-            expect(typeof mockClientInstance.onclose).toBe("function");
-        });
     });
     describe("handlers (elicitation/sampling)", () => {
         it("sets handlers on create when provided", async () => {

package/dist/mcp/__tests__/mcp-client.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"mcp-client.test.js","sourceRoot":"","sources":["../../../src/mcp/__tests__/mcp-client.test.ts"],"names":[],"mappings":";;AAAA,8CAAwD;AAExD,2BAA2B;AAC3B,IAAI,CAAC,IAAI,CAAC,2CAA2C,EAAE,GAAG,EAAE,CAAC,CAAC;IAC5D,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,CAAC;QAC1C,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE;QAClB,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;QAChB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;QACpB,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;QACnB,iBAAiB,EAAE,IAAI,CAAC,EAAE,EAAE;QAC5B,oBAAoB,EAAE,IAAI,CAAC,EAAE,EAAE;QAC/B,OAAO,EAAE,IAAI;KACd,CAAC,CAAC;CACJ,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,yCAAyC,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1D,kBAAkB,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,CAAC;IACtD,uCAAuC;KACxC,CAAC,CAAC;CACJ,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,oDAAoD,EAAE,GAAG,EAAE,CAAC,CAAC;IACrE,6BAA6B,EAAE,IAAI;SAChC,EAAE,EAAE;SACJ,kBAAkB,CAAC,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;QACrC,SAAS,EAAE,OAAO,EAAE,SAAS;KAC9B,CAAC,CAAC;CACN,CAAC,CAAC,CAAC;AAEJ,wEAAmE;AACnE,oEAA6E;AAC7E,0FAAmG;AAEnG,0BAA0B;AAC1B,MAAM,YAAY,GAAG,iBAAyC,CAAC;AAC/D,MAAM,wBAAwB,GAAG,2BAEhC,CAAC;AACF,MAAM,mCAAmC,GACvC,iDAEC,CAAC;AAEJ,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,IAAI,kBAAuB,CAAC;IAC5B,IAAI,qBAA0B,CAAC;IAE/B,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;QAErB,wBAAwB;QACxB,kBAAkB,GAAG;YACnB,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;YAC/C,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;YAC7C,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;YACpB,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;YACnB,iBAAiB,EAAE,IAAI,CAAC,EAAE,EAAE;YAC5B,oBAAoB,EAAE,IAAI,CAAC,EAAE,EAAE;YAC/B,OAAO,EAAE,IAAI;SACd,CAAC;QAEF,qBAAqB,GAAG;YACtB,SAAS,EAAE,iBAAiB;SAC7B,CAAC;QAEF,cAAc;QACd,YAAY,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,CAAC;QAC1D,mCAAmC,CAAC,kBAAkB,CACpD,GAAG,EAAE,CAAC,qBAAqB,CAC5B,CAAC;QACF,wBAAwB,CAAC,kBAAkB,CACzC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAuB,CACjC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;QACtB,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;YACrF,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,OAAO,GAAG,EAAE,aAAa,EAAE,cAAc,EAAE,CAAC;YAElD,MAAM,MAAM,GAAG,MAAM,sBAAS,CAAC,MAAM,CACnC,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,OAAO,EACP,SAAS,EACT,SAAS,CACV,CAAC;YAEF,MAAM,CAAC,mCAAmC,CAAC,CAAC,oBAAoB,CAC9D,IAAI,GAAG,CAAC,QAAQ,CAAC,EACjB,EAAE,SAAS,EAAE,SAAS,EAAE,WAAW,EAAE,EAAE,OAAO,EAAE,EAAE,CACnD,CAAC;YACF,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACvC;gBACE,IAAI,EAAE,kBAAkB;gBACxB,OAAO,EAAE,OAAO;aACjB,EACD,EAAE,YAAY,EAAE,EAAE,EAAE,CACrB,CAAC;YACF,MAAM,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,oBAAoB,CACrD,qBAAqB,CACtB,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,sBAAS,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAE/C,MAAM,MAAM,GAAG,MAAM,sBAAS,CAAC,MAAM,CACnC,QAAQ,EACR,yBAAY,CAAC,GAAG,EAChB,SAAS,EACT,SAAS,EACT,SAAS,CACV,CAAC;YAEF,MAAM,CAAC,wBAAwB,CAAC,CAAC,oBAAoB,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,EAAE;gBACvE,WAAW,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE;aAC7B,CAAC,CAAC;YACH,MAAM,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAAC,EAAE,CAAC,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,sBAAS,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;YAC5E,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAE/C,MAAM,sBAAS,CAAC,MAAM,CACpB,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,SAAS,EACT,SAAS,EACT,SAAS,CACV,CAAC;YAEF,MAAM,CAAC,mCAAmC,CAAC,CAAC,oBAAoB,CAC9D,IAAI,GAAG,CAAC,QAAQ,CAAC,EACjB,EAAE,SAAS,EAAE,SAAS,EAAE,WAAW,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,CACvD,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,iHAAiH,EAAE,KAAK,IAAI,EAAE;YAC/H,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,sBAAS,CAAC,MAAM,CACnC,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,SAAS,EACT,SAAS,EACT,SAAS,CACV,CAAC;YAEF,sDAAsD;YACtD,IAAI,CAAC,aAAa,EAAE,CAAC;YAErB,gEAAgE;YAChE,MAAM,qBAAqB,GAAG;gBAC5B,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;gBAC/C,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;gBAC7C,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;gBACpB,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;gBACnB,OAAO,EAAE,IAAI;aACd,CAAC;YAEF,MAAM,wBAAwB,GAAG;gBAC/B,SAAS,EAAE,gBAAgB;aAC5B,CAAC;YAEF,gDAAgD;YAChD,YAAY,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,qBAA4B,CAAC,CAAC;YACpE,mCAAmC,CAAC,kBAAkB,CACpD,GAAG,EAAE,CAAC,wBAA+B,CACtC,CAAC;YAEF,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,0BAA0B;YAEpD,+BAA+B;YAC/B,MAAM,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAEpD,gFAAgF;YAChF,MAAM,CAAC,mCAAmC,CAAC,CAAC,oBAAoB,CAC9D,IAAI,GAAG,CAAC,QAAQ,CAAC,EACjB,EAAE,SAAS,EAAE,iBAAiB,EAAE,WAAW,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,CAC/D,CAAC;YAEF,gCAAgC;YAChC,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACvC;gBACE,IAAI,EAAE,kBAAkB;gBACxB,OAAO,EAAE,OAAO;aACjB,EACD,EAAE,YAAY,EAAE,EAAE,EAAE,CACrB,CAAC;YAEF,4DAA4D;YAC5D,MAAM,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,oBAAoB,CACxD,wBAAwB,CACzB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,sBAAS,CAAC,MAAM,CACnC,QAAQ,EACR,yBAAY,CAAC,GAAG,EAChB,SAAS,EACT,SAAS,EACT,SAAS,CACV,CAAC;YAEF,uBAAuB;YACvB,IAAI,CAAC,aAAa,EAAE,CAAC;YAErB,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;YAEzB,MAAM,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,gBAAgB,EAAE,CAAC;YACpD,MAAM,CAAC,wBAAwB,CAAC,CAAC,oBAAoB,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,EAAE;gBACvE,WAAW,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE;aAC7B,CAAC,CAAC;YACH,MAAM,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAAC,EAAE,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC1E,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,sBAAS,CAAC,MAAM,CACnC,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,SAAS,EACT,SAAS,EACT,SAAS,CACV,CAAC;YACF,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,kBAAkB,EAAE,CAAC;YAErE,4BAA4B;YAC5B,kBAAkB,CAAC,KAAK,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC;YAEtE,MAAM,MAAM,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;YAEpC,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CACrC,iCAAiC,EACjC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAClB,CAAC;YACF,UAAU,CAAC,WAAW,EAAE,CAAC;QAC3B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;YAC5E,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,sBAAS,CAAC,MAAM,CACnC,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,SAAS,EACT,SAAS,EACT,SAAS,CACV,CAAC;YACF,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,kBAAkB,EAAE,CAAC;YAErE,4BAA4B;YAC5B,kBAAkB,CAAC,KAAK,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC;YAEtE,MAAM,MAAM,CAAC,SAAS,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;YAErC,MAAM,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YAC1C,UAAU,CAAC,WAAW,EAAE,CAAC;QAC3B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,sBAAS,CAAC,MAAM,CACnC,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,SAAS,EACT,SAAS,EACT,SAAS,CACV,CAAC;YAEF,sDAAsD;YACtD,IAAI,CAAC,aAAa,EAAE,CAAC;YAErB,gEAAgE;YAChE,MAAM,qBAAqB,GAAG;gBAC5B,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;gBAC/C,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;gBAC7C,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;gBACpB,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;gBACnB,OAAO,EAAE,IAAI;aACd,CAAC;YAEF,MAAM,wBAAwB,GAAG;gBAC/B,SAAS,EAAE,gBAAgB;aAC5B,CAAC;YAEF,gDAAgD;YAChD,YAAY,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,qBAA4B,CAAC,CAAC;YACpE,mCAAmC,CAAC,kBAAkB,CACpD,GAAG,EAAE,CAAC,wBAA+B,CACtC,CAAC;YAEF,MAAM,MAAM,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC,+CAA+C;YAEnF,+BAA+B;YAC/B,MAAM,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAEpD,2EAA2E;YAC3E,MAAM,CAAC,mCAAmC,CAAC,CAAC,oBAAoB,CAC9D,IAAI,GAAG,CAAC,QAAQ,CAAC,EACjB,EAAE,SAAS,EAAE,SAAS,EAAE,WAAW,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,CACvD,CAAC;YAEF,gCAAgC;YAChC,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACvC;gBACE,IAAI,EAAE,kBAAkB;gBACxB,OAAO,EAAE,OAAO;aACjB,EACD,EAAE,YAAY,EAAE,EAAE,EAAE,CACrB,CAAC;YAEF,4DAA4D;YAC5D,MAAM,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,oBAAoB,CACxD,wBAAwB,CACzB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;YAChF,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,sBAAS,CAAC,MAAM,CACnC,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,SAAS,EACT,SAAS,EACT,SAAS,CACV,CAAC;YAEF,sDAAsD;YACtD,IAAI,CAAC,aAAa,EAAE,CAAC;YAErB,gEAAgE;YAChE,MAAM,qBAAqB,GAAG;gBAC5B,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;gBAC/C,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;gBAC7C,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;gBACpB,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;gBACnB,OAAO,EAAE,IAAI;aACd,CAAC;YAEF,MAAM,wBAAwB,GAAG;gBAC/B,SAAS,EAAE,mBAAmB;aAC/B,CAAC;YAEF,gDAAgD;YAChD,YAAY,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,qBAA4B,CAAC,CAAC;YACpE,mCAAmC,CAAC,kBAAkB,CACpD,GAAG,EAAE,CAAC,wBAA+B,CACtC,CAAC;YAEF,MAAM,MAAM,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC,gDAAgD;YAErF,+BAA+B;YAC/B,MAAM,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAEpD,6DAA6D;YAC7D,MAAM,CAAC,mCAAmC,CAAC,CAAC,oBAAoB,CAC9D,IAAI,GAAG,CAAC,QAAQ,CAAC,EACjB,EAAE,SAAS,EAAE,iBAAiB,EAAE,WAAW,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,CAC/D,CAAC;YAEF,gCAAgC;YAChC,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACvC;gBACE,IAAI,EAAE,kBAAkB;gBACxB,OAAO,EAAE,OAAO;aACjB,EACD,EAAE,YAAY,EAAE,EAAE,EAAE,CACrB,CAAC;YAEF,4DAA4D;YAC5D,MAAM,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,oBAAoB,CACxD,wBAAwB,CACzB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,EAAE,CAAC,sGAAsG,EAAE,KAAK,IAAI,EAAE;YACpH,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,sBAAS,CAAC,MAAM,CACnC,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,SAAS,EACT,SAAS,EACT,SAAS,CACV,CAAC;YACF,IAAI,CAAC,aAAa,EAAE,CAAC;YACrB,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,kBAAkB,EAAE,CAAC;YAEpE,yEAAyE;YACzE,MAAM,qBAAqB,GAAG;gBAC5B,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;gBAC/C,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;gBAC7C,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;gBACpB,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;gBACnB,OAAO,EAAE,IAAI;aACd,CAAC;YAEF,MAAM,wBAAwB,GAAG;gBAC/B,SAAS,EAAE,wBAAwB;aACpC,CAAC;YAEF,iEAAiE;YACjE,YAAY,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,qBAA4B,CAAC,CAAC;YACpE,mCAAmC,CAAC,kBAAkB,CACpD,GAAG,EAAE,CAAC,wBAA+B,CACtC,CAAC;YAEF,sCAAsC;YACtC,IAAI,CAAC,aAAa,EAAE,CAAC;YAErB,4DAA4D;YAC3D,MAAc,CAAC,OAAO,EAAE,CAAC;YAC1B,kEAAkE;YAClE,MAAM,gBAAgB,GAAG,MAAM,CAAC,SAAS,EAAE,CAAC;YAC5C,sDAAsD;YACtD,MAAM,gBAAgB,CAAC;YAEvB,mCAAmC;YACnC,MAAM,CAAC,UAAU,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAEtC,+BAA+B;YAC/B,MAAM,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAEpD,6DAA6D;YAC7D,MAAM,CAAC,mCAAmC,CAAC,CAAC,oBAAoB,CAC9D,IAAI,GAAG,CAAC,QAAQ,CAAC,EACjB,EAAE,SAAS,EAAE,iBAAiB,EAAE,WAAW,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,CAC/D,CAAC;YAEF,gCAAgC;YAChC,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACvC;gBACE,IAAI,EAAE,kBAAkB;gBACxB,OAAO,EAAE,OAAO;aACjB,EACD,EAAE,YAAY,EAAE,EAAE,EAAE,CACrB,CAAC;YAEF,4DAA4D;YAC5D,MAAM,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,oBAAoB,CACxD,wBAAwB,CACzB,CAAC;YAEF,kDAAkD;YAClD,MAAM,CAAC,YAAY,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC9C,MAAM,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAE/D,UAAU,CAAC,WAAW,EAAE,CAAC;YACzB,IAAI,CAAC,aAAa,EAAE,CAAC;QACvB,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACvD,EAAE,CAAC,oFAAoF,EAAE,KAAK,IAAI,EAAE;YAClG,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,sBAAS,CAAC,MAAM,CACnC,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,SAAS,EACT,SAAS,EACT,SAAS,CACV,CAAC;YAEF,yFAAyF;YACzF,MAAM,SAAS,GAAG,IAAI,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE;gBACnC,IAAI,OAAO,kBAAkB,CAAC,OAAO,KAAK,UAAU,EAAE,CAAC;oBACrD,wCAAwC;oBACvC,kBAAkB,CAAC,OAAiC,EAAE,CAAC;gBAC1D,CAAC;gBACD,OAAO;YACT,CAAC,CAAC,CAAC;YACH,kBAAkB,CAAC,KAAK,GAAG,SAAS,CAAC;YAErC,0CAA0C;YAC1C,MAAM,qBAAqB,GAAG;gBAC5B,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;gBAC/C,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;gBAC7C,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;gBACpB,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;gBACnB,OAAO,EAAE,IAAI;aACd,CAAC;YACF,YAAY,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,qBAA4B,CAAC,CAAC;YAEpE,sCAAsC;YACtC,IAAI,CAAC,aAAa,EAAE,CAAC;YAErB,kEAAkE;YACjE,MAAc,CAAC,OAAO,EAAE,CAAC;YAC1B,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;YAEzB,wEAAwE;YACxE,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC3C,MAAM,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,CAAC;YAEnD,6CAA6C;YAC7C,MAAM,CAAC,YAAY,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC9C,MAAM,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,wCAAwC,EAAE,GAAG,EAAE;QACtD,EAAE,CAAC,0FAA0F,EAAE,KAAK,IAAI,EAAE;YACxG,IAAI,CAAC,aAAa,EAAE,CAAC;YACrB,MAAM,IAAI,GAAG,sBAAS,CAAC,kBAAkB,CAAC;YAC1C,MAAM,KAAK,GAAG,sBAAS,CAAC,oBAAoB,CAAC;YAC7C,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC;YAC3C,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC;YAC3C,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YACvD,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,qBAAqB;YAEtF,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,sBAAS,CAAC,MAAM,CACnC,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,SAAS,EACT,SAAS,EACT,SAAS,CACV,CAAC;YAEF,8DAA8D;YAC9D,YAAY,CAAC,kBAAkB,CAC7B,GAAG,EAAE,CACH,CAAC;gBACC,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;gBAC/C,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;gBAC7C,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;gBACpB,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;gBACnB,OAAO,EAAE,IAAI;aACd,CAAQ,CACZ,CAAC;YAEF,oCAAoC;YACnC,MAAc,CAAC,OAAO,EAAE,CAAC;YAC1B,MAAM,CAAC,aAAa,CAAC,CAAC,gBAAgB,EAAE,CAAC;YACzC,MAAM,aAAa,GAAG,aAAa,CAAC,IAAI,CAAC,KAAK;iBAC3C,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;iBAChB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAa,CAAC;YACpD,MAAM,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAChD,MAAM,CAAC,GAAG,aAAa,CAAC,CAAC,CAAE,CAAC;YAC5B,MAAM,CAAC,CAAC,CAAC,CAAC,sBAAsB,CAAC,GAAG,CAAC,CAAC;YACtC,MAAM,CAAC,CAAC,CAAC,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC;YAEnC,8DAA8D;YAC9D,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;YAEzB,qEAAqE;YACpE,MAAc,CAAC,OAAO,EAAE,CAAC;YAC1B,MAAM,cAAc,GAAG,aAAa,CAAC,IAAI,CAAC,KAAK;iBAC5C,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;iBAChB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAa,CAAC;YACpD,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;YACxD,MAAM,MAAM,GAAG,cAAc,CAAC,CAAC,CAAE,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,CAAC,sBAAsB,CAAC,GAAG,CAAC,CAAC;YAC3C,MAAM,CAAC,MAAM,CAAC,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC;YAExC,IAAI,CAAC,aAAa,EAAE,CAAC;YACrB,aAAa,CAAC,WAAW,EAAE,CAAC;YAC5B,OAAO,CAAC,WAAW,EAAE,CAAC;QACxB,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,sBAAS,CAAC,MAAM,CACnC,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,SAAS,EACT,SAAS,EACT,SAAS,CACV,CAAC;YAEF,MAAM,SAAS,GAAG;gBAChB;oBACE,IAAI,EAAE,OAAO;oBACb,WAAW,EAAE,YAAY;oBACzB,WAAW,EAAE;wBACX,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;qBACzC;iBACF;gBACD;oBACE,IAAI,EAAE,OAAO;oBACb,WAAW,EAAE,aAAa;oBAC1B,WAAW,EAAE;wBACX,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;qBACzC;iBACF;aACF,CAAC;YAEF,MAAM,aAAa,GAAG;gBACpB,KAAK,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;gBACrB,UAAU,EAAE,SAAS;aACtB,CAAC;YAEF,MAAM,aAAa,GAAG;gBACpB,KAAK,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;gBACrB,UAAU,EAAE,SAAS;aACtB,CAAC;YAEF,kBAAkB,CAAC,SAAS;iBACzB,qBAAqB,CAAC,aAAa,CAAC;iBACpC,qBAAqB,CAAC,aAAa,CAAC,CAAC;YAExC,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;YAExC,MAAM,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC9D,MAAM,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,uBAAuB,CAC1D,CAAC,EACD,EAAE,MAAM,EAAE,SAAS,EAAE,EACrB,EAAE,CACH,CAAC;YACF,MAAM,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,uBAAuB,CAC1D,CAAC,EACD,EAAE,MAAM,EAAE,SAAS,EAAE,EACrB,EAAE,CACH,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB;oBACE,IAAI,EAAE,OAAO;oBACb,WAAW,EAAE,YAAY;oBACzB,WAAW,EAAE;wBACX,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;qBACzC;iBACF;gBACD;oBACE,IAAI,EAAE,OAAO;oBACb,WAAW,EAAE,aAAa;oBAC1B,WAAW,EAAE;wBACX,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;qBACzC;iBACF;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,sBAAS,CAAC,MAAM,CACnC,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,SAAS,EACT,SAAS,EACT,SAAS,CACV,CAAC;YAEF,MAAM,SAAS,GAAG;gBAChB;oBACE,IAAI,EAAE,OAAO;oBACb,WAAW,EAAE,WAAW;oBACxB,WAAW,EAAE;wBACX,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;qBACzC;iBACF;aACF,CAAC;YAEF,MAAM,YAAY,GAAG;gBACnB,KAAK,EAAE,SAAS;gBAChB,UAAU,EAAE,SAAS;aACtB,CAAC;YAEF,kBAAkB,CAAC,SAAS,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YAE7D,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;YAExC,MAAM,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC9D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB;oBACE,IAAI,EAAE,OAAO;oBACb,WAAW,EAAE,WAAW;oBACxB,WAAW,EAAE;wBACX,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;qBACzC;iBACF;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,sBAAS,CAAC,MAAM,CACnC,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,SAAS,EACT,SAAS,EACT,SAAS,CACV,CAAC;YAEF,MAAM,SAAS,GAAG;gBAChB;oBACE,IAAI,EAAE,cAAc;oBACpB,WAAW,EAAE,0BAA0B;oBACvC,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,6BAA6B;iBAC/D;aACF,CAAC;YAEF,MAAM,YAAY,GAAG;gBACnB,KAAK,EAAE,SAAS;gBAChB,UAAU,EAAE,SAAS;aACtB,CAAC;YAEF,kBAAkB,CAAC,SAAS,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YAE7D,MAAM,MAAM,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAC9C,qDAAqD,CACtD,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YACjD,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,sBAAS,CAAC,MAAM,CACnC,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,SAAS,EACT,SAAS,EACT,SAAS,CACV,CAAC;YAEF,MAAM,UAAU,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;YAC1D,kBAAkB,CAAC,QAAQ,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;YAE1D,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE;gBAC/C,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,EAAE;aACT,CAAC,CAAC;YAEH,MAAM,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC;gBACvD,IAAI,EAAE,UAAU;gBAChB,SAAS,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,EAAE;aACxC,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,sBAAS,CAAC,MAAM,CACnC,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,SAAS,EACT,SAAS,EACT,SAAS,CACV,CAAC;YACF,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;YAC5C,kBAAkB,CAAC,QAAQ,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;YAErD,MAAM,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC3D,kBAAkB,CACnB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;QACxC,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,OAAO,GAAG,EAAE,aAAa,EAAE,cAAc,EAAE,CAAC;YAElD,MAAM,sBAAS,CAAC,MAAM,CACpB,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,OAAO,EACP,SAAS,EACT,SAAS,CACV,CAAC;YAEF,MAAM,CAAC,mCAAmC,CAAC,CAAC,oBAAoB,CAC9D,IAAI,GAAG,CAAC,QAAQ,CAAC,EACjB,EAAE,SAAS,EAAE,SAAS,EAAE,WAAW,EAAE,EAAE,OAAO,EAAE,EAAE,CACnD,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAClE,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,OAAO,GAAG,EAAE,aAAa,EAAE,cAAc,EAAE,CAAC;YAElD,MAAM,sBAAS,CAAC,MAAM,CACpB,QAAQ,EACR,yBAAY,CAAC,GAAG,EAChB,OAAO,EACP,SAAS,EACT,SAAS,CACV,CAAC;YAEF,MAAM,CAAC,wBAAwB,CAAC,CAAC,oBAAoB,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,EAAE;gBACvE,WAAW,EAAE,EAAE,OAAO,EAAE;aACzB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAE/C,MAAM,sBAAS,CAAC,MAAM,CACpB,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,SAAS,EACT,SAAS,EACT,SAAS,CACV,CAAC;YAEF,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACvC;gBACE,IAAI,EAAE,kBAAkB;gBACxB,OAAO,EAAE,OAAO;aACjB,EACD,EAAE,YAAY,EAAE,EAAE,EAAE,CACrB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;YAC1C,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,OAAO,GAAG,MAAM,sBAAS,CAAC,MAAM,CACpC,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,SAAS,EACT,SAAS,EACT,SAAS,CACV,CAAC;YAEF,MAAM,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC;YACjD,MAAM,CAAC,OAAO,kBAAkB,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;QAC/C,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,WAAW,GAAG,IAAI,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAQ,CAAC,CAAC;YACrD,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAQ,CAAC,CAAC;YAElD,MAAM,sBAAS,CAAC,MAAM,CACpB,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,SAAS,EACT,SAAS,EACT,SAAS,EACT,EAAE,WAAW,EAAE,QAAQ,EAAE,CAC1B,CAAC;YAEF,MAAM,CAAC,YAAY,CAAC,CAAC,wBAAwB,CAC3C;gBACE,IAAI,EAAE,kBAAkB;gBACxB,OAAO,EAAE,OAAO;aACjB,EACD;gBACE,YAAY,EAAE;oBACZ,WAAW,EAAE,EAAE;oBACf,QAAQ,EAAE,EAAE;iBACb;aACF,CACF,CAAC;YAEF,0CAA0C;YAC1C,MAAM,CAAC,kBAAkB,CAAC,iBAAiB,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAChE,MAAM,CACH,kBAAkB,CAAC,iBAA+B,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CACtE,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,sBAAS,CAAC,MAAM,CACnC,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,SAAS,EACT,SAAS,EACT,SAAS,EACT;gBACE,WAAW,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAQ;aACrC,CACF,CAAC;YAEF,MAAM,SAAS,GAAG,kBAAkB,CAAC,oBAAiC,CAAC;YAEvE,cAAc;YACd,SAAS,CAAC,SAAS,EAAE,CAAC;YACtB,MAAM,MAAM,CAAC,wBAAwB,CAAC,SAAS,CAAC,CAAC;YACjD,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,sBAAS,CAAC,MAAM,CACnC,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,SAAS,EACT,SAAS,EACT,SAAS,EACT;gBACE,QAAQ,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAQ;aAClC,CACF,CAAC;YAEF,MAAM,SAAS,GAAG,kBAAkB,CAAC,oBAAiC,CAAC;YAEvE,cAAc;YACd,SAAS,CAAC,SAAS,EAAE,CAAC;YACtB,MAAM,MAAM,CAAC,qBAAqB,CAAC,SAAS,CAAC,CAAC;YAC9C,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { MCPClient, MCPTransport } from \"../mcp-client\";\n\n// Mock the MCP SDK modules\njest.mock(\"@modelcontextprotocol/sdk/client/index.js\", () => ({\n  Client: jest.fn().mockImplementation(() => ({\n    connect: jest.fn(),\n    close: jest.fn(),\n    listTools: jest.fn(),\n    callTool: jest.fn(),\n    setRequestHandler: jest.fn(),\n    removeRequestHandler: jest.fn(),\n    onclose: null,\n  })),\n}));\n\njest.mock(\"@modelcontextprotocol/sdk/client/sse.js\", () => ({\n  SSEClientTransport: jest.fn().mockImplementation(() => ({\n    // SSE transport doesn't have sessionId\n  })),\n}));\n\njest.mock(\"@modelcontextprotocol/sdk/client/streamableHttp.js\", () => ({\n  StreamableHTTPClientTransport: jest\n    .fn()\n    .mockImplementation((url, options) => ({\n      sessionId: options?.sessionId,\n    })),\n}));\n\nimport { Client } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport { SSEClientTransport } from \"@modelcontextprotocol/sdk/client/sse.js\";\nimport { StreamableHTTPClientTransport } from \"@modelcontextprotocol/sdk/client/streamableHttp.js\";\n\n// Type the mocked modules\nconst MockedClient = Client as jest.MockedClass<typeof Client>;\nconst MockedSSEClientTransport = SSEClientTransport as jest.MockedClass<\n  typeof SSEClientTransport\n>;\nconst MockedStreamableHTTPClientTransport =\n  StreamableHTTPClientTransport as jest.MockedClass<\n    typeof StreamableHTTPClientTransport\n  >;\n\ndescribe(\"MCPClient\", () => {\n  let mockClientInstance: any;\n  let mockTransportInstance: any;\n\n  beforeEach(() => {\n    jest.clearAllMocks();\n\n    // Create mock instances\n    mockClientInstance = {\n      connect: jest.fn().mockResolvedValue(undefined),\n      close: jest.fn().mockResolvedValue(undefined),\n      listTools: jest.fn(),\n      callTool: jest.fn(),\n      setRequestHandler: jest.fn(),\n      removeRequestHandler: jest.fn(),\n      onclose: null,\n    };\n\n    mockTransportInstance = {\n      sessionId: \"test-session-id\",\n    };\n\n    // Setup mocks\n    MockedClient.mockImplementation(() => mockClientInstance);\n    MockedStreamableHTTPClientTransport.mockImplementation(\n      () => mockTransportInstance,\n    );\n    MockedSSEClientTransport.mockImplementation(\n      () => ({}) as SSEClientTransport,\n    );\n  });\n\n  describe(\"create\", () => {\n    it(\"should create and connect an MCPClient with HTTP transport by default\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n      const headers = { Authorization: \"Bearer token\" };\n\n      const client = await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        headers,\n        undefined,\n        undefined,\n      );\n\n      expect(MockedStreamableHTTPClientTransport).toHaveBeenCalledWith(\n        new URL(endpoint),\n        { sessionId: undefined, requestInit: { headers } },\n      );\n      expect(MockedClient).toHaveBeenCalledWith(\n        {\n          name: \"tambo-mcp-client\",\n          version: \"1.0.0\",\n        },\n        { capabilities: {} },\n      );\n      expect(mockClientInstance.connect).toHaveBeenCalledWith(\n        mockTransportInstance,\n      );\n      expect(client).toBeInstanceOf(MCPClient);\n    });\n\n    it(\"should create and connect an MCPClient with SSE transport\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n\n      const client = await MCPClient.create(\n        endpoint,\n        MCPTransport.SSE,\n        undefined,\n        undefined,\n        undefined,\n      );\n\n      expect(MockedSSEClientTransport).toHaveBeenCalledWith(new URL(endpoint), {\n        requestInit: { headers: {} },\n      });\n      expect(mockClientInstance.connect).toHaveBeenCalledWith({});\n      expect(client).toBeInstanceOf(MCPClient);\n    });\n\n    it(\"should create client with default headers when none provided\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n\n      await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        undefined,\n        undefined,\n        undefined,\n      );\n\n      expect(MockedStreamableHTTPClientTransport).toHaveBeenCalledWith(\n        new URL(endpoint),\n        { sessionId: undefined, requestInit: { headers: {} } },\n      );\n    });\n  });\n\n  describe(\"reconnect\", () => {\n    it(\"should create new transport and client instances and call connect when reconnect() is called (default behavior)\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n      const client = await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        undefined,\n        undefined,\n        undefined,\n      );\n\n      // Clear previous calls to focus on reconnect behavior\n      jest.clearAllMocks();\n\n      // Create new mock instances to verify new instances are created\n      const newMockClientInstance = {\n        connect: jest.fn().mockResolvedValue(undefined),\n        close: jest.fn().mockResolvedValue(undefined),\n        listTools: jest.fn(),\n        callTool: jest.fn(),\n        onclose: null,\n      };\n\n      const newMockTransportInstance = {\n        sessionId: \"new-session-id\",\n      };\n\n      // Mock the constructors to return new instances\n      MockedClient.mockImplementation(() => newMockClientInstance as any);\n      MockedStreamableHTTPClientTransport.mockImplementation(\n        () => newMockTransportInstance as any,\n      );\n\n      await client.reconnect(); // Uses default parameters\n\n      // Verify old client was closed\n      expect(mockClientInstance.close).toHaveBeenCalled();\n\n      // Verify new transport was created with preserved session ID (default behavior)\n      expect(MockedStreamableHTTPClientTransport).toHaveBeenCalledWith(\n        new URL(endpoint),\n        { sessionId: \"test-session-id\", requestInit: { headers: {} } },\n      );\n\n      // Verify new client was created\n      expect(MockedClient).toHaveBeenCalledWith(\n        {\n          name: \"tambo-mcp-client\",\n          version: \"1.0.0\",\n        },\n        { capabilities: {} },\n      );\n\n      // Verify new client's connect was called with new transport\n      expect(newMockClientInstance.connect).toHaveBeenCalledWith(\n        newMockTransportInstance,\n      );\n    });\n\n    it(\"should reconnect without session ID for SSE transport\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n      const client = await MCPClient.create(\n        endpoint,\n        MCPTransport.SSE,\n        undefined,\n        undefined,\n        undefined,\n      );\n\n      // Clear previous calls\n      jest.clearAllMocks();\n\n      await client.reconnect();\n\n      expect(mockClientInstance.close).toHaveBeenCalled();\n      expect(MockedSSEClientTransport).toHaveBeenCalledWith(new URL(endpoint), {\n        requestInit: { headers: {} },\n      });\n      expect(mockClientInstance.connect).toHaveBeenCalledWith({});\n    });\n\n    it(\"should handle close errors when reportErrorOnClose is true\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n      const client = await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        undefined,\n        undefined,\n        undefined,\n      );\n      const consoleSpy = jest.spyOn(console, \"error\").mockImplementation();\n\n      // Make close throw an error\n      mockClientInstance.close.mockRejectedValue(new Error(\"Close failed\"));\n\n      await client.reconnect(false, true);\n\n      expect(consoleSpy).toHaveBeenCalledWith(\n        \"Error closing Tambo MCP Client:\",\n        expect.any(Error),\n      );\n      consoleSpy.mockRestore();\n    });\n\n    it(\"should not log close errors when reportErrorOnClose is false\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n      const client = await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        undefined,\n        undefined,\n        undefined,\n      );\n      const consoleSpy = jest.spyOn(console, \"error\").mockImplementation();\n\n      // Make close throw an error\n      mockClientInstance.close.mockRejectedValue(new Error(\"Close failed\"));\n\n      await client.reconnect(false, false);\n\n      expect(consoleSpy).not.toHaveBeenCalled();\n      consoleSpy.mockRestore();\n    });\n\n    it(\"should create new session when newSession is true\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n      const client = await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        undefined,\n        undefined,\n        undefined,\n      );\n\n      // Clear previous calls to focus on reconnect behavior\n      jest.clearAllMocks();\n\n      // Create new mock instances to verify new instances are created\n      const newMockClientInstance = {\n        connect: jest.fn().mockResolvedValue(undefined),\n        close: jest.fn().mockResolvedValue(undefined),\n        listTools: jest.fn(),\n        callTool: jest.fn(),\n        onclose: null,\n      };\n\n      const newMockTransportInstance = {\n        sessionId: \"new-session-id\",\n      };\n\n      // Mock the constructors to return new instances\n      MockedClient.mockImplementation(() => newMockClientInstance as any);\n      MockedStreamableHTTPClientTransport.mockImplementation(\n        () => newMockTransportInstance as any,\n      );\n\n      await client.reconnect(true, true); // newSession = true, reportErrorOnClose = true\n\n      // Verify old client was closed\n      expect(mockClientInstance.close).toHaveBeenCalled();\n\n      // Verify new transport was created with undefined session ID (new session)\n      expect(MockedStreamableHTTPClientTransport).toHaveBeenCalledWith(\n        new URL(endpoint),\n        { sessionId: undefined, requestInit: { headers: {} } },\n      );\n\n      // Verify new client was created\n      expect(MockedClient).toHaveBeenCalledWith(\n        {\n          name: \"tambo-mcp-client\",\n          version: \"1.0.0\",\n        },\n        { capabilities: {} },\n      );\n\n      // Verify new client's connect was called with new transport\n      expect(newMockClientInstance.connect).toHaveBeenCalledWith(\n        newMockTransportInstance,\n      );\n    });\n\n    it(\"should reuse existing session when newSession is false (default)\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n      const client = await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        undefined,\n        undefined,\n        undefined,\n      );\n\n      // Clear previous calls to focus on reconnect behavior\n      jest.clearAllMocks();\n\n      // Create new mock instances to verify new instances are created\n      const newMockClientInstance = {\n        connect: jest.fn().mockResolvedValue(undefined),\n        close: jest.fn().mockResolvedValue(undefined),\n        listTools: jest.fn(),\n        callTool: jest.fn(),\n        onclose: null,\n      };\n\n      const newMockTransportInstance = {\n        sessionId: \"reused-session-id\",\n      };\n\n      // Mock the constructors to return new instances\n      MockedClient.mockImplementation(() => newMockClientInstance as any);\n      MockedStreamableHTTPClientTransport.mockImplementation(\n        () => newMockTransportInstance as any,\n      );\n\n      await client.reconnect(false, true); // newSession = false, reportErrorOnClose = true\n\n      // Verify old client was closed\n      expect(mockClientInstance.close).toHaveBeenCalled();\n\n      // Verify new transport was created with preserved session ID\n      expect(MockedStreamableHTTPClientTransport).toHaveBeenCalledWith(\n        new URL(endpoint),\n        { sessionId: \"test-session-id\", requestInit: { headers: {} } },\n      );\n\n      // Verify new client was created\n      expect(MockedClient).toHaveBeenCalledWith(\n        {\n          name: \"tambo-mcp-client\",\n          version: \"1.0.0\",\n        },\n        { capabilities: {} },\n      );\n\n      // Verify new client's connect was called with new transport\n      expect(newMockClientInstance.connect).toHaveBeenCalledWith(\n        newMockTransportInstance,\n      );\n    });\n  });\n\n  describe(\"onclose\", () => {\n    it(\"should reconnect MCPClient when client is closed by external means (no backoff on manual preemption)\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n      const client = await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        undefined,\n        undefined,\n        undefined,\n      );\n      jest.useFakeTimers();\n      const consoleSpy = jest.spyOn(console, \"warn\").mockImplementation();\n\n      // Create new mock instances to verify reconnection creates new instances\n      const newMockClientInstance = {\n        connect: jest.fn().mockResolvedValue(undefined),\n        close: jest.fn().mockResolvedValue(undefined),\n        listTools: jest.fn(),\n        callTool: jest.fn(),\n        onclose: null,\n      };\n\n      const newMockTransportInstance = {\n        sessionId: \"reconnected-session-id\",\n      };\n\n      // Mock the constructors to return new instances for reconnection\n      MockedClient.mockImplementation(() => newMockClientInstance as any);\n      MockedStreamableHTTPClientTransport.mockImplementation(\n        () => newMockTransportInstance as any,\n      );\n\n      // Reset counts after initial creation\n      jest.clearAllMocks();\n\n      // Trigger automatic onclose (schedules a delayed reconnect)\n      (client as any).onclose();\n      // Manual reconnect should preempt the scheduled automatic attempt\n      const reconnectPromise = client.reconnect();\n      // No timers should be pending after manual preemption\n      await reconnectPromise;\n\n      // Verify warning message is logged\n      expect(consoleSpy).toHaveBeenCalled();\n\n      // Verify old client was closed\n      expect(mockClientInstance.close).toHaveBeenCalled();\n\n      // Verify new transport was created with preserved session ID\n      expect(MockedStreamableHTTPClientTransport).toHaveBeenCalledWith(\n        new URL(endpoint),\n        { sessionId: \"test-session-id\", requestInit: { headers: {} } },\n      );\n\n      // Verify new client was created\n      expect(MockedClient).toHaveBeenCalledWith(\n        {\n          name: \"tambo-mcp-client\",\n          version: \"1.0.0\",\n        },\n        { capabilities: {} },\n      );\n\n      // Verify new client's connect was called with new transport\n      expect(newMockClientInstance.connect).toHaveBeenCalledWith(\n        newMockTransportInstance,\n      );\n\n      // Ensure only a single reconnect attempt occurred\n      expect(MockedClient).toHaveBeenCalledTimes(1);\n      expect(newMockClientInstance.connect).toHaveBeenCalledTimes(1);\n\n      consoleSpy.mockRestore();\n      jest.useRealTimers();\n    });\n  });\n\n  describe(\"reconnect re-entrancy and single-flight\", () => {\n    it(\"prevents re-entrant onclose during deliberate close and coalesces concurrent calls\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n      const client = await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        undefined,\n        undefined,\n        undefined,\n      );\n\n      // Simulate an implementation where closing the client would call its own onclose handler\n      const closeImpl = jest.fn(async () => {\n        if (typeof mockClientInstance.onclose === \"function\") {\n          // would cause recursion if not detached\n          (mockClientInstance.onclose as unknown as () => void)();\n        }\n        return;\n      });\n      mockClientInstance.close = closeImpl;\n\n      // Prepare new instances for the reconnect\n      const newMockClientInstance = {\n        connect: jest.fn().mockResolvedValue(undefined),\n        close: jest.fn().mockResolvedValue(undefined),\n        listTools: jest.fn(),\n        callTool: jest.fn(),\n        onclose: null,\n      };\n      MockedClient.mockImplementation(() => newMockClientInstance as any);\n\n      // Reset counts after initial creation\n      jest.clearAllMocks();\n\n      // Trigger auto onclose and manual reconnect nearly simultaneously\n      (client as any).onclose();\n      await client.reconnect();\n\n      // Should have detached onclose before calling close, avoiding recursion\n      expect(closeImpl).toHaveBeenCalledTimes(1);\n      expect(mockClientInstance.onclose).toBeUndefined();\n\n      // Single-flight: only one new client/connect\n      expect(MockedClient).toHaveBeenCalledTimes(1);\n      expect(newMockClientInstance.connect).toHaveBeenCalledTimes(1);\n    });\n  });\n\n  describe(\"backoff + jitter (automatic reconnect)\", () => {\n    it(\"applies jitter and resets to initial delay after a successful reconnect (manual preempt)\", async () => {\n      jest.useFakeTimers();\n      const base = MCPClient.BACKOFF_INITIAL_MS;\n      const ratio = MCPClient.BACKOFF_JITTER_RATIO;\n      const min = Math.round(base * (1 - ratio));\n      const max = Math.round(base * (1 + ratio));\n      const setTimeoutSpy = jest.spyOn(global, \"setTimeout\");\n      const randSpy = jest.spyOn(Math, \"random\").mockReturnValue(0.0); // extreme low jitter\n\n      const endpoint = \"https://api.example.com/mcp\";\n      const client = await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        undefined,\n        undefined,\n        undefined,\n      );\n\n      // Prepare one attempt that will succeed to avoid rescheduling\n      MockedClient.mockImplementation(\n        () =>\n          ({\n            connect: jest.fn().mockResolvedValue(undefined),\n            close: jest.fn().mockResolvedValue(undefined),\n            listTools: jest.fn(),\n            callTool: jest.fn(),\n            onclose: null,\n          }) as any,\n      );\n\n      // Trigger once to capture the delay\n      (client as any).onclose();\n      expect(setTimeoutSpy).toHaveBeenCalled();\n      const numericDelays = setTimeoutSpy.mock.calls\n        .map((c) => c[1])\n        .filter((v) => typeof v === \"number\") as number[];\n      expect(numericDelays.length).toBeGreaterThan(0);\n      const d = numericDelays[0]!;\n      expect(d).toBeGreaterThanOrEqual(min);\n      expect(d).toBeLessThanOrEqual(max);\n\n      // Manual reconnect succeeds and should reset backoff attempts\n      await client.reconnect();\n\n      // Trigger onclose again and ensure we start from initial range again\n      (client as any).onclose();\n      const numericDelays2 = setTimeoutSpy.mock.calls\n        .map((c) => c[1])\n        .filter((v) => typeof v === \"number\") as number[];\n      expect(numericDelays2.length).toBeGreaterThanOrEqual(2);\n      const dAgain = numericDelays2[1]!;\n      expect(dAgain).toBeGreaterThanOrEqual(min);\n      expect(dAgain).toBeLessThanOrEqual(max);\n\n      jest.useRealTimers();\n      setTimeoutSpy.mockRestore();\n      randSpy.mockRestore();\n    });\n  });\n\n  describe(\"listTools\", () => {\n    it(\"should list all tools with pagination\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n      const client = await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        undefined,\n        undefined,\n        undefined,\n      );\n\n      const mockTools = [\n        {\n          name: \"tool1\",\n          description: \"First tool\",\n          inputSchema: {\n            type: \"object\",\n            properties: { arg1: { type: \"string\" } },\n          },\n        },\n        {\n          name: \"tool2\",\n          description: \"Second tool\",\n          inputSchema: {\n            type: \"object\",\n            properties: { arg2: { type: \"number\" } },\n          },\n        },\n      ];\n\n      const mockResponse1 = {\n        tools: [mockTools[0]],\n        nextCursor: \"cursor1\",\n      };\n\n      const mockResponse2 = {\n        tools: [mockTools[1]],\n        nextCursor: undefined,\n      };\n\n      mockClientInstance.listTools\n        .mockResolvedValueOnce(mockResponse1)\n        .mockResolvedValueOnce(mockResponse2);\n\n      const result = await client.listTools();\n\n      expect(mockClientInstance.listTools).toHaveBeenCalledTimes(2);\n      expect(mockClientInstance.listTools).toHaveBeenNthCalledWith(\n        1,\n        { cursor: undefined },\n        {},\n      );\n      expect(mockClientInstance.listTools).toHaveBeenNthCalledWith(\n        2,\n        { cursor: \"cursor1\" },\n        {},\n      );\n      expect(result).toEqual([\n        {\n          name: \"tool1\",\n          description: \"First tool\",\n          inputSchema: {\n            type: \"object\",\n            properties: { arg1: { type: \"string\" } },\n          },\n        },\n        {\n          name: \"tool2\",\n          description: \"Second tool\",\n          inputSchema: {\n            type: \"object\",\n            properties: { arg2: { type: \"number\" } },\n          },\n        },\n      ]);\n    });\n\n    it(\"should handle single page of tools\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n      const client = await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        undefined,\n        undefined,\n        undefined,\n      );\n\n      const mockTools = [\n        {\n          name: \"tool1\",\n          description: \"Only tool\",\n          inputSchema: {\n            type: \"object\",\n            properties: { arg1: { type: \"string\" } },\n          },\n        },\n      ];\n\n      const mockResponse = {\n        tools: mockTools,\n        nextCursor: undefined,\n      };\n\n      mockClientInstance.listTools.mockResolvedValue(mockResponse);\n\n      const result = await client.listTools();\n\n      expect(mockClientInstance.listTools).toHaveBeenCalledTimes(1);\n      expect(result).toEqual([\n        {\n          name: \"tool1\",\n          description: \"Only tool\",\n          inputSchema: {\n            type: \"object\",\n            properties: { arg1: { type: \"string\" } },\n          },\n        },\n      ]);\n    });\n\n    it(\"should throw error for invalid input schema\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n      const client = await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        undefined,\n        undefined,\n        undefined,\n      );\n\n      const mockTools = [\n        {\n          name: \"invalid-tool\",\n          description: \"Tool with invalid schema\",\n          inputSchema: { type: \"string\" }, // Invalid - should be object\n        },\n      ];\n\n      const mockResponse = {\n        tools: mockTools,\n        nextCursor: undefined,\n      };\n\n      mockClientInstance.listTools.mockResolvedValue(mockResponse);\n\n      await expect(client.listTools()).rejects.toThrow(\n        \"Input schema for tool invalid-tool is not an object\",\n      );\n    });\n  });\n\n  describe(\"callTool\", () => {\n    it(\"should call a tool with arguments\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n      const client = await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        undefined,\n        undefined,\n        undefined,\n      );\n\n      const mockResult = { success: true, data: \"test result\" };\n      mockClientInstance.callTool.mockResolvedValue(mockResult);\n\n      const result = await client.callTool(\"testTool\", {\n        arg1: \"value1\",\n        arg2: 42,\n      });\n\n      expect(mockClientInstance.callTool).toHaveBeenCalledWith({\n        name: \"testTool\",\n        arguments: { arg1: \"value1\", arg2: 42 },\n      });\n      expect(result).toBe(mockResult);\n    });\n\n    it(\"should handle tool call errors\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n      const client = await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        undefined,\n        undefined,\n        undefined,\n      );\n      const error = new Error(\"Tool call failed\");\n      mockClientInstance.callTool.mockRejectedValue(error);\n\n      await expect(client.callTool(\"testTool\", {})).rejects.toThrow(\n        \"Tool call failed\",\n      );\n    });\n  });\n\n  describe(\"transport initialization\", () => {\n    it(\"should initialize HTTP transport with session ID\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n      const headers = { Authorization: \"Bearer token\" };\n\n      await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        headers,\n        undefined,\n        undefined,\n      );\n\n      expect(MockedStreamableHTTPClientTransport).toHaveBeenCalledWith(\n        new URL(endpoint),\n        { sessionId: undefined, requestInit: { headers } },\n      );\n    });\n\n    it(\"should initialize SSE transport without session ID\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n      const headers = { Authorization: \"Bearer token\" };\n\n      await MCPClient.create(\n        endpoint,\n        MCPTransport.SSE,\n        headers,\n        undefined,\n        undefined,\n      );\n\n      expect(MockedSSEClientTransport).toHaveBeenCalledWith(new URL(endpoint), {\n        requestInit: { headers },\n      });\n    });\n  });\n\n  describe(\"client initialization\", () => {\n    it(\"should initialize client with correct name and version\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n\n      await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        undefined,\n        undefined,\n        undefined,\n      );\n\n      expect(MockedClient).toHaveBeenCalledWith(\n        {\n          name: \"tambo-mcp-client\",\n          version: \"1.0.0\",\n        },\n        { capabilities: {} },\n      );\n    });\n\n    it(\"should set onclose handler\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n      const _client = await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        undefined,\n        undefined,\n        undefined,\n      );\n\n      expect(mockClientInstance.onclose).toBeDefined();\n      expect(typeof mockClientInstance.onclose).toBe(\"function\");\n    });\n  });\n\n  describe(\"handlers (elicitation/sampling)\", () => {\n    it(\"sets handlers on create when provided\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n      const elicitation = jest.fn(async () => ({}) as any);\n      const sampling = jest.fn(async () => ({}) as any);\n\n      await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        undefined,\n        undefined,\n        undefined,\n        { elicitation, sampling },\n      );\n\n      expect(MockedClient).toHaveBeenLastCalledWith(\n        {\n          name: \"tambo-mcp-client\",\n          version: \"1.0.0\",\n        },\n        {\n          capabilities: {\n            elicitation: {},\n            sampling: {},\n          },\n        },\n      );\n\n      // Request handlers should be set for both\n      expect(mockClientInstance.setRequestHandler).toHaveBeenCalled();\n      expect(\n        (mockClientInstance.setRequestHandler as jest.Mock).mock.calls.length,\n      ).toBeGreaterThanOrEqual(2);\n    });\n\n    it(\"removes elicitation handler when set to undefined\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n      const client = await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        undefined,\n        undefined,\n        undefined,\n        {\n          elicitation: async () => ({}) as any,\n        },\n      );\n\n      const removeSpy = mockClientInstance.removeRequestHandler as jest.Mock;\n\n      // then remove\n      removeSpy.mockClear();\n      await client.updateElicitationHandler(undefined);\n      expect(removeSpy).toHaveBeenCalledWith(expect.any(String));\n    });\n\n    it(\"removes sampling handler when set to undefined\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n      const client = await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        undefined,\n        undefined,\n        undefined,\n        {\n          sampling: async () => ({}) as any,\n        },\n      );\n\n      const removeSpy = mockClientInstance.removeRequestHandler as jest.Mock;\n\n      // then remove\n      removeSpy.mockClear();\n      await client.updateSamplingHandler(undefined);\n      expect(removeSpy).toHaveBeenCalledWith(expect.any(String));\n    });\n  });\n});\n"]}
+{"version":3,"file":"mcp-client.test.js","sourceRoot":"","sources":["../../../src/mcp/__tests__/mcp-client.test.ts"],"names":[],"mappings":";;AAAA,8CAAwD;AAExD,2BAA2B;AAC3B,IAAI,CAAC,IAAI,CAAC,2CAA2C,EAAE,GAAG,EAAE,CAAC,CAAC;IAC5D,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,CAAC;QAC1C,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE;QAClB,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;QAChB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;QACpB,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;QACnB,iBAAiB,EAAE,IAAI,CAAC,EAAE,EAAE;QAC5B,oBAAoB,EAAE,IAAI,CAAC,EAAE,EAAE;KAChC,CAAC,CAAC;CACJ,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,yCAAyC,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1D,kBAAkB,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,CAAC;IACtD,uCAAuC;KACxC,CAAC,CAAC;CACJ,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,oDAAoD,EAAE,GAAG,EAAE,CAAC,CAAC;IACrE,6BAA6B,EAAE,IAAI;SAChC,EAAE,EAAE;SACJ,kBAAkB,CAAC,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;QACrC,SAAS,EAAE,OAAO,EAAE,SAAS;KAC9B,CAAC,CAAC;CACN,CAAC,CAAC,CAAC;AAEJ,wEAAmE;AACnE,oEAA6E;AAC7E,0FAAmG;AAEnG,0BAA0B;AAC1B,MAAM,YAAY,GAAG,iBAAyC,CAAC;AAC/D,MAAM,wBAAwB,GAAG,2BAEhC,CAAC;AACF,MAAM,mCAAmC,GACvC,iDAEC,CAAC;AAEJ,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,IAAI,kBAAuB,CAAC;IAC5B,IAAI,qBAA0B,CAAC;IAE/B,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;QAErB,wBAAwB;QACxB,kBAAkB,GAAG;YACnB,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;YAC/C,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;YAC7C,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;YACpB,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;YACnB,iBAAiB,EAAE,IAAI,CAAC,EAAE,EAAE;YAC5B,oBAAoB,EAAE,IAAI,CAAC,EAAE,EAAE;SAChC,CAAC;QAEF,qBAAqB,GAAG;YACtB,SAAS,EAAE,iBAAiB;SAC7B,CAAC;QAEF,cAAc;QACd,YAAY,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,CAAC;QAC1D,mCAAmC,CAAC,kBAAkB,CACpD,GAAG,EAAE,CAAC,qBAAqB,CAC5B,CAAC;QACF,wBAAwB,CAAC,kBAAkB,CACzC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAuB,CACjC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;QACtB,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;YACrF,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,OAAO,GAAG,EAAE,aAAa,EAAE,cAAc,EAAE,CAAC;YAElD,MAAM,MAAM,GAAG,MAAM,sBAAS,CAAC,MAAM,CACnC,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,OAAO,EACP,SAAS,EACT,SAAS,CACV,CAAC;YAEF,MAAM,CAAC,mCAAmC,CAAC,CAAC,oBAAoB,CAC9D,IAAI,GAAG,CAAC,QAAQ,CAAC,EACjB,EAAE,SAAS,EAAE,SAAS,EAAE,WAAW,EAAE,EAAE,OAAO,EAAE,EAAE,CACnD,CAAC;YACF,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACvC;gBACE,IAAI,EAAE,kBAAkB;gBACxB,OAAO,EAAE,OAAO;aACjB,EACD,EAAE,YAAY,EAAE,EAAE,EAAE,CACrB,CAAC;YACF,MAAM,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,oBAAoB,CACrD,qBAAqB,CACtB,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,sBAAS,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAE/C,MAAM,MAAM,GAAG,MAAM,sBAAS,CAAC,MAAM,CACnC,QAAQ,EACR,yBAAY,CAAC,GAAG,EAChB,SAAS,EACT,SAAS,EACT,SAAS,CACV,CAAC;YAEF,MAAM,CAAC,wBAAwB,CAAC,CAAC,oBAAoB,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,EAAE;gBACvE,WAAW,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE;aAC7B,CAAC,CAAC;YACH,MAAM,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAAC,EAAE,CAAC,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,sBAAS,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;YAC5E,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAE/C,MAAM,sBAAS,CAAC,MAAM,CACpB,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,SAAS,EACT,SAAS,EACT,SAAS,CACV,CAAC;YAEF,MAAM,CAAC,mCAAmC,CAAC,CAAC,oBAAoB,CAC9D,IAAI,GAAG,CAAC,QAAQ,CAAC,EACjB,EAAE,SAAS,EAAE,SAAS,EAAE,WAAW,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,CACvD,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,sBAAS,CAAC,MAAM,CACnC,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,SAAS,EACT,SAAS,EACT,SAAS,CACV,CAAC;YAEF,MAAM,SAAS,GAAG;gBAChB;oBACE,IAAI,EAAE,OAAO;oBACb,WAAW,EAAE,YAAY;oBACzB,WAAW,EAAE;wBACX,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;qBACzC;iBACF;gBACD;oBACE,IAAI,EAAE,OAAO;oBACb,WAAW,EAAE,aAAa;oBAC1B,WAAW,EAAE;wBACX,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;qBACzC;iBACF;aACF,CAAC;YAEF,MAAM,aAAa,GAAG;gBACpB,KAAK,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;gBACrB,UAAU,EAAE,SAAS;aACtB,CAAC;YAEF,MAAM,aAAa,GAAG;gBACpB,KAAK,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;gBACrB,UAAU,EAAE,SAAS;aACtB,CAAC;YAEF,kBAAkB,CAAC,SAAS;iBACzB,qBAAqB,CAAC,aAAa,CAAC;iBACpC,qBAAqB,CAAC,aAAa,CAAC,CAAC;YAExC,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;YAExC,MAAM,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC9D,MAAM,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,uBAAuB,CAC1D,CAAC,EACD,EAAE,MAAM,EAAE,SAAS,EAAE,EACrB,EAAE,CACH,CAAC;YACF,MAAM,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,uBAAuB,CAC1D,CAAC,EACD,EAAE,MAAM,EAAE,SAAS,EAAE,EACrB,EAAE,CACH,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB;oBACE,IAAI,EAAE,OAAO;oBACb,WAAW,EAAE,YAAY;oBACzB,WAAW,EAAE;wBACX,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;qBACzC;iBACF;gBACD;oBACE,IAAI,EAAE,OAAO;oBACb,WAAW,EAAE,aAAa;oBAC1B,WAAW,EAAE;wBACX,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;qBACzC;iBACF;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,sBAAS,CAAC,MAAM,CACnC,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,SAAS,EACT,SAAS,EACT,SAAS,CACV,CAAC;YAEF,MAAM,SAAS,GAAG;gBAChB;oBACE,IAAI,EAAE,OAAO;oBACb,WAAW,EAAE,WAAW;oBACxB,WAAW,EAAE;wBACX,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;qBACzC;iBACF;aACF,CAAC;YAEF,MAAM,YAAY,GAAG;gBACnB,KAAK,EAAE,SAAS;gBAChB,UAAU,EAAE,SAAS;aACtB,CAAC;YAEF,kBAAkB,CAAC,SAAS,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YAE7D,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;YAExC,MAAM,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC9D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB;oBACE,IAAI,EAAE,OAAO;oBACb,WAAW,EAAE,WAAW;oBACxB,WAAW,EAAE;wBACX,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;qBACzC;iBACF;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,sBAAS,CAAC,MAAM,CACnC,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,SAAS,EACT,SAAS,EACT,SAAS,CACV,CAAC;YAEF,MAAM,SAAS,GAAG;gBAChB;oBACE,IAAI,EAAE,cAAc;oBACpB,WAAW,EAAE,0BAA0B;oBACvC,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,6BAA6B;iBAC/D;aACF,CAAC;YAEF,MAAM,YAAY,GAAG;gBACnB,KAAK,EAAE,SAAS;gBAChB,UAAU,EAAE,SAAS;aACtB,CAAC;YAEF,kBAAkB,CAAC,SAAS,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YAE7D,MAAM,MAAM,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAC9C,qDAAqD,CACtD,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YACjD,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,sBAAS,CAAC,MAAM,CACnC,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,SAAS,EACT,SAAS,EACT,SAAS,CACV,CAAC;YAEF,MAAM,UAAU,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;YAC1D,kBAAkB,CAAC,QAAQ,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;YAE1D,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE;gBAC/C,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,EAAE;aACT,CAAC,CAAC;YAEH,MAAM,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC;gBACvD,IAAI,EAAE,UAAU;gBAChB,SAAS,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,EAAE;aACxC,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,sBAAS,CAAC,MAAM,CACnC,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,SAAS,EACT,SAAS,EACT,SAAS,CACV,CAAC;YACF,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;YAC5C,kBAAkB,CAAC,QAAQ,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;YAErD,MAAM,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC3D,kBAAkB,CACnB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;QACxC,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,OAAO,GAAG,EAAE,aAAa,EAAE,cAAc,EAAE,CAAC;YAElD,MAAM,sBAAS,CAAC,MAAM,CACpB,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,OAAO,EACP,SAAS,EACT,SAAS,CACV,CAAC;YAEF,MAAM,CAAC,mCAAmC,CAAC,CAAC,oBAAoB,CAC9D,IAAI,GAAG,CAAC,QAAQ,CAAC,EACjB,EAAE,SAAS,EAAE,SAAS,EAAE,WAAW,EAAE,EAAE,OAAO,EAAE,EAAE,CACnD,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAClE,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,OAAO,GAAG,EAAE,aAAa,EAAE,cAAc,EAAE,CAAC;YAElD,MAAM,sBAAS,CAAC,MAAM,CACpB,QAAQ,EACR,yBAAY,CAAC,GAAG,EAChB,OAAO,EACP,SAAS,EACT,SAAS,CACV,CAAC;YAEF,MAAM,CAAC,wBAAwB,CAAC,CAAC,oBAAoB,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,EAAE;gBACvE,WAAW,EAAE,EAAE,OAAO,EAAE;aACzB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAE/C,MAAM,sBAAS,CAAC,MAAM,CACpB,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,SAAS,EACT,SAAS,EACT,SAAS,CACV,CAAC;YAEF,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACvC;gBACE,IAAI,EAAE,kBAAkB;gBACxB,OAAO,EAAE,OAAO;aACjB,EACD,EAAE,YAAY,EAAE,EAAE,EAAE,CACrB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;QAC/C,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,WAAW,GAAG,IAAI,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAQ,CAAC,CAAC;YACrD,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAQ,CAAC,CAAC;YAElD,MAAM,sBAAS,CAAC,MAAM,CACpB,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,SAAS,EACT,SAAS,EACT,SAAS,EACT,EAAE,WAAW,EAAE,QAAQ,EAAE,CAC1B,CAAC;YAEF,MAAM,CAAC,YAAY,CAAC,CAAC,wBAAwB,CAC3C;gBACE,IAAI,EAAE,kBAAkB;gBACxB,OAAO,EAAE,OAAO;aACjB,EACD;gBACE,YAAY,EAAE;oBACZ,WAAW,EAAE,EAAE;oBACf,QAAQ,EAAE,EAAE;iBACb;aACF,CACF,CAAC;YAEF,0CAA0C;YAC1C,MAAM,CAAC,kBAAkB,CAAC,iBAAiB,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAChE,MAAM,CACH,kBAAkB,CAAC,iBAA+B,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CACtE,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,sBAAS,CAAC,MAAM,CACnC,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,SAAS,EACT,SAAS,EACT,SAAS,EACT;gBACE,WAAW,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAQ;aACrC,CACF,CAAC;YAEF,MAAM,SAAS,GAAG,kBAAkB,CAAC,oBAAiC,CAAC;YAEvE,cAAc;YACd,SAAS,CAAC,SAAS,EAAE,CAAC;YACtB,MAAM,MAAM,CAAC,wBAAwB,CAAC,SAAS,CAAC,CAAC;YACjD,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,sBAAS,CAAC,MAAM,CACnC,QAAQ,EACR,yBAAY,CAAC,IAAI,EACjB,SAAS,EACT,SAAS,EACT,SAAS,EACT;gBACE,QAAQ,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAQ;aAClC,CACF,CAAC;YAEF,MAAM,SAAS,GAAG,kBAAkB,CAAC,oBAAiC,CAAC;YAEvE,cAAc;YACd,SAAS,CAAC,SAAS,EAAE,CAAC;YACtB,MAAM,MAAM,CAAC,qBAAqB,CAAC,SAAS,CAAC,CAAC;YAC9C,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { MCPClient, MCPTransport } from \"../mcp-client\";\n\n// Mock the MCP SDK modules\njest.mock(\"@modelcontextprotocol/sdk/client/index.js\", () => ({\n  Client: jest.fn().mockImplementation(() => ({\n    connect: jest.fn(),\n    close: jest.fn(),\n    listTools: jest.fn(),\n    callTool: jest.fn(),\n    setRequestHandler: jest.fn(),\n    removeRequestHandler: jest.fn(),\n  })),\n}));\n\njest.mock(\"@modelcontextprotocol/sdk/client/sse.js\", () => ({\n  SSEClientTransport: jest.fn().mockImplementation(() => ({\n    // SSE transport doesn't have sessionId\n  })),\n}));\n\njest.mock(\"@modelcontextprotocol/sdk/client/streamableHttp.js\", () => ({\n  StreamableHTTPClientTransport: jest\n    .fn()\n    .mockImplementation((url, options) => ({\n      sessionId: options?.sessionId,\n    })),\n}));\n\nimport { Client } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport { SSEClientTransport } from \"@modelcontextprotocol/sdk/client/sse.js\";\nimport { StreamableHTTPClientTransport } from \"@modelcontextprotocol/sdk/client/streamableHttp.js\";\n\n// Type the mocked modules\nconst MockedClient = Client as jest.MockedClass<typeof Client>;\nconst MockedSSEClientTransport = SSEClientTransport as jest.MockedClass<\n  typeof SSEClientTransport\n>;\nconst MockedStreamableHTTPClientTransport =\n  StreamableHTTPClientTransport as jest.MockedClass<\n    typeof StreamableHTTPClientTransport\n  >;\n\ndescribe(\"MCPClient\", () => {\n  let mockClientInstance: any;\n  let mockTransportInstance: any;\n\n  beforeEach(() => {\n    jest.clearAllMocks();\n\n    // Create mock instances\n    mockClientInstance = {\n      connect: jest.fn().mockResolvedValue(undefined),\n      close: jest.fn().mockResolvedValue(undefined),\n      listTools: jest.fn(),\n      callTool: jest.fn(),\n      setRequestHandler: jest.fn(),\n      removeRequestHandler: jest.fn(),\n    };\n\n    mockTransportInstance = {\n      sessionId: \"test-session-id\",\n    };\n\n    // Setup mocks\n    MockedClient.mockImplementation(() => mockClientInstance);\n    MockedStreamableHTTPClientTransport.mockImplementation(\n      () => mockTransportInstance,\n    );\n    MockedSSEClientTransport.mockImplementation(\n      () => ({}) as SSEClientTransport,\n    );\n  });\n\n  describe(\"create\", () => {\n    it(\"should create and connect an MCPClient with HTTP transport by default\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n      const headers = { Authorization: \"Bearer token\" };\n\n      const client = await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        headers,\n        undefined,\n        undefined,\n      );\n\n      expect(MockedStreamableHTTPClientTransport).toHaveBeenCalledWith(\n        new URL(endpoint),\n        { sessionId: undefined, requestInit: { headers } },\n      );\n      expect(MockedClient).toHaveBeenCalledWith(\n        {\n          name: \"tambo-mcp-client\",\n          version: \"1.0.0\",\n        },\n        { capabilities: {} },\n      );\n      expect(mockClientInstance.connect).toHaveBeenCalledWith(\n        mockTransportInstance,\n      );\n      expect(client).toBeInstanceOf(MCPClient);\n    });\n\n    it(\"should create and connect an MCPClient with SSE transport\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n\n      const client = await MCPClient.create(\n        endpoint,\n        MCPTransport.SSE,\n        undefined,\n        undefined,\n        undefined,\n      );\n\n      expect(MockedSSEClientTransport).toHaveBeenCalledWith(new URL(endpoint), {\n        requestInit: { headers: {} },\n      });\n      expect(mockClientInstance.connect).toHaveBeenCalledWith({});\n      expect(client).toBeInstanceOf(MCPClient);\n    });\n\n    it(\"should create client with default headers when none provided\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n\n      await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        undefined,\n        undefined,\n        undefined,\n      );\n\n      expect(MockedStreamableHTTPClientTransport).toHaveBeenCalledWith(\n        new URL(endpoint),\n        { sessionId: undefined, requestInit: { headers: {} } },\n      );\n    });\n  });\n\n  describe(\"listTools\", () => {\n    it(\"should list all tools with pagination\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n      const client = await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        undefined,\n        undefined,\n        undefined,\n      );\n\n      const mockTools = [\n        {\n          name: \"tool1\",\n          description: \"First tool\",\n          inputSchema: {\n            type: \"object\",\n            properties: { arg1: { type: \"string\" } },\n          },\n        },\n        {\n          name: \"tool2\",\n          description: \"Second tool\",\n          inputSchema: {\n            type: \"object\",\n            properties: { arg2: { type: \"number\" } },\n          },\n        },\n      ];\n\n      const mockResponse1 = {\n        tools: [mockTools[0]],\n        nextCursor: \"cursor1\",\n      };\n\n      const mockResponse2 = {\n        tools: [mockTools[1]],\n        nextCursor: undefined,\n      };\n\n      mockClientInstance.listTools\n        .mockResolvedValueOnce(mockResponse1)\n        .mockResolvedValueOnce(mockResponse2);\n\n      const result = await client.listTools();\n\n      expect(mockClientInstance.listTools).toHaveBeenCalledTimes(2);\n      expect(mockClientInstance.listTools).toHaveBeenNthCalledWith(\n        1,\n        { cursor: undefined },\n        {},\n      );\n      expect(mockClientInstance.listTools).toHaveBeenNthCalledWith(\n        2,\n        { cursor: \"cursor1\" },\n        {},\n      );\n      expect(result).toEqual([\n        {\n          name: \"tool1\",\n          description: \"First tool\",\n          inputSchema: {\n            type: \"object\",\n            properties: { arg1: { type: \"string\" } },\n          },\n        },\n        {\n          name: \"tool2\",\n          description: \"Second tool\",\n          inputSchema: {\n            type: \"object\",\n            properties: { arg2: { type: \"number\" } },\n          },\n        },\n      ]);\n    });\n\n    it(\"should handle single page of tools\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n      const client = await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        undefined,\n        undefined,\n        undefined,\n      );\n\n      const mockTools = [\n        {\n          name: \"tool1\",\n          description: \"Only tool\",\n          inputSchema: {\n            type: \"object\",\n            properties: { arg1: { type: \"string\" } },\n          },\n        },\n      ];\n\n      const mockResponse = {\n        tools: mockTools,\n        nextCursor: undefined,\n      };\n\n      mockClientInstance.listTools.mockResolvedValue(mockResponse);\n\n      const result = await client.listTools();\n\n      expect(mockClientInstance.listTools).toHaveBeenCalledTimes(1);\n      expect(result).toEqual([\n        {\n          name: \"tool1\",\n          description: \"Only tool\",\n          inputSchema: {\n            type: \"object\",\n            properties: { arg1: { type: \"string\" } },\n          },\n        },\n      ]);\n    });\n\n    it(\"should throw error for invalid input schema\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n      const client = await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        undefined,\n        undefined,\n        undefined,\n      );\n\n      const mockTools = [\n        {\n          name: \"invalid-tool\",\n          description: \"Tool with invalid schema\",\n          inputSchema: { type: \"string\" }, // Invalid - should be object\n        },\n      ];\n\n      const mockResponse = {\n        tools: mockTools,\n        nextCursor: undefined,\n      };\n\n      mockClientInstance.listTools.mockResolvedValue(mockResponse);\n\n      await expect(client.listTools()).rejects.toThrow(\n        \"Input schema for tool invalid-tool is not an object\",\n      );\n    });\n  });\n\n  describe(\"callTool\", () => {\n    it(\"should call a tool with arguments\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n      const client = await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        undefined,\n        undefined,\n        undefined,\n      );\n\n      const mockResult = { success: true, data: \"test result\" };\n      mockClientInstance.callTool.mockResolvedValue(mockResult);\n\n      const result = await client.callTool(\"testTool\", {\n        arg1: \"value1\",\n        arg2: 42,\n      });\n\n      expect(mockClientInstance.callTool).toHaveBeenCalledWith({\n        name: \"testTool\",\n        arguments: { arg1: \"value1\", arg2: 42 },\n      });\n      expect(result).toBe(mockResult);\n    });\n\n    it(\"should handle tool call errors\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n      const client = await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        undefined,\n        undefined,\n        undefined,\n      );\n      const error = new Error(\"Tool call failed\");\n      mockClientInstance.callTool.mockRejectedValue(error);\n\n      await expect(client.callTool(\"testTool\", {})).rejects.toThrow(\n        \"Tool call failed\",\n      );\n    });\n  });\n\n  describe(\"transport initialization\", () => {\n    it(\"should initialize HTTP transport with session ID\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n      const headers = { Authorization: \"Bearer token\" };\n\n      await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        headers,\n        undefined,\n        undefined,\n      );\n\n      expect(MockedStreamableHTTPClientTransport).toHaveBeenCalledWith(\n        new URL(endpoint),\n        { sessionId: undefined, requestInit: { headers } },\n      );\n    });\n\n    it(\"should initialize SSE transport without session ID\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n      const headers = { Authorization: \"Bearer token\" };\n\n      await MCPClient.create(\n        endpoint,\n        MCPTransport.SSE,\n        headers,\n        undefined,\n        undefined,\n      );\n\n      expect(MockedSSEClientTransport).toHaveBeenCalledWith(new URL(endpoint), {\n        requestInit: { headers },\n      });\n    });\n  });\n\n  describe(\"client initialization\", () => {\n    it(\"should initialize client with correct name and version\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n\n      await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        undefined,\n        undefined,\n        undefined,\n      );\n\n      expect(MockedClient).toHaveBeenCalledWith(\n        {\n          name: \"tambo-mcp-client\",\n          version: \"1.0.0\",\n        },\n        { capabilities: {} },\n      );\n    });\n  });\n\n  describe(\"handlers (elicitation/sampling)\", () => {\n    it(\"sets handlers on create when provided\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n      const elicitation = jest.fn(async () => ({}) as any);\n      const sampling = jest.fn(async () => ({}) as any);\n\n      await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        undefined,\n        undefined,\n        undefined,\n        { elicitation, sampling },\n      );\n\n      expect(MockedClient).toHaveBeenLastCalledWith(\n        {\n          name: \"tambo-mcp-client\",\n          version: \"1.0.0\",\n        },\n        {\n          capabilities: {\n            elicitation: {},\n            sampling: {},\n          },\n        },\n      );\n\n      // Request handlers should be set for both\n      expect(mockClientInstance.setRequestHandler).toHaveBeenCalled();\n      expect(\n        (mockClientInstance.setRequestHandler as jest.Mock).mock.calls.length,\n      ).toBeGreaterThanOrEqual(2);\n    });\n\n    it(\"removes elicitation handler when set to undefined\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n      const client = await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        undefined,\n        undefined,\n        undefined,\n        {\n          elicitation: async () => ({}) as any,\n        },\n      );\n\n      const removeSpy = mockClientInstance.removeRequestHandler as jest.Mock;\n\n      // then remove\n      removeSpy.mockClear();\n      await client.updateElicitationHandler(undefined);\n      expect(removeSpy).toHaveBeenCalledWith(expect.any(String));\n    });\n\n    it(\"removes sampling handler when set to undefined\", async () => {\n      const endpoint = \"https://api.example.com/mcp\";\n      const client = await MCPClient.create(\n        endpoint,\n        MCPTransport.HTTP,\n        undefined,\n        undefined,\n        undefined,\n        {\n          sampling: async () => ({}) as any,\n        },\n      );\n\n      const removeSpy = mockClientInstance.removeRequestHandler as jest.Mock;\n\n      // then remove\n      removeSpy.mockClear();\n      await client.updateSamplingHandler(undefined);\n      expect(removeSpy).toHaveBeenCalledWith(expect.any(String));\n    });\n  });\n});\n"]}