@tambo-ai/react 0.21.1 → 0.21.3

package/dist/hooks/use-suggestions.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"use-suggestions.d.ts","sourceRoot":"","sources":["../../src/hooks/use-suggestions.ts"],"names":[],"mappings":"AAAA,OAAO,OAAO,MAAM,0BAA0B,CAAC;AAO/C,OAAO,EACL,sBAAsB,EAEvB,MAAM,qBAAqB,CAAC;AAE7B,OAAO,EACL,sBAAsB,EACtB,mBAAmB,EAGpB,MAAM,qBAAqB,CAAC;AAG7B;;GAEG;AACH,MAAM,WAAW,0BAA0B;IACzC,kEAAkE;IAClE,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB;AAED;;GAEG;AACH,MAAM,WAAW,iCAAiC;IAChD,4EAA4E;IAC5E,WAAW,EAAE,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;IAC/C,8CAA8C;IAC9C,oBAAoB,EAAE,MAAM,GAAG,IAAI,CAAC;IACpC;;;;OAIG;IACH,MAAM,EAAE,CAAC,aAAa,EAAE;QACtB,UAAU,EAAE,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;QAC5C,YAAY,CAAC,EAAE,OAAO,CAAC;KACxB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAEpB,0DAA0D;IAC1D,YAAY,EAAE,sBAAsB,CAClC,IAAI,EACJ,KAAK,EACL;QAAE,UAAU,EAAE,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;QAAC,YAAY,CAAC,EAAE,OAAO,CAAA;KAAE,CACxE,CAAC;IAEF,0DAA0D;IAC1D,cAAc,EAAE,sBAAsB,CACpC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,0BAA0B,GAAG,SAAS,EACvE,KAAK,EACL,eAAe,CAChB,CAAC;IAEF,yDAAyD;IACzD,iBAAiB,EAAE,mBAAmB,CACpC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,0BAA0B,GAAG,SAAS,EACvE,KAAK,CACN,CAAC;CACH;AAED,KAAK,yBAAyB,GAAG,sBAAsB,CAAC,GAAG,EAAE,KAAK,CAAC,GACjE,iCAAiC,CAAC;AAEpC;;;;GAIG;AACH,wBAAgB,mBAAmB,CACjC,OAAO,GAAE,0BAA+B,GACvC,yBAAyB,CAiI3B"}
+{"version":3,"file":"use-suggestions.d.ts","sourceRoot":"","sources":["../../src/hooks/use-suggestions.ts"],"names":[],"mappings":"AAAA,OAAO,OAAO,MAAM,0BAA0B,CAAC;AAQ/C,OAAO,EACL,sBAAsB,EAEvB,MAAM,qBAAqB,CAAC;AAE7B,OAAO,EACL,sBAAsB,EACtB,mBAAmB,EAGpB,MAAM,qBAAqB,CAAC;AAG7B;;GAEG;AACH,MAAM,WAAW,0BAA0B;IACzC,kEAAkE;IAClE,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB;AAED;;GAEG;AACH,MAAM,WAAW,iCAAiC;IAChD,4EAA4E;IAC5E,WAAW,EAAE,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;IAC/C,8CAA8C;IAC9C,oBAAoB,EAAE,MAAM,GAAG,IAAI,CAAC;IACpC;;;;OAIG;IACH,MAAM,EAAE,CAAC,aAAa,EAAE;QACtB,UAAU,EAAE,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;QAC5C,YAAY,CAAC,EAAE,OAAO,CAAC;KACxB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAEpB,0DAA0D;IAC1D,YAAY,EAAE,sBAAsB,CAClC,IAAI,EACJ,KAAK,EACL;QAAE,UAAU,EAAE,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;QAAC,YAAY,CAAC,EAAE,OAAO,CAAA;KAAE,CACxE,CAAC;IAEF,0DAA0D;IAC1D,cAAc,EAAE,sBAAsB,CACpC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,0BAA0B,GAAG,SAAS,EACvE,KAAK,EACL,eAAe,CAChB,CAAC;IAEF,yDAAyD;IACzD,iBAAiB,EAAE,mBAAmB,CACpC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,0BAA0B,GAAG,SAAS,EACvE,KAAK,CACN,CAAC;CACH;AAED,KAAK,yBAAyB,GAAG,sBAAsB,CAAC,GAAG,EAAE,KAAK,CAAC,GACjE,iCAAiC,CAAC;AAEpC;;;;GAIG;AACH,wBAAgB,mBAAmB,CACjC,OAAO,GAAE,0BAA+B,GACvC,yBAAyB,CAsI3B"}

package/dist/hooks/use-suggestions.js

@@ -2,6 +2,7 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.useTamboSuggestions = useTamboSuggestions;
 const react_1 = require("react");
+const generate_component_response_1 = require("../model/generate-component-response");
 const validate_input_1 = require("../model/validate-input");
 const tambo_client_provider_1 = require("../providers/tambo-client-provider");
 const tambo_provider_1 = require("../providers/tambo-provider");
@@ -18,7 +19,7 @@ const use_thread_input_1 = require("./use-thread-input");
  */
 function useTamboSuggestions(options = {}) {
     const { maxSuggestions = 3 } = options;
-    const { thread } = (0, tambo_thread_provider_1.useTamboThread)();
+    const { thread, generationStage } = (0, tambo_thread_provider_1.useTamboThread)();
     const { sendThreadMessage } = (0, tambo_provider_1.useTambo)();
     const tamboClient = (0, tambo_client_provider_1.useTamboClient)();
     const { componentList, toolRegistry, componentToolAssociations } = (0, tambo_registry_provider_1.useTamboRegistry)();
@@ -31,12 +32,17 @@ function useTamboSuggestions(options = {}) {
     (0, react_1.useEffect)(() => {
         setSelectedSuggestionId(null);
     }, [latestMessageId]);
+    const shouldGenerateSuggestions = latestMessageId && isLatestFromTambo && (0, generate_component_response_1.isIdleStage)(generationStage);
     // Use React Query to fetch suggestions when a new hydra message is received
     const suggestionsResult = (0, react_query_hooks_1.useTamboQuery)({
-        // Only include latestMessageId in the queryKey if the message is from hydra
-        queryKey: ["suggestions", isLatestFromTambo ? latestMessageId : null],
+        // Make sure the query key changes when the message changes, so that we are
+        // always generating suggestions when there is a new message
+        queryKey: [
+            "suggestions",
+            shouldGenerateSuggestions ? latestMessageId : null,
+        ],
         queryFn: async () => {
-            if (!latestMessageId || !isLatestFromTambo) {
+            if (!shouldGenerateSuggestions) {
                 return [];
             }
             // Get registered components from the registry
@@ -78,7 +84,7 @@ function useTamboSuggestions(options = {}) {
     // Generate suggestions mutation
     const generateMutationState = (0, react_query_hooks_1.useTamboMutation)({
         mutationFn: async (abortController) => {
-            if (!latestMessageId || !isLatestFromTambo) {
+            if (!shouldGenerateSuggestions) {
                 return undefined;
             }
             // Get registered components from the registry

package/dist/hooks/use-suggestions.js.map

@@ -1 +1 @@
-{"version":3,"file":"use-suggestions.js","sourceRoot":"","sources":["../../src/hooks/use-suggestions.ts"],"names":[],"mappings":";;AA2EA,kDAmIC;AA7MD,iCAA4C;AAC5C,4DAAwD;AACxD,8EAAoE;AACpE,gEAAuD;AACvD,kFAAwE;AACxE,8EAAoE;AACpE,qDAG6B;AAC7B,+CAA0D;AAC1D,2DAK6B;AAC7B,yDAA+E;AAoD/E;;;;GAIG;AACH,SAAgB,mBAAmB,CACjC,UAAsC,EAAE;IAExC,MAAM,EAAE,cAAc,GAAG,CAAC,EAAE,GAAG,OAAO,CAAC;IACvC,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,sCAAc,GAAE,CAAC;IACpC,MAAM,EAAE,iBAAiB,EAAE,GAAG,IAAA,yBAAQ,GAAE,CAAC;IACzC,MAAM,WAAW,GAAG,IAAA,sCAAc,GAAE,CAAC;IACrC,MAAM,EAAE,aAAa,EAAE,YAAY,EAAE,yBAAyB,EAAE,GAC9D,IAAA,0CAAgB,GAAE,CAAC;IAErB,MAAM,CAAC,oBAAoB,EAAE,uBAAuB,CAAC,GAAG,IAAA,gBAAQ,EAE9D,IAAI,CAAC,CAAC;IACR,MAAM,EAAE,QAAQ,EAAE,aAAa,EAAE,GAAG,IAAA,sCAAmB,GAAE,CAAC;IAE1D,MAAM,aAAa,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAClE,MAAM,iBAAiB,GAAG,aAAa,EAAE,IAAI,KAAK,WAAW,CAAC;IAC9D,MAAM,eAAe,GAAG,aAAa,EAAE,EAAE,CAAC;IAE1C,qDAAqD;IACrD,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,uBAAuB,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC,EAAE,CAAC,eAAe,CAAC,CAAC,CAAC;IAEtB,4EAA4E;IAC5E,MAAM,iBAAiB,GAAG,IAAA,iCAAa,EAAC;QACtC,4EAA4E;QAC5E,QAAQ,EAAE,CAAC,aAAa,EAAE,iBAAiB,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC;QACrE,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,IAAI,CAAC,eAAe,IAAI,CAAC,iBAAiB,EAAE,CAAC;gBAC3C,OAAO,EAAE,CAAC;YACZ,CAAC;YAED,8CAA8C;YAC9C,MAAM,UAAU,GAAG,IAAA,iCAAsB,EACvC,aAAa,EACb,YAAY,EACZ,yBAAyB,CAC1B,CAAC;YAEF,OAAO,MAAM,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,CACxD,MAAM,CAAC,EAAE,EACT,eAAe,EACf;gBACE,cAAc;gBACd,mBAAmB,EAAE,UAAU;aAChC,CACF,CAAC;QACJ,CAAC;QACD,2DAA2D;QAC3D,OAAO,EAAE,OAAO,CAAC,eAAe,IAAI,iBAAiB,CAAC;QACtD,6CAA6C;QAC7C,oBAAoB,EAAE,KAAK;QAC3B,kBAAkB,EAAE,KAAK;QACzB,yBAAyB;QACzB,KAAK,EAAE,KAAK;KACb,CAAC,CAAC;IAEH,6BAA6B;IAC7B,MAAM,mBAAmB,GAAG,IAAA,oCAAgB,EAI1C;QACA,UAAU,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,YAAY,GAAG,KAAK,EAAE,EAAE,EAAE;YACzD,MAAM,UAAU,GAAG,IAAA,8BAAa,EAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC;YAChE,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;gBACxB,IAAI,UAAU,CAAC,KAAK,EAAE,CAAC;oBACrB,MAAM,UAAU,CAAC,KAAK,CAAC;gBACzB,CAAC;gBACD,MAAM,IAAI,KAAK,CAAC,uCAAoB,CAAC,UAAU,CAAC,CAAC;YACnD,CAAC;YAED,IAAI,YAAY,EAAE,CAAC;gBACjB,MAAM,iBAAiB,CAAC,UAAU,CAAC,cAAc,EAAE;oBACjD,QAAQ,EAAE,MAAM,CAAC,EAAE;iBACpB,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,aAAa,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC;YAC3C,CAAC;YACD,uBAAuB,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QACzC,CAAC;KACF,CAAC,CAAC;IAEH,gCAAgC;IAChC,MAAM,qBAAqB,GAAG,IAAA,oCAAgB,EAI5C;QACA,UAAU,EAAE,KAAK,EAAE,eAAgC,EAAE,EAAE;YACrD,IAAI,CAAC,eAAe,IAAI,CAAC,iBAAiB,EAAE,CAAC;gBAC3C,OAAO,SAAS,CAAC;YACnB,CAAC;YAED,8CAA8C;YAC9C,MAAM,UAAU,GAAG,IAAA,iCAAsB,EACvC,aAAa,EACb,YAAY,EACZ,yBAAyB,CAC1B,CAAC;YAEF,OAAO,MAAM,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,CACxD,MAAM,CAAC,EAAE,EACT,eAAe,EACf;gBACE,cAAc;gBACd,mBAAmB,EAAE,UAAU;aAChC,EACD,EAAE,MAAM,EAAE,eAAe,CAAC,MAAM,EAAE,CACnC,CAAC;QACJ,CAAC;QACD,yBAAyB;QACzB,KAAK,EAAE,KAAK;KACb,CAAC,CAAC;IAEH,mEAAmE;IACnE,8DAA8D;IAC9D,MAAM,WAAW,GAAG,iBAAiB;QACnC,CAAC,CAAC,CAAC,iBAAiB,CAAC,IAAI,IAAI,qBAAqB,CAAC,IAAI,IAAI,EAAE,CAAC;QAC9D,CAAC,CAAC,EAAE,CAAC;IAEP,OAAO;QACL,WAAW;QACX,MAAM,EAAE,mBAAmB,CAAC,WAAW;QACvC,oBAAoB;QACpB,YAAY,EAAE,mBAAmB;QACjC,cAAc,EAAE,qBAAqB;QACrC,iBAAiB;QACjB,GAAG,IAAA,oCAAsB,EAAC,mBAAmB,EAAE,qBAAqB,CAAC;KACtE,CAAC;AACJ,CAAC","sourcesContent":["import TamboAI from \"@tambo-ai/typescript-sdk\";\nimport { useEffect, useState } from \"react\";\nimport { validateInput } from \"../model/validate-input\";\nimport { useTamboClient } from \"../providers/tambo-client-provider\";\nimport { useTambo } from \"../providers/tambo-provider\";\nimport { useTamboRegistry } from \"../providers/tambo-registry-provider\";\nimport { useTamboThread } from \"../providers/tambo-thread-provider\";\nimport {\n  CombinedMutationResult,\n  combineMutationResults,\n} from \"../util/query-utils\";\nimport { getAvailableComponents } from \"../util/registry\";\nimport {\n  UseTamboMutationResult,\n  UseTamboQueryResult,\n  useTamboMutation,\n  useTamboQuery,\n} from \"./react-query-hooks\";\nimport { INPUT_ERROR_MESSAGES, useTamboThreadInput } from \"./use-thread-input\";\n\n/**\n * Configuration options for the useTamboSuggestions hook\n */\nexport interface useTamboSuggestionsOptions {\n  /** Maximum number of suggestions to generate (1-10, default 3) */\n  maxSuggestions?: number;\n}\n\n/**\n * Return value interface for useTamboSuggestions hook\n */\nexport interface useTamboSuggestionsResultInternal {\n  /** List of available suggestions (also available in generateResult.data) */\n  suggestions: TamboAI.Beta.Threads.Suggestion[];\n  /** ID of the currently selected suggestion */\n  selectedSuggestionId: string | null;\n  /**\n   * Accept and apply a suggestion (also available in acceptResult.mutateAsync)\n   * @param suggestion - The suggestion to accept\n   * @param shouldSubmit - Whether to automatically submit after accepting (default: false)\n   */\n  accept: (acceptOptions: {\n    suggestion: TamboAI.Beta.Threads.Suggestion;\n    shouldSubmit?: boolean;\n  }) => Promise<void>;\n\n  /** Result and network state for accepting a suggestion */\n  acceptResult: UseTamboMutationResult<\n    void,\n    Error,\n    { suggestion: TamboAI.Beta.Threads.Suggestion; shouldSubmit?: boolean }\n  >;\n\n  /** Result and network state for generating suggestions */\n  generateResult: UseTamboMutationResult<\n    TamboAI.Beta.Threads.Suggestions.SuggestionGenerateResponse | undefined,\n    Error,\n    AbortController\n  >;\n\n  /** The full suggestions query object from React Query */\n  suggestionsResult: UseTamboQueryResult<\n    TamboAI.Beta.Threads.Suggestions.SuggestionGenerateResponse | undefined,\n    Error\n  >;\n}\n\ntype useTamboSuggestionsResult = CombinedMutationResult<any, Error> &\n  useTamboSuggestionsResultInternal;\n\n/**\n * Hook for managing Tambo AI suggestions in a thread\n * @param options - Configuration options for suggestion generation\n * @returns Object containing suggestions state and control functions\n */\nexport function useTamboSuggestions(\n  options: useTamboSuggestionsOptions = {},\n): useTamboSuggestionsResult {\n  const { maxSuggestions = 3 } = options;\n  const { thread } = useTamboThread();\n  const { sendThreadMessage } = useTambo();\n  const tamboClient = useTamboClient();\n  const { componentList, toolRegistry, componentToolAssociations } =\n    useTamboRegistry();\n\n  const [selectedSuggestionId, setSelectedSuggestionId] = useState<\n    string | null\n  >(null);\n  const { setValue: setInputValue } = useTamboThreadInput();\n\n  const latestMessage = thread.messages[thread.messages.length - 1];\n  const isLatestFromTambo = latestMessage?.role === \"assistant\";\n  const latestMessageId = latestMessage?.id;\n\n  // Reset selected suggestion when the message changes\n  useEffect(() => {\n    setSelectedSuggestionId(null);\n  }, [latestMessageId]);\n\n  // Use React Query to fetch suggestions when a new hydra message is received\n  const suggestionsResult = useTamboQuery({\n    // Only include latestMessageId in the queryKey if the message is from hydra\n    queryKey: [\"suggestions\", isLatestFromTambo ? latestMessageId : null],\n    queryFn: async () => {\n      if (!latestMessageId || !isLatestFromTambo) {\n        return [];\n      }\n\n      // Get registered components from the registry\n      const components = getAvailableComponents(\n        componentList,\n        toolRegistry,\n        componentToolAssociations,\n      );\n\n      return await tamboClient.beta.threads.suggestions.generate(\n        thread.id,\n        latestMessageId,\n        {\n          maxSuggestions,\n          availableComponents: components,\n        },\n      );\n    },\n    // Only run the query if we have a valid message from hydra\n    enabled: Boolean(latestMessageId && isLatestFromTambo),\n    // Don't refetch on window focus or reconnect\n    refetchOnWindowFocus: false,\n    refetchOnReconnect: false,\n    // Don't retry on failure\n    retry: false,\n  });\n\n  // Accept suggestion mutation\n  const acceptMutationState = useTamboMutation<\n    void,\n    Error,\n    { suggestion: TamboAI.Beta.Threads.Suggestion; shouldSubmit?: boolean }\n  >({\n    mutationFn: async ({ suggestion, shouldSubmit = false }) => {\n      const validation = validateInput(suggestion.detailedSuggestion);\n      if (!validation.isValid) {\n        if (validation.error) {\n          throw validation.error;\n        }\n        throw new Error(INPUT_ERROR_MESSAGES.VALIDATION);\n      }\n\n      if (shouldSubmit) {\n        await sendThreadMessage(validation.sanitizedInput, {\n          threadId: thread.id,\n        });\n      } else {\n        setInputValue(validation.sanitizedInput);\n      }\n      setSelectedSuggestionId(suggestion.id);\n    },\n  });\n\n  // Generate suggestions mutation\n  const generateMutationState = useTamboMutation<\n    TamboAI.Beta.Threads.Suggestions.SuggestionGenerateResponse | undefined,\n    Error,\n    AbortController\n  >({\n    mutationFn: async (abortController: AbortController) => {\n      if (!latestMessageId || !isLatestFromTambo) {\n        return undefined;\n      }\n\n      // Get registered components from the registry\n      const components = getAvailableComponents(\n        componentList,\n        toolRegistry,\n        componentToolAssociations,\n      );\n\n      return await tamboClient.beta.threads.suggestions.generate(\n        thread.id,\n        latestMessageId,\n        {\n          maxSuggestions,\n          availableComponents: components,\n        },\n        { signal: abortController.signal },\n      );\n    },\n    // Don't retry on failure\n    retry: false,\n  });\n\n  // Use the query data if available, otherwise use the mutation data\n  // Only return suggestions if the latest message is from hydra\n  const suggestions = isLatestFromTambo\n    ? (suggestionsResult.data ?? generateMutationState.data ?? [])\n    : [];\n\n  return {\n    suggestions,\n    accept: acceptMutationState.mutateAsync,\n    selectedSuggestionId,\n    acceptResult: acceptMutationState,\n    generateResult: generateMutationState,\n    suggestionsResult,\n    ...combineMutationResults(acceptMutationState, generateMutationState),\n  };\n}\n"]}
+{"version":3,"file":"use-suggestions.js","sourceRoot":"","sources":["../../src/hooks/use-suggestions.ts"],"names":[],"mappings":";;AA4EA,kDAwIC;AAnND,iCAA4C;AAC5C,sFAAmE;AACnE,4DAAwD;AACxD,8EAAoE;AACpE,gEAAuD;AACvD,kFAAwE;AACxE,8EAAoE;AACpE,qDAG6B;AAC7B,+CAA0D;AAC1D,2DAK6B;AAC7B,yDAA+E;AAoD/E;;;;GAIG;AACH,SAAgB,mBAAmB,CACjC,UAAsC,EAAE;IAExC,MAAM,EAAE,cAAc,GAAG,CAAC,EAAE,GAAG,OAAO,CAAC;IACvC,MAAM,EAAE,MAAM,EAAE,eAAe,EAAE,GAAG,IAAA,sCAAc,GAAE,CAAC;IACrD,MAAM,EAAE,iBAAiB,EAAE,GAAG,IAAA,yBAAQ,GAAE,CAAC;IACzC,MAAM,WAAW,GAAG,IAAA,sCAAc,GAAE,CAAC;IACrC,MAAM,EAAE,aAAa,EAAE,YAAY,EAAE,yBAAyB,EAAE,GAC9D,IAAA,0CAAgB,GAAE,CAAC;IAErB,MAAM,CAAC,oBAAoB,EAAE,uBAAuB,CAAC,GAAG,IAAA,gBAAQ,EAE9D,IAAI,CAAC,CAAC;IACR,MAAM,EAAE,QAAQ,EAAE,aAAa,EAAE,GAAG,IAAA,sCAAmB,GAAE,CAAC;IAE1D,MAAM,aAAa,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAClE,MAAM,iBAAiB,GAAG,aAAa,EAAE,IAAI,KAAK,WAAW,CAAC;IAC9D,MAAM,eAAe,GAAG,aAAa,EAAE,EAAE,CAAC;IAE1C,qDAAqD;IACrD,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,uBAAuB,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC,EAAE,CAAC,eAAe,CAAC,CAAC,CAAC;IAEtB,MAAM,yBAAyB,GAC7B,eAAe,IAAI,iBAAiB,IAAI,IAAA,yCAAW,EAAC,eAAe,CAAC,CAAC;IACvE,4EAA4E;IAC5E,MAAM,iBAAiB,GAAG,IAAA,iCAAa,EAAC;QACtC,2EAA2E;QAC3E,4DAA4D;QAC5D,QAAQ,EAAE;YACR,aAAa;YACb,yBAAyB,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI;SACnD;QACD,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,IAAI,CAAC,yBAAyB,EAAE,CAAC;gBAC/B,OAAO,EAAE,CAAC;YACZ,CAAC;YAED,8CAA8C;YAC9C,MAAM,UAAU,GAAG,IAAA,iCAAsB,EACvC,aAAa,EACb,YAAY,EACZ,yBAAyB,CAC1B,CAAC;YAEF,OAAO,MAAM,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,CACxD,MAAM,CAAC,EAAE,EACT,eAAe,EACf;gBACE,cAAc;gBACd,mBAAmB,EAAE,UAAU;aAChC,CACF,CAAC;QACJ,CAAC;QACD,2DAA2D;QAC3D,OAAO,EAAE,OAAO,CAAC,eAAe,IAAI,iBAAiB,CAAC;QACtD,6CAA6C;QAC7C,oBAAoB,EAAE,KAAK;QAC3B,kBAAkB,EAAE,KAAK;QACzB,yBAAyB;QACzB,KAAK,EAAE,KAAK;KACb,CAAC,CAAC;IAEH,6BAA6B;IAC7B,MAAM,mBAAmB,GAAG,IAAA,oCAAgB,EAI1C;QACA,UAAU,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,YAAY,GAAG,KAAK,EAAE,EAAE,EAAE;YACzD,MAAM,UAAU,GAAG,IAAA,8BAAa,EAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC;YAChE,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;gBACxB,IAAI,UAAU,CAAC,KAAK,EAAE,CAAC;oBACrB,MAAM,UAAU,CAAC,KAAK,CAAC;gBACzB,CAAC;gBACD,MAAM,IAAI,KAAK,CAAC,uCAAoB,CAAC,UAAU,CAAC,CAAC;YACnD,CAAC;YAED,IAAI,YAAY,EAAE,CAAC;gBACjB,MAAM,iBAAiB,CAAC,UAAU,CAAC,cAAc,EAAE;oBACjD,QAAQ,EAAE,MAAM,CAAC,EAAE;iBACpB,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,aAAa,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC;YAC3C,CAAC;YACD,uBAAuB,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QACzC,CAAC;KACF,CAAC,CAAC;IAEH,gCAAgC;IAChC,MAAM,qBAAqB,GAAG,IAAA,oCAAgB,EAI5C;QACA,UAAU,EAAE,KAAK,EAAE,eAAgC,EAAE,EAAE;YACrD,IAAI,CAAC,yBAAyB,EAAE,CAAC;gBAC/B,OAAO,SAAS,CAAC;YACnB,CAAC;YAED,8CAA8C;YAC9C,MAAM,UAAU,GAAG,IAAA,iCAAsB,EACvC,aAAa,EACb,YAAY,EACZ,yBAAyB,CAC1B,CAAC;YACF,OAAO,MAAM,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,CACxD,MAAM,CAAC,EAAE,EACT,eAAe,EACf;gBACE,cAAc;gBACd,mBAAmB,EAAE,UAAU;aAChC,EACD,EAAE,MAAM,EAAE,eAAe,CAAC,MAAM,EAAE,CACnC,CAAC;QACJ,CAAC;QACD,yBAAyB;QACzB,KAAK,EAAE,KAAK;KACb,CAAC,CAAC;IAEH,mEAAmE;IACnE,8DAA8D;IAC9D,MAAM,WAAW,GAAG,iBAAiB;QACnC,CAAC,CAAC,CAAC,iBAAiB,CAAC,IAAI,IAAI,qBAAqB,CAAC,IAAI,IAAI,EAAE,CAAC;QAC9D,CAAC,CAAC,EAAE,CAAC;IAEP,OAAO;QACL,WAAW;QACX,MAAM,EAAE,mBAAmB,CAAC,WAAW;QACvC,oBAAoB;QACpB,YAAY,EAAE,mBAAmB;QACjC,cAAc,EAAE,qBAAqB;QACrC,iBAAiB;QACjB,GAAG,IAAA,oCAAsB,EAAC,mBAAmB,EAAE,qBAAqB,CAAC;KACtE,CAAC;AACJ,CAAC","sourcesContent":["import TamboAI from \"@tambo-ai/typescript-sdk\";\nimport { useEffect, useState } from \"react\";\nimport { isIdleStage } from \"../model/generate-component-response\";\nimport { validateInput } from \"../model/validate-input\";\nimport { useTamboClient } from \"../providers/tambo-client-provider\";\nimport { useTambo } from \"../providers/tambo-provider\";\nimport { useTamboRegistry } from \"../providers/tambo-registry-provider\";\nimport { useTamboThread } from \"../providers/tambo-thread-provider\";\nimport {\n  CombinedMutationResult,\n  combineMutationResults,\n} from \"../util/query-utils\";\nimport { getAvailableComponents } from \"../util/registry\";\nimport {\n  UseTamboMutationResult,\n  UseTamboQueryResult,\n  useTamboMutation,\n  useTamboQuery,\n} from \"./react-query-hooks\";\nimport { INPUT_ERROR_MESSAGES, useTamboThreadInput } from \"./use-thread-input\";\n\n/**\n * Configuration options for the useTamboSuggestions hook\n */\nexport interface useTamboSuggestionsOptions {\n  /** Maximum number of suggestions to generate (1-10, default 3) */\n  maxSuggestions?: number;\n}\n\n/**\n * Return value interface for useTamboSuggestions hook\n */\nexport interface useTamboSuggestionsResultInternal {\n  /** List of available suggestions (also available in generateResult.data) */\n  suggestions: TamboAI.Beta.Threads.Suggestion[];\n  /** ID of the currently selected suggestion */\n  selectedSuggestionId: string | null;\n  /**\n   * Accept and apply a suggestion (also available in acceptResult.mutateAsync)\n   * @param suggestion - The suggestion to accept\n   * @param shouldSubmit - Whether to automatically submit after accepting (default: false)\n   */\n  accept: (acceptOptions: {\n    suggestion: TamboAI.Beta.Threads.Suggestion;\n    shouldSubmit?: boolean;\n  }) => Promise<void>;\n\n  /** Result and network state for accepting a suggestion */\n  acceptResult: UseTamboMutationResult<\n    void,\n    Error,\n    { suggestion: TamboAI.Beta.Threads.Suggestion; shouldSubmit?: boolean }\n  >;\n\n  /** Result and network state for generating suggestions */\n  generateResult: UseTamboMutationResult<\n    TamboAI.Beta.Threads.Suggestions.SuggestionGenerateResponse | undefined,\n    Error,\n    AbortController\n  >;\n\n  /** The full suggestions query object from React Query */\n  suggestionsResult: UseTamboQueryResult<\n    TamboAI.Beta.Threads.Suggestions.SuggestionGenerateResponse | undefined,\n    Error\n  >;\n}\n\ntype useTamboSuggestionsResult = CombinedMutationResult<any, Error> &\n  useTamboSuggestionsResultInternal;\n\n/**\n * Hook for managing Tambo AI suggestions in a thread\n * @param options - Configuration options for suggestion generation\n * @returns Object containing suggestions state and control functions\n */\nexport function useTamboSuggestions(\n  options: useTamboSuggestionsOptions = {},\n): useTamboSuggestionsResult {\n  const { maxSuggestions = 3 } = options;\n  const { thread, generationStage } = useTamboThread();\n  const { sendThreadMessage } = useTambo();\n  const tamboClient = useTamboClient();\n  const { componentList, toolRegistry, componentToolAssociations } =\n    useTamboRegistry();\n\n  const [selectedSuggestionId, setSelectedSuggestionId] = useState<\n    string | null\n  >(null);\n  const { setValue: setInputValue } = useTamboThreadInput();\n\n  const latestMessage = thread.messages[thread.messages.length - 1];\n  const isLatestFromTambo = latestMessage?.role === \"assistant\";\n  const latestMessageId = latestMessage?.id;\n\n  // Reset selected suggestion when the message changes\n  useEffect(() => {\n    setSelectedSuggestionId(null);\n  }, [latestMessageId]);\n\n  const shouldGenerateSuggestions =\n    latestMessageId && isLatestFromTambo && isIdleStage(generationStage);\n  // Use React Query to fetch suggestions when a new hydra message is received\n  const suggestionsResult = useTamboQuery({\n    // Make sure the query key changes when the message changes, so that we are\n    // always generating suggestions when there is a new message\n    queryKey: [\n      \"suggestions\",\n      shouldGenerateSuggestions ? latestMessageId : null,\n    ],\n    queryFn: async () => {\n      if (!shouldGenerateSuggestions) {\n        return [];\n      }\n\n      // Get registered components from the registry\n      const components = getAvailableComponents(\n        componentList,\n        toolRegistry,\n        componentToolAssociations,\n      );\n\n      return await tamboClient.beta.threads.suggestions.generate(\n        thread.id,\n        latestMessageId,\n        {\n          maxSuggestions,\n          availableComponents: components,\n        },\n      );\n    },\n    // Only run the query if we have a valid message from hydra\n    enabled: Boolean(latestMessageId && isLatestFromTambo),\n    // Don't refetch on window focus or reconnect\n    refetchOnWindowFocus: false,\n    refetchOnReconnect: false,\n    // Don't retry on failure\n    retry: false,\n  });\n\n  // Accept suggestion mutation\n  const acceptMutationState = useTamboMutation<\n    void,\n    Error,\n    { suggestion: TamboAI.Beta.Threads.Suggestion; shouldSubmit?: boolean }\n  >({\n    mutationFn: async ({ suggestion, shouldSubmit = false }) => {\n      const validation = validateInput(suggestion.detailedSuggestion);\n      if (!validation.isValid) {\n        if (validation.error) {\n          throw validation.error;\n        }\n        throw new Error(INPUT_ERROR_MESSAGES.VALIDATION);\n      }\n\n      if (shouldSubmit) {\n        await sendThreadMessage(validation.sanitizedInput, {\n          threadId: thread.id,\n        });\n      } else {\n        setInputValue(validation.sanitizedInput);\n      }\n      setSelectedSuggestionId(suggestion.id);\n    },\n  });\n\n  // Generate suggestions mutation\n  const generateMutationState = useTamboMutation<\n    TamboAI.Beta.Threads.Suggestions.SuggestionGenerateResponse | undefined,\n    Error,\n    AbortController\n  >({\n    mutationFn: async (abortController: AbortController) => {\n      if (!shouldGenerateSuggestions) {\n        return undefined;\n      }\n\n      // Get registered components from the registry\n      const components = getAvailableComponents(\n        componentList,\n        toolRegistry,\n        componentToolAssociations,\n      );\n      return await tamboClient.beta.threads.suggestions.generate(\n        thread.id,\n        latestMessageId,\n        {\n          maxSuggestions,\n          availableComponents: components,\n        },\n        { signal: abortController.signal },\n      );\n    },\n    // Don't retry on failure\n    retry: false,\n  });\n\n  // Use the query data if available, otherwise use the mutation data\n  // Only return suggestions if the latest message is from hydra\n  const suggestions = isLatestFromTambo\n    ? (suggestionsResult.data ?? generateMutationState.data ?? [])\n    : [];\n\n  return {\n    suggestions,\n    accept: acceptMutationState.mutateAsync,\n    selectedSuggestionId,\n    acceptResult: acceptMutationState,\n    generateResult: generateMutationState,\n    suggestionsResult,\n    ...combineMutationResults(acceptMutationState, generateMutationState),\n  };\n}\n"]}

package/dist/providers/tambo-provider.d.ts

@@ -12,6 +12,7 @@ import { TamboThreadContextProps } from "./tambo-thread-provider";
  * @param props.apiKey - The API key for the Tambo API
  * @param props.components - The components to register
  * @param props.environment - The environment to use for the Tambo API
+ * @param props.tools - The tools to register
  * @returns The TamboProvider component
  */
 export declare const TamboProvider: React.FC<PropsWithChildren<TamboClientProviderProps & TamboRegistryProviderProps>>;

package/dist/providers/tambo-provider.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"tambo-provider.d.ts","sourceRoot":"","sources":["../../src/providers/tambo-provider.tsx"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,EAAE,iBAAiB,EAA6B,MAAM,OAAO,CAAC;AAC5E,OAAO,EACL,uBAAuB,EAEvB,wBAAwB,EAGzB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EACL,0BAA0B,EAG3B,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAEL,0BAA0B,EAC3B,MAAM,2BAA2B,CAAC;AACnC,OAAO,EACL,uBAAuB,EAGxB,MAAM,yBAAyB,CAAC;AAEjC;;;;;;;;;;GAUG;AACH,eAAO,MAAM,aAAa,EAAE,KAAK,CAAC,EAAE,CAClC,iBAAiB,CAAC,wBAAwB,GAAG,0BAA0B,CAAC,CAsBzE,CAAC;AACF,MAAM,MAAM,iBAAiB,GAAG,uBAAuB,GACrD,uBAAuB,GACvB,0BAA0B,CAAC;AAE7B,eAAO,MAAM,YAAY,kCAExB,CAAC;AA6BF;;;;;;GAMG;AACH,eAAO,MAAM,QAAQ,yBAEpB,CAAC"}
+{"version":3,"file":"tambo-provider.d.ts","sourceRoot":"","sources":["../../src/providers/tambo-provider.tsx"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,EAAE,iBAAiB,EAA6B,MAAM,OAAO,CAAC;AAC5E,OAAO,EACL,uBAAuB,EAEvB,wBAAwB,EAGzB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EACL,0BAA0B,EAG3B,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAEL,0BAA0B,EAC3B,MAAM,2BAA2B,CAAC;AACnC,OAAO,EACL,uBAAuB,EAGxB,MAAM,yBAAyB,CAAC;AAEjC;;;;;;;;;;;GAWG;AACH,eAAO,MAAM,aAAa,EAAE,KAAK,CAAC,EAAE,CAClC,iBAAiB,CAAC,wBAAwB,GAAG,0BAA0B,CAAC,CAsBzE,CAAC;AACF,MAAM,MAAM,iBAAiB,GAAG,uBAAuB,GACrD,uBAAuB,GACvB,0BAA0B,CAAC;AAE7B,eAAO,MAAM,YAAY,kCAExB,CAAC;AA6BF;;;;;;GAMG;AACH,eAAO,MAAM,QAAQ,yBAEpB,CAAC"}

package/dist/providers/tambo-provider.js

@@ -49,15 +49,16 @@ const tambo_thread_provider_1 = require("./tambo-thread-provider");
  * @param props.apiKey - The API key for the Tambo API
  * @param props.components - The components to register
  * @param props.environment - The environment to use for the Tambo API
+ * @param props.tools - The tools to register
  * @returns The TamboProvider component
  */
-const TamboProvider = ({ children, tamboUrl, apiKey, components, environment }) => {
+const TamboProvider = ({ children, tamboUrl, apiKey, components, environment, tools }) => {
     // Should only be used in browser
     if (typeof window === "undefined") {
         console.error("TamboProvider must be used within a browser");
     }
     return (react_1.default.createElement(tambo_client_provider_1.TamboClientProvider, { tamboUrl: tamboUrl, apiKey: apiKey, environment: environment },
-        react_1.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: components },
+        react_1.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: components, tools: tools },
             react_1.default.createElement(tambo_thread_provider_1.TamboThreadProvider, null,
                 react_1.default.createElement(tambo_component_provider_1.TamboComponentProvider, null,
                     react_1.default.createElement(TamboCompositeProvider, null, children))))));

package/dist/providers/tambo-provider.js.map

@@ -1 +1 @@
-{"version":3,"file":"tambo-provider.js","sourceRoot":"","sources":["../../src/providers/tambo-provider.tsx"],"names":[],"mappings":";AAAA,YAAY,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AACb,+CAA4E;AAC5E,mEAMiC;AACjC,yEAIoC;AACpC,uEAGmC;AACnC,mEAIiC;AAEjC;;;;;;;;;;GAUG;AACI,MAAM,aAAa,GAEtB,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,WAAW,EAAE,EAAE,EAAE;IAC9D,iCAAiC;IACjC,IAAI,OAAO,MAAM,KAAK,WAAW,EAAE,CAAC;QAClC,OAAO,CAAC,KAAK,CAAC,6CAA6C,CAAC,CAAC;IAC/D,CAAC;IAED,OAAO,CACL,8BAAC,2CAAmB,IAClB,QAAQ,EAAE,QAAQ,EAClB,MAAM,EAAE,MAAM,EACd,WAAW,EAAE,WAAW;QAExB,8BAAC,+CAAqB,IAAC,UAAU,EAAE,UAAU;YAC3C,8BAAC,2CAAmB;gBAClB,8BAAC,iDAAsB;oBACrB,8BAAC,sBAAsB,QAAE,QAAQ,CAA0B,CACpC,CACL,CACA,CACJ,CACvB,CAAC;AACJ,CAAC,CAAC;AAvBW,QAAA,aAAa,iBAuBxB;AAKW,QAAA,YAAY,GAAG,IAAA,qBAAa,EACvC,EAAuB,CACxB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,sBAAsB,GAAgC,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE;IAC3E,MAAM,OAAO,GAAG,IAAA,sCAAc,GAAE,CAAC;IACjC,MAAM,MAAM,GAAG,IAAA,sCAAc,GAAE,CAAC;IAChC,MAAM,WAAW,GAAG,IAAA,2CAAmB,GAAE,CAAC;IAC1C,MAAM,iBAAiB,GAAG,IAAA,4CAAiB,GAAE,CAAC;IAE9C,OAAO,CACL,8BAAC,oBAAY,CAAC,QAAQ,IACpB,KAAK,EAAE;YACL,MAAM;YACN,WAAW;YACX,GAAG,iBAAiB;YACpB,GAAG,OAAO;SACX,IAEA,QAAQ,CACa,CACzB,CAAC;AACJ,CAAC,CAAC;AAEF;;;;;;GAMG;AACI,MAAM,QAAQ,GAAG,GAAG,EAAE;IAC3B,OAAO,IAAA,kBAAU,EAAC,oBAAY,CAAC,CAAC;AAClC,CAAC,CAAC;AAFW,QAAA,QAAQ,YAEnB","sourcesContent":["\"use client\";\nimport React, { PropsWithChildren, createContext, useContext } from \"react\";\nimport {\n  TamboClientContextProps,\n  TamboClientProvider,\n  TamboClientProviderProps,\n  useTamboClient,\n  useTamboQueryClient,\n} from \"./tambo-client-provider\";\nimport {\n  TamboComponentContextProps,\n  TamboComponentProvider,\n  useTamboComponent,\n} from \"./tambo-component-provider\";\nimport {\n  TamboRegistryProvider,\n  TamboRegistryProviderProps,\n} from \"./tambo-registry-provider\";\nimport {\n  TamboThreadContextProps,\n  TamboThreadProvider,\n  useTamboThread,\n} from \"./tambo-thread-provider\";\n\n/**\n * The TamboProvider gives full access to the whole Tambo API. This includes the\n * TamboAI client, the component registry, and the current thread context.\n * @param props - The props for the TamboProvider\n * @param props.children - The children to wrap\n * @param props.tamboUrl - The URL of the Tambo API\n * @param props.apiKey - The API key for the Tambo API\n * @param props.components - The components to register\n * @param props.environment - The environment to use for the Tambo API\n * @returns The TamboProvider component\n */\nexport const TamboProvider: React.FC<\n  PropsWithChildren<TamboClientProviderProps & TamboRegistryProviderProps>\n> = ({ children, tamboUrl, apiKey, components, environment }) => {\n  // Should only be used in browser\n  if (typeof window === \"undefined\") {\n    console.error(\"TamboProvider must be used within a browser\");\n  }\n\n  return (\n    <TamboClientProvider\n      tamboUrl={tamboUrl}\n      apiKey={apiKey}\n      environment={environment}\n    >\n      <TamboRegistryProvider components={components}>\n        <TamboThreadProvider>\n          <TamboComponentProvider>\n            <TamboCompositeProvider>{children}</TamboCompositeProvider>\n          </TamboComponentProvider>\n        </TamboThreadProvider>\n      </TamboRegistryProvider>\n    </TamboClientProvider>\n  );\n};\nexport type TamboContextProps = TamboClientContextProps &\n  TamboThreadContextProps &\n  TamboComponentContextProps;\n\nexport const TamboContext = createContext<TamboContextProps>(\n  {} as TamboContextProps,\n);\n\n/**\n * TamboCompositeProvider is a provider that combines the TamboClient,\n * TamboThread, and TamboComponent providers\n * @param props - The props for the TamboCompositeProvider\n * @param props.children - The children to wrap\n * @returns The wrapped component\n */\nconst TamboCompositeProvider: React.FC<PropsWithChildren> = ({ children }) => {\n  const threads = useTamboThread();\n  const client = useTamboClient();\n  const queryClient = useTamboQueryClient();\n  const componentRegistry = useTamboComponent();\n\n  return (\n    <TamboContext.Provider\n      value={{\n        client,\n        queryClient,\n        ...componentRegistry,\n        ...threads,\n      }}\n    >\n      {children}\n    </TamboContext.Provider>\n  );\n};\n\n/**\n * The useTambo hook provides access to the Tambo API. This is the primary entrypoint\n * for the Tambo React SDK.\n *\n * This includes the TamboAI client, the component registry, and the current thread context.\n * @returns The Tambo API\n */\nexport const useTambo = () => {\n  return useContext(TamboContext);\n};\n"]}
+{"version":3,"file":"tambo-provider.js","sourceRoot":"","sources":["../../src/providers/tambo-provider.tsx"],"names":[],"mappings":";AAAA,YAAY,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AACb,+CAA4E;AAC5E,mEAMiC;AACjC,yEAIoC;AACpC,uEAGmC;AACnC,mEAIiC;AAEjC;;;;;;;;;;;GAWG;AACI,MAAM,aAAa,GAEtB,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,WAAW,EAAE,KAAK,EAAE,EAAE,EAAE;IACrE,iCAAiC;IACjC,IAAI,OAAO,MAAM,KAAK,WAAW,EAAE,CAAC;QAClC,OAAO,CAAC,KAAK,CAAC,6CAA6C,CAAC,CAAC;IAC/D,CAAC;IAED,OAAO,CACL,8BAAC,2CAAmB,IAClB,QAAQ,EAAE,QAAQ,EAClB,MAAM,EAAE,MAAM,EACd,WAAW,EAAE,WAAW;QAExB,8BAAC,+CAAqB,IAAC,UAAU,EAAE,UAAU,EAAE,KAAK,EAAE,KAAK;YACzD,8BAAC,2CAAmB;gBAClB,8BAAC,iDAAsB;oBACrB,8BAAC,sBAAsB,QAAE,QAAQ,CAA0B,CACpC,CACL,CACA,CACJ,CACvB,CAAC;AACJ,CAAC,CAAC;AAvBW,QAAA,aAAa,iBAuBxB;AAKW,QAAA,YAAY,GAAG,IAAA,qBAAa,EACvC,EAAuB,CACxB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,sBAAsB,GAAgC,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE;IAC3E,MAAM,OAAO,GAAG,IAAA,sCAAc,GAAE,CAAC;IACjC,MAAM,MAAM,GAAG,IAAA,sCAAc,GAAE,CAAC;IAChC,MAAM,WAAW,GAAG,IAAA,2CAAmB,GAAE,CAAC;IAC1C,MAAM,iBAAiB,GAAG,IAAA,4CAAiB,GAAE,CAAC;IAE9C,OAAO,CACL,8BAAC,oBAAY,CAAC,QAAQ,IACpB,KAAK,EAAE;YACL,MAAM;YACN,WAAW;YACX,GAAG,iBAAiB;YACpB,GAAG,OAAO;SACX,IAEA,QAAQ,CACa,CACzB,CAAC;AACJ,CAAC,CAAC;AAEF;;;;;;GAMG;AACI,MAAM,QAAQ,GAAG,GAAG,EAAE;IAC3B,OAAO,IAAA,kBAAU,EAAC,oBAAY,CAAC,CAAC;AAClC,CAAC,CAAC;AAFW,QAAA,QAAQ,YAEnB","sourcesContent":["\"use client\";\nimport React, { PropsWithChildren, createContext, useContext } from \"react\";\nimport {\n  TamboClientContextProps,\n  TamboClientProvider,\n  TamboClientProviderProps,\n  useTamboClient,\n  useTamboQueryClient,\n} from \"./tambo-client-provider\";\nimport {\n  TamboComponentContextProps,\n  TamboComponentProvider,\n  useTamboComponent,\n} from \"./tambo-component-provider\";\nimport {\n  TamboRegistryProvider,\n  TamboRegistryProviderProps,\n} from \"./tambo-registry-provider\";\nimport {\n  TamboThreadContextProps,\n  TamboThreadProvider,\n  useTamboThread,\n} from \"./tambo-thread-provider\";\n\n/**\n * The TamboProvider gives full access to the whole Tambo API. This includes the\n * TamboAI client, the component registry, and the current thread context.\n * @param props - The props for the TamboProvider\n * @param props.children - The children to wrap\n * @param props.tamboUrl - The URL of the Tambo API\n * @param props.apiKey - The API key for the Tambo API\n * @param props.components - The components to register\n * @param props.environment - The environment to use for the Tambo API\n * @param props.tools - The tools to register\n * @returns The TamboProvider component\n */\nexport const TamboProvider: React.FC<\n  PropsWithChildren<TamboClientProviderProps & TamboRegistryProviderProps>\n> = ({ children, tamboUrl, apiKey, components, environment, tools }) => {\n  // Should only be used in browser\n  if (typeof window === \"undefined\") {\n    console.error(\"TamboProvider must be used within a browser\");\n  }\n\n  return (\n    <TamboClientProvider\n      tamboUrl={tamboUrl}\n      apiKey={apiKey}\n      environment={environment}\n    >\n      <TamboRegistryProvider components={components} tools={tools}>\n        <TamboThreadProvider>\n          <TamboComponentProvider>\n            <TamboCompositeProvider>{children}</TamboCompositeProvider>\n          </TamboComponentProvider>\n        </TamboThreadProvider>\n      </TamboRegistryProvider>\n    </TamboClientProvider>\n  );\n};\nexport type TamboContextProps = TamboClientContextProps &\n  TamboThreadContextProps &\n  TamboComponentContextProps;\n\nexport const TamboContext = createContext<TamboContextProps>(\n  {} as TamboContextProps,\n);\n\n/**\n * TamboCompositeProvider is a provider that combines the TamboClient,\n * TamboThread, and TamboComponent providers\n * @param props - The props for the TamboCompositeProvider\n * @param props.children - The children to wrap\n * @returns The wrapped component\n */\nconst TamboCompositeProvider: React.FC<PropsWithChildren> = ({ children }) => {\n  const threads = useTamboThread();\n  const client = useTamboClient();\n  const queryClient = useTamboQueryClient();\n  const componentRegistry = useTamboComponent();\n\n  return (\n    <TamboContext.Provider\n      value={{\n        client,\n        queryClient,\n        ...componentRegistry,\n        ...threads,\n      }}\n    >\n      {children}\n    </TamboContext.Provider>\n  );\n};\n\n/**\n * The useTambo hook provides access to the Tambo API. This is the primary entrypoint\n * for the Tambo React SDK.\n *\n * This includes the TamboAI client, the component registry, and the current thread context.\n * @returns The Tambo API\n */\nexport const useTambo = () => {\n  return useContext(TamboContext);\n};\n"]}

package/dist/providers/tambo-registry-provider.d.ts

@@ -12,6 +12,8 @@ export interface TamboRegistryContext {
 export interface TamboRegistryProviderProps {
     /** The components to register */
     components?: TamboComponent[];
+    /** The tools to register */
+    tools?: TamboTool[];
 }
 /**
  * The TamboRegistryProvider is a React provider that provides a component
@@ -19,6 +21,7 @@ export interface TamboRegistryProviderProps {
  * @param props - The props for the TamboRegistryProvider
  * @param props.children - The children to wrap
  * @param props.components - The components to register
+ * @param props.tools - The tools to register
  * @returns The TamboRegistryProvider component
  */
 export declare const TamboRegistryProvider: React.FC<PropsWithChildren<TamboRegistryProviderProps>>;

package/dist/providers/tambo-registry-provider.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"tambo-registry-provider.d.ts","sourceRoot":"","sources":["../../src/providers/tambo-registry-provider.tsx"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,EAEZ,iBAAiB,EAKlB,MAAM,OAAO,CAAC;AAGf,OAAO,EACL,iBAAiB,EACjB,cAAc,EACd,SAAS,EACV,MAAM,6BAA6B,CAAC;AAErC,MAAM,WAAW,oBAAoB;IACnC,aAAa,EAAE,iBAAiB,CAAC;IACjC,YAAY,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IACxC,yBAAyB,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;IACpD,iBAAiB,EAAE,CAAC,OAAO,EAAE,cAAc,KAAK,IAAI,CAAC;IACrD,YAAY,EAAE,CAAC,IAAI,EAAE,SAAS,KAAK,IAAI,CAAC;IACxC,aAAa,EAAE,CAAC,KAAK,EAAE,SAAS,EAAE,KAAK,IAAI,CAAC;IAC5C,kBAAkB,EAAE,CAAC,aAAa,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,KAAK,IAAI,CAAC;CACtE;AAwBD,MAAM,WAAW,0BAA0B;IACzC,iCAAiC;IACjC,UAAU,CAAC,EAAE,cAAc,EAAE,CAAC;CAC/B;AAED;;;;;;;GAOG;AACH,eAAO,MAAM,qBAAqB,EAAE,KAAK,CAAC,EAAE,CAC1C,iBAAiB,CAAC,0BAA0B,CAAC,CA2H9C,CAAC;AAEF;;;;GAIG;AACH,eAAO,MAAM,gBAAgB,4BAE5B,CAAC"}
+{"version":3,"file":"tambo-registry-provider.d.ts","sourceRoot":"","sources":["../../src/providers/tambo-registry-provider.tsx"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,EAEZ,iBAAiB,EAKlB,MAAM,OAAO,CAAC;AAGf,OAAO,EACL,iBAAiB,EACjB,cAAc,EACd,SAAS,EACV,MAAM,6BAA6B,CAAC;AAErC,MAAM,WAAW,oBAAoB;IACnC,aAAa,EAAE,iBAAiB,CAAC;IACjC,YAAY,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IACxC,yBAAyB,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;IACpD,iBAAiB,EAAE,CAAC,OAAO,EAAE,cAAc,KAAK,IAAI,CAAC;IACrD,YAAY,EAAE,CAAC,IAAI,EAAE,SAAS,KAAK,IAAI,CAAC;IACxC,aAAa,EAAE,CAAC,KAAK,EAAE,SAAS,EAAE,KAAK,IAAI,CAAC;IAC5C,kBAAkB,EAAE,CAAC,aAAa,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,KAAK,IAAI,CAAC;CACtE;AAwBD,MAAM,WAAW,0BAA0B;IACzC,iCAAiC;IACjC,UAAU,CAAC,EAAE,cAAc,EAAE,CAAC;IAC9B,4BAA4B;IAC5B,KAAK,CAAC,EAAE,SAAS,EAAE,CAAC;CACrB;AAED;;;;;;;;GAQG;AACH,eAAO,MAAM,qBAAqB,EAAE,KAAK,CAAC,EAAE,CAC1C,iBAAiB,CAAC,0BAA0B,CAAC,CAiI9C,CAAC;AAEF;;;;GAIG;AACH,eAAO,MAAM,gBAAgB,4BAE5B,CAAC"}

package/dist/providers/tambo-registry-provider.js

@@ -68,9 +68,10 @@ const TamboRegistryContext = (0, react_1.createContext)({
  * @param props - The props for the TamboRegistryProvider
  * @param props.children - The children to wrap
  * @param props.components - The components to register
+ * @param props.tools - The tools to register
  * @returns The TamboRegistryProvider component
  */
-const TamboRegistryProvider = ({ children, components: userComponents }) => {
+const TamboRegistryProvider = ({ children, components: userComponents, tools: userTools }) => {
     const [componentList, setComponentList] = (0, react_1.useState)({});
     const [toolRegistry, setToolRegistry] = (0, react_1.useState)({});
     const [componentToolAssociations, setComponentToolAssociations] = (0, react_1.useState)({});
@@ -140,6 +141,11 @@ const TamboRegistryProvider = ({ children, components: userComponents }) => {
             });
         }
     }, [registerComponent, userComponents]);
+    (0, react_1.useEffect)(() => {
+        if (userTools) {
+            registerTools(userTools, false);
+        }
+    }, [registerTools, userTools]);
     const value = {
         componentList,
         toolRegistry,

package/dist/providers/tambo-registry-provider.js.map

@@ -1 +1 @@
-{"version":3,"file":"tambo-registry-provider.js","sourceRoot":"","sources":["../../src/providers/tambo-registry-provider.tsx"],"names":[],"mappings":";AAAA,YAAY,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AACb,+CAOe;AACf,6BAAgC;AAChC,4EAAiD;AAiBjD,MAAM,oBAAoB,GAAG,IAAA,qBAAa,EAAuB;IAC/D,aAAa,EAAE,EAAE;IACjB,YAAY,EAAE,EAAE;IAChB,yBAAyB,EAAE,EAAE;IAC7B;;OAEG;IACH,iBAAiB,EAAE,GAAG,EAAE,GAAE,CAAC;IAC3B;;OAEG;IACH,YAAY,EAAE,GAAG,EAAE,GAAE,CAAC;IACtB;;OAEG;IACH,aAAa,EAAE,GAAG,EAAE,GAAE,CAAC;IACvB;;OAEG;IACH,kBAAkB,EAAE,GAAG,EAAE,GAAE,CAAC;CAC7B,CAAC,CAAC;AAOH;;;;;;;GAOG;AACI,MAAM,qBAAqB,GAE9B,CAAC,EAAE,QAAQ,EAAE,UAAU,EAAE,cAAc,EAAE,EAAE,EAAE;IAC/C,MAAM,CAAC,aAAa,EAAE,gBAAgB,CAAC,GAAG,IAAA,gBAAQ,EAAoB,EAAE,CAAC,CAAC;IAC1E,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,IAAA,gBAAQ,EAC9C,EAAE,CACH,CAAC;IACF,MAAM,CAAC,yBAAyB,EAAE,4BAA4B,CAAC,GAAG,IAAA,gBAAQ,EAExE,EAAE,CAAC,CAAC;IAEN,MAAM,YAAY,GAAG,IAAA,mBAAW,EAC9B,CAAC,IAAe,EAAE,eAAe,GAAG,IAAI,EAAE,EAAE;QAC1C,eAAe,CAAC,CAAC,IAAI,EAAE,EAAE;YACvB,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,eAAe,EAAE,CAAC;gBACvC,OAAO,CAAC,IAAI,CAAC,oBAAoB,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;YAChD,CAAC;YACD,OAAO;gBACL,GAAG,IAAI;gBACP,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI;aAClB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,EACD,EAAE,CACH,CAAC;IAEF,MAAM,aAAa,GAAG,IAAA,mBAAW,EAC/B,CAAC,KAAkB,EAAE,eAAe,GAAG,IAAI,EAAE,EAAE;QAC7C,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC,CAAC;IAC/D,CAAC,EACD,CAAC,YAAY,CAAC,CACf,CAAC;IAEF,MAAM,kBAAkB,GAAG,IAAA,mBAAW,EACpC,CAAC,aAAqB,EAAE,IAAe,EAAE,EAAE;QACzC,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,EAAE,CAAC;YAClC,MAAM,IAAI,KAAK,CAAC,aAAa,aAAa,wBAAwB,CAAC,CAAC;QACtE,CAAC;QACD,4BAA4B,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACtC,GAAG,IAAI;YACP,CAAC,aAAa,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC;SAC7D,CAAC,CAAC,CAAC;IACN,CAAC,EACD,CAAC,aAAa,CAAC,CAChB,CAAC;IAEF,MAAM,iBAAiB,GAAG,IAAA,mBAAW,EACnC,CAAC,OAAuB,EAAE,eAAe,GAAG,IAAI,EAAE,EAAE;QAClD,MAAM,EACJ,IAAI,EACJ,WAAW,EACX,SAAS,EACT,WAAW,EACX,eAAe,EACf,gBAAgB,EAChB,eAAe,GAChB,GAAG,OAAO,CAAC;QAEZ,0DAA0D;QAC1D,IAAI,CAAC,WAAW,IAAI,CAAC,eAAe,EAAE,CAAC;YACrC,MAAM,IAAI,KAAK,CACb,aAAa,IAAI,wEAAwE,CAC1F,CAAC;QACJ,CAAC;QAED,sDAAsD;QACtD,IAAI,WAAW,IAAI,eAAe,EAAE,CAAC;YACnC,MAAM,IAAI,KAAK,CACb,aAAa,IAAI,0GAA0G,CAC5H,CAAC;QACJ,CAAC;QAED,kDAAkD;QAClD,MAAM,KAAK,GAAG,kBAAkB,CAAC,eAAe,EAAE,WAAW,EAAE,IAAI,CAAC,CAAC;QAErE,gBAAgB,CAAC,CAAC,IAAI,EAAE,EAAE;YACxB,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,eAAe,EAAE,CAAC;gBAClC,OAAO,CAAC,IAAI,CAAC,yBAAyB,IAAI,EAAE,CAAC,CAAC;YAChD,CAAC;YACD,OAAO;gBACL,GAAG,IAAI;gBACP,CAAC,IAAI,CAAC,EAAE;oBACN,SAAS;oBACT,gBAAgB;oBAChB,IAAI;oBACJ,WAAW;oBACX,KAAK;oBACL,YAAY,EAAE,EAAE;iBACjB;aACF,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,IAAI,eAAe,EAAE,CAAC;YACpB,aAAa,CAAC,eAAe,CAAC,CAAC;YAC/B,4BAA4B,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBACtC,GAAG,IAAI;gBACP,CAAC,IAAI,CAAC,EAAE,eAAe,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;aACjD,CAAC,CAAC,CAAC;QACN,CAAC;IACH,CAAC,EACD,CAAC,aAAa,CAAC,CAChB,CAAC;IACF,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,IAAI,cAAc,EAAE,CAAC;YACnB,cAAc,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;gBACnC,iBAAiB,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;YACtC,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC,EAAE,CAAC,iBAAiB,EAAE,cAAc,CAAC,CAAC,CAAC;IAExC,MAAM,KAAK,GAAG;QACZ,aAAa;QACb,YAAY;QACZ,yBAAyB;QACzB,iBAAiB;QACjB,YAAY;QACZ,aAAa;QACb,kBAAkB;KACnB,CAAC;IAEF,OAAO,CACL,8BAAC,oBAAoB,CAAC,QAAQ,IAAC,KAAK,EAAE,KAAK,IACxC,QAAQ,CACqB,CACjC,CAAC;AACJ,CAAC,CAAC;AA5HW,QAAA,qBAAqB,yBA4HhC;AAEF;;;;GAIG;AACI,MAAM,gBAAgB,GAAG,GAAG,EAAE;IACnC,OAAO,IAAA,kBAAU,EAAC,oBAAoB,CAAC,CAAC;AAC1C,CAAC,CAAC;AAFW,QAAA,gBAAgB,oBAE3B;AACF,SAAS,kBAAkB,CACzB,eAAoB,EACpB,WAAgB,EAChB,IAAY;IAEZ,IAAI,eAAe,EAAE,CAAC;QACpB,OAAO,CAAC,IAAI,CAAC,yDAAyD,CAAC,CAAC;QACxE,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,IAAI,WAAW,CAAC,WAAW,CAAC,EAAE,CAAC;QAC7B,IAAI,CAAC;YACH,OAAO,IAAA,4BAAe,EAAC,WAAW,CAAC,CAAC;QACtC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CACX,oBAAoB,IAAI,+BAA+B,EACvD,KAAK,CACN,CAAC;QACJ,CAAC;IACH,CAAC;IACD,qFAAqF;IACrF,IAAI,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC;QAC9B,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,4BAA4B,IAAI,EAAE,CAAC,CAAC;AACtD,CAAC;AAED;;;;;GAKG;AACH,SAAS,YAAY,CAAC,WAAgB;IACpC,OAAO,CACL,WAAW;QACX,OAAO,WAAW,KAAK,QAAQ;QAC/B,WAAW,CAAC,IAAI,KAAK,QAAQ;QAC7B,WAAW,CAAC,UAAU,CACvB,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,SAAS,WAAW,CAAC,GAAY;IAC/B,IAAI,GAAG,YAAY,eAAS,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC;IACd,CAAC;IACD,6CAA6C;IAC7C,OAAO,CACL,OAAO,GAAG,KAAK,QAAQ;QACvB,GAAG,KAAK,IAAI;QACZ,OAAQ,GAAW,CAAC,SAAS,KAAK,UAAU;QAC5C,OAAQ,GAAW,CAAC,IAAI,KAAK,QAAQ,CACtC,CAAC;AACJ,CAAC","sourcesContent":["\"use client\";\nimport React, {\n  createContext,\n  PropsWithChildren,\n  useCallback,\n  useContext,\n  useEffect,\n  useState,\n} from \"react\";\nimport { ZodSchema } from \"zod\";\nimport zodToJsonSchema from \"zod-to-json-schema\";\nimport {\n  ComponentRegistry,\n  TamboComponent,\n  TamboTool,\n} from \"../model/component-metadata\";\n\nexport interface TamboRegistryContext {\n  componentList: ComponentRegistry;\n  toolRegistry: Record<string, TamboTool>;\n  componentToolAssociations: Record<string, string[]>;\n  registerComponent: (options: TamboComponent) => void;\n  registerTool: (tool: TamboTool) => void;\n  registerTools: (tools: TamboTool[]) => void;\n  addToolAssociation: (componentName: string, tool: TamboTool) => void;\n}\n\nconst TamboRegistryContext = createContext<TamboRegistryContext>({\n  componentList: {},\n  toolRegistry: {},\n  componentToolAssociations: {},\n  /**\n   *\n   */\n  registerComponent: () => {},\n  /**\n   *\n   */\n  registerTool: () => {},\n  /**\n   *\n   */\n  registerTools: () => {},\n  /**\n   *\n   */\n  addToolAssociation: () => {},\n});\n\nexport interface TamboRegistryProviderProps {\n  /** The components to register */\n  components?: TamboComponent[];\n}\n\n/**\n * The TamboRegistryProvider is a React provider that provides a component\n * registry to the descendants of the provider.\n * @param props - The props for the TamboRegistryProvider\n * @param props.children - The children to wrap\n * @param props.components - The components to register\n * @returns The TamboRegistryProvider component\n */\nexport const TamboRegistryProvider: React.FC<\n  PropsWithChildren<TamboRegistryProviderProps>\n> = ({ children, components: userComponents }) => {\n  const [componentList, setComponentList] = useState<ComponentRegistry>({});\n  const [toolRegistry, setToolRegistry] = useState<Record<string, TamboTool>>(\n    {},\n  );\n  const [componentToolAssociations, setComponentToolAssociations] = useState<\n    Record<string, string[]>\n  >({});\n\n  const registerTool = useCallback(\n    (tool: TamboTool, warnOnOverwrite = true) => {\n      setToolRegistry((prev) => {\n        if (prev[tool.name] && warnOnOverwrite) {\n          console.warn(`Overwriting tool ${tool.name}`);\n        }\n        return {\n          ...prev,\n          [tool.name]: tool,\n        };\n      });\n    },\n    [],\n  );\n\n  const registerTools = useCallback(\n    (tools: TamboTool[], warnOnOverwrite = true) => {\n      tools.forEach((tool) => registerTool(tool, warnOnOverwrite));\n    },\n    [registerTool],\n  );\n\n  const addToolAssociation = useCallback(\n    (componentName: string, tool: TamboTool) => {\n      if (!componentList[componentName]) {\n        throw new Error(`Component ${componentName} not found in registry`);\n      }\n      setComponentToolAssociations((prev) => ({\n        ...prev,\n        [componentName]: [...(prev[componentName] || []), tool.name],\n      }));\n    },\n    [componentList],\n  );\n\n  const registerComponent = useCallback(\n    (options: TamboComponent, warnOnOverwrite = true) => {\n      const {\n        name,\n        description,\n        component,\n        propsSchema,\n        propsDefinition,\n        loadingComponent,\n        associatedTools,\n      } = options;\n\n      // Validate that at least one props definition is provided\n      if (!propsSchema && !propsDefinition) {\n        throw new Error(\n          `Component ${name} must have either propsSchema (recommended) or propsDefinition defined`,\n        );\n      }\n\n      // Validate that only one props definition is provided\n      if (propsSchema && propsDefinition) {\n        throw new Error(\n          `Component ${name} cannot have both propsSchema and propsDefinition defined. Use only one. We recommend using propsSchema.`,\n        );\n      }\n\n      // Convert propsSchema to JSON Schema if it exists\n      const props = getSerializedProps(propsDefinition, propsSchema, name);\n\n      setComponentList((prev) => {\n        if (prev[name] && warnOnOverwrite) {\n          console.warn(`overwriting component ${name}`);\n        }\n        return {\n          ...prev,\n          [name]: {\n            component,\n            loadingComponent,\n            name,\n            description,\n            props,\n            contextTools: [],\n          },\n        };\n      });\n      if (associatedTools) {\n        registerTools(associatedTools);\n        setComponentToolAssociations((prev) => ({\n          ...prev,\n          [name]: associatedTools.map((tool) => tool.name),\n        }));\n      }\n    },\n    [registerTools],\n  );\n  useEffect(() => {\n    if (userComponents) {\n      userComponents.forEach((component) => {\n        registerComponent(component, false);\n      });\n    }\n  }, [registerComponent, userComponents]);\n\n  const value = {\n    componentList,\n    toolRegistry,\n    componentToolAssociations,\n    registerComponent,\n    registerTool,\n    registerTools,\n    addToolAssociation,\n  };\n\n  return (\n    <TamboRegistryContext.Provider value={value}>\n      {children}\n    </TamboRegistryContext.Provider>\n  );\n};\n\n/**\n * The useTamboRegistry hook provides access to the component registry\n * to the descendants of the TamboRegistryProvider.\n * @returns The component registry\n */\nexport const useTamboRegistry = () => {\n  return useContext(TamboRegistryContext);\n};\nfunction getSerializedProps(\n  propsDefinition: any,\n  propsSchema: any,\n  name: string,\n) {\n  if (propsDefinition) {\n    console.warn(`propsDefinition is deprecated. Use propsSchema instead.`);\n    return propsDefinition;\n  }\n\n  if (isZodSchema(propsSchema)) {\n    try {\n      return zodToJsonSchema(propsSchema);\n    } catch (error) {\n      console.error(\n        `Error converting ${name} props schema to JSON Schema:`,\n        error,\n      );\n    }\n  }\n  // try to roughly detect JSONSchema, should always be an object with a properties key\n  if (isJSONSchema(propsSchema)) {\n    return propsSchema;\n  }\n\n  throw new Error(`Invalid props schema for ${name}`);\n}\n\n/**\n * Checks if the propsSchema is a JSON Schema. This is a rough check, and the\n * server will provide the definitive check.\n * @param propsSchema - The props schema to check\n * @returns True if the props schema is a JSON Schema, false otherwise\n */\nfunction isJSONSchema(propsSchema: any) {\n  return (\n    propsSchema &&\n    typeof propsSchema === \"object\" &&\n    propsSchema.type === \"object\" &&\n    propsSchema.properties\n  );\n}\n\n/**\n * Since we require a certain zod version, we need to check if the object is a ZodSchema\n * @param obj - The object to check\n * @returns True if the object is a ZodSchema, false otherwise\n */\nfunction isZodSchema(obj: unknown): obj is ZodSchema {\n  if (obj instanceof ZodSchema) {\n    return true;\n  }\n  // try to detect if the object is a ZodSchema\n  return (\n    typeof obj === \"object\" &&\n    obj !== null &&\n    typeof (obj as any).safeParse === \"function\" &&\n    typeof (obj as any)._def === \"object\"\n  );\n}\n"]}
+{"version":3,"file":"tambo-registry-provider.js","sourceRoot":"","sources":["../../src/providers/tambo-registry-provider.tsx"],"names":[],"mappings":";AAAA,YAAY,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AACb,+CAOe;AACf,6BAAgC;AAChC,4EAAiD;AAiBjD,MAAM,oBAAoB,GAAG,IAAA,qBAAa,EAAuB;IAC/D,aAAa,EAAE,EAAE;IACjB,YAAY,EAAE,EAAE;IAChB,yBAAyB,EAAE,EAAE;IAC7B;;OAEG;IACH,iBAAiB,EAAE,GAAG,EAAE,GAAE,CAAC;IAC3B;;OAEG;IACH,YAAY,EAAE,GAAG,EAAE,GAAE,CAAC;IACtB;;OAEG;IACH,aAAa,EAAE,GAAG,EAAE,GAAE,CAAC;IACvB;;OAEG;IACH,kBAAkB,EAAE,GAAG,EAAE,GAAE,CAAC;CAC7B,CAAC,CAAC;AASH;;;;;;;;GAQG;AACI,MAAM,qBAAqB,GAE9B,CAAC,EAAE,QAAQ,EAAE,UAAU,EAAE,cAAc,EAAE,KAAK,EAAE,SAAS,EAAE,EAAE,EAAE;IACjE,MAAM,CAAC,aAAa,EAAE,gBAAgB,CAAC,GAAG,IAAA,gBAAQ,EAAoB,EAAE,CAAC,CAAC;IAC1E,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,IAAA,gBAAQ,EAC9C,EAAE,CACH,CAAC;IACF,MAAM,CAAC,yBAAyB,EAAE,4BAA4B,CAAC,GAAG,IAAA,gBAAQ,EAExE,EAAE,CAAC,CAAC;IAEN,MAAM,YAAY,GAAG,IAAA,mBAAW,EAC9B,CAAC,IAAe,EAAE,eAAe,GAAG,IAAI,EAAE,EAAE;QAC1C,eAAe,CAAC,CAAC,IAAI,EAAE,EAAE;YACvB,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,eAAe,EAAE,CAAC;gBACvC,OAAO,CAAC,IAAI,CAAC,oBAAoB,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;YAChD,CAAC;YACD,OAAO;gBACL,GAAG,IAAI;gBACP,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI;aAClB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,EACD,EAAE,CACH,CAAC;IAEF,MAAM,aAAa,GAAG,IAAA,mBAAW,EAC/B,CAAC,KAAkB,EAAE,eAAe,GAAG,IAAI,EAAE,EAAE;QAC7C,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC,CAAC;IAC/D,CAAC,EACD,CAAC,YAAY,CAAC,CACf,CAAC;IAEF,MAAM,kBAAkB,GAAG,IAAA,mBAAW,EACpC,CAAC,aAAqB,EAAE,IAAe,EAAE,EAAE;QACzC,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,EAAE,CAAC;YAClC,MAAM,IAAI,KAAK,CAAC,aAAa,aAAa,wBAAwB,CAAC,CAAC;QACtE,CAAC;QACD,4BAA4B,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACtC,GAAG,IAAI;YACP,CAAC,aAAa,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC;SAC7D,CAAC,CAAC,CAAC;IACN,CAAC,EACD,CAAC,aAAa,CAAC,CAChB,CAAC;IAEF,MAAM,iBAAiB,GAAG,IAAA,mBAAW,EACnC,CAAC,OAAuB,EAAE,eAAe,GAAG,IAAI,EAAE,EAAE;QAClD,MAAM,EACJ,IAAI,EACJ,WAAW,EACX,SAAS,EACT,WAAW,EACX,eAAe,EACf,gBAAgB,EAChB,eAAe,GAChB,GAAG,OAAO,CAAC;QAEZ,0DAA0D;QAC1D,IAAI,CAAC,WAAW,IAAI,CAAC,eAAe,EAAE,CAAC;YACrC,MAAM,IAAI,KAAK,CACb,aAAa,IAAI,wEAAwE,CAC1F,CAAC;QACJ,CAAC;QAED,sDAAsD;QACtD,IAAI,WAAW,IAAI,eAAe,EAAE,CAAC;YACnC,MAAM,IAAI,KAAK,CACb,aAAa,IAAI,0GAA0G,CAC5H,CAAC;QACJ,CAAC;QAED,kDAAkD;QAClD,MAAM,KAAK,GAAG,kBAAkB,CAAC,eAAe,EAAE,WAAW,EAAE,IAAI,CAAC,CAAC;QAErE,gBAAgB,CAAC,CAAC,IAAI,EAAE,EAAE;YACxB,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,eAAe,EAAE,CAAC;gBAClC,OAAO,CAAC,IAAI,CAAC,yBAAyB,IAAI,EAAE,CAAC,CAAC;YAChD,CAAC;YACD,OAAO;gBACL,GAAG,IAAI;gBACP,CAAC,IAAI,CAAC,EAAE;oBACN,SAAS;oBACT,gBAAgB;oBAChB,IAAI;oBACJ,WAAW;oBACX,KAAK;oBACL,YAAY,EAAE,EAAE;iBACjB;aACF,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,IAAI,eAAe,EAAE,CAAC;YACpB,aAAa,CAAC,eAAe,CAAC,CAAC;YAC/B,4BAA4B,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBACtC,GAAG,IAAI;gBACP,CAAC,IAAI,CAAC,EAAE,eAAe,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;aACjD,CAAC,CAAC,CAAC;QACN,CAAC;IACH,CAAC,EACD,CAAC,aAAa,CAAC,CAChB,CAAC;IACF,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,IAAI,cAAc,EAAE,CAAC;YACnB,cAAc,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;gBACnC,iBAAiB,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;YACtC,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC,EAAE,CAAC,iBAAiB,EAAE,cAAc,CAAC,CAAC,CAAC;IAExC,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,IAAI,SAAS,EAAE,CAAC;YACd,aAAa,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QAClC,CAAC;IACH,CAAC,EAAE,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC,CAAC;IAE/B,MAAM,KAAK,GAAG;QACZ,aAAa;QACb,YAAY;QACZ,yBAAyB;QACzB,iBAAiB;QACjB,YAAY;QACZ,aAAa;QACb,kBAAkB;KACnB,CAAC;IAEF,OAAO,CACL,8BAAC,oBAAoB,CAAC,QAAQ,IAAC,KAAK,EAAE,KAAK,IACxC,QAAQ,CACqB,CACjC,CAAC;AACJ,CAAC,CAAC;AAlIW,QAAA,qBAAqB,yBAkIhC;AAEF;;;;GAIG;AACI,MAAM,gBAAgB,GAAG,GAAG,EAAE;IACnC,OAAO,IAAA,kBAAU,EAAC,oBAAoB,CAAC,CAAC;AAC1C,CAAC,CAAC;AAFW,QAAA,gBAAgB,oBAE3B;AACF,SAAS,kBAAkB,CACzB,eAAoB,EACpB,WAAgB,EAChB,IAAY;IAEZ,IAAI,eAAe,EAAE,CAAC;QACpB,OAAO,CAAC,IAAI,CAAC,yDAAyD,CAAC,CAAC;QACxE,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,IAAI,WAAW,CAAC,WAAW,CAAC,EAAE,CAAC;QAC7B,IAAI,CAAC;YACH,OAAO,IAAA,4BAAe,EAAC,WAAW,CAAC,CAAC;QACtC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CACX,oBAAoB,IAAI,+BAA+B,EACvD,KAAK,CACN,CAAC;QACJ,CAAC;IACH,CAAC;IACD,qFAAqF;IACrF,IAAI,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC;QAC9B,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,4BAA4B,IAAI,EAAE,CAAC,CAAC;AACtD,CAAC;AAED;;;;;GAKG;AACH,SAAS,YAAY,CAAC,WAAgB;IACpC,OAAO,CACL,WAAW;QACX,OAAO,WAAW,KAAK,QAAQ;QAC/B,WAAW,CAAC,IAAI,KAAK,QAAQ;QAC7B,WAAW,CAAC,UAAU,CACvB,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,SAAS,WAAW,CAAC,GAAY;IAC/B,IAAI,GAAG,YAAY,eAAS,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC;IACd,CAAC;IACD,6CAA6C;IAC7C,OAAO,CACL,OAAO,GAAG,KAAK,QAAQ;QACvB,GAAG,KAAK,IAAI;QACZ,OAAQ,GAAW,CAAC,SAAS,KAAK,UAAU;QAC5C,OAAQ,GAAW,CAAC,IAAI,KAAK,QAAQ,CACtC,CAAC;AACJ,CAAC","sourcesContent":["\"use client\";\nimport React, {\n  createContext,\n  PropsWithChildren,\n  useCallback,\n  useContext,\n  useEffect,\n  useState,\n} from \"react\";\nimport { ZodSchema } from \"zod\";\nimport zodToJsonSchema from \"zod-to-json-schema\";\nimport {\n  ComponentRegistry,\n  TamboComponent,\n  TamboTool,\n} from \"../model/component-metadata\";\n\nexport interface TamboRegistryContext {\n  componentList: ComponentRegistry;\n  toolRegistry: Record<string, TamboTool>;\n  componentToolAssociations: Record<string, string[]>;\n  registerComponent: (options: TamboComponent) => void;\n  registerTool: (tool: TamboTool) => void;\n  registerTools: (tools: TamboTool[]) => void;\n  addToolAssociation: (componentName: string, tool: TamboTool) => void;\n}\n\nconst TamboRegistryContext = createContext<TamboRegistryContext>({\n  componentList: {},\n  toolRegistry: {},\n  componentToolAssociations: {},\n  /**\n   *\n   */\n  registerComponent: () => {},\n  /**\n   *\n   */\n  registerTool: () => {},\n  /**\n   *\n   */\n  registerTools: () => {},\n  /**\n   *\n   */\n  addToolAssociation: () => {},\n});\n\nexport interface TamboRegistryProviderProps {\n  /** The components to register */\n  components?: TamboComponent[];\n  /** The tools to register */\n  tools?: TamboTool[];\n}\n\n/**\n * The TamboRegistryProvider is a React provider that provides a component\n * registry to the descendants of the provider.\n * @param props - The props for the TamboRegistryProvider\n * @param props.children - The children to wrap\n * @param props.components - The components to register\n * @param props.tools - The tools to register\n * @returns The TamboRegistryProvider component\n */\nexport const TamboRegistryProvider: React.FC<\n  PropsWithChildren<TamboRegistryProviderProps>\n> = ({ children, components: userComponents, tools: userTools }) => {\n  const [componentList, setComponentList] = useState<ComponentRegistry>({});\n  const [toolRegistry, setToolRegistry] = useState<Record<string, TamboTool>>(\n    {},\n  );\n  const [componentToolAssociations, setComponentToolAssociations] = useState<\n    Record<string, string[]>\n  >({});\n\n  const registerTool = useCallback(\n    (tool: TamboTool, warnOnOverwrite = true) => {\n      setToolRegistry((prev) => {\n        if (prev[tool.name] && warnOnOverwrite) {\n          console.warn(`Overwriting tool ${tool.name}`);\n        }\n        return {\n          ...prev,\n          [tool.name]: tool,\n        };\n      });\n    },\n    [],\n  );\n\n  const registerTools = useCallback(\n    (tools: TamboTool[], warnOnOverwrite = true) => {\n      tools.forEach((tool) => registerTool(tool, warnOnOverwrite));\n    },\n    [registerTool],\n  );\n\n  const addToolAssociation = useCallback(\n    (componentName: string, tool: TamboTool) => {\n      if (!componentList[componentName]) {\n        throw new Error(`Component ${componentName} not found in registry`);\n      }\n      setComponentToolAssociations((prev) => ({\n        ...prev,\n        [componentName]: [...(prev[componentName] || []), tool.name],\n      }));\n    },\n    [componentList],\n  );\n\n  const registerComponent = useCallback(\n    (options: TamboComponent, warnOnOverwrite = true) => {\n      const {\n        name,\n        description,\n        component,\n        propsSchema,\n        propsDefinition,\n        loadingComponent,\n        associatedTools,\n      } = options;\n\n      // Validate that at least one props definition is provided\n      if (!propsSchema && !propsDefinition) {\n        throw new Error(\n          `Component ${name} must have either propsSchema (recommended) or propsDefinition defined`,\n        );\n      }\n\n      // Validate that only one props definition is provided\n      if (propsSchema && propsDefinition) {\n        throw new Error(\n          `Component ${name} cannot have both propsSchema and propsDefinition defined. Use only one. We recommend using propsSchema.`,\n        );\n      }\n\n      // Convert propsSchema to JSON Schema if it exists\n      const props = getSerializedProps(propsDefinition, propsSchema, name);\n\n      setComponentList((prev) => {\n        if (prev[name] && warnOnOverwrite) {\n          console.warn(`overwriting component ${name}`);\n        }\n        return {\n          ...prev,\n          [name]: {\n            component,\n            loadingComponent,\n            name,\n            description,\n            props,\n            contextTools: [],\n          },\n        };\n      });\n      if (associatedTools) {\n        registerTools(associatedTools);\n        setComponentToolAssociations((prev) => ({\n          ...prev,\n          [name]: associatedTools.map((tool) => tool.name),\n        }));\n      }\n    },\n    [registerTools],\n  );\n  useEffect(() => {\n    if (userComponents) {\n      userComponents.forEach((component) => {\n        registerComponent(component, false);\n      });\n    }\n  }, [registerComponent, userComponents]);\n\n  useEffect(() => {\n    if (userTools) {\n      registerTools(userTools, false);\n    }\n  }, [registerTools, userTools]);\n\n  const value = {\n    componentList,\n    toolRegistry,\n    componentToolAssociations,\n    registerComponent,\n    registerTool,\n    registerTools,\n    addToolAssociation,\n  };\n\n  return (\n    <TamboRegistryContext.Provider value={value}>\n      {children}\n    </TamboRegistryContext.Provider>\n  );\n};\n\n/**\n * The useTamboRegistry hook provides access to the component registry\n * to the descendants of the TamboRegistryProvider.\n * @returns The component registry\n */\nexport const useTamboRegistry = () => {\n  return useContext(TamboRegistryContext);\n};\nfunction getSerializedProps(\n  propsDefinition: any,\n  propsSchema: any,\n  name: string,\n) {\n  if (propsDefinition) {\n    console.warn(`propsDefinition is deprecated. Use propsSchema instead.`);\n    return propsDefinition;\n  }\n\n  if (isZodSchema(propsSchema)) {\n    try {\n      return zodToJsonSchema(propsSchema);\n    } catch (error) {\n      console.error(\n        `Error converting ${name} props schema to JSON Schema:`,\n        error,\n      );\n    }\n  }\n  // try to roughly detect JSONSchema, should always be an object with a properties key\n  if (isJSONSchema(propsSchema)) {\n    return propsSchema;\n  }\n\n  throw new Error(`Invalid props schema for ${name}`);\n}\n\n/**\n * Checks if the propsSchema is a JSON Schema. This is a rough check, and the\n * server will provide the definitive check.\n * @param propsSchema - The props schema to check\n * @returns True if the props schema is a JSON Schema, false otherwise\n */\nfunction isJSONSchema(propsSchema: any) {\n  return (\n    propsSchema &&\n    typeof propsSchema === \"object\" &&\n    propsSchema.type === \"object\" &&\n    propsSchema.properties\n  );\n}\n\n/**\n * Since we require a certain zod version, we need to check if the object is a ZodSchema\n * @param obj - The object to check\n * @returns True if the object is a ZodSchema, false otherwise\n */\nfunction isZodSchema(obj: unknown): obj is ZodSchema {\n  if (obj instanceof ZodSchema) {\n    return true;\n  }\n  // try to detect if the object is a ZodSchema\n  return (\n    typeof obj === \"object\" &&\n    obj !== null &&\n    typeof (obj as any).safeParse === \"function\" &&\n    typeof (obj as any)._def === \"object\"\n  );\n}\n"]}

package/dist/providers/tambo-thread-provider.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"tambo-thread-provider.d.ts","sourceRoot":"","sources":["../../src/providers/tambo-thread-provider.tsx"],"names":[],"mappings":"AACA,OAAO,OAA0B,MAAM,0BAA0B,CAAC;AAClE,OAAO,KAAK,EAAE,EAEZ,iBAAiB,EAMlB,MAAM,OAAO,CAAC;AACf,OAAO,EACL,eAAe,EAEf,kBAAkB,EACnB,MAAM,sCAAsC,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAYpD,MAAM,WAAW,uBAAuB;IACtC,yBAAyB;IACzB,MAAM,EAAE,WAAW,CAAC;IACpB,mCAAmC;IACnC,mBAAmB,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,OAAO,KAAK,IAAI,CAAC;IACjE,yBAAyB;IACzB,cAAc,EAAE,MAAM,IAAI,CAAC;IAC3B,0CAA0C;IAC1C,gBAAgB,EAAE,CAChB,OAAO,EAAE,kBAAkB,EAC3B,YAAY,EAAE,OAAO,KAClB,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC,CAAC;IACnD,6CAA6C;IAC7C,mBAAmB,EAAE,CACnB,EAAE,EAAE,MAAM,EACV,OAAO,EAAE,kBAAkB,EAC3B,YAAY,EAAE,OAAO,KAClB,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,4CAA4C;IAC5C,UAAU,EAAE,MAAM,CAAC;IACnB,gDAAgD;IAChD,aAAa,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;IACvC,2CAA2C;IAC3C,iBAAiB,EAAE,CACjB,OAAO,EAAE,MAAM,EACf,OAAO,EAAE;QACP,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,cAAc,CAAC,EAAE,OAAO,CAAC;QACzB,UAAU,CAAC,EAAE,MAAM,CAAC;KACrB,KACE,OAAO,CAAC,kBAAkB,CAAC,CAAC;IACjC,oFAAoF;IACpF,eAAe,EAAE,eAAe,CAAC;IACjC,6FAA6F;IAC7F,uBAAuB,EAAE,MAAM,CAAC;IAChC,iCAAiC;IACjC,MAAM,EAAE,OAAO,CAAC;CACjB;AAED;;;;;;GAMG;AACH,eAAO,MAAM,kBAAkB,EAAE,WAOhC,CAAC;AAEF,eAAO,MAAM,kBAAkB,wCA0C7B,CAAC;AAEH;;;;;;GAMG;AACH,eAAO,MAAM,mBAAmB,EAAE,KAAK,CAAC,EAAE,CAAC,iBAAiB,CAgf3D,CAAC;AAEF;;;;GAIG;AACH,eAAO,MAAM,cAAc,+BAM1B,CAAC"}
+{"version":3,"file":"tambo-thread-provider.d.ts","sourceRoot":"","sources":["../../src/providers/tambo-thread-provider.tsx"],"names":[],"mappings":"AACA,OAAO,OAA0B,MAAM,0BAA0B,CAAC;AAClE,OAAO,KAAK,EAAE,EAEZ,iBAAiB,EAMlB,MAAM,OAAO,CAAC;AACf,OAAO,EACL,eAAe,EAEf,kBAAkB,EACnB,MAAM,sCAAsC,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAYpD,MAAM,WAAW,uBAAuB;IACtC,yBAAyB;IACzB,MAAM,EAAE,WAAW,CAAC;IACpB,mCAAmC;IACnC,mBAAmB,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,OAAO,KAAK,IAAI,CAAC;IACjE,yBAAyB;IACzB,cAAc,EAAE,MAAM,IAAI,CAAC;IAC3B,0CAA0C;IAC1C,gBAAgB,EAAE,CAChB,OAAO,EAAE,kBAAkB,EAC3B,YAAY,EAAE,OAAO,KAClB,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC,CAAC;IACnD,6CAA6C;IAC7C,mBAAmB,EAAE,CACnB,EAAE,EAAE,MAAM,EACV,OAAO,EAAE,kBAAkB,EAC3B,YAAY,EAAE,OAAO,KAClB,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,4CAA4C;IAC5C,UAAU,EAAE,MAAM,CAAC;IACnB,gDAAgD;IAChD,aAAa,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;IACvC,2CAA2C;IAC3C,iBAAiB,EAAE,CACjB,OAAO,EAAE,MAAM,EACf,OAAO,EAAE;QACP,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,cAAc,CAAC,EAAE,OAAO,CAAC;QACzB,UAAU,CAAC,EAAE,MAAM,CAAC;KACrB,KACE,OAAO,CAAC,kBAAkB,CAAC,CAAC;IACjC,oFAAoF;IACpF,eAAe,EAAE,eAAe,CAAC;IACjC,6FAA6F;IAC7F,uBAAuB,EAAE,MAAM,CAAC;IAChC,iCAAiC;IACjC,MAAM,EAAE,OAAO,CAAC;CACjB;AAED;;;;;;GAMG;AACH,eAAO,MAAM,kBAAkB,EAAE,WAOhC,CAAC;AAEF,eAAO,MAAM,kBAAkB,wCA0C7B,CAAC;AAEH;;;;;;GAMG;AACH,eAAO,MAAM,mBAAmB,EAAE,KAAK,CAAC,EAAE,CAAC,iBAAiB,CAwe3D,CAAC;AAEF;;;;GAIG;AACH,eAAO,MAAM,cAAc,+BAM1B,CAAC"}

package/dist/providers/tambo-thread-provider.js

@@ -172,13 +172,23 @@ const TamboThreadProvider = ({ children, }) => {
             createdAt,
         };
         const threadId = message.threadId;
+        const messageId = chatMessage.id;
         // optimistically update the thread in the local state
         setThreadMap((prevMap) => {
             if (!threadId) {
                 return prevMap;
             }
             const prevMessages = prevMap[threadId]?.messages || [];
-            const updatedMessages = [...prevMessages, chatMessage];
+            const haveMessage = prevMessages.find((msg) => msg.id === messageId);
+            // Update in place if the message already exists
+            const updatedMessages = haveMessage
+                ? prevMessages.map((msg) => {
+                    if (msg.id === messageId) {
+                        return chatMessage;
+                    }
+                    return msg;
+                })
+                : [...prevMessages, chatMessage];
             const updatedThreadMap = {
                 ...prevMap,
                 [threadId]: {
@@ -231,17 +241,6 @@ const TamboThreadProvider = ({ children, }) => {
             });
         }
     }, [client.beta.threads.messages]);
-    const deleteThreadMessage = (0, react_1.useCallback)((messageId, threadId) => {
-        if (!threadMap[threadId])
-            return;
-        setThreadMap((prevMap) => ({
-            ...prevMap,
-            [threadId]: {
-                ...prevMap[threadId],
-                messages: prevMap[threadId].messages.filter((msg) => msg.id !== messageId),
-            },
-        }));
-    }, [threadMap]);
     const startNewThread = (0, react_1.useCallback)(() => {
         setCurrentThreadId(exports.PLACEHOLDER_THREAD.id);
         setThreadMap((prevMap) => {
@@ -274,7 +273,7 @@ const TamboThreadProvider = ({ children, }) => {
         if (fetch) {
             await fetchThread(threadId);
         }
-    }, [fetchThread, threadMap]);
+    }, [fetchThread]);
     const updateThreadStatus = (0, react_1.useCallback)((threadId, stage, statusMessage) => {
         setThreadMap((prevMap) => {
             const updatedThreadMap = {
@@ -288,16 +287,11 @@ const TamboThreadProvider = ({ children, }) => {
             return updatedThreadMap;
         });
     }, []);
-    const handleAdvanceStream = (0, react_1.useCallback)(async (stream, params, threadId, messageIdToRemove) => {
+    const handleAdvanceStream = (0, react_1.useCallback)(async (stream, params, threadId) => {
         let finalMessage;
         let hasSetThreadId = false;
-        let isFirstChunk = true;
         updateThreadStatus(threadId, generate_component_response_1.GenerationStage.STREAMING_RESPONSE);
         for await (const chunk of stream) {
-            if (isFirstChunk && messageIdToRemove) {
-                deleteThreadMessage(messageIdToRemove, threadId);
-            }
-            isFirstChunk = false;
             if (chunk.responseMessageDto.toolCallRequest) {
                 updateThreadStatus(chunk.responseMessageDto.threadId, generate_component_response_1.GenerationStage.FETCHING_CONTEXT);
                 const toolCallResponse = await (0, tool_caller_1.handleToolCall)(chunk.responseMessageDto, toolRegistry);
@@ -323,20 +317,26 @@ const TamboThreadProvider = ({ children, }) => {
                     chunk.responseMessageDto.threadId &&
                     chunk.responseMessageDto.threadId !== currentThread?.id) {
                     hasSetThreadId = true;
-                    switchCurrentThread(chunk.responseMessageDto.threadId, false);
+                    await switchCurrentThread(chunk.responseMessageDto.threadId, false);
                 }
                 if (!finalMessage) {
                     finalMessage = chunk.responseMessageDto.component?.componentName
                         ? (0, generate_component_1.renderComponentIntoMessage)(chunk.responseMessageDto, componentList)
                         : chunk.responseMessageDto;
-                    addThreadMessage(finalMessage, false);
+                    await addThreadMessage(finalMessage, false);
                 }
                 else {
-                    const previousId = finalMessage.id;
+                    // if we start getting a new message mid-stream, put the previous one on screen
+                    const isNewMessage = chunk.responseMessageDto.id !== finalMessage.id;
                     finalMessage = chunk.responseMessageDto.component?.componentName
                         ? (0, generate_component_1.renderComponentIntoMessage)(chunk.responseMessageDto, componentList)
                         : chunk.responseMessageDto;
-                    updateThreadMessage(previousId, finalMessage, false);
+                    if (isNewMessage) {
+                        await addThreadMessage(finalMessage, false);
+                    }
+                    else {
+                        await updateThreadMessage(finalMessage.id, finalMessage, false);
+                    }
                 }
             }
         }
@@ -344,7 +344,7 @@ const TamboThreadProvider = ({ children, }) => {
         return (finalMessage ?? {
             threadId: "",
             content: [{ type: "text", text: `Error processing stream` }],
-            role: "hydra",
+            role: "assistant",
             createdAt: new Date().toISOString(),
             id: crypto.randomUUID(),
             componentState: {},
@@ -354,8 +354,6 @@ const TamboThreadProvider = ({ children, }) => {
         client,
         componentList,
         currentThread?.id,
-        deleteThreadMessage,
-        fetchThread,
         switchCurrentThread,
         toolRegistry,
         updateThreadMessage,

package/dist/providers/tambo-thread-provider.js.map

@@ -1 +1 @@
-{"version":3,"file":"tambo-thread-provider.js","sourceRoot":"","sources":["../../src/providers/tambo-thread-provider.tsx"],"names":[],"mappings":";AAAA,YAAY,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AACb,6DAAkE;AAClE,+CAQe;AACf,sFAI8C;AAE9C,mEAAwE;AACxE,+CAK0B;AAC1B,qDAAqD;AACrD,mEAAyD;AACzD,uEAA6D;AAyC7D;;;;;;GAMG;AACU,QAAA,kBAAkB,GAAgB;IAC7C,EAAE,EAAE,aAAa;IACjB,QAAQ,EAAE,EAAE;IACZ,SAAS,EAAE,EAAE;IACb,SAAS,EAAE,EAAE;IACb,SAAS,EAAE,EAAE;IACb,QAAQ,EAAE,EAAE;CACb,CAAC;AAEW,QAAA,kBAAkB,GAAG,IAAA,qBAAa,EAA0B;IACvE,MAAM,EAAE,0BAAkB;IAC1B;;OAEG;IACH,mBAAmB,EAAE,GAAG,EAAE;QACxB,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;IACzD,CAAC;IACD;;OAEG;IACH,cAAc,EAAE,GAAG,EAAE;QACnB,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;IACpD,CAAC;IACD;;OAEG;IACH,gBAAgB,EAAE,GAAG,EAAE;QACrB,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;IAChE,CAAC;IACD,UAAU,EAAE,EAAE;IACd;;OAEG;IACH,aAAa,EAAE,GAAG,EAAE;QAClB,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;IACnD,CAAC;IACD;;OAEG;IACH,mBAAmB,EAAE,GAAG,EAAE;QACxB,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;IACzD,CAAC;IACD;;OAEG;IACH,iBAAiB,EAAE,GAAG,EAAE;QACtB,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC7C,CAAC;IACD,eAAe,EAAE,6CAAe,CAAC,IAAI;IACrC,uBAAuB,EAAE,EAAE;IAC3B,MAAM,EAAE,IAAI;CACb,CAAC,CAAC;AAEH;;;;;;GAMG;AACI,MAAM,mBAAmB,GAAgC,CAAC,EAC/D,QAAQ,GACT,EAAE,EAAE;IACH,MAAM,CAAC,SAAS,EAAE,YAAY,CAAC,GAAG,IAAA,gBAAQ,EAA8B;QACtE,CAAC,0BAAkB,CAAC,EAAE,CAAC,EAAE,0BAAkB;KAC5C,CAAC,CAAC;IACH,MAAM,MAAM,GAAG,IAAA,sCAAc,GAAE,CAAC;IAChC,MAAM,EAAE,aAAa,EAAE,YAAY,EAAE,yBAAyB,EAAE,GAC9D,IAAA,0CAAgB,GAAE,CAAC;IACrB,MAAM,CAAC,UAAU,EAAE,aAAa,CAAC,GAAG,IAAA,gBAAQ,EAAC,EAAE,CAAC,CAAC;IAEjD,MAAM,CAAC,eAAe,EAAE,kBAAkB,CAAC,GAAG,IAAA,gBAAQ,EACpD,0BAAkB,CAAC,EAAE,CACtB,CAAC;IACF,MAAM,aAAa,GAA4B,SAAS,CAAC,eAAe,CAAC,CAAC;IAE1E,sFAAsF;IACtF,MAAM,mBAAmB,GAAG,IAAA,eAAO,EAAC,GAAG,EAAE;QACvC,MAAM,YAAY,GAAG,IAAI,GAAG,EAA8B,CAAC;QAC3D,IAAI,aAAa,EAAE,CAAC;YAClB,KAAK,MAAM,OAAO,IAAI,aAAa,CAAC,QAAQ,EAAE,CAAC;gBAC7C,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YACxC,CAAC;QACH,CAAC;QACD,OAAO,YAAY,CAAC;IACtB,CAAC,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC;IAEpB,MAAM,WAAW,GAAG,IAAA,mBAAW,EAC7B,KAAK,EAAE,QAAgB,EAAE,EAAE;QACzB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAC5D,MAAM,4BAA4B,GAAG;YACnC,GAAG,MAAM;YACT,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;gBACxC,IAAI,mBAAmB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,CAAC;oBACxC,MAAM,eAAe,GAAG,mBAAmB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;oBAC5D,OAAO;wBACL,GAAG,eAAe;wBAClB,GAAG,OAAO;qBACX,CAAC;gBACJ,CAAC;gBACD,IAAI,OAAO,CAAC,SAAS,EAAE,aAAa,EAAE,CAAC;oBACrC,MAAM,oBAAoB,GAAG,IAAA,+CAA0B,EACrD,OAAO,EACP,aAAa,CACd,CAAC;oBACF,OAAO,oBAAoB,CAAC;gBAC9B,CAAC;gBACD,OAAO,OAAO,CAAC;YACjB,CAAC,CAAC;SACH,CAAC;QAEF,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE;YACvB,MAAM,gBAAgB,GAAG;gBACvB,GAAG,OAAO;gBACV,CAAC,QAAQ,CAAC,EAAE,4BAA4B;aACzC,CAAC;YACF,OAAO,gBAAgB,CAAC;QAC1B,CAAC,CAAC,CAAC;IACL,CAAC,EACD,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,aAAa,EAAE,mBAAmB,CAAC,CAC1D,CAAC;IAEF,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,IACE,eAAe;YACf,eAAe,KAAK,0BAAkB,CAAC,EAAE;YACzC,CAAC,SAAS,CAAC,eAAe,CAAC,EAC3B,CAAC;YACD,WAAW,CAAC,eAAe,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC,EAAE,CAAC,eAAe,EAAE,WAAW,EAAE,SAAS,CAAC,CAAC,CAAC;IAE9C,MAAM,gBAAgB,GAAG,IAAA,mBAAW,EAClC,KAAK,EACH,OAA2B,EAC3B,YAAY,GAAG,IAAI,EACnB,YAAoB,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAC5C,EAAE;QACF,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,OAAO,CAAC,IAAI,CAAC,wDAAwD,CAAC,CAAC;YACvE,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,MAAM,WAAW,GAEb;YACF,GAAG,OAAO;YACV,iBAAiB,EACf,OAAO,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,IAAA,2BAAgB,GAAE,CAAC,CAAC,CAAC,SAAS;YAC1D,SAAS;SACV,CAAC;QACF,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;QAClC,sDAAsD;QACtD,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE;YACvB,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,OAAO,OAAO,CAAC;YACjB,CAAC;YACD,MAAM,YAAY,GAAG,OAAO,CAAC,QAAQ,CAAC,EAAE,QAAQ,IAAI,EAAE,CAAC;YACvD,MAAM,eAAe,GAAG,CAAC,GAAG,YAAY,EAAE,WAAW,CAAC,CAAC;YAEvD,MAAM,gBAAgB,GAAG;gBACvB,GAAG,OAAO;gBACV,CAAC,QAAQ,CAAC,EAAE;oBACV,GAAG,OAAO,CAAC,QAAQ,CAAC;oBACpB,QAAQ,EAAE,eAAe;iBAC1B;aACF,CAAC;YACF,OAAO,gBAAgB,CAAC;QAC1B,CAAC,CAAC,CAAC;QAEH,IAAI,YAAY,EAAE,CAAC;YACjB,gEAAgE;YAChE,MAAM,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE;gBAC1D,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,oDAAoD;aACrD,CAAC,CAAC;QACL,CAAC;QACD,OAAO,SAAS,CAAC,QAAQ,CAAC,EAAE,QAAQ,IAAI,EAAE,CAAC;IAC7C,CAAC,EACD,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,aAAa,EAAE,SAAS,CAAC,CACzD,CAAC;IAEF,MAAM,mBAAmB,GAAG,IAAA,mBAAW,EACrC,KAAK,EACH,EAAU,EACV,OAA2B,EAC3B,YAAY,GAAG,IAAI,EACnB,YAAoB,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAC5C,EAAE;QACF,MAAM,WAAW,GAAuB;YACtC,GAAG,OAAO;YACV,SAAS;SACV,CAAC;QAEF,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE;YACvB,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;gBACtB,OAAO,OAAO,CAAC;YACjB,CAAC;YACD,MAAM,YAAY,GAAG,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,QAAQ,IAAI,EAAE,CAAC;YAC/D,MAAM,eAAe,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;gBAC/C,IAAI,GAAG,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC;oBAClB,OAAO,WAAW,CAAC;gBACrB,CAAC;gBACD,OAAO,GAAG,CAAC;YACb,CAAC,CAAC,CAAC;YACH,OAAO;gBACL,GAAG,OAAO;gBACV,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE;oBAClB,GAAG,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC;oBAC5B,QAAQ,EAAE,eAAe;iBAC1B;aACF,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,IAAI,YAAY,EAAE,CAAC;YACjB,gEAAgE;YAChE,MAAM,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE;gBAC1D,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,oDAAoD;aACrD,CAAC,CAAC;QACL,CAAC;IACH,CAAC,EACD,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAC/B,CAAC;IAEF,MAAM,mBAAmB,GAAG,IAAA,mBAAW,EACrC,CAAC,SAAiB,EAAE,QAAgB,EAAE,EAAE;QACtC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC;YAAE,OAAO;QAEjC,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;YACzB,GAAG,OAAO;YACV,CAAC,QAAQ,CAAC,EAAE;gBACV,GAAG,OAAO,CAAC,QAAQ,CAAC;gBACpB,QAAQ,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC,QAAQ,CAAC,MAAM,CACzC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,KAAK,SAAS,CAC9B;aACF;SACF,CAAC,CAAC,CAAC;IACN,CAAC,EACD,CAAC,SAAS,CAAC,CACZ,CAAC;IAEF,MAAM,cAAc,GAAG,IAAA,mBAAW,EAAC,GAAG,EAAE;QACtC,kBAAkB,CAAC,0BAAkB,CAAC,EAAE,CAAC,CAAC;QAC1C,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE;YACvB,OAAO;gBACL,GAAG,OAAO;gBACV,CAAC,0BAAkB,CAAC,EAAE,CAAC,EAAE,0BAAkB;aAC5C,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,mBAAmB,GAAG,IAAA,mBAAW,EACrC,KAAK,EAAE,QAAgB,EAAE,KAAK,GAAG,IAAI,EAAE,EAAE;QACvC,IAAI,QAAQ,KAAK,0BAAkB,CAAC,EAAE,EAAE,CAAC;YACvC,OAAO,CAAC,IAAI,CAAC,gDAAgD,CAAC,CAAC;YAC/D,OAAO;QACT,CAAC;QACD,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QAC7B,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE;YACvB,IAAI,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACtB,OAAO,OAAO,CAAC;YACjB,CAAC;YACD,yEAAyE;YACzE,MAAM,gBAAgB,GAAG;gBACvB,GAAG,OAAO;gBACV,CAAC,QAAQ,CAAC,EAAE;oBACV,GAAG,OAAO,CAAC,0BAAkB,CAAC,EAAE,CAAC;oBACjC,EAAE,EAAE,QAAQ;iBACb;aACF,CAAC;YACF,OAAO,gBAAgB,CAAC;QAC1B,CAAC,CAAC,CAAC;QACH,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,WAAW,CAAC,QAAQ,CAAC,CAAC;QAC9B,CAAC;IACH,CAAC,EACD,CAAC,WAAW,EAAE,SAAS,CAAC,CACzB,CAAC;IAEF,MAAM,kBAAkB,GAAG,IAAA,mBAAW,EACpC,CAAC,QAAgB,EAAE,KAAsB,EAAE,aAAsB,EAAE,EAAE;QACnE,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE;YACvB,MAAM,gBAAgB,GAAG;gBACvB,GAAG,OAAO;gBACV,CAAC,QAAQ,CAAC,EAAE;oBACV,GAAG,OAAO,CAAC,QAAQ,CAAC;oBACpB,eAAe,EAAE,KAAK;oBACtB,aAAa,EAAE,aAAa;iBAC7B;aACF,CAAC;YACF,OAAO,gBAAgB,CAAC;QAC1B,CAAC,CAAC,CAAC;IACL,CAAC,EACD,EAAE,CACH,CAAC;IAEF,MAAM,mBAAmB,GAAG,IAAA,mBAAW,EACrC,KAAK,EACH,MAAiE,EACjE,MAAgD,EAChD,QAAgB,EAChB,iBAA0B,EACG,EAAE;QAC/B,IAAI,YAA4C,CAAC;QACjD,IAAI,cAAc,GAAG,KAAK,CAAC;QAC3B,IAAI,YAAY,GAAG,IAAI,CAAC;QACxB,kBAAkB,CAAC,QAAQ,EAAE,6CAAe,CAAC,kBAAkB,CAAC,CAAC;QAEjE,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YACjC,IAAI,YAAY,IAAI,iBAAiB,EAAE,CAAC;gBACtC,mBAAmB,CAAC,iBAAiB,EAAE,QAAQ,CAAC,CAAC;YACnD,CAAC;YACD,YAAY,GAAG,KAAK,CAAC;YAErB,IAAI,KAAK,CAAC,kBAAkB,CAAC,eAAe,EAAE,CAAC;gBAC7C,kBAAkB,CAChB,KAAK,CAAC,kBAAkB,CAAC,QAAQ,EACjC,6CAAe,CAAC,gBAAgB,CACjC,CAAC;gBACF,MAAM,gBAAgB,GAAG,MAAM,IAAA,4BAAc,EAC3C,KAAK,CAAC,kBAAkB,EACxB,YAAY,CACb,CAAC;gBACF,MAAM,sBAAsB,GAC1B,OAAO,gBAAgB,KAAK,QAAQ;oBAClC,CAAC,CAAC,gBAAgB;oBAClB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;gBACvC,MAAM,sBAAsB,GAC1B;oBACE,GAAG,MAAM;oBACT,eAAe,EAAE;wBACf,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,sBAAsB,EAAE,CAAC;wBACzD,IAAI,EAAE,MAAM;wBACZ,UAAU,EAAE,eAAe;wBAC3B,SAAS,EAAE,KAAK,CAAC,kBAAkB,CAAC,SAAS;wBAC7C,YAAY,EAAE,KAAK,CAAC,kBAAkB,CAAC,YAAY;qBACpD;iBACF,CAAC;gBACJ,kBAAkB,CAChB,KAAK,CAAC,kBAAkB,CAAC,QAAQ,EACjC,6CAAe,CAAC,kBAAkB,CACnC,CAAC;gBACF,MAAM,sBAAsB,GAAG,MAAM,IAAA,8BAAa,EAChD,MAAM,EACN,sBAAsB,EACtB,KAAK,CAAC,kBAAkB,CAAC,QAAQ,CAClC,CAAC;gBAEF,OAAO,MAAM,mBAAmB,CAC9B,sBAAsB,EACtB,sBAAsB,EACtB,KAAK,CAAC,kBAAkB,CAAC,QAAQ,CAClC,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,IACE,CAAC,cAAc;oBACf,KAAK,CAAC,kBAAkB,CAAC,QAAQ;oBACjC,KAAK,CAAC,kBAAkB,CAAC,QAAQ,KAAK,aAAa,EAAE,EAAE,EACvD,CAAC;oBACD,cAAc,GAAG,IAAI,CAAC;oBACtB,mBAAmB,CAAC,KAAK,CAAC,kBAAkB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;gBAChE,CAAC;gBAED,IAAI,CAAC,YAAY,EAAE,CAAC;oBAClB,YAAY,GAAG,KAAK,CAAC,kBAAkB,CAAC,SAAS,EAAE,aAAa;wBAC9D,CAAC,CAAC,IAAA,+CAA0B,EACxB,KAAK,CAAC,kBAAkB,EACxB,aAAa,CACd;wBACH,CAAC,CAAC,KAAK,CAAC,kBAAkB,CAAC;oBAC7B,gBAAgB,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC;gBACxC,CAAC;qBAAM,CAAC;oBACN,MAAM,UAAU,GAAG,YAAY,CAAC,EAAE,CAAC;oBACnC,YAAY,GAAG,KAAK,CAAC,kBAAkB,CAAC,SAAS,EAAE,aAAa;wBAC9D,CAAC,CAAC,IAAA,+CAA0B,EACxB,KAAK,CAAC,kBAAkB,EACxB,aAAa,CACd;wBACH,CAAC,CAAC,KAAK,CAAC,kBAAkB,CAAC;oBAC7B,mBAAmB,CAAC,UAAU,EAAE,YAAY,EAAE,KAAK,CAAC,CAAC;gBACvD,CAAC;YACH,CAAC;QACH,CAAC;QAED,kBAAkB,CAChB,YAAY,EAAE,QAAQ,IAAI,QAAQ,EAClC,6CAAe,CAAC,QAAQ,CACzB,CAAC;QACF,OAAO,CACL,YAAY,IAAI;YACd,QAAQ,EAAE,EAAE;YACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,yBAAyB,EAAE,CAAC;YAC5D,IAAI,EAAE,OAAO;YACb,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE;YACvB,cAAc,EAAE,EAAE;SACnB,CACF,CAAC;IACJ,CAAC,EACD;QACE,gBAAgB;QAChB,MAAM;QACN,aAAa;QACb,aAAa,EAAE,EAAE;QACjB,mBAAmB;QACnB,WAAW;QACX,mBAAmB;QACnB,YAAY;QACZ,mBAAmB;QACnB,kBAAkB;KACnB,CACF,CAAC;IAEF,MAAM,iBAAiB,GAAG,IAAA,mBAAW,EACnC,KAAK,EACH,OAAe,EACf,UAII,EAAE,QAAQ,EAAE,0BAAkB,CAAC,EAAE,EAAE,EACV,EAAE;QAC/B,MAAM,EAAE,QAAQ,GAAG,aAAa,CAAC,EAAE,EAAE,cAAc,EAAE,GAAG,OAAO,CAAC;QAChE,kBAAkB,CAAC,QAAQ,EAAE,6CAAe,CAAC,kBAAkB,CAAC,CAAC;QAEjE,gBAAgB,CACd;YACE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;YAC1C,iBAAiB,EAAE,IAAI;YACvB,IAAI,EAAE,MAAM;YACZ,QAAQ,EAAE,QAAQ;YAClB,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE;YACvB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,cAAc,EAAE,EAAE;SACnB,EACD,KAAK,CACN,CAAC;QAEF,MAAM,mBAAmB,GAAG,IAAA,iCAAsB,EAChD,aAAa,EACb,YAAY,EACZ,yBAAyB,CAC1B,CAAC;QACF,MAAM,iBAAiB,GAAG,IAAA,+BAAoB,EAC5C,YAAY,EACZ,yBAAyB,CAC1B,CAAC;QACF,MAAM,MAAM,GAA6C;YACvD,eAAe,EAAE;gBACf,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;gBAC1C,IAAI,EAAE,MAAM;aACb;YACD,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,mBAAmB,EAAE,mBAAmB;YACxC,WAAW,EAAE,iBAAiB,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAC1C,IAAA,oCAAyB,EAAC,IAAI,CAAC,CAChC;SACF,CAAC;QAEF,IAAI,cAAc,EAAE,CAAC;YACnB,MAAM,qBAAqB,GAAG,MAAM,IAAA,8BAAa,EAC/C,MAAM,EACN,MAAM,EACN,QAAQ,KAAK,0BAAkB,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAC1D,CAAC;YACF,OAAO,MAAM,mBAAmB,CAC9B,qBAAqB,EACrB,MAAM,EACN,QAAQ,CACT,CAAC;QACJ,CAAC;QACD,IAAI,eAAe,GAAG,MAAM,CAAC,QAAQ,KAAK,0BAAkB,CAAC,EAAE;YAC7D,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC;YACrC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAC;QAEvD,mBAAmB;QACnB,OAAO,eAAe,CAAC,kBAAkB,CAAC,eAAe,EAAE,CAAC;YAC1D,kBAAkB,CAAC,QAAQ,EAAE,6CAAe,CAAC,gBAAgB,CAAC,CAAC;YAC/D,MAAM,gBAAgB,GAAG,MAAM,IAAA,4BAAc,EAC3C,eAAe,CAAC,kBAAkB,EAClC,YAAY,CACb,CAAC;YACF,MAAM,kBAAkB,GACtB,OAAO,gBAAgB,KAAK,QAAQ;gBAClC,CAAC,CAAC,gBAAgB;gBAClB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YACvC,MAAM,sBAAsB,GAC1B;gBACE,GAAG,MAAM;gBACT,eAAe,EAAE;oBACf,GAAG,MAAM,CAAC,eAAe;oBACzB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC;oBACrD,IAAI,EAAE,MAAM;oBACZ,UAAU,EAAE,eAAe;oBAC3B,SAAS,EAAE,eAAe,CAAC,kBAAkB,CAAC,SAAS;oBACvD,YAAY,EAAE,eAAe,CAAC,kBAAkB,CAAC,YAAY;iBAC9D;aACF,CAAC;YACJ,kBAAkB,CAAC,QAAQ,EAAE,6CAAe,CAAC,mBAAmB,CAAC,CAAC;YAClE,eAAe,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CACrD,eAAe,CAAC,kBAAkB,CAAC,QAAQ,EAC3C,sBAAsB,CACvB,CAAC;QACJ,CAAC;QAED,MAAM,YAAY,GAAG,eAAe,CAAC,kBAAkB,CAAC,SAAS;YAC/D,EAAE,aAAa;YACf,CAAC,CAAC,IAAA,+CAA0B,EACxB,eAAe,CAAC,kBAAkB,EAClC,aAAa,CACd;YACH,CAAC,CAAC,eAAe,CAAC,kBAAkB,CAAC;QACvC,MAAM,mBAAmB,CAAC,eAAe,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QACvE,kBAAkB,CAChB,eAAe,CAAC,kBAAkB,CAAC,QAAQ,EAC3C,6CAAe,CAAC,QAAQ,CACzB,CAAC;QACF,OAAO,YAAY,CAAC;IACtB,CAAC,EACD;QACE,aAAa;QACb,YAAY;QACZ,yBAAyB;QACzB,aAAa,CAAC,EAAE;QAChB,mBAAmB;QACnB,gBAAgB;QAChB,MAAM;QACN,kBAAkB;QAClB,mBAAmB;KACpB,CACF,CAAC;IAEF,OAAO,CACL,8BAAC,0BAAkB,CAAC,QAAQ,IAC1B,KAAK,EAAE;YACL,MAAM,EAAE,aAAa;YACrB,mBAAmB;YACnB,cAAc;YACd,gBAAgB;YAChB,mBAAmB;YACnB,UAAU;YACV,aAAa;YACb,iBAAiB;YACjB,eAAe,EAAE,CAAC,aAAa,EAAE,eAAe;gBAC9C,6CAAe,CAAC,IAAI,CAAoB;YAC1C,uBAAuB,EAAE,aAAa,EAAE,aAAa,IAAI,EAAE;YAC3D,MAAM,EAAE,IAAA,yCAAW,EACjB,CAAC,aAAa,EAAE,eAAe;gBAC7B,6CAAe,CAAC,IAAI,CAAoB,CAC3C;SACF,IAEA,QAAQ,CACmB,CAC/B,CAAC;AACJ,CAAC,CAAC;AAhfW,QAAA,mBAAmB,uBAgf9B;AAEF;;;;GAIG;AACI,MAAM,cAAc,GAAG,GAAG,EAAE;IACjC,MAAM,OAAO,GAAG,IAAA,kBAAU,EAAC,0BAAkB,CAAC,CAAC;IAC/C,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;QAC1B,MAAM,IAAI,KAAK,CAAC,0DAA0D,CAAC,CAAC;IAC9E,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC;AANW,QAAA,cAAc,kBAMzB","sourcesContent":["\"use client\";\nimport TamboAI, { advanceStream } from \"@tambo-ai/typescript-sdk\";\nimport React, {\n  createContext,\n  PropsWithChildren,\n  useCallback,\n  useContext,\n  useEffect,\n  useMemo,\n  useState,\n} from \"react\";\nimport {\n  GenerationStage,\n  isIdleStage,\n  TamboThreadMessage,\n} from \"../model/generate-component-response\";\nimport { TamboThread } from \"../model/tambo-thread\";\nimport { renderComponentIntoMessage } from \"../util/generate-component\";\nimport {\n  getAvailableComponents,\n  getClientContext,\n  getUnassociatedTools,\n  mapTamboToolToContextTool,\n} from \"../util/registry\";\nimport { handleToolCall } from \"../util/tool-caller\";\nimport { useTamboClient } from \"./tambo-client-provider\";\nimport { useTamboRegistry } from \"./tambo-registry-provider\";\n\nexport interface TamboThreadContextProps {\n  /** The current thread */\n  thread: TamboThread;\n  /** Switch to a different thread */\n  switchCurrentThread: (threadId: string, fetch?: boolean) => void;\n  /** Start a new thread */\n  startNewThread: () => void;\n  /** Add a message to the current thread */\n  addThreadMessage: (\n    message: TamboThreadMessage,\n    sendToServer: boolean,\n  ) => Promise<TamboAI.Beta.Threads.ThreadMessage[]>;\n  /** Update a message in the current thread */\n  updateThreadMessage: (\n    id: string,\n    message: TamboThreadMessage,\n    sendToServer: boolean,\n  ) => Promise<void>;\n  /** The input value of the current thread */\n  inputValue: string;\n  /** Set the input value of the current thread */\n  setInputValue: (value: string) => void;\n  /** Send a message to the current thread */\n  sendThreadMessage: (\n    message: string,\n    options: {\n      threadId?: string;\n      streamResponse?: boolean;\n      contextKey?: string;\n    },\n  ) => Promise<TamboThreadMessage>;\n  /** The generation stage of the current thread - updated as the thread progresses */\n  generationStage: GenerationStage;\n  /** The generation status message of the current thread - updated as the thread progresses */\n  generationStatusMessage: string;\n  /** Whether the thread is idle */\n  isIdle: boolean;\n}\n\n/**\n * This is a stub entry for when the thread is not yet created, the first time\n * the user sends a message\n *\n * Note that the consumer needs to be careful never to send `PLACEHOLDER_THREAD.id` to the server,\n * as this doesn't really exist on the server side.\n */\nexport const PLACEHOLDER_THREAD: TamboThread = {\n  id: \"placeholder\",\n  messages: [],\n  createdAt: \"\",\n  projectId: \"\",\n  updatedAt: \"\",\n  metadata: {},\n};\n\nexport const TamboThreadContext = createContext<TamboThreadContextProps>({\n  thread: PLACEHOLDER_THREAD,\n  /**\n   *\n   */\n  switchCurrentThread: () => {\n    throw new Error(\"switchCurrentThread not implemented\");\n  },\n  /**\n   *\n   */\n  startNewThread: () => {\n    throw new Error(\"startNewThread not implemented\");\n  },\n  /**\n   *\n   */\n  addThreadMessage: () => {\n    throw new Error(\"updateThreadMessageHistory not implemented\");\n  },\n  inputValue: \"\",\n  /**\n   *\n   */\n  setInputValue: () => {\n    throw new Error(\"setInputValue not implemented\");\n  },\n  /**\n   *\n   */\n  updateThreadMessage: () => {\n    throw new Error(\"updateThreadMessage not implemented\");\n  },\n  /**\n   *\n   */\n  sendThreadMessage: () => {\n    throw new Error(\"advance not implemented\");\n  },\n  generationStage: GenerationStage.IDLE,\n  generationStatusMessage: \"\",\n  isIdle: true,\n});\n\n/**\n * The TamboThreadProvider is a React provider that provides a thread context\n * to the descendants of the provider.\n * @param props - The props for the TamboThreadProvider\n * @param props.children - The children to wrap\n * @returns The TamboThreadProvider component\n */\nexport const TamboThreadProvider: React.FC<PropsWithChildren> = ({\n  children,\n}) => {\n  const [threadMap, setThreadMap] = useState<Record<string, TamboThread>>({\n    [PLACEHOLDER_THREAD.id]: PLACEHOLDER_THREAD,\n  });\n  const client = useTamboClient();\n  const { componentList, toolRegistry, componentToolAssociations } =\n    useTamboRegistry();\n  const [inputValue, setInputValue] = useState(\"\");\n\n  const [currentThreadId, setCurrentThreadId] = useState<string>(\n    PLACEHOLDER_THREAD.id,\n  );\n  const currentThread: TamboThread | undefined = threadMap[currentThreadId];\n\n  // Use existing messages from the current thread to avoid re-generating any components\n  const currentMessageCache = useMemo(() => {\n    const messageCache = new Map<string, TamboThreadMessage>();\n    if (currentThread) {\n      for (const message of currentThread.messages) {\n        messageCache.set(message.id, message);\n      }\n    }\n    return messageCache;\n  }, [currentThread]);\n\n  const fetchThread = useCallback(\n    async (threadId: string) => {\n      const thread = await client.beta.threads.retrieve(threadId);\n      const threadWithRenderedComponents = {\n        ...thread,\n        messages: thread.messages.map((message) => {\n          if (currentMessageCache.has(message.id)) {\n            const renderedMessage = currentMessageCache.get(message.id);\n            return {\n              ...renderedMessage,\n              ...message,\n            };\n          }\n          if (message.component?.componentName) {\n            const messageWithComponent = renderComponentIntoMessage(\n              message,\n              componentList,\n            );\n            return messageWithComponent;\n          }\n          return message;\n        }),\n      };\n\n      setThreadMap((prevMap) => {\n        const updatedThreadMap = {\n          ...prevMap,\n          [threadId]: threadWithRenderedComponents,\n        };\n        return updatedThreadMap;\n      });\n    },\n    [client.beta.threads, componentList, currentMessageCache],\n  );\n\n  useEffect(() => {\n    if (\n      currentThreadId &&\n      currentThreadId !== PLACEHOLDER_THREAD.id &&\n      !threadMap[currentThreadId]\n    ) {\n      fetchThread(currentThreadId);\n    }\n  }, [currentThreadId, fetchThread, threadMap]);\n\n  const addThreadMessage = useCallback(\n    async (\n      message: TamboThreadMessage,\n      sendToServer = true,\n      createdAt: string = new Date().toISOString(),\n    ) => {\n      if (!currentThread) {\n        console.warn(\"Cannot add messages if we do not have a current thread\");\n        return [];\n      }\n      const chatMessage: TamboThreadMessage & {\n        additionalContext?: string;\n      } = {\n        ...message,\n        additionalContext:\n          message.role === \"user\" ? getClientContext() : undefined,\n        createdAt,\n      };\n      const threadId = message.threadId;\n      // optimistically update the thread in the local state\n      setThreadMap((prevMap) => {\n        if (!threadId) {\n          return prevMap;\n        }\n        const prevMessages = prevMap[threadId]?.messages || [];\n        const updatedMessages = [...prevMessages, chatMessage];\n\n        const updatedThreadMap = {\n          ...prevMap,\n          [threadId]: {\n            ...prevMap[threadId],\n            messages: updatedMessages,\n          },\n        };\n        return updatedThreadMap;\n      });\n\n      if (sendToServer) {\n        // TODO: if this fails, we need to revert the local state update\n        await client.beta.threads.messages.create(message.threadId, {\n          content: message.content,\n          role: message.role,\n          // additionalContext: chatMessage.additionalContext,\n        });\n      }\n      return threadMap[threadId]?.messages || [];\n    },\n    [client.beta.threads.messages, currentThread, threadMap],\n  );\n\n  const updateThreadMessage = useCallback(\n    async (\n      id: string,\n      message: TamboThreadMessage,\n      sendToServer = true,\n      createdAt: string = new Date().toISOString(),\n    ) => {\n      const chatMessage: TamboThreadMessage = {\n        ...message,\n        createdAt,\n      };\n\n      setThreadMap((prevMap) => {\n        if (!message.threadId) {\n          return prevMap;\n        }\n        const prevMessages = prevMap[message.threadId]?.messages || [];\n        const updatedMessages = prevMessages.map((msg) => {\n          if (msg.id === id) {\n            return chatMessage;\n          }\n          return msg;\n        });\n        return {\n          ...prevMap,\n          [message.threadId]: {\n            ...prevMap[message.threadId],\n            messages: updatedMessages,\n          },\n        };\n      });\n      if (sendToServer) {\n        // TODO: if this fails, we need to revert the local state update\n        await client.beta.threads.messages.create(message.threadId, {\n          content: message.content,\n          role: message.role,\n          // additionalContext: chatMessage.additionalContext,\n        });\n      }\n    },\n    [client.beta.threads.messages],\n  );\n\n  const deleteThreadMessage = useCallback(\n    (messageId: string, threadId: string) => {\n      if (!threadMap[threadId]) return;\n\n      setThreadMap((prevMap) => ({\n        ...prevMap,\n        [threadId]: {\n          ...prevMap[threadId],\n          messages: prevMap[threadId].messages.filter(\n            (msg) => msg.id !== messageId,\n          ),\n        },\n      }));\n    },\n    [threadMap],\n  );\n\n  const startNewThread = useCallback(() => {\n    setCurrentThreadId(PLACEHOLDER_THREAD.id);\n    setThreadMap((prevMap) => {\n      return {\n        ...prevMap,\n        [PLACEHOLDER_THREAD.id]: PLACEHOLDER_THREAD,\n      };\n    });\n  }, []);\n\n  const switchCurrentThread = useCallback(\n    async (threadId: string, fetch = true) => {\n      if (threadId === PLACEHOLDER_THREAD.id) {\n        console.warn(\"Switching to placeholder thread, may be a bug.\");\n        return;\n      }\n      setCurrentThreadId(threadId);\n      setThreadMap((prevMap) => {\n        if (prevMap[threadId]) {\n          return prevMap;\n        }\n        // If this is a new thread, add placeholder thread messages to the thread\n        const updatedThreadMap = {\n          ...prevMap,\n          [threadId]: {\n            ...prevMap[PLACEHOLDER_THREAD.id],\n            id: threadId,\n          },\n        };\n        return updatedThreadMap;\n      });\n      if (fetch) {\n        await fetchThread(threadId);\n      }\n    },\n    [fetchThread, threadMap],\n  );\n\n  const updateThreadStatus = useCallback(\n    (threadId: string, stage: GenerationStage, statusMessage?: string) => {\n      setThreadMap((prevMap) => {\n        const updatedThreadMap = {\n          ...prevMap,\n          [threadId]: {\n            ...prevMap[threadId],\n            generationStage: stage,\n            statusMessage: statusMessage,\n          },\n        };\n        return updatedThreadMap;\n      });\n    },\n    [],\n  );\n\n  const handleAdvanceStream = useCallback(\n    async (\n      stream: AsyncIterable<TamboAI.Beta.Threads.ThreadAdvanceResponse>,\n      params: TamboAI.Beta.Threads.ThreadAdvanceParams,\n      threadId: string,\n      messageIdToRemove?: string,\n    ): Promise<TamboThreadMessage> => {\n      let finalMessage: TamboThreadMessage | undefined;\n      let hasSetThreadId = false;\n      let isFirstChunk = true;\n      updateThreadStatus(threadId, GenerationStage.STREAMING_RESPONSE);\n\n      for await (const chunk of stream) {\n        if (isFirstChunk && messageIdToRemove) {\n          deleteThreadMessage(messageIdToRemove, threadId);\n        }\n        isFirstChunk = false;\n\n        if (chunk.responseMessageDto.toolCallRequest) {\n          updateThreadStatus(\n            chunk.responseMessageDto.threadId,\n            GenerationStage.FETCHING_CONTEXT,\n          );\n          const toolCallResponse = await handleToolCall(\n            chunk.responseMessageDto,\n            toolRegistry,\n          );\n          const toolCallResponseString =\n            typeof toolCallResponse === \"string\"\n              ? toolCallResponse\n              : JSON.stringify(toolCallResponse);\n          const toolCallResponseParams: TamboAI.Beta.Threads.ThreadAdvanceParams =\n            {\n              ...params,\n              messageToAppend: {\n                content: [{ type: \"text\", text: toolCallResponseString }],\n                role: \"tool\",\n                actionType: \"tool_response\",\n                component: chunk.responseMessageDto.component,\n                tool_call_id: chunk.responseMessageDto.tool_call_id,\n              },\n            };\n          updateThreadStatus(\n            chunk.responseMessageDto.threadId,\n            GenerationStage.STREAMING_RESPONSE,\n          );\n          const toolCallResponseStream = await advanceStream(\n            client,\n            toolCallResponseParams,\n            chunk.responseMessageDto.threadId,\n          );\n\n          return await handleAdvanceStream(\n            toolCallResponseStream,\n            toolCallResponseParams,\n            chunk.responseMessageDto.threadId,\n          );\n        } else {\n          if (\n            !hasSetThreadId &&\n            chunk.responseMessageDto.threadId &&\n            chunk.responseMessageDto.threadId !== currentThread?.id\n          ) {\n            hasSetThreadId = true;\n            switchCurrentThread(chunk.responseMessageDto.threadId, false);\n          }\n\n          if (!finalMessage) {\n            finalMessage = chunk.responseMessageDto.component?.componentName\n              ? renderComponentIntoMessage(\n                  chunk.responseMessageDto,\n                  componentList,\n                )\n              : chunk.responseMessageDto;\n            addThreadMessage(finalMessage, false);\n          } else {\n            const previousId = finalMessage.id;\n            finalMessage = chunk.responseMessageDto.component?.componentName\n              ? renderComponentIntoMessage(\n                  chunk.responseMessageDto,\n                  componentList,\n                )\n              : chunk.responseMessageDto;\n            updateThreadMessage(previousId, finalMessage, false);\n          }\n        }\n      }\n\n      updateThreadStatus(\n        finalMessage?.threadId ?? threadId,\n        GenerationStage.COMPLETE,\n      );\n      return (\n        finalMessage ?? {\n          threadId: \"\",\n          content: [{ type: \"text\", text: `Error processing stream` }],\n          role: \"hydra\",\n          createdAt: new Date().toISOString(),\n          id: crypto.randomUUID(),\n          componentState: {},\n        }\n      );\n    },\n    [\n      addThreadMessage,\n      client,\n      componentList,\n      currentThread?.id,\n      deleteThreadMessage,\n      fetchThread,\n      switchCurrentThread,\n      toolRegistry,\n      updateThreadMessage,\n      updateThreadStatus,\n    ],\n  );\n\n  const sendThreadMessage = useCallback(\n    async (\n      message: string,\n      options: {\n        threadId?: string;\n        streamResponse?: boolean;\n        contextKey?: string;\n      } = { threadId: PLACEHOLDER_THREAD.id },\n    ): Promise<TamboThreadMessage> => {\n      const { threadId = currentThread.id, streamResponse } = options;\n      updateThreadStatus(threadId, GenerationStage.CHOOSING_COMPONENT);\n\n      addThreadMessage(\n        {\n          content: [{ type: \"text\", text: message }],\n          renderedComponent: null,\n          role: \"user\",\n          threadId: threadId,\n          id: crypto.randomUUID(),\n          createdAt: new Date().toISOString(),\n          componentState: {},\n        },\n        false,\n      );\n\n      const availableComponents = getAvailableComponents(\n        componentList,\n        toolRegistry,\n        componentToolAssociations,\n      );\n      const unassociatedTools = getUnassociatedTools(\n        toolRegistry,\n        componentToolAssociations,\n      );\n      const params: TamboAI.Beta.Threads.ThreadAdvanceParams = {\n        messageToAppend: {\n          content: [{ type: \"text\", text: message }],\n          role: \"user\",\n        },\n        contextKey: options.contextKey,\n        availableComponents: availableComponents,\n        clientTools: unassociatedTools.map((tool) =>\n          mapTamboToolToContextTool(tool),\n        ),\n      };\n\n      if (streamResponse) {\n        const advanceStreamResponse = await advanceStream(\n          client,\n          params,\n          threadId === PLACEHOLDER_THREAD.id ? undefined : threadId,\n        );\n        return await handleAdvanceStream(\n          advanceStreamResponse,\n          params,\n          threadId,\n        );\n      }\n      let advanceResponse = await (threadId === PLACEHOLDER_THREAD.id\n        ? client.beta.threads.advance(params)\n        : client.beta.threads.advanceById(threadId, params));\n\n      //handle tool calls\n      while (advanceResponse.responseMessageDto.toolCallRequest) {\n        updateThreadStatus(threadId, GenerationStage.FETCHING_CONTEXT);\n        const toolCallResponse = await handleToolCall(\n          advanceResponse.responseMessageDto,\n          toolRegistry,\n        );\n        const toolResponseString =\n          typeof toolCallResponse === \"string\"\n            ? toolCallResponse\n            : JSON.stringify(toolCallResponse);\n        const toolCallResponseParams: TamboAI.Beta.Threads.ThreadAdvanceParams =\n          {\n            ...params,\n            messageToAppend: {\n              ...params.messageToAppend,\n              content: [{ type: \"text\", text: toolResponseString }],\n              role: \"tool\",\n              actionType: \"tool_response\",\n              component: advanceResponse.responseMessageDto.component,\n              tool_call_id: advanceResponse.responseMessageDto.tool_call_id,\n            },\n          };\n        updateThreadStatus(threadId, GenerationStage.HYDRATING_COMPONENT);\n        advanceResponse = await client.beta.threads.advanceById(\n          advanceResponse.responseMessageDto.threadId,\n          toolCallResponseParams,\n        );\n      }\n\n      const finalMessage = advanceResponse.responseMessageDto.component\n        ?.componentName\n        ? renderComponentIntoMessage(\n            advanceResponse.responseMessageDto,\n            componentList,\n          )\n        : advanceResponse.responseMessageDto;\n      await switchCurrentThread(advanceResponse.responseMessageDto.threadId);\n      updateThreadStatus(\n        advanceResponse.responseMessageDto.threadId,\n        GenerationStage.COMPLETE,\n      );\n      return finalMessage;\n    },\n    [\n      componentList,\n      toolRegistry,\n      componentToolAssociations,\n      currentThread.id,\n      switchCurrentThread,\n      addThreadMessage,\n      client,\n      updateThreadStatus,\n      handleAdvanceStream,\n    ],\n  );\n\n  return (\n    <TamboThreadContext.Provider\n      value={{\n        thread: currentThread,\n        switchCurrentThread,\n        startNewThread,\n        addThreadMessage,\n        updateThreadMessage,\n        inputValue,\n        setInputValue,\n        sendThreadMessage,\n        generationStage: (currentThread?.generationStage ??\n          GenerationStage.IDLE) as GenerationStage,\n        generationStatusMessage: currentThread?.statusMessage ?? \"\",\n        isIdle: isIdleStage(\n          (currentThread?.generationStage ??\n            GenerationStage.IDLE) as GenerationStage,\n        ),\n      }}\n    >\n      {children}\n    </TamboThreadContext.Provider>\n  );\n};\n\n/**\n * The useTamboThread hook provides access to the current thread context\n * to the descendants of the TamboThreadProvider.\n * @returns All state and actions for the current thread\n */\nexport const useTamboThread = () => {\n  const context = useContext(TamboThreadContext);\n  if (context === undefined) {\n    throw new Error(\"useTamboThread must be used within a TamboThreadProvider\");\n  }\n  return context;\n};\n"]}
+{"version":3,"file":"tambo-thread-provider.js","sourceRoot":"","sources":["../../src/providers/tambo-thread-provider.tsx"],"names":[],"mappings":";AAAA,YAAY,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AACb,6DAAkE;AAClE,+CAQe;AACf,sFAI8C;AAE9C,mEAAwE;AACxE,+CAK0B;AAC1B,qDAAqD;AACrD,mEAAyD;AACzD,uEAA6D;AAyC7D;;;;;;GAMG;AACU,QAAA,kBAAkB,GAAgB;IAC7C,EAAE,EAAE,aAAa;IACjB,QAAQ,EAAE,EAAE;IACZ,SAAS,EAAE,EAAE;IACb,SAAS,EAAE,EAAE;IACb,SAAS,EAAE,EAAE;IACb,QAAQ,EAAE,EAAE;CACb,CAAC;AAEW,QAAA,kBAAkB,GAAG,IAAA,qBAAa,EAA0B;IACvE,MAAM,EAAE,0BAAkB;IAC1B;;OAEG;IACH,mBAAmB,EAAE,GAAG,EAAE;QACxB,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;IACzD,CAAC;IACD;;OAEG;IACH,cAAc,EAAE,GAAG,EAAE;QACnB,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;IACpD,CAAC;IACD;;OAEG;IACH,gBAAgB,EAAE,GAAG,EAAE;QACrB,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;IAChE,CAAC;IACD,UAAU,EAAE,EAAE;IACd;;OAEG;IACH,aAAa,EAAE,GAAG,EAAE;QAClB,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;IACnD,CAAC;IACD;;OAEG;IACH,mBAAmB,EAAE,GAAG,EAAE;QACxB,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;IACzD,CAAC;IACD;;OAEG;IACH,iBAAiB,EAAE,GAAG,EAAE;QACtB,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC7C,CAAC;IACD,eAAe,EAAE,6CAAe,CAAC,IAAI;IACrC,uBAAuB,EAAE,EAAE;IAC3B,MAAM,EAAE,IAAI;CACb,CAAC,CAAC;AAEH;;;;;;GAMG;AACI,MAAM,mBAAmB,GAAgC,CAAC,EAC/D,QAAQ,GACT,EAAE,EAAE;IACH,MAAM,CAAC,SAAS,EAAE,YAAY,CAAC,GAAG,IAAA,gBAAQ,EAA8B;QACtE,CAAC,0BAAkB,CAAC,EAAE,CAAC,EAAE,0BAAkB;KAC5C,CAAC,CAAC;IACH,MAAM,MAAM,GAAG,IAAA,sCAAc,GAAE,CAAC;IAChC,MAAM,EAAE,aAAa,EAAE,YAAY,EAAE,yBAAyB,EAAE,GAC9D,IAAA,0CAAgB,GAAE,CAAC;IACrB,MAAM,CAAC,UAAU,EAAE,aAAa,CAAC,GAAG,IAAA,gBAAQ,EAAC,EAAE,CAAC,CAAC;IAEjD,MAAM,CAAC,eAAe,EAAE,kBAAkB,CAAC,GAAG,IAAA,gBAAQ,EACpD,0BAAkB,CAAC,EAAE,CACtB,CAAC;IACF,MAAM,aAAa,GAA4B,SAAS,CAAC,eAAe,CAAC,CAAC;IAE1E,sFAAsF;IACtF,MAAM,mBAAmB,GAAG,IAAA,eAAO,EAAC,GAAG,EAAE;QACvC,MAAM,YAAY,GAAG,IAAI,GAAG,EAA8B,CAAC;QAC3D,IAAI,aAAa,EAAE,CAAC;YAClB,KAAK,MAAM,OAAO,IAAI,aAAa,CAAC,QAAQ,EAAE,CAAC;gBAC7C,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YACxC,CAAC;QACH,CAAC;QACD,OAAO,YAAY,CAAC;IACtB,CAAC,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC;IAEpB,MAAM,WAAW,GAAG,IAAA,mBAAW,EAC7B,KAAK,EAAE,QAAgB,EAAE,EAAE;QACzB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAC5D,MAAM,4BAA4B,GAAG;YACnC,GAAG,MAAM;YACT,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;gBACxC,IAAI,mBAAmB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,CAAC;oBACxC,MAAM,eAAe,GAAG,mBAAmB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;oBAC5D,OAAO;wBACL,GAAG,eAAe;wBAClB,GAAG,OAAO;qBACX,CAAC;gBACJ,CAAC;gBACD,IAAI,OAAO,CAAC,SAAS,EAAE,aAAa,EAAE,CAAC;oBACrC,MAAM,oBAAoB,GAAG,IAAA,+CAA0B,EACrD,OAAO,EACP,aAAa,CACd,CAAC;oBACF,OAAO,oBAAoB,CAAC;gBAC9B,CAAC;gBACD,OAAO,OAAO,CAAC;YACjB,CAAC,CAAC;SACH,CAAC;QAEF,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE;YACvB,MAAM,gBAAgB,GAAG;gBACvB,GAAG,OAAO;gBACV,CAAC,QAAQ,CAAC,EAAE,4BAA4B;aACzC,CAAC;YACF,OAAO,gBAAgB,CAAC;QAC1B,CAAC,CAAC,CAAC;IACL,CAAC,EACD,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,aAAa,EAAE,mBAAmB,CAAC,CAC1D,CAAC;IAEF,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,IACE,eAAe;YACf,eAAe,KAAK,0BAAkB,CAAC,EAAE;YACzC,CAAC,SAAS,CAAC,eAAe,CAAC,EAC3B,CAAC;YACD,WAAW,CAAC,eAAe,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC,EAAE,CAAC,eAAe,EAAE,WAAW,EAAE,SAAS,CAAC,CAAC,CAAC;IAE9C,MAAM,gBAAgB,GAAG,IAAA,mBAAW,EAClC,KAAK,EACH,OAA2B,EAC3B,YAAY,GAAG,IAAI,EACnB,YAAoB,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAC5C,EAAE;QACF,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,OAAO,CAAC,IAAI,CAAC,wDAAwD,CAAC,CAAC;YACvE,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,MAAM,WAAW,GAEb;YACF,GAAG,OAAO;YACV,iBAAiB,EACf,OAAO,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,IAAA,2BAAgB,GAAE,CAAC,CAAC,CAAC,SAAS;YAC1D,SAAS;SACV,CAAC;QACF,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;QAClC,MAAM,SAAS,GAAG,WAAW,CAAC,EAAE,CAAC;QACjC,sDAAsD;QACtD,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE;YACvB,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,OAAO,OAAO,CAAC;YACjB,CAAC;YACD,MAAM,YAAY,GAAG,OAAO,CAAC,QAAQ,CAAC,EAAE,QAAQ,IAAI,EAAE,CAAC;YACvD,MAAM,WAAW,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,KAAK,SAAS,CAAC,CAAC;YACrE,gDAAgD;YAChD,MAAM,eAAe,GAAG,WAAW;gBACjC,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;oBACvB,IAAI,GAAG,CAAC,EAAE,KAAK,SAAS,EAAE,CAAC;wBACzB,OAAO,WAAW,CAAC;oBACrB,CAAC;oBACD,OAAO,GAAG,CAAC;gBACb,CAAC,CAAC;gBACJ,CAAC,CAAC,CAAC,GAAG,YAAY,EAAE,WAAW,CAAC,CAAC;YAEnC,MAAM,gBAAgB,GAAG;gBACvB,GAAG,OAAO;gBACV,CAAC,QAAQ,CAAC,EAAE;oBACV,GAAG,OAAO,CAAC,QAAQ,CAAC;oBACpB,QAAQ,EAAE,eAAe;iBAC1B;aACF,CAAC;YACF,OAAO,gBAAgB,CAAC;QAC1B,CAAC,CAAC,CAAC;QAEH,IAAI,YAAY,EAAE,CAAC;YACjB,gEAAgE;YAChE,MAAM,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE;gBAC1D,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,oDAAoD;aACrD,CAAC,CAAC;QACL,CAAC;QACD,OAAO,SAAS,CAAC,QAAQ,CAAC,EAAE,QAAQ,IAAI,EAAE,CAAC;IAC7C,CAAC,EACD,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,aAAa,EAAE,SAAS,CAAC,CACzD,CAAC;IAEF,MAAM,mBAAmB,GAAG,IAAA,mBAAW,EACrC,KAAK,EACH,EAAU,EACV,OAA2B,EAC3B,YAAY,GAAG,IAAI,EACnB,YAAoB,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAC5C,EAAE;QACF,MAAM,WAAW,GAAuB;YACtC,GAAG,OAAO;YACV,SAAS;SACV,CAAC;QAEF,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE;YACvB,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;gBACtB,OAAO,OAAO,CAAC;YACjB,CAAC;YACD,MAAM,YAAY,GAAG,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,QAAQ,IAAI,EAAE,CAAC;YAC/D,MAAM,eAAe,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;gBAC/C,IAAI,GAAG,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC;oBAClB,OAAO,WAAW,CAAC;gBACrB,CAAC;gBACD,OAAO,GAAG,CAAC;YACb,CAAC,CAAC,CAAC;YACH,OAAO;gBACL,GAAG,OAAO;gBACV,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE;oBAClB,GAAG,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC;oBAC5B,QAAQ,EAAE,eAAe;iBAC1B;aACF,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,IAAI,YAAY,EAAE,CAAC;YACjB,gEAAgE;YAChE,MAAM,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE;gBAC1D,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,oDAAoD;aACrD,CAAC,CAAC;QACL,CAAC;IACH,CAAC,EACD,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAC/B,CAAC;IAEF,MAAM,cAAc,GAAG,IAAA,mBAAW,EAAC,GAAG,EAAE;QACtC,kBAAkB,CAAC,0BAAkB,CAAC,EAAE,CAAC,CAAC;QAC1C,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE;YACvB,OAAO;gBACL,GAAG,OAAO;gBACV,CAAC,0BAAkB,CAAC,EAAE,CAAC,EAAE,0BAAkB;aAC5C,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,mBAAmB,GAAG,IAAA,mBAAW,EACrC,KAAK,EAAE,QAAgB,EAAE,KAAK,GAAG,IAAI,EAAE,EAAE;QACvC,IAAI,QAAQ,KAAK,0BAAkB,CAAC,EAAE,EAAE,CAAC;YACvC,OAAO,CAAC,IAAI,CAAC,gDAAgD,CAAC,CAAC;YAC/D,OAAO;QACT,CAAC;QACD,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QAC7B,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE;YACvB,IAAI,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACtB,OAAO,OAAO,CAAC;YACjB,CAAC;YACD,yEAAyE;YACzE,MAAM,gBAAgB,GAAG;gBACvB,GAAG,OAAO;gBACV,CAAC,QAAQ,CAAC,EAAE;oBACV,GAAG,OAAO,CAAC,0BAAkB,CAAC,EAAE,CAAC;oBACjC,EAAE,EAAE,QAAQ;iBACb;aACF,CAAC;YACF,OAAO,gBAAgB,CAAC;QAC1B,CAAC,CAAC,CAAC;QACH,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,WAAW,CAAC,QAAQ,CAAC,CAAC;QAC9B,CAAC;IACH,CAAC,EACD,CAAC,WAAW,CAAC,CACd,CAAC;IAEF,MAAM,kBAAkB,GAAG,IAAA,mBAAW,EACpC,CAAC,QAAgB,EAAE,KAAsB,EAAE,aAAsB,EAAE,EAAE;QACnE,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE;YACvB,MAAM,gBAAgB,GAAG;gBACvB,GAAG,OAAO;gBACV,CAAC,QAAQ,CAAC,EAAE;oBACV,GAAG,OAAO,CAAC,QAAQ,CAAC;oBACpB,eAAe,EAAE,KAAK;oBACtB,aAAa,EAAE,aAAa;iBAC7B;aACF,CAAC;YACF,OAAO,gBAAgB,CAAC;QAC1B,CAAC,CAAC,CAAC;IACL,CAAC,EACD,EAAE,CACH,CAAC;IAEF,MAAM,mBAAmB,GAAG,IAAA,mBAAW,EACrC,KAAK,EACH,MAAiE,EACjE,MAAgD,EAChD,QAAgB,EACa,EAAE;QAC/B,IAAI,YAAsD,CAAC;QAC3D,IAAI,cAAc,GAAG,KAAK,CAAC;QAC3B,kBAAkB,CAAC,QAAQ,EAAE,6CAAe,CAAC,kBAAkB,CAAC,CAAC;QAEjE,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YACjC,IAAI,KAAK,CAAC,kBAAkB,CAAC,eAAe,EAAE,CAAC;gBAC7C,kBAAkB,CAChB,KAAK,CAAC,kBAAkB,CAAC,QAAQ,EACjC,6CAAe,CAAC,gBAAgB,CACjC,CAAC;gBACF,MAAM,gBAAgB,GAAG,MAAM,IAAA,4BAAc,EAC3C,KAAK,CAAC,kBAAkB,EACxB,YAAY,CACb,CAAC;gBACF,MAAM,sBAAsB,GAC1B,OAAO,gBAAgB,KAAK,QAAQ;oBAClC,CAAC,CAAC,gBAAgB;oBAClB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;gBACvC,MAAM,sBAAsB,GAC1B;oBACE,GAAG,MAAM;oBACT,eAAe,EAAE;wBACf,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,sBAAsB,EAAE,CAAC;wBACzD,IAAI,EAAE,MAAM;wBACZ,UAAU,EAAE,eAAe;wBAC3B,SAAS,EAAE,KAAK,CAAC,kBAAkB,CAAC,SAAS;wBAC7C,YAAY,EAAE,KAAK,CAAC,kBAAkB,CAAC,YAAY;qBACpD;iBACF,CAAC;gBACJ,kBAAkB,CAChB,KAAK,CAAC,kBAAkB,CAAC,QAAQ,EACjC,6CAAe,CAAC,kBAAkB,CACnC,CAAC;gBACF,MAAM,sBAAsB,GAAG,MAAM,IAAA,8BAAa,EAChD,MAAM,EACN,sBAAsB,EACtB,KAAK,CAAC,kBAAkB,CAAC,QAAQ,CAClC,CAAC;gBAEF,OAAO,MAAM,mBAAmB,CAC9B,sBAAsB,EACtB,sBAAsB,EACtB,KAAK,CAAC,kBAAkB,CAAC,QAAQ,CAClC,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,IACE,CAAC,cAAc;oBACf,KAAK,CAAC,kBAAkB,CAAC,QAAQ;oBACjC,KAAK,CAAC,kBAAkB,CAAC,QAAQ,KAAK,aAAa,EAAE,EAAE,EACvD,CAAC;oBACD,cAAc,GAAG,IAAI,CAAC;oBACtB,MAAM,mBAAmB,CAAC,KAAK,CAAC,kBAAkB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;gBACtE,CAAC;gBAED,IAAI,CAAC,YAAY,EAAE,CAAC;oBAClB,YAAY,GAAG,KAAK,CAAC,kBAAkB,CAAC,SAAS,EAAE,aAAa;wBAC9D,CAAC,CAAC,IAAA,+CAA0B,EACxB,KAAK,CAAC,kBAAkB,EACxB,aAAa,CACd;wBACH,CAAC,CAAC,KAAK,CAAC,kBAAkB,CAAC;oBAC7B,MAAM,gBAAgB,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC;gBAC9C,CAAC;qBAAM,CAAC;oBACN,+EAA+E;oBAC/E,MAAM,YAAY,GAChB,KAAK,CAAC,kBAAkB,CAAC,EAAE,KAAK,YAAY,CAAC,EAAE,CAAC;oBAElD,YAAY,GAAG,KAAK,CAAC,kBAAkB,CAAC,SAAS,EAAE,aAAa;wBAC9D,CAAC,CAAC,IAAA,+CAA0B,EACxB,KAAK,CAAC,kBAAkB,EACxB,aAAa,CACd;wBACH,CAAC,CAAC,KAAK,CAAC,kBAAkB,CAAC;oBAE7B,IAAI,YAAY,EAAE,CAAC;wBACjB,MAAM,gBAAgB,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC;oBAC9C,CAAC;yBAAM,CAAC;wBACN,MAAM,mBAAmB,CAAC,YAAY,CAAC,EAAE,EAAE,YAAY,EAAE,KAAK,CAAC,CAAC;oBAClE,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,kBAAkB,CAChB,YAAY,EAAE,QAAQ,IAAI,QAAQ,EAClC,6CAAe,CAAC,QAAQ,CACzB,CAAC;QACF,OAAO,CACL,YAAY,IAAI;YACd,QAAQ,EAAE,EAAE;YACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,yBAAyB,EAAE,CAAC;YAC5D,IAAI,EAAE,WAAW;YACjB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE;YACvB,cAAc,EAAE,EAAE;SACnB,CACF,CAAC;IACJ,CAAC,EACD;QACE,gBAAgB;QAChB,MAAM;QACN,aAAa;QACb,aAAa,EAAE,EAAE;QACjB,mBAAmB;QACnB,YAAY;QACZ,mBAAmB;QACnB,kBAAkB;KACnB,CACF,CAAC;IAEF,MAAM,iBAAiB,GAAG,IAAA,mBAAW,EACnC,KAAK,EACH,OAAe,EACf,UAII,EAAE,QAAQ,EAAE,0BAAkB,CAAC,EAAE,EAAE,EACV,EAAE;QAC/B,MAAM,EAAE,QAAQ,GAAG,aAAa,CAAC,EAAE,EAAE,cAAc,EAAE,GAAG,OAAO,CAAC;QAChE,kBAAkB,CAAC,QAAQ,EAAE,6CAAe,CAAC,kBAAkB,CAAC,CAAC;QAEjE,gBAAgB,CACd;YACE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;YAC1C,iBAAiB,EAAE,IAAI;YACvB,IAAI,EAAE,MAAM;YACZ,QAAQ,EAAE,QAAQ;YAClB,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE;YACvB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,cAAc,EAAE,EAAE;SACnB,EACD,KAAK,CACN,CAAC;QAEF,MAAM,mBAAmB,GAAG,IAAA,iCAAsB,EAChD,aAAa,EACb,YAAY,EACZ,yBAAyB,CAC1B,CAAC;QACF,MAAM,iBAAiB,GAAG,IAAA,+BAAoB,EAC5C,YAAY,EACZ,yBAAyB,CAC1B,CAAC;QACF,MAAM,MAAM,GAA6C;YACvD,eAAe,EAAE;gBACf,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;gBAC1C,IAAI,EAAE,MAAM;aACb;YACD,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,mBAAmB,EAAE,mBAAmB;YACxC,WAAW,EAAE,iBAAiB,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAC1C,IAAA,oCAAyB,EAAC,IAAI,CAAC,CAChC;SACF,CAAC;QAEF,IAAI,cAAc,EAAE,CAAC;YACnB,MAAM,qBAAqB,GAAG,MAAM,IAAA,8BAAa,EAC/C,MAAM,EACN,MAAM,EACN,QAAQ,KAAK,0BAAkB,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAC1D,CAAC;YACF,OAAO,MAAM,mBAAmB,CAC9B,qBAAqB,EACrB,MAAM,EACN,QAAQ,CACT,CAAC;QACJ,CAAC;QACD,IAAI,eAAe,GAAG,MAAM,CAAC,QAAQ,KAAK,0BAAkB,CAAC,EAAE;YAC7D,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC;YACrC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAC;QAEvD,mBAAmB;QACnB,OAAO,eAAe,CAAC,kBAAkB,CAAC,eAAe,EAAE,CAAC;YAC1D,kBAAkB,CAAC,QAAQ,EAAE,6CAAe,CAAC,gBAAgB,CAAC,CAAC;YAC/D,MAAM,gBAAgB,GAAG,MAAM,IAAA,4BAAc,EAC3C,eAAe,CAAC,kBAAkB,EAClC,YAAY,CACb,CAAC;YACF,MAAM,kBAAkB,GACtB,OAAO,gBAAgB,KAAK,QAAQ;gBAClC,CAAC,CAAC,gBAAgB;gBAClB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YACvC,MAAM,sBAAsB,GAC1B;gBACE,GAAG,MAAM;gBACT,eAAe,EAAE;oBACf,GAAG,MAAM,CAAC,eAAe;oBACzB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC;oBACrD,IAAI,EAAE,MAAM;oBACZ,UAAU,EAAE,eAAe;oBAC3B,SAAS,EAAE,eAAe,CAAC,kBAAkB,CAAC,SAAS;oBACvD,YAAY,EAAE,eAAe,CAAC,kBAAkB,CAAC,YAAY;iBAC9D;aACF,CAAC;YACJ,kBAAkB,CAAC,QAAQ,EAAE,6CAAe,CAAC,mBAAmB,CAAC,CAAC;YAClE,eAAe,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CACrD,eAAe,CAAC,kBAAkB,CAAC,QAAQ,EAC3C,sBAAsB,CACvB,CAAC;QACJ,CAAC;QAED,MAAM,YAAY,GAAG,eAAe,CAAC,kBAAkB,CAAC,SAAS;YAC/D,EAAE,aAAa;YACf,CAAC,CAAC,IAAA,+CAA0B,EACxB,eAAe,CAAC,kBAAkB,EAClC,aAAa,CACd;YACH,CAAC,CAAC,eAAe,CAAC,kBAAkB,CAAC;QACvC,MAAM,mBAAmB,CAAC,eAAe,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QACvE,kBAAkB,CAChB,eAAe,CAAC,kBAAkB,CAAC,QAAQ,EAC3C,6CAAe,CAAC,QAAQ,CACzB,CAAC;QACF,OAAO,YAAY,CAAC;IACtB,CAAC,EACD;QACE,aAAa;QACb,YAAY;QACZ,yBAAyB;QACzB,aAAa,CAAC,EAAE;QAChB,mBAAmB;QACnB,gBAAgB;QAChB,MAAM;QACN,kBAAkB;QAClB,mBAAmB;KACpB,CACF,CAAC;IAEF,OAAO,CACL,8BAAC,0BAAkB,CAAC,QAAQ,IAC1B,KAAK,EAAE;YACL,MAAM,EAAE,aAAa;YACrB,mBAAmB;YACnB,cAAc;YACd,gBAAgB;YAChB,mBAAmB;YACnB,UAAU;YACV,aAAa;YACb,iBAAiB;YACjB,eAAe,EAAE,CAAC,aAAa,EAAE,eAAe;gBAC9C,6CAAe,CAAC,IAAI,CAAoB;YAC1C,uBAAuB,EAAE,aAAa,EAAE,aAAa,IAAI,EAAE;YAC3D,MAAM,EAAE,IAAA,yCAAW,EACjB,CAAC,aAAa,EAAE,eAAe;gBAC7B,6CAAe,CAAC,IAAI,CAAoB,CAC3C;SACF,IAEA,QAAQ,CACmB,CAC/B,CAAC;AACJ,CAAC,CAAC;AAxeW,QAAA,mBAAmB,uBAwe9B;AAEF;;;;GAIG;AACI,MAAM,cAAc,GAAG,GAAG,EAAE;IACjC,MAAM,OAAO,GAAG,IAAA,kBAAU,EAAC,0BAAkB,CAAC,CAAC;IAC/C,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;QAC1B,MAAM,IAAI,KAAK,CAAC,0DAA0D,CAAC,CAAC;IAC9E,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC;AANW,QAAA,cAAc,kBAMzB","sourcesContent":["\"use client\";\nimport TamboAI, { advanceStream } from \"@tambo-ai/typescript-sdk\";\nimport React, {\n  createContext,\n  PropsWithChildren,\n  useCallback,\n  useContext,\n  useEffect,\n  useMemo,\n  useState,\n} from \"react\";\nimport {\n  GenerationStage,\n  isIdleStage,\n  TamboThreadMessage,\n} from \"../model/generate-component-response\";\nimport { TamboThread } from \"../model/tambo-thread\";\nimport { renderComponentIntoMessage } from \"../util/generate-component\";\nimport {\n  getAvailableComponents,\n  getClientContext,\n  getUnassociatedTools,\n  mapTamboToolToContextTool,\n} from \"../util/registry\";\nimport { handleToolCall } from \"../util/tool-caller\";\nimport { useTamboClient } from \"./tambo-client-provider\";\nimport { useTamboRegistry } from \"./tambo-registry-provider\";\n\nexport interface TamboThreadContextProps {\n  /** The current thread */\n  thread: TamboThread;\n  /** Switch to a different thread */\n  switchCurrentThread: (threadId: string, fetch?: boolean) => void;\n  /** Start a new thread */\n  startNewThread: () => void;\n  /** Add a message to the current thread */\n  addThreadMessage: (\n    message: TamboThreadMessage,\n    sendToServer: boolean,\n  ) => Promise<TamboAI.Beta.Threads.ThreadMessage[]>;\n  /** Update a message in the current thread */\n  updateThreadMessage: (\n    id: string,\n    message: TamboThreadMessage,\n    sendToServer: boolean,\n  ) => Promise<void>;\n  /** The input value of the current thread */\n  inputValue: string;\n  /** Set the input value of the current thread */\n  setInputValue: (value: string) => void;\n  /** Send a message to the current thread */\n  sendThreadMessage: (\n    message: string,\n    options: {\n      threadId?: string;\n      streamResponse?: boolean;\n      contextKey?: string;\n    },\n  ) => Promise<TamboThreadMessage>;\n  /** The generation stage of the current thread - updated as the thread progresses */\n  generationStage: GenerationStage;\n  /** The generation status message of the current thread - updated as the thread progresses */\n  generationStatusMessage: string;\n  /** Whether the thread is idle */\n  isIdle: boolean;\n}\n\n/**\n * This is a stub entry for when the thread is not yet created, the first time\n * the user sends a message\n *\n * Note that the consumer needs to be careful never to send `PLACEHOLDER_THREAD.id` to the server,\n * as this doesn't really exist on the server side.\n */\nexport const PLACEHOLDER_THREAD: TamboThread = {\n  id: \"placeholder\",\n  messages: [],\n  createdAt: \"\",\n  projectId: \"\",\n  updatedAt: \"\",\n  metadata: {},\n};\n\nexport const TamboThreadContext = createContext<TamboThreadContextProps>({\n  thread: PLACEHOLDER_THREAD,\n  /**\n   *\n   */\n  switchCurrentThread: () => {\n    throw new Error(\"switchCurrentThread not implemented\");\n  },\n  /**\n   *\n   */\n  startNewThread: () => {\n    throw new Error(\"startNewThread not implemented\");\n  },\n  /**\n   *\n   */\n  addThreadMessage: () => {\n    throw new Error(\"updateThreadMessageHistory not implemented\");\n  },\n  inputValue: \"\",\n  /**\n   *\n   */\n  setInputValue: () => {\n    throw new Error(\"setInputValue not implemented\");\n  },\n  /**\n   *\n   */\n  updateThreadMessage: () => {\n    throw new Error(\"updateThreadMessage not implemented\");\n  },\n  /**\n   *\n   */\n  sendThreadMessage: () => {\n    throw new Error(\"advance not implemented\");\n  },\n  generationStage: GenerationStage.IDLE,\n  generationStatusMessage: \"\",\n  isIdle: true,\n});\n\n/**\n * The TamboThreadProvider is a React provider that provides a thread context\n * to the descendants of the provider.\n * @param props - The props for the TamboThreadProvider\n * @param props.children - The children to wrap\n * @returns The TamboThreadProvider component\n */\nexport const TamboThreadProvider: React.FC<PropsWithChildren> = ({\n  children,\n}) => {\n  const [threadMap, setThreadMap] = useState<Record<string, TamboThread>>({\n    [PLACEHOLDER_THREAD.id]: PLACEHOLDER_THREAD,\n  });\n  const client = useTamboClient();\n  const { componentList, toolRegistry, componentToolAssociations } =\n    useTamboRegistry();\n  const [inputValue, setInputValue] = useState(\"\");\n\n  const [currentThreadId, setCurrentThreadId] = useState<string>(\n    PLACEHOLDER_THREAD.id,\n  );\n  const currentThread: TamboThread | undefined = threadMap[currentThreadId];\n\n  // Use existing messages from the current thread to avoid re-generating any components\n  const currentMessageCache = useMemo(() => {\n    const messageCache = new Map<string, TamboThreadMessage>();\n    if (currentThread) {\n      for (const message of currentThread.messages) {\n        messageCache.set(message.id, message);\n      }\n    }\n    return messageCache;\n  }, [currentThread]);\n\n  const fetchThread = useCallback(\n    async (threadId: string) => {\n      const thread = await client.beta.threads.retrieve(threadId);\n      const threadWithRenderedComponents = {\n        ...thread,\n        messages: thread.messages.map((message) => {\n          if (currentMessageCache.has(message.id)) {\n            const renderedMessage = currentMessageCache.get(message.id);\n            return {\n              ...renderedMessage,\n              ...message,\n            };\n          }\n          if (message.component?.componentName) {\n            const messageWithComponent = renderComponentIntoMessage(\n              message,\n              componentList,\n            );\n            return messageWithComponent;\n          }\n          return message;\n        }),\n      };\n\n      setThreadMap((prevMap) => {\n        const updatedThreadMap = {\n          ...prevMap,\n          [threadId]: threadWithRenderedComponents,\n        };\n        return updatedThreadMap;\n      });\n    },\n    [client.beta.threads, componentList, currentMessageCache],\n  );\n\n  useEffect(() => {\n    if (\n      currentThreadId &&\n      currentThreadId !== PLACEHOLDER_THREAD.id &&\n      !threadMap[currentThreadId]\n    ) {\n      fetchThread(currentThreadId);\n    }\n  }, [currentThreadId, fetchThread, threadMap]);\n\n  const addThreadMessage = useCallback(\n    async (\n      message: TamboThreadMessage,\n      sendToServer = true,\n      createdAt: string = new Date().toISOString(),\n    ) => {\n      if (!currentThread) {\n        console.warn(\"Cannot add messages if we do not have a current thread\");\n        return [];\n      }\n      const chatMessage: TamboThreadMessage & {\n        additionalContext?: string;\n      } = {\n        ...message,\n        additionalContext:\n          message.role === \"user\" ? getClientContext() : undefined,\n        createdAt,\n      };\n      const threadId = message.threadId;\n      const messageId = chatMessage.id;\n      // optimistically update the thread in the local state\n      setThreadMap((prevMap) => {\n        if (!threadId) {\n          return prevMap;\n        }\n        const prevMessages = prevMap[threadId]?.messages || [];\n        const haveMessage = prevMessages.find((msg) => msg.id === messageId);\n        // Update in place if the message already exists\n        const updatedMessages = haveMessage\n          ? prevMessages.map((msg) => {\n              if (msg.id === messageId) {\n                return chatMessage;\n              }\n              return msg;\n            })\n          : [...prevMessages, chatMessage];\n\n        const updatedThreadMap = {\n          ...prevMap,\n          [threadId]: {\n            ...prevMap[threadId],\n            messages: updatedMessages,\n          },\n        };\n        return updatedThreadMap;\n      });\n\n      if (sendToServer) {\n        // TODO: if this fails, we need to revert the local state update\n        await client.beta.threads.messages.create(message.threadId, {\n          content: message.content,\n          role: message.role,\n          // additionalContext: chatMessage.additionalContext,\n        });\n      }\n      return threadMap[threadId]?.messages || [];\n    },\n    [client.beta.threads.messages, currentThread, threadMap],\n  );\n\n  const updateThreadMessage = useCallback(\n    async (\n      id: string,\n      message: TamboThreadMessage,\n      sendToServer = true,\n      createdAt: string = new Date().toISOString(),\n    ) => {\n      const chatMessage: TamboThreadMessage = {\n        ...message,\n        createdAt,\n      };\n\n      setThreadMap((prevMap) => {\n        if (!message.threadId) {\n          return prevMap;\n        }\n        const prevMessages = prevMap[message.threadId]?.messages || [];\n        const updatedMessages = prevMessages.map((msg) => {\n          if (msg.id === id) {\n            return chatMessage;\n          }\n          return msg;\n        });\n        return {\n          ...prevMap,\n          [message.threadId]: {\n            ...prevMap[message.threadId],\n            messages: updatedMessages,\n          },\n        };\n      });\n      if (sendToServer) {\n        // TODO: if this fails, we need to revert the local state update\n        await client.beta.threads.messages.create(message.threadId, {\n          content: message.content,\n          role: message.role,\n          // additionalContext: chatMessage.additionalContext,\n        });\n      }\n    },\n    [client.beta.threads.messages],\n  );\n\n  const startNewThread = useCallback(() => {\n    setCurrentThreadId(PLACEHOLDER_THREAD.id);\n    setThreadMap((prevMap) => {\n      return {\n        ...prevMap,\n        [PLACEHOLDER_THREAD.id]: PLACEHOLDER_THREAD,\n      };\n    });\n  }, []);\n\n  const switchCurrentThread = useCallback(\n    async (threadId: string, fetch = true) => {\n      if (threadId === PLACEHOLDER_THREAD.id) {\n        console.warn(\"Switching to placeholder thread, may be a bug.\");\n        return;\n      }\n      setCurrentThreadId(threadId);\n      setThreadMap((prevMap) => {\n        if (prevMap[threadId]) {\n          return prevMap;\n        }\n        // If this is a new thread, add placeholder thread messages to the thread\n        const updatedThreadMap = {\n          ...prevMap,\n          [threadId]: {\n            ...prevMap[PLACEHOLDER_THREAD.id],\n            id: threadId,\n          },\n        };\n        return updatedThreadMap;\n      });\n      if (fetch) {\n        await fetchThread(threadId);\n      }\n    },\n    [fetchThread],\n  );\n\n  const updateThreadStatus = useCallback(\n    (threadId: string, stage: GenerationStage, statusMessage?: string) => {\n      setThreadMap((prevMap) => {\n        const updatedThreadMap = {\n          ...prevMap,\n          [threadId]: {\n            ...prevMap[threadId],\n            generationStage: stage,\n            statusMessage: statusMessage,\n          },\n        };\n        return updatedThreadMap;\n      });\n    },\n    [],\n  );\n\n  const handleAdvanceStream = useCallback(\n    async (\n      stream: AsyncIterable<TamboAI.Beta.Threads.ThreadAdvanceResponse>,\n      params: TamboAI.Beta.Threads.ThreadAdvanceParams,\n      threadId: string,\n    ): Promise<TamboThreadMessage> => {\n      let finalMessage: Readonly<TamboThreadMessage> | undefined;\n      let hasSetThreadId = false;\n      updateThreadStatus(threadId, GenerationStage.STREAMING_RESPONSE);\n\n      for await (const chunk of stream) {\n        if (chunk.responseMessageDto.toolCallRequest) {\n          updateThreadStatus(\n            chunk.responseMessageDto.threadId,\n            GenerationStage.FETCHING_CONTEXT,\n          );\n          const toolCallResponse = await handleToolCall(\n            chunk.responseMessageDto,\n            toolRegistry,\n          );\n          const toolCallResponseString =\n            typeof toolCallResponse === \"string\"\n              ? toolCallResponse\n              : JSON.stringify(toolCallResponse);\n          const toolCallResponseParams: TamboAI.Beta.Threads.ThreadAdvanceParams =\n            {\n              ...params,\n              messageToAppend: {\n                content: [{ type: \"text\", text: toolCallResponseString }],\n                role: \"tool\",\n                actionType: \"tool_response\",\n                component: chunk.responseMessageDto.component,\n                tool_call_id: chunk.responseMessageDto.tool_call_id,\n              },\n            };\n          updateThreadStatus(\n            chunk.responseMessageDto.threadId,\n            GenerationStage.STREAMING_RESPONSE,\n          );\n          const toolCallResponseStream = await advanceStream(\n            client,\n            toolCallResponseParams,\n            chunk.responseMessageDto.threadId,\n          );\n\n          return await handleAdvanceStream(\n            toolCallResponseStream,\n            toolCallResponseParams,\n            chunk.responseMessageDto.threadId,\n          );\n        } else {\n          if (\n            !hasSetThreadId &&\n            chunk.responseMessageDto.threadId &&\n            chunk.responseMessageDto.threadId !== currentThread?.id\n          ) {\n            hasSetThreadId = true;\n            await switchCurrentThread(chunk.responseMessageDto.threadId, false);\n          }\n\n          if (!finalMessage) {\n            finalMessage = chunk.responseMessageDto.component?.componentName\n              ? renderComponentIntoMessage(\n                  chunk.responseMessageDto,\n                  componentList,\n                )\n              : chunk.responseMessageDto;\n            await addThreadMessage(finalMessage, false);\n          } else {\n            // if we start getting a new message mid-stream, put the previous one on screen\n            const isNewMessage =\n              chunk.responseMessageDto.id !== finalMessage.id;\n\n            finalMessage = chunk.responseMessageDto.component?.componentName\n              ? renderComponentIntoMessage(\n                  chunk.responseMessageDto,\n                  componentList,\n                )\n              : chunk.responseMessageDto;\n\n            if (isNewMessage) {\n              await addThreadMessage(finalMessage, false);\n            } else {\n              await updateThreadMessage(finalMessage.id, finalMessage, false);\n            }\n          }\n        }\n      }\n\n      updateThreadStatus(\n        finalMessage?.threadId ?? threadId,\n        GenerationStage.COMPLETE,\n      );\n      return (\n        finalMessage ?? {\n          threadId: \"\",\n          content: [{ type: \"text\", text: `Error processing stream` }],\n          role: \"assistant\",\n          createdAt: new Date().toISOString(),\n          id: crypto.randomUUID(),\n          componentState: {},\n        }\n      );\n    },\n    [\n      addThreadMessage,\n      client,\n      componentList,\n      currentThread?.id,\n      switchCurrentThread,\n      toolRegistry,\n      updateThreadMessage,\n      updateThreadStatus,\n    ],\n  );\n\n  const sendThreadMessage = useCallback(\n    async (\n      message: string,\n      options: {\n        threadId?: string;\n        streamResponse?: boolean;\n        contextKey?: string;\n      } = { threadId: PLACEHOLDER_THREAD.id },\n    ): Promise<TamboThreadMessage> => {\n      const { threadId = currentThread.id, streamResponse } = options;\n      updateThreadStatus(threadId, GenerationStage.CHOOSING_COMPONENT);\n\n      addThreadMessage(\n        {\n          content: [{ type: \"text\", text: message }],\n          renderedComponent: null,\n          role: \"user\",\n          threadId: threadId,\n          id: crypto.randomUUID(),\n          createdAt: new Date().toISOString(),\n          componentState: {},\n        },\n        false,\n      );\n\n      const availableComponents = getAvailableComponents(\n        componentList,\n        toolRegistry,\n        componentToolAssociations,\n      );\n      const unassociatedTools = getUnassociatedTools(\n        toolRegistry,\n        componentToolAssociations,\n      );\n      const params: TamboAI.Beta.Threads.ThreadAdvanceParams = {\n        messageToAppend: {\n          content: [{ type: \"text\", text: message }],\n          role: \"user\",\n        },\n        contextKey: options.contextKey,\n        availableComponents: availableComponents,\n        clientTools: unassociatedTools.map((tool) =>\n          mapTamboToolToContextTool(tool),\n        ),\n      };\n\n      if (streamResponse) {\n        const advanceStreamResponse = await advanceStream(\n          client,\n          params,\n          threadId === PLACEHOLDER_THREAD.id ? undefined : threadId,\n        );\n        return await handleAdvanceStream(\n          advanceStreamResponse,\n          params,\n          threadId,\n        );\n      }\n      let advanceResponse = await (threadId === PLACEHOLDER_THREAD.id\n        ? client.beta.threads.advance(params)\n        : client.beta.threads.advanceById(threadId, params));\n\n      //handle tool calls\n      while (advanceResponse.responseMessageDto.toolCallRequest) {\n        updateThreadStatus(threadId, GenerationStage.FETCHING_CONTEXT);\n        const toolCallResponse = await handleToolCall(\n          advanceResponse.responseMessageDto,\n          toolRegistry,\n        );\n        const toolResponseString =\n          typeof toolCallResponse === \"string\"\n            ? toolCallResponse\n            : JSON.stringify(toolCallResponse);\n        const toolCallResponseParams: TamboAI.Beta.Threads.ThreadAdvanceParams =\n          {\n            ...params,\n            messageToAppend: {\n              ...params.messageToAppend,\n              content: [{ type: \"text\", text: toolResponseString }],\n              role: \"tool\",\n              actionType: \"tool_response\",\n              component: advanceResponse.responseMessageDto.component,\n              tool_call_id: advanceResponse.responseMessageDto.tool_call_id,\n            },\n          };\n        updateThreadStatus(threadId, GenerationStage.HYDRATING_COMPONENT);\n        advanceResponse = await client.beta.threads.advanceById(\n          advanceResponse.responseMessageDto.threadId,\n          toolCallResponseParams,\n        );\n      }\n\n      const finalMessage = advanceResponse.responseMessageDto.component\n        ?.componentName\n        ? renderComponentIntoMessage(\n            advanceResponse.responseMessageDto,\n            componentList,\n          )\n        : advanceResponse.responseMessageDto;\n      await switchCurrentThread(advanceResponse.responseMessageDto.threadId);\n      updateThreadStatus(\n        advanceResponse.responseMessageDto.threadId,\n        GenerationStage.COMPLETE,\n      );\n      return finalMessage;\n    },\n    [\n      componentList,\n      toolRegistry,\n      componentToolAssociations,\n      currentThread.id,\n      switchCurrentThread,\n      addThreadMessage,\n      client,\n      updateThreadStatus,\n      handleAdvanceStream,\n    ],\n  );\n\n  return (\n    <TamboThreadContext.Provider\n      value={{\n        thread: currentThread,\n        switchCurrentThread,\n        startNewThread,\n        addThreadMessage,\n        updateThreadMessage,\n        inputValue,\n        setInputValue,\n        sendThreadMessage,\n        generationStage: (currentThread?.generationStage ??\n          GenerationStage.IDLE) as GenerationStage,\n        generationStatusMessage: currentThread?.statusMessage ?? \"\",\n        isIdle: isIdleStage(\n          (currentThread?.generationStage ??\n            GenerationStage.IDLE) as GenerationStage,\n        ),\n      }}\n    >\n      {children}\n    </TamboThreadContext.Provider>\n  );\n};\n\n/**\n * The useTamboThread hook provides access to the current thread context\n * to the descendants of the TamboThreadProvider.\n * @returns All state and actions for the current thread\n */\nexport const useTamboThread = () => {\n  const context = useContext(TamboThreadContext);\n  if (context === undefined) {\n    throw new Error(\"useTamboThread must be used within a TamboThreadProvider\");\n  }\n  return context;\n};\n"]}

package/esm/hooks/use-suggestions.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"use-suggestions.d.ts","sourceRoot":"","sources":["../../src/hooks/use-suggestions.ts"],"names":[],"mappings":"AAAA,OAAO,OAAO,MAAM,0BAA0B,CAAC;AAO/C,OAAO,EACL,sBAAsB,EAEvB,MAAM,qBAAqB,CAAC;AAE7B,OAAO,EACL,sBAAsB,EACtB,mBAAmB,EAGpB,MAAM,qBAAqB,CAAC;AAG7B;;GAEG;AACH,MAAM,WAAW,0BAA0B;IACzC,kEAAkE;IAClE,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB;AAED;;GAEG;AACH,MAAM,WAAW,iCAAiC;IAChD,4EAA4E;IAC5E,WAAW,EAAE,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;IAC/C,8CAA8C;IAC9C,oBAAoB,EAAE,MAAM,GAAG,IAAI,CAAC;IACpC;;;;OAIG;IACH,MAAM,EAAE,CAAC,aAAa,EAAE;QACtB,UAAU,EAAE,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;QAC5C,YAAY,CAAC,EAAE,OAAO,CAAC;KACxB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAEpB,0DAA0D;IAC1D,YAAY,EAAE,sBAAsB,CAClC,IAAI,EACJ,KAAK,EACL;QAAE,UAAU,EAAE,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;QAAC,YAAY,CAAC,EAAE,OAAO,CAAA;KAAE,CACxE,CAAC;IAEF,0DAA0D;IAC1D,cAAc,EAAE,sBAAsB,CACpC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,0BAA0B,GAAG,SAAS,EACvE,KAAK,EACL,eAAe,CAChB,CAAC;IAEF,yDAAyD;IACzD,iBAAiB,EAAE,mBAAmB,CACpC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,0BAA0B,GAAG,SAAS,EACvE,KAAK,CACN,CAAC;CACH;AAED,KAAK,yBAAyB,GAAG,sBAAsB,CAAC,GAAG,EAAE,KAAK,CAAC,GACjE,iCAAiC,CAAC;AAEpC;;;;GAIG;AACH,wBAAgB,mBAAmB,CACjC,OAAO,GAAE,0BAA+B,GACvC,yBAAyB,CAiI3B"}
+{"version":3,"file":"use-suggestions.d.ts","sourceRoot":"","sources":["../../src/hooks/use-suggestions.ts"],"names":[],"mappings":"AAAA,OAAO,OAAO,MAAM,0BAA0B,CAAC;AAQ/C,OAAO,EACL,sBAAsB,EAEvB,MAAM,qBAAqB,CAAC;AAE7B,OAAO,EACL,sBAAsB,EACtB,mBAAmB,EAGpB,MAAM,qBAAqB,CAAC;AAG7B;;GAEG;AACH,MAAM,WAAW,0BAA0B;IACzC,kEAAkE;IAClE,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB;AAED;;GAEG;AACH,MAAM,WAAW,iCAAiC;IAChD,4EAA4E;IAC5E,WAAW,EAAE,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;IAC/C,8CAA8C;IAC9C,oBAAoB,EAAE,MAAM,GAAG,IAAI,CAAC;IACpC;;;;OAIG;IACH,MAAM,EAAE,CAAC,aAAa,EAAE;QACtB,UAAU,EAAE,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;QAC5C,YAAY,CAAC,EAAE,OAAO,CAAC;KACxB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAEpB,0DAA0D;IAC1D,YAAY,EAAE,sBAAsB,CAClC,IAAI,EACJ,KAAK,EACL;QAAE,UAAU,EAAE,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;QAAC,YAAY,CAAC,EAAE,OAAO,CAAA;KAAE,CACxE,CAAC;IAEF,0DAA0D;IAC1D,cAAc,EAAE,sBAAsB,CACpC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,0BAA0B,GAAG,SAAS,EACvE,KAAK,EACL,eAAe,CAChB,CAAC;IAEF,yDAAyD;IACzD,iBAAiB,EAAE,mBAAmB,CACpC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,0BAA0B,GAAG,SAAS,EACvE,KAAK,CACN,CAAC;CACH;AAED,KAAK,yBAAyB,GAAG,sBAAsB,CAAC,GAAG,EAAE,KAAK,CAAC,GACjE,iCAAiC,CAAC;AAEpC;;;;GAIG;AACH,wBAAgB,mBAAmB,CACjC,OAAO,GAAE,0BAA+B,GACvC,yBAAyB,CAsI3B"}

package/esm/hooks/use-suggestions.js

@@ -1,4 +1,5 @@
 import { useEffect, useState } from "react";
+import { isIdleStage } from "../model/generate-component-response";
 import { validateInput } from "../model/validate-input";
 import { useTamboClient } from "../providers/tambo-client-provider";
 import { useTambo } from "../providers/tambo-provider";
@@ -15,7 +16,7 @@ import { INPUT_ERROR_MESSAGES, useTamboThreadInput } from "./use-thread-input";
  */
 export function useTamboSuggestions(options = {}) {
     const { maxSuggestions = 3 } = options;
-    const { thread } = useTamboThread();
+    const { thread, generationStage } = useTamboThread();
     const { sendThreadMessage } = useTambo();
     const tamboClient = useTamboClient();
     const { componentList, toolRegistry, componentToolAssociations } = useTamboRegistry();
@@ -28,12 +29,17 @@ export function useTamboSuggestions(options = {}) {
     useEffect(() => {
         setSelectedSuggestionId(null);
     }, [latestMessageId]);
+    const shouldGenerateSuggestions = latestMessageId && isLatestFromTambo && isIdleStage(generationStage);
     // Use React Query to fetch suggestions when a new hydra message is received
     const suggestionsResult = useTamboQuery({
-        // Only include latestMessageId in the queryKey if the message is from hydra
-        queryKey: ["suggestions", isLatestFromTambo ? latestMessageId : null],
+        // Make sure the query key changes when the message changes, so that we are
+        // always generating suggestions when there is a new message
+        queryKey: [
+            "suggestions",
+            shouldGenerateSuggestions ? latestMessageId : null,
+        ],
         queryFn: async () => {
-            if (!latestMessageId || !isLatestFromTambo) {
+            if (!shouldGenerateSuggestions) {
                 return [];
             }
             // Get registered components from the registry
@@ -75,7 +81,7 @@ export function useTamboSuggestions(options = {}) {
     // Generate suggestions mutation
     const generateMutationState = useTamboMutation({
         mutationFn: async (abortController) => {
-            if (!latestMessageId || !isLatestFromTambo) {
+            if (!shouldGenerateSuggestions) {
                 return undefined;
             }
             // Get registered components from the registry

package/esm/hooks/use-suggestions.js.map

@@ -1 +1 @@
-{"version":3,"file":"use-suggestions.js","sourceRoot":"","sources":["../../src/hooks/use-suggestions.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AAC5C,OAAO,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AACxD,OAAO,EAAE,cAAc,EAAE,MAAM,oCAAoC,CAAC;AACpE,OAAO,EAAE,QAAQ,EAAE,MAAM,6BAA6B,CAAC;AACvD,OAAO,EAAE,gBAAgB,EAAE,MAAM,sCAAsC,CAAC;AACxE,OAAO,EAAE,cAAc,EAAE,MAAM,oCAAoC,CAAC;AACpE,OAAO,EAEL,sBAAsB,GACvB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,sBAAsB,EAAE,MAAM,kBAAkB,CAAC;AAC1D,OAAO,EAGL,gBAAgB,EAChB,aAAa,GACd,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,oBAAoB,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AAoD/E;;;;GAIG;AACH,MAAM,UAAU,mBAAmB,CACjC,UAAsC,EAAE;IAExC,MAAM,EAAE,cAAc,GAAG,CAAC,EAAE,GAAG,OAAO,CAAC;IACvC,MAAM,EAAE,MAAM,EAAE,GAAG,cAAc,EAAE,CAAC;IACpC,MAAM,EAAE,iBAAiB,EAAE,GAAG,QAAQ,EAAE,CAAC;IACzC,MAAM,WAAW,GAAG,cAAc,EAAE,CAAC;IACrC,MAAM,EAAE,aAAa,EAAE,YAAY,EAAE,yBAAyB,EAAE,GAC9D,gBAAgB,EAAE,CAAC;IAErB,MAAM,CAAC,oBAAoB,EAAE,uBAAuB,CAAC,GAAG,QAAQ,CAE9D,IAAI,CAAC,CAAC;IACR,MAAM,EAAE,QAAQ,EAAE,aAAa,EAAE,GAAG,mBAAmB,EAAE,CAAC;IAE1D,MAAM,aAAa,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAClE,MAAM,iBAAiB,GAAG,aAAa,EAAE,IAAI,KAAK,WAAW,CAAC;IAC9D,MAAM,eAAe,GAAG,aAAa,EAAE,EAAE,CAAC;IAE1C,qDAAqD;IACrD,SAAS,CAAC,GAAG,EAAE;QACb,uBAAuB,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC,EAAE,CAAC,eAAe,CAAC,CAAC,CAAC;IAEtB,4EAA4E;IAC5E,MAAM,iBAAiB,GAAG,aAAa,CAAC;QACtC,4EAA4E;QAC5E,QAAQ,EAAE,CAAC,aAAa,EAAE,iBAAiB,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC;QACrE,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,IAAI,CAAC,eAAe,IAAI,CAAC,iBAAiB,EAAE,CAAC;gBAC3C,OAAO,EAAE,CAAC;YACZ,CAAC;YAED,8CAA8C;YAC9C,MAAM,UAAU,GAAG,sBAAsB,CACvC,aAAa,EACb,YAAY,EACZ,yBAAyB,CAC1B,CAAC;YAEF,OAAO,MAAM,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,CACxD,MAAM,CAAC,EAAE,EACT,eAAe,EACf;gBACE,cAAc;gBACd,mBAAmB,EAAE,UAAU;aAChC,CACF,CAAC;QACJ,CAAC;QACD,2DAA2D;QAC3D,OAAO,EAAE,OAAO,CAAC,eAAe,IAAI,iBAAiB,CAAC;QACtD,6CAA6C;QAC7C,oBAAoB,EAAE,KAAK;QAC3B,kBAAkB,EAAE,KAAK;QACzB,yBAAyB;QACzB,KAAK,EAAE,KAAK;KACb,CAAC,CAAC;IAEH,6BAA6B;IAC7B,MAAM,mBAAmB,GAAG,gBAAgB,CAI1C;QACA,UAAU,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,YAAY,GAAG,KAAK,EAAE,EAAE,EAAE;YACzD,MAAM,UAAU,GAAG,aAAa,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC;YAChE,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;gBACxB,IAAI,UAAU,CAAC,KAAK,EAAE,CAAC;oBACrB,MAAM,UAAU,CAAC,KAAK,CAAC;gBACzB,CAAC;gBACD,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC;YACnD,CAAC;YAED,IAAI,YAAY,EAAE,CAAC;gBACjB,MAAM,iBAAiB,CAAC,UAAU,CAAC,cAAc,EAAE;oBACjD,QAAQ,EAAE,MAAM,CAAC,EAAE;iBACpB,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,aAAa,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC;YAC3C,CAAC;YACD,uBAAuB,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QACzC,CAAC;KACF,CAAC,CAAC;IAEH,gCAAgC;IAChC,MAAM,qBAAqB,GAAG,gBAAgB,CAI5C;QACA,UAAU,EAAE,KAAK,EAAE,eAAgC,EAAE,EAAE;YACrD,IAAI,CAAC,eAAe,IAAI,CAAC,iBAAiB,EAAE,CAAC;gBAC3C,OAAO,SAAS,CAAC;YACnB,CAAC;YAED,8CAA8C;YAC9C,MAAM,UAAU,GAAG,sBAAsB,CACvC,aAAa,EACb,YAAY,EACZ,yBAAyB,CAC1B,CAAC;YAEF,OAAO,MAAM,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,CACxD,MAAM,CAAC,EAAE,EACT,eAAe,EACf;gBACE,cAAc;gBACd,mBAAmB,EAAE,UAAU;aAChC,EACD,EAAE,MAAM,EAAE,eAAe,CAAC,MAAM,EAAE,CACnC,CAAC;QACJ,CAAC;QACD,yBAAyB;QACzB,KAAK,EAAE,KAAK;KACb,CAAC,CAAC;IAEH,mEAAmE;IACnE,8DAA8D;IAC9D,MAAM,WAAW,GAAG,iBAAiB;QACnC,CAAC,CAAC,CAAC,iBAAiB,CAAC,IAAI,IAAI,qBAAqB,CAAC,IAAI,IAAI,EAAE,CAAC;QAC9D,CAAC,CAAC,EAAE,CAAC;IAEP,OAAO;QACL,WAAW;QACX,MAAM,EAAE,mBAAmB,CAAC,WAAW;QACvC,oBAAoB;QACpB,YAAY,EAAE,mBAAmB;QACjC,cAAc,EAAE,qBAAqB;QACrC,iBAAiB;QACjB,GAAG,sBAAsB,CAAC,mBAAmB,EAAE,qBAAqB,CAAC;KACtE,CAAC;AACJ,CAAC","sourcesContent":["import TamboAI from \"@tambo-ai/typescript-sdk\";\nimport { useEffect, useState } from \"react\";\nimport { validateInput } from \"../model/validate-input\";\nimport { useTamboClient } from \"../providers/tambo-client-provider\";\nimport { useTambo } from \"../providers/tambo-provider\";\nimport { useTamboRegistry } from \"../providers/tambo-registry-provider\";\nimport { useTamboThread } from \"../providers/tambo-thread-provider\";\nimport {\n  CombinedMutationResult,\n  combineMutationResults,\n} from \"../util/query-utils\";\nimport { getAvailableComponents } from \"../util/registry\";\nimport {\n  UseTamboMutationResult,\n  UseTamboQueryResult,\n  useTamboMutation,\n  useTamboQuery,\n} from \"./react-query-hooks\";\nimport { INPUT_ERROR_MESSAGES, useTamboThreadInput } from \"./use-thread-input\";\n\n/**\n * Configuration options for the useTamboSuggestions hook\n */\nexport interface useTamboSuggestionsOptions {\n  /** Maximum number of suggestions to generate (1-10, default 3) */\n  maxSuggestions?: number;\n}\n\n/**\n * Return value interface for useTamboSuggestions hook\n */\nexport interface useTamboSuggestionsResultInternal {\n  /** List of available suggestions (also available in generateResult.data) */\n  suggestions: TamboAI.Beta.Threads.Suggestion[];\n  /** ID of the currently selected suggestion */\n  selectedSuggestionId: string | null;\n  /**\n   * Accept and apply a suggestion (also available in acceptResult.mutateAsync)\n   * @param suggestion - The suggestion to accept\n   * @param shouldSubmit - Whether to automatically submit after accepting (default: false)\n   */\n  accept: (acceptOptions: {\n    suggestion: TamboAI.Beta.Threads.Suggestion;\n    shouldSubmit?: boolean;\n  }) => Promise<void>;\n\n  /** Result and network state for accepting a suggestion */\n  acceptResult: UseTamboMutationResult<\n    void,\n    Error,\n    { suggestion: TamboAI.Beta.Threads.Suggestion; shouldSubmit?: boolean }\n  >;\n\n  /** Result and network state for generating suggestions */\n  generateResult: UseTamboMutationResult<\n    TamboAI.Beta.Threads.Suggestions.SuggestionGenerateResponse | undefined,\n    Error,\n    AbortController\n  >;\n\n  /** The full suggestions query object from React Query */\n  suggestionsResult: UseTamboQueryResult<\n    TamboAI.Beta.Threads.Suggestions.SuggestionGenerateResponse | undefined,\n    Error\n  >;\n}\n\ntype useTamboSuggestionsResult = CombinedMutationResult<any, Error> &\n  useTamboSuggestionsResultInternal;\n\n/**\n * Hook for managing Tambo AI suggestions in a thread\n * @param options - Configuration options for suggestion generation\n * @returns Object containing suggestions state and control functions\n */\nexport function useTamboSuggestions(\n  options: useTamboSuggestionsOptions = {},\n): useTamboSuggestionsResult {\n  const { maxSuggestions = 3 } = options;\n  const { thread } = useTamboThread();\n  const { sendThreadMessage } = useTambo();\n  const tamboClient = useTamboClient();\n  const { componentList, toolRegistry, componentToolAssociations } =\n    useTamboRegistry();\n\n  const [selectedSuggestionId, setSelectedSuggestionId] = useState<\n    string | null\n  >(null);\n  const { setValue: setInputValue } = useTamboThreadInput();\n\n  const latestMessage = thread.messages[thread.messages.length - 1];\n  const isLatestFromTambo = latestMessage?.role === \"assistant\";\n  const latestMessageId = latestMessage?.id;\n\n  // Reset selected suggestion when the message changes\n  useEffect(() => {\n    setSelectedSuggestionId(null);\n  }, [latestMessageId]);\n\n  // Use React Query to fetch suggestions when a new hydra message is received\n  const suggestionsResult = useTamboQuery({\n    // Only include latestMessageId in the queryKey if the message is from hydra\n    queryKey: [\"suggestions\", isLatestFromTambo ? latestMessageId : null],\n    queryFn: async () => {\n      if (!latestMessageId || !isLatestFromTambo) {\n        return [];\n      }\n\n      // Get registered components from the registry\n      const components = getAvailableComponents(\n        componentList,\n        toolRegistry,\n        componentToolAssociations,\n      );\n\n      return await tamboClient.beta.threads.suggestions.generate(\n        thread.id,\n        latestMessageId,\n        {\n          maxSuggestions,\n          availableComponents: components,\n        },\n      );\n    },\n    // Only run the query if we have a valid message from hydra\n    enabled: Boolean(latestMessageId && isLatestFromTambo),\n    // Don't refetch on window focus or reconnect\n    refetchOnWindowFocus: false,\n    refetchOnReconnect: false,\n    // Don't retry on failure\n    retry: false,\n  });\n\n  // Accept suggestion mutation\n  const acceptMutationState = useTamboMutation<\n    void,\n    Error,\n    { suggestion: TamboAI.Beta.Threads.Suggestion; shouldSubmit?: boolean }\n  >({\n    mutationFn: async ({ suggestion, shouldSubmit = false }) => {\n      const validation = validateInput(suggestion.detailedSuggestion);\n      if (!validation.isValid) {\n        if (validation.error) {\n          throw validation.error;\n        }\n        throw new Error(INPUT_ERROR_MESSAGES.VALIDATION);\n      }\n\n      if (shouldSubmit) {\n        await sendThreadMessage(validation.sanitizedInput, {\n          threadId: thread.id,\n        });\n      } else {\n        setInputValue(validation.sanitizedInput);\n      }\n      setSelectedSuggestionId(suggestion.id);\n    },\n  });\n\n  // Generate suggestions mutation\n  const generateMutationState = useTamboMutation<\n    TamboAI.Beta.Threads.Suggestions.SuggestionGenerateResponse | undefined,\n    Error,\n    AbortController\n  >({\n    mutationFn: async (abortController: AbortController) => {\n      if (!latestMessageId || !isLatestFromTambo) {\n        return undefined;\n      }\n\n      // Get registered components from the registry\n      const components = getAvailableComponents(\n        componentList,\n        toolRegistry,\n        componentToolAssociations,\n      );\n\n      return await tamboClient.beta.threads.suggestions.generate(\n        thread.id,\n        latestMessageId,\n        {\n          maxSuggestions,\n          availableComponents: components,\n        },\n        { signal: abortController.signal },\n      );\n    },\n    // Don't retry on failure\n    retry: false,\n  });\n\n  // Use the query data if available, otherwise use the mutation data\n  // Only return suggestions if the latest message is from hydra\n  const suggestions = isLatestFromTambo\n    ? (suggestionsResult.data ?? generateMutationState.data ?? [])\n    : [];\n\n  return {\n    suggestions,\n    accept: acceptMutationState.mutateAsync,\n    selectedSuggestionId,\n    acceptResult: acceptMutationState,\n    generateResult: generateMutationState,\n    suggestionsResult,\n    ...combineMutationResults(acceptMutationState, generateMutationState),\n  };\n}\n"]}
+{"version":3,"file":"use-suggestions.js","sourceRoot":"","sources":["../../src/hooks/use-suggestions.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AAC5C,OAAO,EAAE,WAAW,EAAE,MAAM,sCAAsC,CAAC;AACnE,OAAO,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AACxD,OAAO,EAAE,cAAc,EAAE,MAAM,oCAAoC,CAAC;AACpE,OAAO,EAAE,QAAQ,EAAE,MAAM,6BAA6B,CAAC;AACvD,OAAO,EAAE,gBAAgB,EAAE,MAAM,sCAAsC,CAAC;AACxE,OAAO,EAAE,cAAc,EAAE,MAAM,oCAAoC,CAAC;AACpE,OAAO,EAEL,sBAAsB,GACvB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,sBAAsB,EAAE,MAAM,kBAAkB,CAAC;AAC1D,OAAO,EAGL,gBAAgB,EAChB,aAAa,GACd,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,oBAAoB,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AAoD/E;;;;GAIG;AACH,MAAM,UAAU,mBAAmB,CACjC,UAAsC,EAAE;IAExC,MAAM,EAAE,cAAc,GAAG,CAAC,EAAE,GAAG,OAAO,CAAC;IACvC,MAAM,EAAE,MAAM,EAAE,eAAe,EAAE,GAAG,cAAc,EAAE,CAAC;IACrD,MAAM,EAAE,iBAAiB,EAAE,GAAG,QAAQ,EAAE,CAAC;IACzC,MAAM,WAAW,GAAG,cAAc,EAAE,CAAC;IACrC,MAAM,EAAE,aAAa,EAAE,YAAY,EAAE,yBAAyB,EAAE,GAC9D,gBAAgB,EAAE,CAAC;IAErB,MAAM,CAAC,oBAAoB,EAAE,uBAAuB,CAAC,GAAG,QAAQ,CAE9D,IAAI,CAAC,CAAC;IACR,MAAM,EAAE,QAAQ,EAAE,aAAa,EAAE,GAAG,mBAAmB,EAAE,CAAC;IAE1D,MAAM,aAAa,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAClE,MAAM,iBAAiB,GAAG,aAAa,EAAE,IAAI,KAAK,WAAW,CAAC;IAC9D,MAAM,eAAe,GAAG,aAAa,EAAE,EAAE,CAAC;IAE1C,qDAAqD;IACrD,SAAS,CAAC,GAAG,EAAE;QACb,uBAAuB,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC,EAAE,CAAC,eAAe,CAAC,CAAC,CAAC;IAEtB,MAAM,yBAAyB,GAC7B,eAAe,IAAI,iBAAiB,IAAI,WAAW,CAAC,eAAe,CAAC,CAAC;IACvE,4EAA4E;IAC5E,MAAM,iBAAiB,GAAG,aAAa,CAAC;QACtC,2EAA2E;QAC3E,4DAA4D;QAC5D,QAAQ,EAAE;YACR,aAAa;YACb,yBAAyB,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI;SACnD;QACD,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,IAAI,CAAC,yBAAyB,EAAE,CAAC;gBAC/B,OAAO,EAAE,CAAC;YACZ,CAAC;YAED,8CAA8C;YAC9C,MAAM,UAAU,GAAG,sBAAsB,CACvC,aAAa,EACb,YAAY,EACZ,yBAAyB,CAC1B,CAAC;YAEF,OAAO,MAAM,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,CACxD,MAAM,CAAC,EAAE,EACT,eAAe,EACf;gBACE,cAAc;gBACd,mBAAmB,EAAE,UAAU;aAChC,CACF,CAAC;QACJ,CAAC;QACD,2DAA2D;QAC3D,OAAO,EAAE,OAAO,CAAC,eAAe,IAAI,iBAAiB,CAAC;QACtD,6CAA6C;QAC7C,oBAAoB,EAAE,KAAK;QAC3B,kBAAkB,EAAE,KAAK;QACzB,yBAAyB;QACzB,KAAK,EAAE,KAAK;KACb,CAAC,CAAC;IAEH,6BAA6B;IAC7B,MAAM,mBAAmB,GAAG,gBAAgB,CAI1C;QACA,UAAU,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,YAAY,GAAG,KAAK,EAAE,EAAE,EAAE;YACzD,MAAM,UAAU,GAAG,aAAa,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC;YAChE,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;gBACxB,IAAI,UAAU,CAAC,KAAK,EAAE,CAAC;oBACrB,MAAM,UAAU,CAAC,KAAK,CAAC;gBACzB,CAAC;gBACD,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC;YACnD,CAAC;YAED,IAAI,YAAY,EAAE,CAAC;gBACjB,MAAM,iBAAiB,CAAC,UAAU,CAAC,cAAc,EAAE;oBACjD,QAAQ,EAAE,MAAM,CAAC,EAAE;iBACpB,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,aAAa,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC;YAC3C,CAAC;YACD,uBAAuB,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QACzC,CAAC;KACF,CAAC,CAAC;IAEH,gCAAgC;IAChC,MAAM,qBAAqB,GAAG,gBAAgB,CAI5C;QACA,UAAU,EAAE,KAAK,EAAE,eAAgC,EAAE,EAAE;YACrD,IAAI,CAAC,yBAAyB,EAAE,CAAC;gBAC/B,OAAO,SAAS,CAAC;YACnB,CAAC;YAED,8CAA8C;YAC9C,MAAM,UAAU,GAAG,sBAAsB,CACvC,aAAa,EACb,YAAY,EACZ,yBAAyB,CAC1B,CAAC;YACF,OAAO,MAAM,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,CACxD,MAAM,CAAC,EAAE,EACT,eAAe,EACf;gBACE,cAAc;gBACd,mBAAmB,EAAE,UAAU;aAChC,EACD,EAAE,MAAM,EAAE,eAAe,CAAC,MAAM,EAAE,CACnC,CAAC;QACJ,CAAC;QACD,yBAAyB;QACzB,KAAK,EAAE,KAAK;KACb,CAAC,CAAC;IAEH,mEAAmE;IACnE,8DAA8D;IAC9D,MAAM,WAAW,GAAG,iBAAiB;QACnC,CAAC,CAAC,CAAC,iBAAiB,CAAC,IAAI,IAAI,qBAAqB,CAAC,IAAI,IAAI,EAAE,CAAC;QAC9D,CAAC,CAAC,EAAE,CAAC;IAEP,OAAO;QACL,WAAW;QACX,MAAM,EAAE,mBAAmB,CAAC,WAAW;QACvC,oBAAoB;QACpB,YAAY,EAAE,mBAAmB;QACjC,cAAc,EAAE,qBAAqB;QACrC,iBAAiB;QACjB,GAAG,sBAAsB,CAAC,mBAAmB,EAAE,qBAAqB,CAAC;KACtE,CAAC;AACJ,CAAC","sourcesContent":["import TamboAI from \"@tambo-ai/typescript-sdk\";\nimport { useEffect, useState } from \"react\";\nimport { isIdleStage } from \"../model/generate-component-response\";\nimport { validateInput } from \"../model/validate-input\";\nimport { useTamboClient } from \"../providers/tambo-client-provider\";\nimport { useTambo } from \"../providers/tambo-provider\";\nimport { useTamboRegistry } from \"../providers/tambo-registry-provider\";\nimport { useTamboThread } from \"../providers/tambo-thread-provider\";\nimport {\n  CombinedMutationResult,\n  combineMutationResults,\n} from \"../util/query-utils\";\nimport { getAvailableComponents } from \"../util/registry\";\nimport {\n  UseTamboMutationResult,\n  UseTamboQueryResult,\n  useTamboMutation,\n  useTamboQuery,\n} from \"./react-query-hooks\";\nimport { INPUT_ERROR_MESSAGES, useTamboThreadInput } from \"./use-thread-input\";\n\n/**\n * Configuration options for the useTamboSuggestions hook\n */\nexport interface useTamboSuggestionsOptions {\n  /** Maximum number of suggestions to generate (1-10, default 3) */\n  maxSuggestions?: number;\n}\n\n/**\n * Return value interface for useTamboSuggestions hook\n */\nexport interface useTamboSuggestionsResultInternal {\n  /** List of available suggestions (also available in generateResult.data) */\n  suggestions: TamboAI.Beta.Threads.Suggestion[];\n  /** ID of the currently selected suggestion */\n  selectedSuggestionId: string | null;\n  /**\n   * Accept and apply a suggestion (also available in acceptResult.mutateAsync)\n   * @param suggestion - The suggestion to accept\n   * @param shouldSubmit - Whether to automatically submit after accepting (default: false)\n   */\n  accept: (acceptOptions: {\n    suggestion: TamboAI.Beta.Threads.Suggestion;\n    shouldSubmit?: boolean;\n  }) => Promise<void>;\n\n  /** Result and network state for accepting a suggestion */\n  acceptResult: UseTamboMutationResult<\n    void,\n    Error,\n    { suggestion: TamboAI.Beta.Threads.Suggestion; shouldSubmit?: boolean }\n  >;\n\n  /** Result and network state for generating suggestions */\n  generateResult: UseTamboMutationResult<\n    TamboAI.Beta.Threads.Suggestions.SuggestionGenerateResponse | undefined,\n    Error,\n    AbortController\n  >;\n\n  /** The full suggestions query object from React Query */\n  suggestionsResult: UseTamboQueryResult<\n    TamboAI.Beta.Threads.Suggestions.SuggestionGenerateResponse | undefined,\n    Error\n  >;\n}\n\ntype useTamboSuggestionsResult = CombinedMutationResult<any, Error> &\n  useTamboSuggestionsResultInternal;\n\n/**\n * Hook for managing Tambo AI suggestions in a thread\n * @param options - Configuration options for suggestion generation\n * @returns Object containing suggestions state and control functions\n */\nexport function useTamboSuggestions(\n  options: useTamboSuggestionsOptions = {},\n): useTamboSuggestionsResult {\n  const { maxSuggestions = 3 } = options;\n  const { thread, generationStage } = useTamboThread();\n  const { sendThreadMessage } = useTambo();\n  const tamboClient = useTamboClient();\n  const { componentList, toolRegistry, componentToolAssociations } =\n    useTamboRegistry();\n\n  const [selectedSuggestionId, setSelectedSuggestionId] = useState<\n    string | null\n  >(null);\n  const { setValue: setInputValue } = useTamboThreadInput();\n\n  const latestMessage = thread.messages[thread.messages.length - 1];\n  const isLatestFromTambo = latestMessage?.role === \"assistant\";\n  const latestMessageId = latestMessage?.id;\n\n  // Reset selected suggestion when the message changes\n  useEffect(() => {\n    setSelectedSuggestionId(null);\n  }, [latestMessageId]);\n\n  const shouldGenerateSuggestions =\n    latestMessageId && isLatestFromTambo && isIdleStage(generationStage);\n  // Use React Query to fetch suggestions when a new hydra message is received\n  const suggestionsResult = useTamboQuery({\n    // Make sure the query key changes when the message changes, so that we are\n    // always generating suggestions when there is a new message\n    queryKey: [\n      \"suggestions\",\n      shouldGenerateSuggestions ? latestMessageId : null,\n    ],\n    queryFn: async () => {\n      if (!shouldGenerateSuggestions) {\n        return [];\n      }\n\n      // Get registered components from the registry\n      const components = getAvailableComponents(\n        componentList,\n        toolRegistry,\n        componentToolAssociations,\n      );\n\n      return await tamboClient.beta.threads.suggestions.generate(\n        thread.id,\n        latestMessageId,\n        {\n          maxSuggestions,\n          availableComponents: components,\n        },\n      );\n    },\n    // Only run the query if we have a valid message from hydra\n    enabled: Boolean(latestMessageId && isLatestFromTambo),\n    // Don't refetch on window focus or reconnect\n    refetchOnWindowFocus: false,\n    refetchOnReconnect: false,\n    // Don't retry on failure\n    retry: false,\n  });\n\n  // Accept suggestion mutation\n  const acceptMutationState = useTamboMutation<\n    void,\n    Error,\n    { suggestion: TamboAI.Beta.Threads.Suggestion; shouldSubmit?: boolean }\n  >({\n    mutationFn: async ({ suggestion, shouldSubmit = false }) => {\n      const validation = validateInput(suggestion.detailedSuggestion);\n      if (!validation.isValid) {\n        if (validation.error) {\n          throw validation.error;\n        }\n        throw new Error(INPUT_ERROR_MESSAGES.VALIDATION);\n      }\n\n      if (shouldSubmit) {\n        await sendThreadMessage(validation.sanitizedInput, {\n          threadId: thread.id,\n        });\n      } else {\n        setInputValue(validation.sanitizedInput);\n      }\n      setSelectedSuggestionId(suggestion.id);\n    },\n  });\n\n  // Generate suggestions mutation\n  const generateMutationState = useTamboMutation<\n    TamboAI.Beta.Threads.Suggestions.SuggestionGenerateResponse | undefined,\n    Error,\n    AbortController\n  >({\n    mutationFn: async (abortController: AbortController) => {\n      if (!shouldGenerateSuggestions) {\n        return undefined;\n      }\n\n      // Get registered components from the registry\n      const components = getAvailableComponents(\n        componentList,\n        toolRegistry,\n        componentToolAssociations,\n      );\n      return await tamboClient.beta.threads.suggestions.generate(\n        thread.id,\n        latestMessageId,\n        {\n          maxSuggestions,\n          availableComponents: components,\n        },\n        { signal: abortController.signal },\n      );\n    },\n    // Don't retry on failure\n    retry: false,\n  });\n\n  // Use the query data if available, otherwise use the mutation data\n  // Only return suggestions if the latest message is from hydra\n  const suggestions = isLatestFromTambo\n    ? (suggestionsResult.data ?? generateMutationState.data ?? [])\n    : [];\n\n  return {\n    suggestions,\n    accept: acceptMutationState.mutateAsync,\n    selectedSuggestionId,\n    acceptResult: acceptMutationState,\n    generateResult: generateMutationState,\n    suggestionsResult,\n    ...combineMutationResults(acceptMutationState, generateMutationState),\n  };\n}\n"]}

package/esm/providers/tambo-provider.d.ts

@@ -12,6 +12,7 @@ import { TamboThreadContextProps } from "./tambo-thread-provider";
  * @param props.apiKey - The API key for the Tambo API
  * @param props.components - The components to register
  * @param props.environment - The environment to use for the Tambo API
+ * @param props.tools - The tools to register
  * @returns The TamboProvider component
  */
 export declare const TamboProvider: React.FC<PropsWithChildren<TamboClientProviderProps & TamboRegistryProviderProps>>;

package/esm/providers/tambo-provider.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"tambo-provider.d.ts","sourceRoot":"","sources":["../../src/providers/tambo-provider.tsx"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,EAAE,iBAAiB,EAA6B,MAAM,OAAO,CAAC;AAC5E,OAAO,EACL,uBAAuB,EAEvB,wBAAwB,EAGzB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EACL,0BAA0B,EAG3B,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAEL,0BAA0B,EAC3B,MAAM,2BAA2B,CAAC;AACnC,OAAO,EACL,uBAAuB,EAGxB,MAAM,yBAAyB,CAAC;AAEjC;;;;;;;;;;GAUG;AACH,eAAO,MAAM,aAAa,EAAE,KAAK,CAAC,EAAE,CAClC,iBAAiB,CAAC,wBAAwB,GAAG,0BAA0B,CAAC,CAsBzE,CAAC;AACF,MAAM,MAAM,iBAAiB,GAAG,uBAAuB,GACrD,uBAAuB,GACvB,0BAA0B,CAAC;AAE7B,eAAO,MAAM,YAAY,kCAExB,CAAC;AA6BF;;;;;;GAMG;AACH,eAAO,MAAM,QAAQ,yBAEpB,CAAC"}
+{"version":3,"file":"tambo-provider.d.ts","sourceRoot":"","sources":["../../src/providers/tambo-provider.tsx"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,EAAE,iBAAiB,EAA6B,MAAM,OAAO,CAAC;AAC5E,OAAO,EACL,uBAAuB,EAEvB,wBAAwB,EAGzB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EACL,0BAA0B,EAG3B,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAEL,0BAA0B,EAC3B,MAAM,2BAA2B,CAAC;AACnC,OAAO,EACL,uBAAuB,EAGxB,MAAM,yBAAyB,CAAC;AAEjC;;;;;;;;;;;GAWG;AACH,eAAO,MAAM,aAAa,EAAE,KAAK,CAAC,EAAE,CAClC,iBAAiB,CAAC,wBAAwB,GAAG,0BAA0B,CAAC,CAsBzE,CAAC;AACF,MAAM,MAAM,iBAAiB,GAAG,uBAAuB,GACrD,uBAAuB,GACvB,0BAA0B,CAAC;AAE7B,eAAO,MAAM,YAAY,kCAExB,CAAC;AA6BF;;;;;;GAMG;AACH,eAAO,MAAM,QAAQ,yBAEpB,CAAC"}

package/esm/providers/tambo-provider.js

@@ -13,15 +13,16 @@ import { TamboThreadProvider, useTamboThread, } from "./tambo-thread-provider";
  * @param props.apiKey - The API key for the Tambo API
  * @param props.components - The components to register
  * @param props.environment - The environment to use for the Tambo API
+ * @param props.tools - The tools to register
  * @returns The TamboProvider component
  */
-export const TamboProvider = ({ children, tamboUrl, apiKey, components, environment }) => {
+export const TamboProvider = ({ children, tamboUrl, apiKey, components, environment, tools }) => {
     // Should only be used in browser
     if (typeof window === "undefined") {
         console.error("TamboProvider must be used within a browser");
     }
     return (React.createElement(TamboClientProvider, { tamboUrl: tamboUrl, apiKey: apiKey, environment: environment },
-        React.createElement(TamboRegistryProvider, { components: components },
+        React.createElement(TamboRegistryProvider, { components: components, tools: tools },
             React.createElement(TamboThreadProvider, null,
                 React.createElement(TamboComponentProvider, null,
                     React.createElement(TamboCompositeProvider, null, children))))));

package/esm/providers/tambo-provider.js.map

@@ -1 +1 @@
-{"version":3,"file":"tambo-provider.js","sourceRoot":"","sources":["../../src/providers/tambo-provider.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;AACb,OAAO,KAAK,EAAE,EAAqB,aAAa,EAAE,UAAU,EAAE,MAAM,OAAO,CAAC;AAC5E,OAAO,EAEL,mBAAmB,EAEnB,cAAc,EACd,mBAAmB,GACpB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAEL,sBAAsB,EACtB,iBAAiB,GAClB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EACL,qBAAqB,GAEtB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAEL,mBAAmB,EACnB,cAAc,GACf,MAAM,yBAAyB,CAAC;AAEjC;;;;;;;;;;GAUG;AACH,MAAM,CAAC,MAAM,aAAa,GAEtB,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,WAAW,EAAE,EAAE,EAAE;IAC9D,iCAAiC;IACjC,IAAI,OAAO,MAAM,KAAK,WAAW,EAAE,CAAC;QAClC,OAAO,CAAC,KAAK,CAAC,6CAA6C,CAAC,CAAC;IAC/D,CAAC;IAED,OAAO,CACL,oBAAC,mBAAmB,IAClB,QAAQ,EAAE,QAAQ,EAClB,MAAM,EAAE,MAAM,EACd,WAAW,EAAE,WAAW;QAExB,oBAAC,qBAAqB,IAAC,UAAU,EAAE,UAAU;YAC3C,oBAAC,mBAAmB;gBAClB,oBAAC,sBAAsB;oBACrB,oBAAC,sBAAsB,QAAE,QAAQ,CAA0B,CACpC,CACL,CACA,CACJ,CACvB,CAAC;AACJ,CAAC,CAAC;AAKF,MAAM,CAAC,MAAM,YAAY,GAAG,aAAa,CACvC,EAAuB,CACxB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,sBAAsB,GAAgC,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE;IAC3E,MAAM,OAAO,GAAG,cAAc,EAAE,CAAC;IACjC,MAAM,MAAM,GAAG,cAAc,EAAE,CAAC;IAChC,MAAM,WAAW,GAAG,mBAAmB,EAAE,CAAC;IAC1C,MAAM,iBAAiB,GAAG,iBAAiB,EAAE,CAAC;IAE9C,OAAO,CACL,oBAAC,YAAY,CAAC,QAAQ,IACpB,KAAK,EAAE;YACL,MAAM;YACN,WAAW;YACX,GAAG,iBAAiB;YACpB,GAAG,OAAO;SACX,IAEA,QAAQ,CACa,CACzB,CAAC;AACJ,CAAC,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,QAAQ,GAAG,GAAG,EAAE;IAC3B,OAAO,UAAU,CAAC,YAAY,CAAC,CAAC;AAClC,CAAC,CAAC","sourcesContent":["\"use client\";\nimport React, { PropsWithChildren, createContext, useContext } from \"react\";\nimport {\n  TamboClientContextProps,\n  TamboClientProvider,\n  TamboClientProviderProps,\n  useTamboClient,\n  useTamboQueryClient,\n} from \"./tambo-client-provider\";\nimport {\n  TamboComponentContextProps,\n  TamboComponentProvider,\n  useTamboComponent,\n} from \"./tambo-component-provider\";\nimport {\n  TamboRegistryProvider,\n  TamboRegistryProviderProps,\n} from \"./tambo-registry-provider\";\nimport {\n  TamboThreadContextProps,\n  TamboThreadProvider,\n  useTamboThread,\n} from \"./tambo-thread-provider\";\n\n/**\n * The TamboProvider gives full access to the whole Tambo API. This includes the\n * TamboAI client, the component registry, and the current thread context.\n * @param props - The props for the TamboProvider\n * @param props.children - The children to wrap\n * @param props.tamboUrl - The URL of the Tambo API\n * @param props.apiKey - The API key for the Tambo API\n * @param props.components - The components to register\n * @param props.environment - The environment to use for the Tambo API\n * @returns The TamboProvider component\n */\nexport const TamboProvider: React.FC<\n  PropsWithChildren<TamboClientProviderProps & TamboRegistryProviderProps>\n> = ({ children, tamboUrl, apiKey, components, environment }) => {\n  // Should only be used in browser\n  if (typeof window === \"undefined\") {\n    console.error(\"TamboProvider must be used within a browser\");\n  }\n\n  return (\n    <TamboClientProvider\n      tamboUrl={tamboUrl}\n      apiKey={apiKey}\n      environment={environment}\n    >\n      <TamboRegistryProvider components={components}>\n        <TamboThreadProvider>\n          <TamboComponentProvider>\n            <TamboCompositeProvider>{children}</TamboCompositeProvider>\n          </TamboComponentProvider>\n        </TamboThreadProvider>\n      </TamboRegistryProvider>\n    </TamboClientProvider>\n  );\n};\nexport type TamboContextProps = TamboClientContextProps &\n  TamboThreadContextProps &\n  TamboComponentContextProps;\n\nexport const TamboContext = createContext<TamboContextProps>(\n  {} as TamboContextProps,\n);\n\n/**\n * TamboCompositeProvider is a provider that combines the TamboClient,\n * TamboThread, and TamboComponent providers\n * @param props - The props for the TamboCompositeProvider\n * @param props.children - The children to wrap\n * @returns The wrapped component\n */\nconst TamboCompositeProvider: React.FC<PropsWithChildren> = ({ children }) => {\n  const threads = useTamboThread();\n  const client = useTamboClient();\n  const queryClient = useTamboQueryClient();\n  const componentRegistry = useTamboComponent();\n\n  return (\n    <TamboContext.Provider\n      value={{\n        client,\n        queryClient,\n        ...componentRegistry,\n        ...threads,\n      }}\n    >\n      {children}\n    </TamboContext.Provider>\n  );\n};\n\n/**\n * The useTambo hook provides access to the Tambo API. This is the primary entrypoint\n * for the Tambo React SDK.\n *\n * This includes the TamboAI client, the component registry, and the current thread context.\n * @returns The Tambo API\n */\nexport const useTambo = () => {\n  return useContext(TamboContext);\n};\n"]}
+{"version":3,"file":"tambo-provider.js","sourceRoot":"","sources":["../../src/providers/tambo-provider.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;AACb,OAAO,KAAK,EAAE,EAAqB,aAAa,EAAE,UAAU,EAAE,MAAM,OAAO,CAAC;AAC5E,OAAO,EAEL,mBAAmB,EAEnB,cAAc,EACd,mBAAmB,GACpB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAEL,sBAAsB,EACtB,iBAAiB,GAClB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EACL,qBAAqB,GAEtB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAEL,mBAAmB,EACnB,cAAc,GACf,MAAM,yBAAyB,CAAC;AAEjC;;;;;;;;;;;GAWG;AACH,MAAM,CAAC,MAAM,aAAa,GAEtB,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,WAAW,EAAE,KAAK,EAAE,EAAE,EAAE;IACrE,iCAAiC;IACjC,IAAI,OAAO,MAAM,KAAK,WAAW,EAAE,CAAC;QAClC,OAAO,CAAC,KAAK,CAAC,6CAA6C,CAAC,CAAC;IAC/D,CAAC;IAED,OAAO,CACL,oBAAC,mBAAmB,IAClB,QAAQ,EAAE,QAAQ,EAClB,MAAM,EAAE,MAAM,EACd,WAAW,EAAE,WAAW;QAExB,oBAAC,qBAAqB,IAAC,UAAU,EAAE,UAAU,EAAE,KAAK,EAAE,KAAK;YACzD,oBAAC,mBAAmB;gBAClB,oBAAC,sBAAsB;oBACrB,oBAAC,sBAAsB,QAAE,QAAQ,CAA0B,CACpC,CACL,CACA,CACJ,CACvB,CAAC;AACJ,CAAC,CAAC;AAKF,MAAM,CAAC,MAAM,YAAY,GAAG,aAAa,CACvC,EAAuB,CACxB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,sBAAsB,GAAgC,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE;IAC3E,MAAM,OAAO,GAAG,cAAc,EAAE,CAAC;IACjC,MAAM,MAAM,GAAG,cAAc,EAAE,CAAC;IAChC,MAAM,WAAW,GAAG,mBAAmB,EAAE,CAAC;IAC1C,MAAM,iBAAiB,GAAG,iBAAiB,EAAE,CAAC;IAE9C,OAAO,CACL,oBAAC,YAAY,CAAC,QAAQ,IACpB,KAAK,EAAE;YACL,MAAM;YACN,WAAW;YACX,GAAG,iBAAiB;YACpB,GAAG,OAAO;SACX,IAEA,QAAQ,CACa,CACzB,CAAC;AACJ,CAAC,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,QAAQ,GAAG,GAAG,EAAE;IAC3B,OAAO,UAAU,CAAC,YAAY,CAAC,CAAC;AAClC,CAAC,CAAC","sourcesContent":["\"use client\";\nimport React, { PropsWithChildren, createContext, useContext } from \"react\";\nimport {\n  TamboClientContextProps,\n  TamboClientProvider,\n  TamboClientProviderProps,\n  useTamboClient,\n  useTamboQueryClient,\n} from \"./tambo-client-provider\";\nimport {\n  TamboComponentContextProps,\n  TamboComponentProvider,\n  useTamboComponent,\n} from \"./tambo-component-provider\";\nimport {\n  TamboRegistryProvider,\n  TamboRegistryProviderProps,\n} from \"./tambo-registry-provider\";\nimport {\n  TamboThreadContextProps,\n  TamboThreadProvider,\n  useTamboThread,\n} from \"./tambo-thread-provider\";\n\n/**\n * The TamboProvider gives full access to the whole Tambo API. This includes the\n * TamboAI client, the component registry, and the current thread context.\n * @param props - The props for the TamboProvider\n * @param props.children - The children to wrap\n * @param props.tamboUrl - The URL of the Tambo API\n * @param props.apiKey - The API key for the Tambo API\n * @param props.components - The components to register\n * @param props.environment - The environment to use for the Tambo API\n * @param props.tools - The tools to register\n * @returns The TamboProvider component\n */\nexport const TamboProvider: React.FC<\n  PropsWithChildren<TamboClientProviderProps & TamboRegistryProviderProps>\n> = ({ children, tamboUrl, apiKey, components, environment, tools }) => {\n  // Should only be used in browser\n  if (typeof window === \"undefined\") {\n    console.error(\"TamboProvider must be used within a browser\");\n  }\n\n  return (\n    <TamboClientProvider\n      tamboUrl={tamboUrl}\n      apiKey={apiKey}\n      environment={environment}\n    >\n      <TamboRegistryProvider components={components} tools={tools}>\n        <TamboThreadProvider>\n          <TamboComponentProvider>\n            <TamboCompositeProvider>{children}</TamboCompositeProvider>\n          </TamboComponentProvider>\n        </TamboThreadProvider>\n      </TamboRegistryProvider>\n    </TamboClientProvider>\n  );\n};\nexport type TamboContextProps = TamboClientContextProps &\n  TamboThreadContextProps &\n  TamboComponentContextProps;\n\nexport const TamboContext = createContext<TamboContextProps>(\n  {} as TamboContextProps,\n);\n\n/**\n * TamboCompositeProvider is a provider that combines the TamboClient,\n * TamboThread, and TamboComponent providers\n * @param props - The props for the TamboCompositeProvider\n * @param props.children - The children to wrap\n * @returns The wrapped component\n */\nconst TamboCompositeProvider: React.FC<PropsWithChildren> = ({ children }) => {\n  const threads = useTamboThread();\n  const client = useTamboClient();\n  const queryClient = useTamboQueryClient();\n  const componentRegistry = useTamboComponent();\n\n  return (\n    <TamboContext.Provider\n      value={{\n        client,\n        queryClient,\n        ...componentRegistry,\n        ...threads,\n      }}\n    >\n      {children}\n    </TamboContext.Provider>\n  );\n};\n\n/**\n * The useTambo hook provides access to the Tambo API. This is the primary entrypoint\n * for the Tambo React SDK.\n *\n * This includes the TamboAI client, the component registry, and the current thread context.\n * @returns The Tambo API\n */\nexport const useTambo = () => {\n  return useContext(TamboContext);\n};\n"]}

package/esm/providers/tambo-registry-provider.d.ts

@@ -12,6 +12,8 @@ export interface TamboRegistryContext {
 export interface TamboRegistryProviderProps {
     /** The components to register */
     components?: TamboComponent[];
+    /** The tools to register */
+    tools?: TamboTool[];
 }
 /**
  * The TamboRegistryProvider is a React provider that provides a component
@@ -19,6 +21,7 @@ export interface TamboRegistryProviderProps {
  * @param props - The props for the TamboRegistryProvider
  * @param props.children - The children to wrap
  * @param props.components - The components to register
+ * @param props.tools - The tools to register
  * @returns The TamboRegistryProvider component
  */
 export declare const TamboRegistryProvider: React.FC<PropsWithChildren<TamboRegistryProviderProps>>;

package/esm/providers/tambo-registry-provider.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"tambo-registry-provider.d.ts","sourceRoot":"","sources":["../../src/providers/tambo-registry-provider.tsx"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,EAEZ,iBAAiB,EAKlB,MAAM,OAAO,CAAC;AAGf,OAAO,EACL,iBAAiB,EACjB,cAAc,EACd,SAAS,EACV,MAAM,6BAA6B,CAAC;AAErC,MAAM,WAAW,oBAAoB;IACnC,aAAa,EAAE,iBAAiB,CAAC;IACjC,YAAY,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IACxC,yBAAyB,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;IACpD,iBAAiB,EAAE,CAAC,OAAO,EAAE,cAAc,KAAK,IAAI,CAAC;IACrD,YAAY,EAAE,CAAC,IAAI,EAAE,SAAS,KAAK,IAAI,CAAC;IACxC,aAAa,EAAE,CAAC,KAAK,EAAE,SAAS,EAAE,KAAK,IAAI,CAAC;IAC5C,kBAAkB,EAAE,CAAC,aAAa,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,KAAK,IAAI,CAAC;CACtE;AAwBD,MAAM,WAAW,0BAA0B;IACzC,iCAAiC;IACjC,UAAU,CAAC,EAAE,cAAc,EAAE,CAAC;CAC/B;AAED;;;;;;;GAOG;AACH,eAAO,MAAM,qBAAqB,EAAE,KAAK,CAAC,EAAE,CAC1C,iBAAiB,CAAC,0BAA0B,CAAC,CA2H9C,CAAC;AAEF;;;;GAIG;AACH,eAAO,MAAM,gBAAgB,4BAE5B,CAAC"}
+{"version":3,"file":"tambo-registry-provider.d.ts","sourceRoot":"","sources":["../../src/providers/tambo-registry-provider.tsx"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,EAEZ,iBAAiB,EAKlB,MAAM,OAAO,CAAC;AAGf,OAAO,EACL,iBAAiB,EACjB,cAAc,EACd,SAAS,EACV,MAAM,6BAA6B,CAAC;AAErC,MAAM,WAAW,oBAAoB;IACnC,aAAa,EAAE,iBAAiB,CAAC;IACjC,YAAY,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IACxC,yBAAyB,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;IACpD,iBAAiB,EAAE,CAAC,OAAO,EAAE,cAAc,KAAK,IAAI,CAAC;IACrD,YAAY,EAAE,CAAC,IAAI,EAAE,SAAS,KAAK,IAAI,CAAC;IACxC,aAAa,EAAE,CAAC,KAAK,EAAE,SAAS,EAAE,KAAK,IAAI,CAAC;IAC5C,kBAAkB,EAAE,CAAC,aAAa,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,KAAK,IAAI,CAAC;CACtE;AAwBD,MAAM,WAAW,0BAA0B;IACzC,iCAAiC;IACjC,UAAU,CAAC,EAAE,cAAc,EAAE,CAAC;IAC9B,4BAA4B;IAC5B,KAAK,CAAC,EAAE,SAAS,EAAE,CAAC;CACrB;AAED;;;;;;;;GAQG;AACH,eAAO,MAAM,qBAAqB,EAAE,KAAK,CAAC,EAAE,CAC1C,iBAAiB,CAAC,0BAA0B,CAAC,CAiI9C,CAAC;AAEF;;;;GAIG;AACH,eAAO,MAAM,gBAAgB,4BAE5B,CAAC"}

package/esm/providers/tambo-registry-provider.js

@@ -29,9 +29,10 @@ const TamboRegistryContext = createContext({
  * @param props - The props for the TamboRegistryProvider
  * @param props.children - The children to wrap
  * @param props.components - The components to register
+ * @param props.tools - The tools to register
  * @returns The TamboRegistryProvider component
  */
-export const TamboRegistryProvider = ({ children, components: userComponents }) => {
+export const TamboRegistryProvider = ({ children, components: userComponents, tools: userTools }) => {
     const [componentList, setComponentList] = useState({});
     const [toolRegistry, setToolRegistry] = useState({});
     const [componentToolAssociations, setComponentToolAssociations] = useState({});
@@ -101,6 +102,11 @@ export const TamboRegistryProvider = ({ children, components: userComponents })
             });
         }
     }, [registerComponent, userComponents]);
+    useEffect(() => {
+        if (userTools) {
+            registerTools(userTools, false);
+        }
+    }, [registerTools, userTools]);
     const value = {
         componentList,
         toolRegistry,

package/esm/providers/tambo-registry-provider.js.map

@@ -1 +1 @@
-{"version":3,"file":"tambo-registry-provider.js","sourceRoot":"","sources":["../../src/providers/tambo-registry-provider.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;AACb,OAAO,KAAK,EAAE,EACZ,aAAa,EAEb,WAAW,EACX,UAAU,EACV,SAAS,EACT,QAAQ,GACT,MAAM,OAAO,CAAC;AACf,OAAO,EAAE,SAAS,EAAE,MAAM,KAAK,CAAC;AAChC,OAAO,eAAe,MAAM,oBAAoB,CAAC;AAiBjD,MAAM,oBAAoB,GAAG,aAAa,CAAuB;IAC/D,aAAa,EAAE,EAAE;IACjB,YAAY,EAAE,EAAE;IAChB,yBAAyB,EAAE,EAAE;IAC7B;;OAEG;IACH,iBAAiB,EAAE,GAAG,EAAE,GAAE,CAAC;IAC3B;;OAEG;IACH,YAAY,EAAE,GAAG,EAAE,GAAE,CAAC;IACtB;;OAEG;IACH,aAAa,EAAE,GAAG,EAAE,GAAE,CAAC;IACvB;;OAEG;IACH,kBAAkB,EAAE,GAAG,EAAE,GAAE,CAAC;CAC7B,CAAC,CAAC;AAOH;;;;;;;GAOG;AACH,MAAM,CAAC,MAAM,qBAAqB,GAE9B,CAAC,EAAE,QAAQ,EAAE,UAAU,EAAE,cAAc,EAAE,EAAE,EAAE;IAC/C,MAAM,CAAC,aAAa,EAAE,gBAAgB,CAAC,GAAG,QAAQ,CAAoB,EAAE,CAAC,CAAC;IAC1E,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,QAAQ,CAC9C,EAAE,CACH,CAAC;IACF,MAAM,CAAC,yBAAyB,EAAE,4BAA4B,CAAC,GAAG,QAAQ,CAExE,EAAE,CAAC,CAAC;IAEN,MAAM,YAAY,GAAG,WAAW,CAC9B,CAAC,IAAe,EAAE,eAAe,GAAG,IAAI,EAAE,EAAE;QAC1C,eAAe,CAAC,CAAC,IAAI,EAAE,EAAE;YACvB,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,eAAe,EAAE,CAAC;gBACvC,OAAO,CAAC,IAAI,CAAC,oBAAoB,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;YAChD,CAAC;YACD,OAAO;gBACL,GAAG,IAAI;gBACP,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI;aAClB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,EACD,EAAE,CACH,CAAC;IAEF,MAAM,aAAa,GAAG,WAAW,CAC/B,CAAC,KAAkB,EAAE,eAAe,GAAG,IAAI,EAAE,EAAE;QAC7C,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC,CAAC;IAC/D,CAAC,EACD,CAAC,YAAY,CAAC,CACf,CAAC;IAEF,MAAM,kBAAkB,GAAG,WAAW,CACpC,CAAC,aAAqB,EAAE,IAAe,EAAE,EAAE;QACzC,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,EAAE,CAAC;YAClC,MAAM,IAAI,KAAK,CAAC,aAAa,aAAa,wBAAwB,CAAC,CAAC;QACtE,CAAC;QACD,4BAA4B,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACtC,GAAG,IAAI;YACP,CAAC,aAAa,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC;SAC7D,CAAC,CAAC,CAAC;IACN,CAAC,EACD,CAAC,aAAa,CAAC,CAChB,CAAC;IAEF,MAAM,iBAAiB,GAAG,WAAW,CACnC,CAAC,OAAuB,EAAE,eAAe,GAAG,IAAI,EAAE,EAAE;QAClD,MAAM,EACJ,IAAI,EACJ,WAAW,EACX,SAAS,EACT,WAAW,EACX,eAAe,EACf,gBAAgB,EAChB,eAAe,GAChB,GAAG,OAAO,CAAC;QAEZ,0DAA0D;QAC1D,IAAI,CAAC,WAAW,IAAI,CAAC,eAAe,EAAE,CAAC;YACrC,MAAM,IAAI,KAAK,CACb,aAAa,IAAI,wEAAwE,CAC1F,CAAC;QACJ,CAAC;QAED,sDAAsD;QACtD,IAAI,WAAW,IAAI,eAAe,EAAE,CAAC;YACnC,MAAM,IAAI,KAAK,CACb,aAAa,IAAI,0GAA0G,CAC5H,CAAC;QACJ,CAAC;QAED,kDAAkD;QAClD,MAAM,KAAK,GAAG,kBAAkB,CAAC,eAAe,EAAE,WAAW,EAAE,IAAI,CAAC,CAAC;QAErE,gBAAgB,CAAC,CAAC,IAAI,EAAE,EAAE;YACxB,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,eAAe,EAAE,CAAC;gBAClC,OAAO,CAAC,IAAI,CAAC,yBAAyB,IAAI,EAAE,CAAC,CAAC;YAChD,CAAC;YACD,OAAO;gBACL,GAAG,IAAI;gBACP,CAAC,IAAI,CAAC,EAAE;oBACN,SAAS;oBACT,gBAAgB;oBAChB,IAAI;oBACJ,WAAW;oBACX,KAAK;oBACL,YAAY,EAAE,EAAE;iBACjB;aACF,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,IAAI,eAAe,EAAE,CAAC;YACpB,aAAa,CAAC,eAAe,CAAC,CAAC;YAC/B,4BAA4B,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBACtC,GAAG,IAAI;gBACP,CAAC,IAAI,CAAC,EAAE,eAAe,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;aACjD,CAAC,CAAC,CAAC;QACN,CAAC;IACH,CAAC,EACD,CAAC,aAAa,CAAC,CAChB,CAAC;IACF,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,cAAc,EAAE,CAAC;YACnB,cAAc,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;gBACnC,iBAAiB,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;YACtC,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC,EAAE,CAAC,iBAAiB,EAAE,cAAc,CAAC,CAAC,CAAC;IAExC,MAAM,KAAK,GAAG;QACZ,aAAa;QACb,YAAY;QACZ,yBAAyB;QACzB,iBAAiB;QACjB,YAAY;QACZ,aAAa;QACb,kBAAkB;KACnB,CAAC;IAEF,OAAO,CACL,oBAAC,oBAAoB,CAAC,QAAQ,IAAC,KAAK,EAAE,KAAK,IACxC,QAAQ,CACqB,CACjC,CAAC;AACJ,CAAC,CAAC;AAEF;;;;GAIG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,GAAG,EAAE;IACnC,OAAO,UAAU,CAAC,oBAAoB,CAAC,CAAC;AAC1C,CAAC,CAAC;AACF,SAAS,kBAAkB,CACzB,eAAoB,EACpB,WAAgB,EAChB,IAAY;IAEZ,IAAI,eAAe,EAAE,CAAC;QACpB,OAAO,CAAC,IAAI,CAAC,yDAAyD,CAAC,CAAC;QACxE,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,IAAI,WAAW,CAAC,WAAW,CAAC,EAAE,CAAC;QAC7B,IAAI,CAAC;YACH,OAAO,eAAe,CAAC,WAAW,CAAC,CAAC;QACtC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CACX,oBAAoB,IAAI,+BAA+B,EACvD,KAAK,CACN,CAAC;QACJ,CAAC;IACH,CAAC;IACD,qFAAqF;IACrF,IAAI,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC;QAC9B,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,4BAA4B,IAAI,EAAE,CAAC,CAAC;AACtD,CAAC;AAED;;;;;GAKG;AACH,SAAS,YAAY,CAAC,WAAgB;IACpC,OAAO,CACL,WAAW;QACX,OAAO,WAAW,KAAK,QAAQ;QAC/B,WAAW,CAAC,IAAI,KAAK,QAAQ;QAC7B,WAAW,CAAC,UAAU,CACvB,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,SAAS,WAAW,CAAC,GAAY;IAC/B,IAAI,GAAG,YAAY,SAAS,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC;IACd,CAAC;IACD,6CAA6C;IAC7C,OAAO,CACL,OAAO,GAAG,KAAK,QAAQ;QACvB,GAAG,KAAK,IAAI;QACZ,OAAQ,GAAW,CAAC,SAAS,KAAK,UAAU;QAC5C,OAAQ,GAAW,CAAC,IAAI,KAAK,QAAQ,CACtC,CAAC;AACJ,CAAC","sourcesContent":["\"use client\";\nimport React, {\n  createContext,\n  PropsWithChildren,\n  useCallback,\n  useContext,\n  useEffect,\n  useState,\n} from \"react\";\nimport { ZodSchema } from \"zod\";\nimport zodToJsonSchema from \"zod-to-json-schema\";\nimport {\n  ComponentRegistry,\n  TamboComponent,\n  TamboTool,\n} from \"../model/component-metadata\";\n\nexport interface TamboRegistryContext {\n  componentList: ComponentRegistry;\n  toolRegistry: Record<string, TamboTool>;\n  componentToolAssociations: Record<string, string[]>;\n  registerComponent: (options: TamboComponent) => void;\n  registerTool: (tool: TamboTool) => void;\n  registerTools: (tools: TamboTool[]) => void;\n  addToolAssociation: (componentName: string, tool: TamboTool) => void;\n}\n\nconst TamboRegistryContext = createContext<TamboRegistryContext>({\n  componentList: {},\n  toolRegistry: {},\n  componentToolAssociations: {},\n  /**\n   *\n   */\n  registerComponent: () => {},\n  /**\n   *\n   */\n  registerTool: () => {},\n  /**\n   *\n   */\n  registerTools: () => {},\n  /**\n   *\n   */\n  addToolAssociation: () => {},\n});\n\nexport interface TamboRegistryProviderProps {\n  /** The components to register */\n  components?: TamboComponent[];\n}\n\n/**\n * The TamboRegistryProvider is a React provider that provides a component\n * registry to the descendants of the provider.\n * @param props - The props for the TamboRegistryProvider\n * @param props.children - The children to wrap\n * @param props.components - The components to register\n * @returns The TamboRegistryProvider component\n */\nexport const TamboRegistryProvider: React.FC<\n  PropsWithChildren<TamboRegistryProviderProps>\n> = ({ children, components: userComponents }) => {\n  const [componentList, setComponentList] = useState<ComponentRegistry>({});\n  const [toolRegistry, setToolRegistry] = useState<Record<string, TamboTool>>(\n    {},\n  );\n  const [componentToolAssociations, setComponentToolAssociations] = useState<\n    Record<string, string[]>\n  >({});\n\n  const registerTool = useCallback(\n    (tool: TamboTool, warnOnOverwrite = true) => {\n      setToolRegistry((prev) => {\n        if (prev[tool.name] && warnOnOverwrite) {\n          console.warn(`Overwriting tool ${tool.name}`);\n        }\n        return {\n          ...prev,\n          [tool.name]: tool,\n        };\n      });\n    },\n    [],\n  );\n\n  const registerTools = useCallback(\n    (tools: TamboTool[], warnOnOverwrite = true) => {\n      tools.forEach((tool) => registerTool(tool, warnOnOverwrite));\n    },\n    [registerTool],\n  );\n\n  const addToolAssociation = useCallback(\n    (componentName: string, tool: TamboTool) => {\n      if (!componentList[componentName]) {\n        throw new Error(`Component ${componentName} not found in registry`);\n      }\n      setComponentToolAssociations((prev) => ({\n        ...prev,\n        [componentName]: [...(prev[componentName] || []), tool.name],\n      }));\n    },\n    [componentList],\n  );\n\n  const registerComponent = useCallback(\n    (options: TamboComponent, warnOnOverwrite = true) => {\n      const {\n        name,\n        description,\n        component,\n        propsSchema,\n        propsDefinition,\n        loadingComponent,\n        associatedTools,\n      } = options;\n\n      // Validate that at least one props definition is provided\n      if (!propsSchema && !propsDefinition) {\n        throw new Error(\n          `Component ${name} must have either propsSchema (recommended) or propsDefinition defined`,\n        );\n      }\n\n      // Validate that only one props definition is provided\n      if (propsSchema && propsDefinition) {\n        throw new Error(\n          `Component ${name} cannot have both propsSchema and propsDefinition defined. Use only one. We recommend using propsSchema.`,\n        );\n      }\n\n      // Convert propsSchema to JSON Schema if it exists\n      const props = getSerializedProps(propsDefinition, propsSchema, name);\n\n      setComponentList((prev) => {\n        if (prev[name] && warnOnOverwrite) {\n          console.warn(`overwriting component ${name}`);\n        }\n        return {\n          ...prev,\n          [name]: {\n            component,\n            loadingComponent,\n            name,\n            description,\n            props,\n            contextTools: [],\n          },\n        };\n      });\n      if (associatedTools) {\n        registerTools(associatedTools);\n        setComponentToolAssociations((prev) => ({\n          ...prev,\n          [name]: associatedTools.map((tool) => tool.name),\n        }));\n      }\n    },\n    [registerTools],\n  );\n  useEffect(() => {\n    if (userComponents) {\n      userComponents.forEach((component) => {\n        registerComponent(component, false);\n      });\n    }\n  }, [registerComponent, userComponents]);\n\n  const value = {\n    componentList,\n    toolRegistry,\n    componentToolAssociations,\n    registerComponent,\n    registerTool,\n    registerTools,\n    addToolAssociation,\n  };\n\n  return (\n    <TamboRegistryContext.Provider value={value}>\n      {children}\n    </TamboRegistryContext.Provider>\n  );\n};\n\n/**\n * The useTamboRegistry hook provides access to the component registry\n * to the descendants of the TamboRegistryProvider.\n * @returns The component registry\n */\nexport const useTamboRegistry = () => {\n  return useContext(TamboRegistryContext);\n};\nfunction getSerializedProps(\n  propsDefinition: any,\n  propsSchema: any,\n  name: string,\n) {\n  if (propsDefinition) {\n    console.warn(`propsDefinition is deprecated. Use propsSchema instead.`);\n    return propsDefinition;\n  }\n\n  if (isZodSchema(propsSchema)) {\n    try {\n      return zodToJsonSchema(propsSchema);\n    } catch (error) {\n      console.error(\n        `Error converting ${name} props schema to JSON Schema:`,\n        error,\n      );\n    }\n  }\n  // try to roughly detect JSONSchema, should always be an object with a properties key\n  if (isJSONSchema(propsSchema)) {\n    return propsSchema;\n  }\n\n  throw new Error(`Invalid props schema for ${name}`);\n}\n\n/**\n * Checks if the propsSchema is a JSON Schema. This is a rough check, and the\n * server will provide the definitive check.\n * @param propsSchema - The props schema to check\n * @returns True if the props schema is a JSON Schema, false otherwise\n */\nfunction isJSONSchema(propsSchema: any) {\n  return (\n    propsSchema &&\n    typeof propsSchema === \"object\" &&\n    propsSchema.type === \"object\" &&\n    propsSchema.properties\n  );\n}\n\n/**\n * Since we require a certain zod version, we need to check if the object is a ZodSchema\n * @param obj - The object to check\n * @returns True if the object is a ZodSchema, false otherwise\n */\nfunction isZodSchema(obj: unknown): obj is ZodSchema {\n  if (obj instanceof ZodSchema) {\n    return true;\n  }\n  // try to detect if the object is a ZodSchema\n  return (\n    typeof obj === \"object\" &&\n    obj !== null &&\n    typeof (obj as any).safeParse === \"function\" &&\n    typeof (obj as any)._def === \"object\"\n  );\n}\n"]}
+{"version":3,"file":"tambo-registry-provider.js","sourceRoot":"","sources":["../../src/providers/tambo-registry-provider.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;AACb,OAAO,KAAK,EAAE,EACZ,aAAa,EAEb,WAAW,EACX,UAAU,EACV,SAAS,EACT,QAAQ,GACT,MAAM,OAAO,CAAC;AACf,OAAO,EAAE,SAAS,EAAE,MAAM,KAAK,CAAC;AAChC,OAAO,eAAe,MAAM,oBAAoB,CAAC;AAiBjD,MAAM,oBAAoB,GAAG,aAAa,CAAuB;IAC/D,aAAa,EAAE,EAAE;IACjB,YAAY,EAAE,EAAE;IAChB,yBAAyB,EAAE,EAAE;IAC7B;;OAEG;IACH,iBAAiB,EAAE,GAAG,EAAE,GAAE,CAAC;IAC3B;;OAEG;IACH,YAAY,EAAE,GAAG,EAAE,GAAE,CAAC;IACtB;;OAEG;IACH,aAAa,EAAE,GAAG,EAAE,GAAE,CAAC;IACvB;;OAEG;IACH,kBAAkB,EAAE,GAAG,EAAE,GAAE,CAAC;CAC7B,CAAC,CAAC;AASH;;;;;;;;GAQG;AACH,MAAM,CAAC,MAAM,qBAAqB,GAE9B,CAAC,EAAE,QAAQ,EAAE,UAAU,EAAE,cAAc,EAAE,KAAK,EAAE,SAAS,EAAE,EAAE,EAAE;IACjE,MAAM,CAAC,aAAa,EAAE,gBAAgB,CAAC,GAAG,QAAQ,CAAoB,EAAE,CAAC,CAAC;IAC1E,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,QAAQ,CAC9C,EAAE,CACH,CAAC;IACF,MAAM,CAAC,yBAAyB,EAAE,4BAA4B,CAAC,GAAG,QAAQ,CAExE,EAAE,CAAC,CAAC;IAEN,MAAM,YAAY,GAAG,WAAW,CAC9B,CAAC,IAAe,EAAE,eAAe,GAAG,IAAI,EAAE,EAAE;QAC1C,eAAe,CAAC,CAAC,IAAI,EAAE,EAAE;YACvB,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,eAAe,EAAE,CAAC;gBACvC,OAAO,CAAC,IAAI,CAAC,oBAAoB,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;YAChD,CAAC;YACD,OAAO;gBACL,GAAG,IAAI;gBACP,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI;aAClB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,EACD,EAAE,CACH,CAAC;IAEF,MAAM,aAAa,GAAG,WAAW,CAC/B,CAAC,KAAkB,EAAE,eAAe,GAAG,IAAI,EAAE,EAAE;QAC7C,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC,CAAC;IAC/D,CAAC,EACD,CAAC,YAAY,CAAC,CACf,CAAC;IAEF,MAAM,kBAAkB,GAAG,WAAW,CACpC,CAAC,aAAqB,EAAE,IAAe,EAAE,EAAE;QACzC,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,EAAE,CAAC;YAClC,MAAM,IAAI,KAAK,CAAC,aAAa,aAAa,wBAAwB,CAAC,CAAC;QACtE,CAAC;QACD,4BAA4B,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACtC,GAAG,IAAI;YACP,CAAC,aAAa,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC;SAC7D,CAAC,CAAC,CAAC;IACN,CAAC,EACD,CAAC,aAAa,CAAC,CAChB,CAAC;IAEF,MAAM,iBAAiB,GAAG,WAAW,CACnC,CAAC,OAAuB,EAAE,eAAe,GAAG,IAAI,EAAE,EAAE;QAClD,MAAM,EACJ,IAAI,EACJ,WAAW,EACX,SAAS,EACT,WAAW,EACX,eAAe,EACf,gBAAgB,EAChB,eAAe,GAChB,GAAG,OAAO,CAAC;QAEZ,0DAA0D;QAC1D,IAAI,CAAC,WAAW,IAAI,CAAC,eAAe,EAAE,CAAC;YACrC,MAAM,IAAI,KAAK,CACb,aAAa,IAAI,wEAAwE,CAC1F,CAAC;QACJ,CAAC;QAED,sDAAsD;QACtD,IAAI,WAAW,IAAI,eAAe,EAAE,CAAC;YACnC,MAAM,IAAI,KAAK,CACb,aAAa,IAAI,0GAA0G,CAC5H,CAAC;QACJ,CAAC;QAED,kDAAkD;QAClD,MAAM,KAAK,GAAG,kBAAkB,CAAC,eAAe,EAAE,WAAW,EAAE,IAAI,CAAC,CAAC;QAErE,gBAAgB,CAAC,CAAC,IAAI,EAAE,EAAE;YACxB,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,eAAe,EAAE,CAAC;gBAClC,OAAO,CAAC,IAAI,CAAC,yBAAyB,IAAI,EAAE,CAAC,CAAC;YAChD,CAAC;YACD,OAAO;gBACL,GAAG,IAAI;gBACP,CAAC,IAAI,CAAC,EAAE;oBACN,SAAS;oBACT,gBAAgB;oBAChB,IAAI;oBACJ,WAAW;oBACX,KAAK;oBACL,YAAY,EAAE,EAAE;iBACjB;aACF,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,IAAI,eAAe,EAAE,CAAC;YACpB,aAAa,CAAC,eAAe,CAAC,CAAC;YAC/B,4BAA4B,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBACtC,GAAG,IAAI;gBACP,CAAC,IAAI,CAAC,EAAE,eAAe,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;aACjD,CAAC,CAAC,CAAC;QACN,CAAC;IACH,CAAC,EACD,CAAC,aAAa,CAAC,CAChB,CAAC;IACF,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,cAAc,EAAE,CAAC;YACnB,cAAc,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;gBACnC,iBAAiB,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;YACtC,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC,EAAE,CAAC,iBAAiB,EAAE,cAAc,CAAC,CAAC,CAAC;IAExC,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,SAAS,EAAE,CAAC;YACd,aAAa,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QAClC,CAAC;IACH,CAAC,EAAE,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC,CAAC;IAE/B,MAAM,KAAK,GAAG;QACZ,aAAa;QACb,YAAY;QACZ,yBAAyB;QACzB,iBAAiB;QACjB,YAAY;QACZ,aAAa;QACb,kBAAkB;KACnB,CAAC;IAEF,OAAO,CACL,oBAAC,oBAAoB,CAAC,QAAQ,IAAC,KAAK,EAAE,KAAK,IACxC,QAAQ,CACqB,CACjC,CAAC;AACJ,CAAC,CAAC;AAEF;;;;GAIG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,GAAG,EAAE;IACnC,OAAO,UAAU,CAAC,oBAAoB,CAAC,CAAC;AAC1C,CAAC,CAAC;AACF,SAAS,kBAAkB,CACzB,eAAoB,EACpB,WAAgB,EAChB,IAAY;IAEZ,IAAI,eAAe,EAAE,CAAC;QACpB,OAAO,CAAC,IAAI,CAAC,yDAAyD,CAAC,CAAC;QACxE,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,IAAI,WAAW,CAAC,WAAW,CAAC,EAAE,CAAC;QAC7B,IAAI,CAAC;YACH,OAAO,eAAe,CAAC,WAAW,CAAC,CAAC;QACtC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CACX,oBAAoB,IAAI,+BAA+B,EACvD,KAAK,CACN,CAAC;QACJ,CAAC;IACH,CAAC;IACD,qFAAqF;IACrF,IAAI,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC;QAC9B,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,4BAA4B,IAAI,EAAE,CAAC,CAAC;AACtD,CAAC;AAED;;;;;GAKG;AACH,SAAS,YAAY,CAAC,WAAgB;IACpC,OAAO,CACL,WAAW;QACX,OAAO,WAAW,KAAK,QAAQ;QAC/B,WAAW,CAAC,IAAI,KAAK,QAAQ;QAC7B,WAAW,CAAC,UAAU,CACvB,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,SAAS,WAAW,CAAC,GAAY;IAC/B,IAAI,GAAG,YAAY,SAAS,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC;IACd,CAAC;IACD,6CAA6C;IAC7C,OAAO,CACL,OAAO,GAAG,KAAK,QAAQ;QACvB,GAAG,KAAK,IAAI;QACZ,OAAQ,GAAW,CAAC,SAAS,KAAK,UAAU;QAC5C,OAAQ,GAAW,CAAC,IAAI,KAAK,QAAQ,CACtC,CAAC;AACJ,CAAC","sourcesContent":["\"use client\";\nimport React, {\n  createContext,\n  PropsWithChildren,\n  useCallback,\n  useContext,\n  useEffect,\n  useState,\n} from \"react\";\nimport { ZodSchema } from \"zod\";\nimport zodToJsonSchema from \"zod-to-json-schema\";\nimport {\n  ComponentRegistry,\n  TamboComponent,\n  TamboTool,\n} from \"../model/component-metadata\";\n\nexport interface TamboRegistryContext {\n  componentList: ComponentRegistry;\n  toolRegistry: Record<string, TamboTool>;\n  componentToolAssociations: Record<string, string[]>;\n  registerComponent: (options: TamboComponent) => void;\n  registerTool: (tool: TamboTool) => void;\n  registerTools: (tools: TamboTool[]) => void;\n  addToolAssociation: (componentName: string, tool: TamboTool) => void;\n}\n\nconst TamboRegistryContext = createContext<TamboRegistryContext>({\n  componentList: {},\n  toolRegistry: {},\n  componentToolAssociations: {},\n  /**\n   *\n   */\n  registerComponent: () => {},\n  /**\n   *\n   */\n  registerTool: () => {},\n  /**\n   *\n   */\n  registerTools: () => {},\n  /**\n   *\n   */\n  addToolAssociation: () => {},\n});\n\nexport interface TamboRegistryProviderProps {\n  /** The components to register */\n  components?: TamboComponent[];\n  /** The tools to register */\n  tools?: TamboTool[];\n}\n\n/**\n * The TamboRegistryProvider is a React provider that provides a component\n * registry to the descendants of the provider.\n * @param props - The props for the TamboRegistryProvider\n * @param props.children - The children to wrap\n * @param props.components - The components to register\n * @param props.tools - The tools to register\n * @returns The TamboRegistryProvider component\n */\nexport const TamboRegistryProvider: React.FC<\n  PropsWithChildren<TamboRegistryProviderProps>\n> = ({ children, components: userComponents, tools: userTools }) => {\n  const [componentList, setComponentList] = useState<ComponentRegistry>({});\n  const [toolRegistry, setToolRegistry] = useState<Record<string, TamboTool>>(\n    {},\n  );\n  const [componentToolAssociations, setComponentToolAssociations] = useState<\n    Record<string, string[]>\n  >({});\n\n  const registerTool = useCallback(\n    (tool: TamboTool, warnOnOverwrite = true) => {\n      setToolRegistry((prev) => {\n        if (prev[tool.name] && warnOnOverwrite) {\n          console.warn(`Overwriting tool ${tool.name}`);\n        }\n        return {\n          ...prev,\n          [tool.name]: tool,\n        };\n      });\n    },\n    [],\n  );\n\n  const registerTools = useCallback(\n    (tools: TamboTool[], warnOnOverwrite = true) => {\n      tools.forEach((tool) => registerTool(tool, warnOnOverwrite));\n    },\n    [registerTool],\n  );\n\n  const addToolAssociation = useCallback(\n    (componentName: string, tool: TamboTool) => {\n      if (!componentList[componentName]) {\n        throw new Error(`Component ${componentName} not found in registry`);\n      }\n      setComponentToolAssociations((prev) => ({\n        ...prev,\n        [componentName]: [...(prev[componentName] || []), tool.name],\n      }));\n    },\n    [componentList],\n  );\n\n  const registerComponent = useCallback(\n    (options: TamboComponent, warnOnOverwrite = true) => {\n      const {\n        name,\n        description,\n        component,\n        propsSchema,\n        propsDefinition,\n        loadingComponent,\n        associatedTools,\n      } = options;\n\n      // Validate that at least one props definition is provided\n      if (!propsSchema && !propsDefinition) {\n        throw new Error(\n          `Component ${name} must have either propsSchema (recommended) or propsDefinition defined`,\n        );\n      }\n\n      // Validate that only one props definition is provided\n      if (propsSchema && propsDefinition) {\n        throw new Error(\n          `Component ${name} cannot have both propsSchema and propsDefinition defined. Use only one. We recommend using propsSchema.`,\n        );\n      }\n\n      // Convert propsSchema to JSON Schema if it exists\n      const props = getSerializedProps(propsDefinition, propsSchema, name);\n\n      setComponentList((prev) => {\n        if (prev[name] && warnOnOverwrite) {\n          console.warn(`overwriting component ${name}`);\n        }\n        return {\n          ...prev,\n          [name]: {\n            component,\n            loadingComponent,\n            name,\n            description,\n            props,\n            contextTools: [],\n          },\n        };\n      });\n      if (associatedTools) {\n        registerTools(associatedTools);\n        setComponentToolAssociations((prev) => ({\n          ...prev,\n          [name]: associatedTools.map((tool) => tool.name),\n        }));\n      }\n    },\n    [registerTools],\n  );\n  useEffect(() => {\n    if (userComponents) {\n      userComponents.forEach((component) => {\n        registerComponent(component, false);\n      });\n    }\n  }, [registerComponent, userComponents]);\n\n  useEffect(() => {\n    if (userTools) {\n      registerTools(userTools, false);\n    }\n  }, [registerTools, userTools]);\n\n  const value = {\n    componentList,\n    toolRegistry,\n    componentToolAssociations,\n    registerComponent,\n    registerTool,\n    registerTools,\n    addToolAssociation,\n  };\n\n  return (\n    <TamboRegistryContext.Provider value={value}>\n      {children}\n    </TamboRegistryContext.Provider>\n  );\n};\n\n/**\n * The useTamboRegistry hook provides access to the component registry\n * to the descendants of the TamboRegistryProvider.\n * @returns The component registry\n */\nexport const useTamboRegistry = () => {\n  return useContext(TamboRegistryContext);\n};\nfunction getSerializedProps(\n  propsDefinition: any,\n  propsSchema: any,\n  name: string,\n) {\n  if (propsDefinition) {\n    console.warn(`propsDefinition is deprecated. Use propsSchema instead.`);\n    return propsDefinition;\n  }\n\n  if (isZodSchema(propsSchema)) {\n    try {\n      return zodToJsonSchema(propsSchema);\n    } catch (error) {\n      console.error(\n        `Error converting ${name} props schema to JSON Schema:`,\n        error,\n      );\n    }\n  }\n  // try to roughly detect JSONSchema, should always be an object with a properties key\n  if (isJSONSchema(propsSchema)) {\n    return propsSchema;\n  }\n\n  throw new Error(`Invalid props schema for ${name}`);\n}\n\n/**\n * Checks if the propsSchema is a JSON Schema. This is a rough check, and the\n * server will provide the definitive check.\n * @param propsSchema - The props schema to check\n * @returns True if the props schema is a JSON Schema, false otherwise\n */\nfunction isJSONSchema(propsSchema: any) {\n  return (\n    propsSchema &&\n    typeof propsSchema === \"object\" &&\n    propsSchema.type === \"object\" &&\n    propsSchema.properties\n  );\n}\n\n/**\n * Since we require a certain zod version, we need to check if the object is a ZodSchema\n * @param obj - The object to check\n * @returns True if the object is a ZodSchema, false otherwise\n */\nfunction isZodSchema(obj: unknown): obj is ZodSchema {\n  if (obj instanceof ZodSchema) {\n    return true;\n  }\n  // try to detect if the object is a ZodSchema\n  return (\n    typeof obj === \"object\" &&\n    obj !== null &&\n    typeof (obj as any).safeParse === \"function\" &&\n    typeof (obj as any)._def === \"object\"\n  );\n}\n"]}

package/esm/providers/tambo-thread-provider.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"tambo-thread-provider.d.ts","sourceRoot":"","sources":["../../src/providers/tambo-thread-provider.tsx"],"names":[],"mappings":"AACA,OAAO,OAA0B,MAAM,0BAA0B,CAAC;AAClE,OAAO,KAAK,EAAE,EAEZ,iBAAiB,EAMlB,MAAM,OAAO,CAAC;AACf,OAAO,EACL,eAAe,EAEf,kBAAkB,EACnB,MAAM,sCAAsC,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAYpD,MAAM,WAAW,uBAAuB;IACtC,yBAAyB;IACzB,MAAM,EAAE,WAAW,CAAC;IACpB,mCAAmC;IACnC,mBAAmB,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,OAAO,KAAK,IAAI,CAAC;IACjE,yBAAyB;IACzB,cAAc,EAAE,MAAM,IAAI,CAAC;IAC3B,0CAA0C;IAC1C,gBAAgB,EAAE,CAChB,OAAO,EAAE,kBAAkB,EAC3B,YAAY,EAAE,OAAO,KAClB,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC,CAAC;IACnD,6CAA6C;IAC7C,mBAAmB,EAAE,CACnB,EAAE,EAAE,MAAM,EACV,OAAO,EAAE,kBAAkB,EAC3B,YAAY,EAAE,OAAO,KAClB,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,4CAA4C;IAC5C,UAAU,EAAE,MAAM,CAAC;IACnB,gDAAgD;IAChD,aAAa,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;IACvC,2CAA2C;IAC3C,iBAAiB,EAAE,CACjB,OAAO,EAAE,MAAM,EACf,OAAO,EAAE;QACP,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,cAAc,CAAC,EAAE,OAAO,CAAC;QACzB,UAAU,CAAC,EAAE,MAAM,CAAC;KACrB,KACE,OAAO,CAAC,kBAAkB,CAAC,CAAC;IACjC,oFAAoF;IACpF,eAAe,EAAE,eAAe,CAAC;IACjC,6FAA6F;IAC7F,uBAAuB,EAAE,MAAM,CAAC;IAChC,iCAAiC;IACjC,MAAM,EAAE,OAAO,CAAC;CACjB;AAED;;;;;;GAMG;AACH,eAAO,MAAM,kBAAkB,EAAE,WAOhC,CAAC;AAEF,eAAO,MAAM,kBAAkB,wCA0C7B,CAAC;AAEH;;;;;;GAMG;AACH,eAAO,MAAM,mBAAmB,EAAE,KAAK,CAAC,EAAE,CAAC,iBAAiB,CAgf3D,CAAC;AAEF;;;;GAIG;AACH,eAAO,MAAM,cAAc,+BAM1B,CAAC"}
+{"version":3,"file":"tambo-thread-provider.d.ts","sourceRoot":"","sources":["../../src/providers/tambo-thread-provider.tsx"],"names":[],"mappings":"AACA,OAAO,OAA0B,MAAM,0BAA0B,CAAC;AAClE,OAAO,KAAK,EAAE,EAEZ,iBAAiB,EAMlB,MAAM,OAAO,CAAC;AACf,OAAO,EACL,eAAe,EAEf,kBAAkB,EACnB,MAAM,sCAAsC,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAYpD,MAAM,WAAW,uBAAuB;IACtC,yBAAyB;IACzB,MAAM,EAAE,WAAW,CAAC;IACpB,mCAAmC;IACnC,mBAAmB,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,OAAO,KAAK,IAAI,CAAC;IACjE,yBAAyB;IACzB,cAAc,EAAE,MAAM,IAAI,CAAC;IAC3B,0CAA0C;IAC1C,gBAAgB,EAAE,CAChB,OAAO,EAAE,kBAAkB,EAC3B,YAAY,EAAE,OAAO,KAClB,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC,CAAC;IACnD,6CAA6C;IAC7C,mBAAmB,EAAE,CACnB,EAAE,EAAE,MAAM,EACV,OAAO,EAAE,kBAAkB,EAC3B,YAAY,EAAE,OAAO,KAClB,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,4CAA4C;IAC5C,UAAU,EAAE,MAAM,CAAC;IACnB,gDAAgD;IAChD,aAAa,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;IACvC,2CAA2C;IAC3C,iBAAiB,EAAE,CACjB,OAAO,EAAE,MAAM,EACf,OAAO,EAAE;QACP,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,cAAc,CAAC,EAAE,OAAO,CAAC;QACzB,UAAU,CAAC,EAAE,MAAM,CAAC;KACrB,KACE,OAAO,CAAC,kBAAkB,CAAC,CAAC;IACjC,oFAAoF;IACpF,eAAe,EAAE,eAAe,CAAC;IACjC,6FAA6F;IAC7F,uBAAuB,EAAE,MAAM,CAAC;IAChC,iCAAiC;IACjC,MAAM,EAAE,OAAO,CAAC;CACjB;AAED;;;;;;GAMG;AACH,eAAO,MAAM,kBAAkB,EAAE,WAOhC,CAAC;AAEF,eAAO,MAAM,kBAAkB,wCA0C7B,CAAC;AAEH;;;;;;GAMG;AACH,eAAO,MAAM,mBAAmB,EAAE,KAAK,CAAC,EAAE,CAAC,iBAAiB,CAwe3D,CAAC;AAEF;;;;GAIG;AACH,eAAO,MAAM,cAAc,+BAM1B,CAAC"}

package/esm/providers/tambo-thread-provider.js

@@ -136,13 +136,23 @@ export const TamboThreadProvider = ({ children, }) => {
             createdAt,
         };
         const threadId = message.threadId;
+        const messageId = chatMessage.id;
         // optimistically update the thread in the local state
         setThreadMap((prevMap) => {
             if (!threadId) {
                 return prevMap;
             }
             const prevMessages = prevMap[threadId]?.messages || [];
-            const updatedMessages = [...prevMessages, chatMessage];
+            const haveMessage = prevMessages.find((msg) => msg.id === messageId);
+            // Update in place if the message already exists
+            const updatedMessages = haveMessage
+                ? prevMessages.map((msg) => {
+                    if (msg.id === messageId) {
+                        return chatMessage;
+                    }
+                    return msg;
+                })
+                : [...prevMessages, chatMessage];
             const updatedThreadMap = {
                 ...prevMap,
                 [threadId]: {
@@ -195,17 +205,6 @@ export const TamboThreadProvider = ({ children, }) => {
             });
         }
     }, [client.beta.threads.messages]);
-    const deleteThreadMessage = useCallback((messageId, threadId) => {
-        if (!threadMap[threadId])
-            return;
-        setThreadMap((prevMap) => ({
-            ...prevMap,
-            [threadId]: {
-                ...prevMap[threadId],
-                messages: prevMap[threadId].messages.filter((msg) => msg.id !== messageId),
-            },
-        }));
-    }, [threadMap]);
     const startNewThread = useCallback(() => {
         setCurrentThreadId(PLACEHOLDER_THREAD.id);
         setThreadMap((prevMap) => {
@@ -238,7 +237,7 @@ export const TamboThreadProvider = ({ children, }) => {
         if (fetch) {
             await fetchThread(threadId);
         }
-    }, [fetchThread, threadMap]);
+    }, [fetchThread]);
     const updateThreadStatus = useCallback((threadId, stage, statusMessage) => {
         setThreadMap((prevMap) => {
             const updatedThreadMap = {
@@ -252,16 +251,11 @@ export const TamboThreadProvider = ({ children, }) => {
             return updatedThreadMap;
         });
     }, []);
-    const handleAdvanceStream = useCallback(async (stream, params, threadId, messageIdToRemove) => {
+    const handleAdvanceStream = useCallback(async (stream, params, threadId) => {
         let finalMessage;
         let hasSetThreadId = false;
-        let isFirstChunk = true;
         updateThreadStatus(threadId, GenerationStage.STREAMING_RESPONSE);
         for await (const chunk of stream) {
-            if (isFirstChunk && messageIdToRemove) {
-                deleteThreadMessage(messageIdToRemove, threadId);
-            }
-            isFirstChunk = false;
             if (chunk.responseMessageDto.toolCallRequest) {
                 updateThreadStatus(chunk.responseMessageDto.threadId, GenerationStage.FETCHING_CONTEXT);
                 const toolCallResponse = await handleToolCall(chunk.responseMessageDto, toolRegistry);
@@ -287,20 +281,26 @@ export const TamboThreadProvider = ({ children, }) => {
                     chunk.responseMessageDto.threadId &&
                     chunk.responseMessageDto.threadId !== currentThread?.id) {
                     hasSetThreadId = true;
-                    switchCurrentThread(chunk.responseMessageDto.threadId, false);
+                    await switchCurrentThread(chunk.responseMessageDto.threadId, false);
                 }
                 if (!finalMessage) {
                     finalMessage = chunk.responseMessageDto.component?.componentName
                         ? renderComponentIntoMessage(chunk.responseMessageDto, componentList)
                         : chunk.responseMessageDto;
-                    addThreadMessage(finalMessage, false);
+                    await addThreadMessage(finalMessage, false);
                 }
                 else {
-                    const previousId = finalMessage.id;
+                    // if we start getting a new message mid-stream, put the previous one on screen
+                    const isNewMessage = chunk.responseMessageDto.id !== finalMessage.id;
                     finalMessage = chunk.responseMessageDto.component?.componentName
                         ? renderComponentIntoMessage(chunk.responseMessageDto, componentList)
                         : chunk.responseMessageDto;
-                    updateThreadMessage(previousId, finalMessage, false);
+                    if (isNewMessage) {
+                        await addThreadMessage(finalMessage, false);
+                    }
+                    else {
+                        await updateThreadMessage(finalMessage.id, finalMessage, false);
+                    }
                 }
             }
         }
@@ -308,7 +308,7 @@ export const TamboThreadProvider = ({ children, }) => {
         return (finalMessage ?? {
             threadId: "",
             content: [{ type: "text", text: `Error processing stream` }],
-            role: "hydra",
+            role: "assistant",
             createdAt: new Date().toISOString(),
             id: crypto.randomUUID(),
             componentState: {},
@@ -318,8 +318,6 @@ export const TamboThreadProvider = ({ children, }) => {
         client,
         componentList,
         currentThread?.id,
-        deleteThreadMessage,
-        fetchThread,
         switchCurrentThread,
         toolRegistry,
         updateThreadMessage,

package/esm/providers/tambo-thread-provider.js.map

@@ -1 +1 @@
-{"version":3,"file":"tambo-thread-provider.js","sourceRoot":"","sources":["../../src/providers/tambo-thread-provider.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;AACb,OAAgB,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AAClE,OAAO,KAAK,EAAE,EACZ,aAAa,EAEb,WAAW,EACX,UAAU,EACV,SAAS,EACT,OAAO,EACP,QAAQ,GACT,MAAM,OAAO,CAAC;AACf,OAAO,EACL,eAAe,EACf,WAAW,GAEZ,MAAM,sCAAsC,CAAC;AAE9C,OAAO,EAAE,0BAA0B,EAAE,MAAM,4BAA4B,CAAC;AACxE,OAAO,EACL,sBAAsB,EACtB,gBAAgB,EAChB,oBAAoB,EACpB,yBAAyB,GAC1B,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AACzD,OAAO,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAyC7D;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAgB;IAC7C,EAAE,EAAE,aAAa;IACjB,QAAQ,EAAE,EAAE;IACZ,SAAS,EAAE,EAAE;IACb,SAAS,EAAE,EAAE;IACb,SAAS,EAAE,EAAE;IACb,QAAQ,EAAE,EAAE;CACb,CAAC;AAEF,MAAM,CAAC,MAAM,kBAAkB,GAAG,aAAa,CAA0B;IACvE,MAAM,EAAE,kBAAkB;IAC1B;;OAEG;IACH,mBAAmB,EAAE,GAAG,EAAE;QACxB,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;IACzD,CAAC;IACD;;OAEG;IACH,cAAc,EAAE,GAAG,EAAE;QACnB,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;IACpD,CAAC;IACD;;OAEG;IACH,gBAAgB,EAAE,GAAG,EAAE;QACrB,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;IAChE,CAAC;IACD,UAAU,EAAE,EAAE;IACd;;OAEG;IACH,aAAa,EAAE,GAAG,EAAE;QAClB,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;IACnD,CAAC;IACD;;OAEG;IACH,mBAAmB,EAAE,GAAG,EAAE;QACxB,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;IACzD,CAAC;IACD;;OAEG;IACH,iBAAiB,EAAE,GAAG,EAAE;QACtB,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC7C,CAAC;IACD,eAAe,EAAE,eAAe,CAAC,IAAI;IACrC,uBAAuB,EAAE,EAAE;IAC3B,MAAM,EAAE,IAAI;CACb,CAAC,CAAC;AAEH;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,mBAAmB,GAAgC,CAAC,EAC/D,QAAQ,GACT,EAAE,EAAE;IACH,MAAM,CAAC,SAAS,EAAE,YAAY,CAAC,GAAG,QAAQ,CAA8B;QACtE,CAAC,kBAAkB,CAAC,EAAE,CAAC,EAAE,kBAAkB;KAC5C,CAAC,CAAC;IACH,MAAM,MAAM,GAAG,cAAc,EAAE,CAAC;IAChC,MAAM,EAAE,aAAa,EAAE,YAAY,EAAE,yBAAyB,EAAE,GAC9D,gBAAgB,EAAE,CAAC;IACrB,MAAM,CAAC,UAAU,EAAE,aAAa,CAAC,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC;IAEjD,MAAM,CAAC,eAAe,EAAE,kBAAkB,CAAC,GAAG,QAAQ,CACpD,kBAAkB,CAAC,EAAE,CACtB,CAAC;IACF,MAAM,aAAa,GAA4B,SAAS,CAAC,eAAe,CAAC,CAAC;IAE1E,sFAAsF;IACtF,MAAM,mBAAmB,GAAG,OAAO,CAAC,GAAG,EAAE;QACvC,MAAM,YAAY,GAAG,IAAI,GAAG,EAA8B,CAAC;QAC3D,IAAI,aAAa,EAAE,CAAC;YAClB,KAAK,MAAM,OAAO,IAAI,aAAa,CAAC,QAAQ,EAAE,CAAC;gBAC7C,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YACxC,CAAC;QACH,CAAC;QACD,OAAO,YAAY,CAAC;IACtB,CAAC,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC;IAEpB,MAAM,WAAW,GAAG,WAAW,CAC7B,KAAK,EAAE,QAAgB,EAAE,EAAE;QACzB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAC5D,MAAM,4BAA4B,GAAG;YACnC,GAAG,MAAM;YACT,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;gBACxC,IAAI,mBAAmB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,CAAC;oBACxC,MAAM,eAAe,GAAG,mBAAmB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;oBAC5D,OAAO;wBACL,GAAG,eAAe;wBAClB,GAAG,OAAO;qBACX,CAAC;gBACJ,CAAC;gBACD,IAAI,OAAO,CAAC,SAAS,EAAE,aAAa,EAAE,CAAC;oBACrC,MAAM,oBAAoB,GAAG,0BAA0B,CACrD,OAAO,EACP,aAAa,CACd,CAAC;oBACF,OAAO,oBAAoB,CAAC;gBAC9B,CAAC;gBACD,OAAO,OAAO,CAAC;YACjB,CAAC,CAAC;SACH,CAAC;QAEF,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE;YACvB,MAAM,gBAAgB,GAAG;gBACvB,GAAG,OAAO;gBACV,CAAC,QAAQ,CAAC,EAAE,4BAA4B;aACzC,CAAC;YACF,OAAO,gBAAgB,CAAC;QAC1B,CAAC,CAAC,CAAC;IACL,CAAC,EACD,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,aAAa,EAAE,mBAAmB,CAAC,CAC1D,CAAC;IAEF,SAAS,CAAC,GAAG,EAAE;QACb,IACE,eAAe;YACf,eAAe,KAAK,kBAAkB,CAAC,EAAE;YACzC,CAAC,SAAS,CAAC,eAAe,CAAC,EAC3B,CAAC;YACD,WAAW,CAAC,eAAe,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC,EAAE,CAAC,eAAe,EAAE,WAAW,EAAE,SAAS,CAAC,CAAC,CAAC;IAE9C,MAAM,gBAAgB,GAAG,WAAW,CAClC,KAAK,EACH,OAA2B,EAC3B,YAAY,GAAG,IAAI,EACnB,YAAoB,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAC5C,EAAE;QACF,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,OAAO,CAAC,IAAI,CAAC,wDAAwD,CAAC,CAAC;YACvE,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,MAAM,WAAW,GAEb;YACF,GAAG,OAAO;YACV,iBAAiB,EACf,OAAO,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC,SAAS;YAC1D,SAAS;SACV,CAAC;QACF,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;QAClC,sDAAsD;QACtD,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE;YACvB,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,OAAO,OAAO,CAAC;YACjB,CAAC;YACD,MAAM,YAAY,GAAG,OAAO,CAAC,QAAQ,CAAC,EAAE,QAAQ,IAAI,EAAE,CAAC;YACvD,MAAM,eAAe,GAAG,CAAC,GAAG,YAAY,EAAE,WAAW,CAAC,CAAC;YAEvD,MAAM,gBAAgB,GAAG;gBACvB,GAAG,OAAO;gBACV,CAAC,QAAQ,CAAC,EAAE;oBACV,GAAG,OAAO,CAAC,QAAQ,CAAC;oBACpB,QAAQ,EAAE,eAAe;iBAC1B;aACF,CAAC;YACF,OAAO,gBAAgB,CAAC;QAC1B,CAAC,CAAC,CAAC;QAEH,IAAI,YAAY,EAAE,CAAC;YACjB,gEAAgE;YAChE,MAAM,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE;gBAC1D,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,oDAAoD;aACrD,CAAC,CAAC;QACL,CAAC;QACD,OAAO,SAAS,CAAC,QAAQ,CAAC,EAAE,QAAQ,IAAI,EAAE,CAAC;IAC7C,CAAC,EACD,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,aAAa,EAAE,SAAS,CAAC,CACzD,CAAC;IAEF,MAAM,mBAAmB,GAAG,WAAW,CACrC,KAAK,EACH,EAAU,EACV,OAA2B,EAC3B,YAAY,GAAG,IAAI,EACnB,YAAoB,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAC5C,EAAE;QACF,MAAM,WAAW,GAAuB;YACtC,GAAG,OAAO;YACV,SAAS;SACV,CAAC;QAEF,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE;YACvB,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;gBACtB,OAAO,OAAO,CAAC;YACjB,CAAC;YACD,MAAM,YAAY,GAAG,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,QAAQ,IAAI,EAAE,CAAC;YAC/D,MAAM,eAAe,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;gBAC/C,IAAI,GAAG,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC;oBAClB,OAAO,WAAW,CAAC;gBACrB,CAAC;gBACD,OAAO,GAAG,CAAC;YACb,CAAC,CAAC,CAAC;YACH,OAAO;gBACL,GAAG,OAAO;gBACV,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE;oBAClB,GAAG,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC;oBAC5B,QAAQ,EAAE,eAAe;iBAC1B;aACF,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,IAAI,YAAY,EAAE,CAAC;YACjB,gEAAgE;YAChE,MAAM,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE;gBAC1D,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,oDAAoD;aACrD,CAAC,CAAC;QACL,CAAC;IACH,CAAC,EACD,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAC/B,CAAC;IAEF,MAAM,mBAAmB,GAAG,WAAW,CACrC,CAAC,SAAiB,EAAE,QAAgB,EAAE,EAAE;QACtC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC;YAAE,OAAO;QAEjC,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;YACzB,GAAG,OAAO;YACV,CAAC,QAAQ,CAAC,EAAE;gBACV,GAAG,OAAO,CAAC,QAAQ,CAAC;gBACpB,QAAQ,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC,QAAQ,CAAC,MAAM,CACzC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,KAAK,SAAS,CAC9B;aACF;SACF,CAAC,CAAC,CAAC;IACN,CAAC,EACD,CAAC,SAAS,CAAC,CACZ,CAAC;IAEF,MAAM,cAAc,GAAG,WAAW,CAAC,GAAG,EAAE;QACtC,kBAAkB,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC;QAC1C,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE;YACvB,OAAO;gBACL,GAAG,OAAO;gBACV,CAAC,kBAAkB,CAAC,EAAE,CAAC,EAAE,kBAAkB;aAC5C,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,mBAAmB,GAAG,WAAW,CACrC,KAAK,EAAE,QAAgB,EAAE,KAAK,GAAG,IAAI,EAAE,EAAE;QACvC,IAAI,QAAQ,KAAK,kBAAkB,CAAC,EAAE,EAAE,CAAC;YACvC,OAAO,CAAC,IAAI,CAAC,gDAAgD,CAAC,CAAC;YAC/D,OAAO;QACT,CAAC;QACD,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QAC7B,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE;YACvB,IAAI,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACtB,OAAO,OAAO,CAAC;YACjB,CAAC;YACD,yEAAyE;YACzE,MAAM,gBAAgB,GAAG;gBACvB,GAAG,OAAO;gBACV,CAAC,QAAQ,CAAC,EAAE;oBACV,GAAG,OAAO,CAAC,kBAAkB,CAAC,EAAE,CAAC;oBACjC,EAAE,EAAE,QAAQ;iBACb;aACF,CAAC;YACF,OAAO,gBAAgB,CAAC;QAC1B,CAAC,CAAC,CAAC;QACH,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,WAAW,CAAC,QAAQ,CAAC,CAAC;QAC9B,CAAC;IACH,CAAC,EACD,CAAC,WAAW,EAAE,SAAS,CAAC,CACzB,CAAC;IAEF,MAAM,kBAAkB,GAAG,WAAW,CACpC,CAAC,QAAgB,EAAE,KAAsB,EAAE,aAAsB,EAAE,EAAE;QACnE,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE;YACvB,MAAM,gBAAgB,GAAG;gBACvB,GAAG,OAAO;gBACV,CAAC,QAAQ,CAAC,EAAE;oBACV,GAAG,OAAO,CAAC,QAAQ,CAAC;oBACpB,eAAe,EAAE,KAAK;oBACtB,aAAa,EAAE,aAAa;iBAC7B;aACF,CAAC;YACF,OAAO,gBAAgB,CAAC;QAC1B,CAAC,CAAC,CAAC;IACL,CAAC,EACD,EAAE,CACH,CAAC;IAEF,MAAM,mBAAmB,GAAG,WAAW,CACrC,KAAK,EACH,MAAiE,EACjE,MAAgD,EAChD,QAAgB,EAChB,iBAA0B,EACG,EAAE;QAC/B,IAAI,YAA4C,CAAC;QACjD,IAAI,cAAc,GAAG,KAAK,CAAC;QAC3B,IAAI,YAAY,GAAG,IAAI,CAAC;QACxB,kBAAkB,CAAC,QAAQ,EAAE,eAAe,CAAC,kBAAkB,CAAC,CAAC;QAEjE,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YACjC,IAAI,YAAY,IAAI,iBAAiB,EAAE,CAAC;gBACtC,mBAAmB,CAAC,iBAAiB,EAAE,QAAQ,CAAC,CAAC;YACnD,CAAC;YACD,YAAY,GAAG,KAAK,CAAC;YAErB,IAAI,KAAK,CAAC,kBAAkB,CAAC,eAAe,EAAE,CAAC;gBAC7C,kBAAkB,CAChB,KAAK,CAAC,kBAAkB,CAAC,QAAQ,EACjC,eAAe,CAAC,gBAAgB,CACjC,CAAC;gBACF,MAAM,gBAAgB,GAAG,MAAM,cAAc,CAC3C,KAAK,CAAC,kBAAkB,EACxB,YAAY,CACb,CAAC;gBACF,MAAM,sBAAsB,GAC1B,OAAO,gBAAgB,KAAK,QAAQ;oBAClC,CAAC,CAAC,gBAAgB;oBAClB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;gBACvC,MAAM,sBAAsB,GAC1B;oBACE,GAAG,MAAM;oBACT,eAAe,EAAE;wBACf,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,sBAAsB,EAAE,CAAC;wBACzD,IAAI,EAAE,MAAM;wBACZ,UAAU,EAAE,eAAe;wBAC3B,SAAS,EAAE,KAAK,CAAC,kBAAkB,CAAC,SAAS;wBAC7C,YAAY,EAAE,KAAK,CAAC,kBAAkB,CAAC,YAAY;qBACpD;iBACF,CAAC;gBACJ,kBAAkB,CAChB,KAAK,CAAC,kBAAkB,CAAC,QAAQ,EACjC,eAAe,CAAC,kBAAkB,CACnC,CAAC;gBACF,MAAM,sBAAsB,GAAG,MAAM,aAAa,CAChD,MAAM,EACN,sBAAsB,EACtB,KAAK,CAAC,kBAAkB,CAAC,QAAQ,CAClC,CAAC;gBAEF,OAAO,MAAM,mBAAmB,CAC9B,sBAAsB,EACtB,sBAAsB,EACtB,KAAK,CAAC,kBAAkB,CAAC,QAAQ,CAClC,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,IACE,CAAC,cAAc;oBACf,KAAK,CAAC,kBAAkB,CAAC,QAAQ;oBACjC,KAAK,CAAC,kBAAkB,CAAC,QAAQ,KAAK,aAAa,EAAE,EAAE,EACvD,CAAC;oBACD,cAAc,GAAG,IAAI,CAAC;oBACtB,mBAAmB,CAAC,KAAK,CAAC,kBAAkB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;gBAChE,CAAC;gBAED,IAAI,CAAC,YAAY,EAAE,CAAC;oBAClB,YAAY,GAAG,KAAK,CAAC,kBAAkB,CAAC,SAAS,EAAE,aAAa;wBAC9D,CAAC,CAAC,0BAA0B,CACxB,KAAK,CAAC,kBAAkB,EACxB,aAAa,CACd;wBACH,CAAC,CAAC,KAAK,CAAC,kBAAkB,CAAC;oBAC7B,gBAAgB,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC;gBACxC,CAAC;qBAAM,CAAC;oBACN,MAAM,UAAU,GAAG,YAAY,CAAC,EAAE,CAAC;oBACnC,YAAY,GAAG,KAAK,CAAC,kBAAkB,CAAC,SAAS,EAAE,aAAa;wBAC9D,CAAC,CAAC,0BAA0B,CACxB,KAAK,CAAC,kBAAkB,EACxB,aAAa,CACd;wBACH,CAAC,CAAC,KAAK,CAAC,kBAAkB,CAAC;oBAC7B,mBAAmB,CAAC,UAAU,EAAE,YAAY,EAAE,KAAK,CAAC,CAAC;gBACvD,CAAC;YACH,CAAC;QACH,CAAC;QAED,kBAAkB,CAChB,YAAY,EAAE,QAAQ,IAAI,QAAQ,EAClC,eAAe,CAAC,QAAQ,CACzB,CAAC;QACF,OAAO,CACL,YAAY,IAAI;YACd,QAAQ,EAAE,EAAE;YACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,yBAAyB,EAAE,CAAC;YAC5D,IAAI,EAAE,OAAO;YACb,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE;YACvB,cAAc,EAAE,EAAE;SACnB,CACF,CAAC;IACJ,CAAC,EACD;QACE,gBAAgB;QAChB,MAAM;QACN,aAAa;QACb,aAAa,EAAE,EAAE;QACjB,mBAAmB;QACnB,WAAW;QACX,mBAAmB;QACnB,YAAY;QACZ,mBAAmB;QACnB,kBAAkB;KACnB,CACF,CAAC;IAEF,MAAM,iBAAiB,GAAG,WAAW,CACnC,KAAK,EACH,OAAe,EACf,UAII,EAAE,QAAQ,EAAE,kBAAkB,CAAC,EAAE,EAAE,EACV,EAAE;QAC/B,MAAM,EAAE,QAAQ,GAAG,aAAa,CAAC,EAAE,EAAE,cAAc,EAAE,GAAG,OAAO,CAAC;QAChE,kBAAkB,CAAC,QAAQ,EAAE,eAAe,CAAC,kBAAkB,CAAC,CAAC;QAEjE,gBAAgB,CACd;YACE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;YAC1C,iBAAiB,EAAE,IAAI;YACvB,IAAI,EAAE,MAAM;YACZ,QAAQ,EAAE,QAAQ;YAClB,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE;YACvB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,cAAc,EAAE,EAAE;SACnB,EACD,KAAK,CACN,CAAC;QAEF,MAAM,mBAAmB,GAAG,sBAAsB,CAChD,aAAa,EACb,YAAY,EACZ,yBAAyB,CAC1B,CAAC;QACF,MAAM,iBAAiB,GAAG,oBAAoB,CAC5C,YAAY,EACZ,yBAAyB,CAC1B,CAAC;QACF,MAAM,MAAM,GAA6C;YACvD,eAAe,EAAE;gBACf,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;gBAC1C,IAAI,EAAE,MAAM;aACb;YACD,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,mBAAmB,EAAE,mBAAmB;YACxC,WAAW,EAAE,iBAAiB,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAC1C,yBAAyB,CAAC,IAAI,CAAC,CAChC;SACF,CAAC;QAEF,IAAI,cAAc,EAAE,CAAC;YACnB,MAAM,qBAAqB,GAAG,MAAM,aAAa,CAC/C,MAAM,EACN,MAAM,EACN,QAAQ,KAAK,kBAAkB,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAC1D,CAAC;YACF,OAAO,MAAM,mBAAmB,CAC9B,qBAAqB,EACrB,MAAM,EACN,QAAQ,CACT,CAAC;QACJ,CAAC;QACD,IAAI,eAAe,GAAG,MAAM,CAAC,QAAQ,KAAK,kBAAkB,CAAC,EAAE;YAC7D,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC;YACrC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAC;QAEvD,mBAAmB;QACnB,OAAO,eAAe,CAAC,kBAAkB,CAAC,eAAe,EAAE,CAAC;YAC1D,kBAAkB,CAAC,QAAQ,EAAE,eAAe,CAAC,gBAAgB,CAAC,CAAC;YAC/D,MAAM,gBAAgB,GAAG,MAAM,cAAc,CAC3C,eAAe,CAAC,kBAAkB,EAClC,YAAY,CACb,CAAC;YACF,MAAM,kBAAkB,GACtB,OAAO,gBAAgB,KAAK,QAAQ;gBAClC,CAAC,CAAC,gBAAgB;gBAClB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YACvC,MAAM,sBAAsB,GAC1B;gBACE,GAAG,MAAM;gBACT,eAAe,EAAE;oBACf,GAAG,MAAM,CAAC,eAAe;oBACzB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC;oBACrD,IAAI,EAAE,MAAM;oBACZ,UAAU,EAAE,eAAe;oBAC3B,SAAS,EAAE,eAAe,CAAC,kBAAkB,CAAC,SAAS;oBACvD,YAAY,EAAE,eAAe,CAAC,kBAAkB,CAAC,YAAY;iBAC9D;aACF,CAAC;YACJ,kBAAkB,CAAC,QAAQ,EAAE,eAAe,CAAC,mBAAmB,CAAC,CAAC;YAClE,eAAe,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CACrD,eAAe,CAAC,kBAAkB,CAAC,QAAQ,EAC3C,sBAAsB,CACvB,CAAC;QACJ,CAAC;QAED,MAAM,YAAY,GAAG,eAAe,CAAC,kBAAkB,CAAC,SAAS;YAC/D,EAAE,aAAa;YACf,CAAC,CAAC,0BAA0B,CACxB,eAAe,CAAC,kBAAkB,EAClC,aAAa,CACd;YACH,CAAC,CAAC,eAAe,CAAC,kBAAkB,CAAC;QACvC,MAAM,mBAAmB,CAAC,eAAe,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QACvE,kBAAkB,CAChB,eAAe,CAAC,kBAAkB,CAAC,QAAQ,EAC3C,eAAe,CAAC,QAAQ,CACzB,CAAC;QACF,OAAO,YAAY,CAAC;IACtB,CAAC,EACD;QACE,aAAa;QACb,YAAY;QACZ,yBAAyB;QACzB,aAAa,CAAC,EAAE;QAChB,mBAAmB;QACnB,gBAAgB;QAChB,MAAM;QACN,kBAAkB;QAClB,mBAAmB;KACpB,CACF,CAAC;IAEF,OAAO,CACL,oBAAC,kBAAkB,CAAC,QAAQ,IAC1B,KAAK,EAAE;YACL,MAAM,EAAE,aAAa;YACrB,mBAAmB;YACnB,cAAc;YACd,gBAAgB;YAChB,mBAAmB;YACnB,UAAU;YACV,aAAa;YACb,iBAAiB;YACjB,eAAe,EAAE,CAAC,aAAa,EAAE,eAAe;gBAC9C,eAAe,CAAC,IAAI,CAAoB;YAC1C,uBAAuB,EAAE,aAAa,EAAE,aAAa,IAAI,EAAE;YAC3D,MAAM,EAAE,WAAW,CACjB,CAAC,aAAa,EAAE,eAAe;gBAC7B,eAAe,CAAC,IAAI,CAAoB,CAC3C;SACF,IAEA,QAAQ,CACmB,CAC/B,CAAC;AACJ,CAAC,CAAC;AAEF;;;;GAIG;AACH,MAAM,CAAC,MAAM,cAAc,GAAG,GAAG,EAAE;IACjC,MAAM,OAAO,GAAG,UAAU,CAAC,kBAAkB,CAAC,CAAC;IAC/C,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;QAC1B,MAAM,IAAI,KAAK,CAAC,0DAA0D,CAAC,CAAC;IAC9E,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC","sourcesContent":["\"use client\";\nimport TamboAI, { advanceStream } from \"@tambo-ai/typescript-sdk\";\nimport React, {\n  createContext,\n  PropsWithChildren,\n  useCallback,\n  useContext,\n  useEffect,\n  useMemo,\n  useState,\n} from \"react\";\nimport {\n  GenerationStage,\n  isIdleStage,\n  TamboThreadMessage,\n} from \"../model/generate-component-response\";\nimport { TamboThread } from \"../model/tambo-thread\";\nimport { renderComponentIntoMessage } from \"../util/generate-component\";\nimport {\n  getAvailableComponents,\n  getClientContext,\n  getUnassociatedTools,\n  mapTamboToolToContextTool,\n} from \"../util/registry\";\nimport { handleToolCall } from \"../util/tool-caller\";\nimport { useTamboClient } from \"./tambo-client-provider\";\nimport { useTamboRegistry } from \"./tambo-registry-provider\";\n\nexport interface TamboThreadContextProps {\n  /** The current thread */\n  thread: TamboThread;\n  /** Switch to a different thread */\n  switchCurrentThread: (threadId: string, fetch?: boolean) => void;\n  /** Start a new thread */\n  startNewThread: () => void;\n  /** Add a message to the current thread */\n  addThreadMessage: (\n    message: TamboThreadMessage,\n    sendToServer: boolean,\n  ) => Promise<TamboAI.Beta.Threads.ThreadMessage[]>;\n  /** Update a message in the current thread */\n  updateThreadMessage: (\n    id: string,\n    message: TamboThreadMessage,\n    sendToServer: boolean,\n  ) => Promise<void>;\n  /** The input value of the current thread */\n  inputValue: string;\n  /** Set the input value of the current thread */\n  setInputValue: (value: string) => void;\n  /** Send a message to the current thread */\n  sendThreadMessage: (\n    message: string,\n    options: {\n      threadId?: string;\n      streamResponse?: boolean;\n      contextKey?: string;\n    },\n  ) => Promise<TamboThreadMessage>;\n  /** The generation stage of the current thread - updated as the thread progresses */\n  generationStage: GenerationStage;\n  /** The generation status message of the current thread - updated as the thread progresses */\n  generationStatusMessage: string;\n  /** Whether the thread is idle */\n  isIdle: boolean;\n}\n\n/**\n * This is a stub entry for when the thread is not yet created, the first time\n * the user sends a message\n *\n * Note that the consumer needs to be careful never to send `PLACEHOLDER_THREAD.id` to the server,\n * as this doesn't really exist on the server side.\n */\nexport const PLACEHOLDER_THREAD: TamboThread = {\n  id: \"placeholder\",\n  messages: [],\n  createdAt: \"\",\n  projectId: \"\",\n  updatedAt: \"\",\n  metadata: {},\n};\n\nexport const TamboThreadContext = createContext<TamboThreadContextProps>({\n  thread: PLACEHOLDER_THREAD,\n  /**\n   *\n   */\n  switchCurrentThread: () => {\n    throw new Error(\"switchCurrentThread not implemented\");\n  },\n  /**\n   *\n   */\n  startNewThread: () => {\n    throw new Error(\"startNewThread not implemented\");\n  },\n  /**\n   *\n   */\n  addThreadMessage: () => {\n    throw new Error(\"updateThreadMessageHistory not implemented\");\n  },\n  inputValue: \"\",\n  /**\n   *\n   */\n  setInputValue: () => {\n    throw new Error(\"setInputValue not implemented\");\n  },\n  /**\n   *\n   */\n  updateThreadMessage: () => {\n    throw new Error(\"updateThreadMessage not implemented\");\n  },\n  /**\n   *\n   */\n  sendThreadMessage: () => {\n    throw new Error(\"advance not implemented\");\n  },\n  generationStage: GenerationStage.IDLE,\n  generationStatusMessage: \"\",\n  isIdle: true,\n});\n\n/**\n * The TamboThreadProvider is a React provider that provides a thread context\n * to the descendants of the provider.\n * @param props - The props for the TamboThreadProvider\n * @param props.children - The children to wrap\n * @returns The TamboThreadProvider component\n */\nexport const TamboThreadProvider: React.FC<PropsWithChildren> = ({\n  children,\n}) => {\n  const [threadMap, setThreadMap] = useState<Record<string, TamboThread>>({\n    [PLACEHOLDER_THREAD.id]: PLACEHOLDER_THREAD,\n  });\n  const client = useTamboClient();\n  const { componentList, toolRegistry, componentToolAssociations } =\n    useTamboRegistry();\n  const [inputValue, setInputValue] = useState(\"\");\n\n  const [currentThreadId, setCurrentThreadId] = useState<string>(\n    PLACEHOLDER_THREAD.id,\n  );\n  const currentThread: TamboThread | undefined = threadMap[currentThreadId];\n\n  // Use existing messages from the current thread to avoid re-generating any components\n  const currentMessageCache = useMemo(() => {\n    const messageCache = new Map<string, TamboThreadMessage>();\n    if (currentThread) {\n      for (const message of currentThread.messages) {\n        messageCache.set(message.id, message);\n      }\n    }\n    return messageCache;\n  }, [currentThread]);\n\n  const fetchThread = useCallback(\n    async (threadId: string) => {\n      const thread = await client.beta.threads.retrieve(threadId);\n      const threadWithRenderedComponents = {\n        ...thread,\n        messages: thread.messages.map((message) => {\n          if (currentMessageCache.has(message.id)) {\n            const renderedMessage = currentMessageCache.get(message.id);\n            return {\n              ...renderedMessage,\n              ...message,\n            };\n          }\n          if (message.component?.componentName) {\n            const messageWithComponent = renderComponentIntoMessage(\n              message,\n              componentList,\n            );\n            return messageWithComponent;\n          }\n          return message;\n        }),\n      };\n\n      setThreadMap((prevMap) => {\n        const updatedThreadMap = {\n          ...prevMap,\n          [threadId]: threadWithRenderedComponents,\n        };\n        return updatedThreadMap;\n      });\n    },\n    [client.beta.threads, componentList, currentMessageCache],\n  );\n\n  useEffect(() => {\n    if (\n      currentThreadId &&\n      currentThreadId !== PLACEHOLDER_THREAD.id &&\n      !threadMap[currentThreadId]\n    ) {\n      fetchThread(currentThreadId);\n    }\n  }, [currentThreadId, fetchThread, threadMap]);\n\n  const addThreadMessage = useCallback(\n    async (\n      message: TamboThreadMessage,\n      sendToServer = true,\n      createdAt: string = new Date().toISOString(),\n    ) => {\n      if (!currentThread) {\n        console.warn(\"Cannot add messages if we do not have a current thread\");\n        return [];\n      }\n      const chatMessage: TamboThreadMessage & {\n        additionalContext?: string;\n      } = {\n        ...message,\n        additionalContext:\n          message.role === \"user\" ? getClientContext() : undefined,\n        createdAt,\n      };\n      const threadId = message.threadId;\n      // optimistically update the thread in the local state\n      setThreadMap((prevMap) => {\n        if (!threadId) {\n          return prevMap;\n        }\n        const prevMessages = prevMap[threadId]?.messages || [];\n        const updatedMessages = [...prevMessages, chatMessage];\n\n        const updatedThreadMap = {\n          ...prevMap,\n          [threadId]: {\n            ...prevMap[threadId],\n            messages: updatedMessages,\n          },\n        };\n        return updatedThreadMap;\n      });\n\n      if (sendToServer) {\n        // TODO: if this fails, we need to revert the local state update\n        await client.beta.threads.messages.create(message.threadId, {\n          content: message.content,\n          role: message.role,\n          // additionalContext: chatMessage.additionalContext,\n        });\n      }\n      return threadMap[threadId]?.messages || [];\n    },\n    [client.beta.threads.messages, currentThread, threadMap],\n  );\n\n  const updateThreadMessage = useCallback(\n    async (\n      id: string,\n      message: TamboThreadMessage,\n      sendToServer = true,\n      createdAt: string = new Date().toISOString(),\n    ) => {\n      const chatMessage: TamboThreadMessage = {\n        ...message,\n        createdAt,\n      };\n\n      setThreadMap((prevMap) => {\n        if (!message.threadId) {\n          return prevMap;\n        }\n        const prevMessages = prevMap[message.threadId]?.messages || [];\n        const updatedMessages = prevMessages.map((msg) => {\n          if (msg.id === id) {\n            return chatMessage;\n          }\n          return msg;\n        });\n        return {\n          ...prevMap,\n          [message.threadId]: {\n            ...prevMap[message.threadId],\n            messages: updatedMessages,\n          },\n        };\n      });\n      if (sendToServer) {\n        // TODO: if this fails, we need to revert the local state update\n        await client.beta.threads.messages.create(message.threadId, {\n          content: message.content,\n          role: message.role,\n          // additionalContext: chatMessage.additionalContext,\n        });\n      }\n    },\n    [client.beta.threads.messages],\n  );\n\n  const deleteThreadMessage = useCallback(\n    (messageId: string, threadId: string) => {\n      if (!threadMap[threadId]) return;\n\n      setThreadMap((prevMap) => ({\n        ...prevMap,\n        [threadId]: {\n          ...prevMap[threadId],\n          messages: prevMap[threadId].messages.filter(\n            (msg) => msg.id !== messageId,\n          ),\n        },\n      }));\n    },\n    [threadMap],\n  );\n\n  const startNewThread = useCallback(() => {\n    setCurrentThreadId(PLACEHOLDER_THREAD.id);\n    setThreadMap((prevMap) => {\n      return {\n        ...prevMap,\n        [PLACEHOLDER_THREAD.id]: PLACEHOLDER_THREAD,\n      };\n    });\n  }, []);\n\n  const switchCurrentThread = useCallback(\n    async (threadId: string, fetch = true) => {\n      if (threadId === PLACEHOLDER_THREAD.id) {\n        console.warn(\"Switching to placeholder thread, may be a bug.\");\n        return;\n      }\n      setCurrentThreadId(threadId);\n      setThreadMap((prevMap) => {\n        if (prevMap[threadId]) {\n          return prevMap;\n        }\n        // If this is a new thread, add placeholder thread messages to the thread\n        const updatedThreadMap = {\n          ...prevMap,\n          [threadId]: {\n            ...prevMap[PLACEHOLDER_THREAD.id],\n            id: threadId,\n          },\n        };\n        return updatedThreadMap;\n      });\n      if (fetch) {\n        await fetchThread(threadId);\n      }\n    },\n    [fetchThread, threadMap],\n  );\n\n  const updateThreadStatus = useCallback(\n    (threadId: string, stage: GenerationStage, statusMessage?: string) => {\n      setThreadMap((prevMap) => {\n        const updatedThreadMap = {\n          ...prevMap,\n          [threadId]: {\n            ...prevMap[threadId],\n            generationStage: stage,\n            statusMessage: statusMessage,\n          },\n        };\n        return updatedThreadMap;\n      });\n    },\n    [],\n  );\n\n  const handleAdvanceStream = useCallback(\n    async (\n      stream: AsyncIterable<TamboAI.Beta.Threads.ThreadAdvanceResponse>,\n      params: TamboAI.Beta.Threads.ThreadAdvanceParams,\n      threadId: string,\n      messageIdToRemove?: string,\n    ): Promise<TamboThreadMessage> => {\n      let finalMessage: TamboThreadMessage | undefined;\n      let hasSetThreadId = false;\n      let isFirstChunk = true;\n      updateThreadStatus(threadId, GenerationStage.STREAMING_RESPONSE);\n\n      for await (const chunk of stream) {\n        if (isFirstChunk && messageIdToRemove) {\n          deleteThreadMessage(messageIdToRemove, threadId);\n        }\n        isFirstChunk = false;\n\n        if (chunk.responseMessageDto.toolCallRequest) {\n          updateThreadStatus(\n            chunk.responseMessageDto.threadId,\n            GenerationStage.FETCHING_CONTEXT,\n          );\n          const toolCallResponse = await handleToolCall(\n            chunk.responseMessageDto,\n            toolRegistry,\n          );\n          const toolCallResponseString =\n            typeof toolCallResponse === \"string\"\n              ? toolCallResponse\n              : JSON.stringify(toolCallResponse);\n          const toolCallResponseParams: TamboAI.Beta.Threads.ThreadAdvanceParams =\n            {\n              ...params,\n              messageToAppend: {\n                content: [{ type: \"text\", text: toolCallResponseString }],\n                role: \"tool\",\n                actionType: \"tool_response\",\n                component: chunk.responseMessageDto.component,\n                tool_call_id: chunk.responseMessageDto.tool_call_id,\n              },\n            };\n          updateThreadStatus(\n            chunk.responseMessageDto.threadId,\n            GenerationStage.STREAMING_RESPONSE,\n          );\n          const toolCallResponseStream = await advanceStream(\n            client,\n            toolCallResponseParams,\n            chunk.responseMessageDto.threadId,\n          );\n\n          return await handleAdvanceStream(\n            toolCallResponseStream,\n            toolCallResponseParams,\n            chunk.responseMessageDto.threadId,\n          );\n        } else {\n          if (\n            !hasSetThreadId &&\n            chunk.responseMessageDto.threadId &&\n            chunk.responseMessageDto.threadId !== currentThread?.id\n          ) {\n            hasSetThreadId = true;\n            switchCurrentThread(chunk.responseMessageDto.threadId, false);\n          }\n\n          if (!finalMessage) {\n            finalMessage = chunk.responseMessageDto.component?.componentName\n              ? renderComponentIntoMessage(\n                  chunk.responseMessageDto,\n                  componentList,\n                )\n              : chunk.responseMessageDto;\n            addThreadMessage(finalMessage, false);\n          } else {\n            const previousId = finalMessage.id;\n            finalMessage = chunk.responseMessageDto.component?.componentName\n              ? renderComponentIntoMessage(\n                  chunk.responseMessageDto,\n                  componentList,\n                )\n              : chunk.responseMessageDto;\n            updateThreadMessage(previousId, finalMessage, false);\n          }\n        }\n      }\n\n      updateThreadStatus(\n        finalMessage?.threadId ?? threadId,\n        GenerationStage.COMPLETE,\n      );\n      return (\n        finalMessage ?? {\n          threadId: \"\",\n          content: [{ type: \"text\", text: `Error processing stream` }],\n          role: \"hydra\",\n          createdAt: new Date().toISOString(),\n          id: crypto.randomUUID(),\n          componentState: {},\n        }\n      );\n    },\n    [\n      addThreadMessage,\n      client,\n      componentList,\n      currentThread?.id,\n      deleteThreadMessage,\n      fetchThread,\n      switchCurrentThread,\n      toolRegistry,\n      updateThreadMessage,\n      updateThreadStatus,\n    ],\n  );\n\n  const sendThreadMessage = useCallback(\n    async (\n      message: string,\n      options: {\n        threadId?: string;\n        streamResponse?: boolean;\n        contextKey?: string;\n      } = { threadId: PLACEHOLDER_THREAD.id },\n    ): Promise<TamboThreadMessage> => {\n      const { threadId = currentThread.id, streamResponse } = options;\n      updateThreadStatus(threadId, GenerationStage.CHOOSING_COMPONENT);\n\n      addThreadMessage(\n        {\n          content: [{ type: \"text\", text: message }],\n          renderedComponent: null,\n          role: \"user\",\n          threadId: threadId,\n          id: crypto.randomUUID(),\n          createdAt: new Date().toISOString(),\n          componentState: {},\n        },\n        false,\n      );\n\n      const availableComponents = getAvailableComponents(\n        componentList,\n        toolRegistry,\n        componentToolAssociations,\n      );\n      const unassociatedTools = getUnassociatedTools(\n        toolRegistry,\n        componentToolAssociations,\n      );\n      const params: TamboAI.Beta.Threads.ThreadAdvanceParams = {\n        messageToAppend: {\n          content: [{ type: \"text\", text: message }],\n          role: \"user\",\n        },\n        contextKey: options.contextKey,\n        availableComponents: availableComponents,\n        clientTools: unassociatedTools.map((tool) =>\n          mapTamboToolToContextTool(tool),\n        ),\n      };\n\n      if (streamResponse) {\n        const advanceStreamResponse = await advanceStream(\n          client,\n          params,\n          threadId === PLACEHOLDER_THREAD.id ? undefined : threadId,\n        );\n        return await handleAdvanceStream(\n          advanceStreamResponse,\n          params,\n          threadId,\n        );\n      }\n      let advanceResponse = await (threadId === PLACEHOLDER_THREAD.id\n        ? client.beta.threads.advance(params)\n        : client.beta.threads.advanceById(threadId, params));\n\n      //handle tool calls\n      while (advanceResponse.responseMessageDto.toolCallRequest) {\n        updateThreadStatus(threadId, GenerationStage.FETCHING_CONTEXT);\n        const toolCallResponse = await handleToolCall(\n          advanceResponse.responseMessageDto,\n          toolRegistry,\n        );\n        const toolResponseString =\n          typeof toolCallResponse === \"string\"\n            ? toolCallResponse\n            : JSON.stringify(toolCallResponse);\n        const toolCallResponseParams: TamboAI.Beta.Threads.ThreadAdvanceParams =\n          {\n            ...params,\n            messageToAppend: {\n              ...params.messageToAppend,\n              content: [{ type: \"text\", text: toolResponseString }],\n              role: \"tool\",\n              actionType: \"tool_response\",\n              component: advanceResponse.responseMessageDto.component,\n              tool_call_id: advanceResponse.responseMessageDto.tool_call_id,\n            },\n          };\n        updateThreadStatus(threadId, GenerationStage.HYDRATING_COMPONENT);\n        advanceResponse = await client.beta.threads.advanceById(\n          advanceResponse.responseMessageDto.threadId,\n          toolCallResponseParams,\n        );\n      }\n\n      const finalMessage = advanceResponse.responseMessageDto.component\n        ?.componentName\n        ? renderComponentIntoMessage(\n            advanceResponse.responseMessageDto,\n            componentList,\n          )\n        : advanceResponse.responseMessageDto;\n      await switchCurrentThread(advanceResponse.responseMessageDto.threadId);\n      updateThreadStatus(\n        advanceResponse.responseMessageDto.threadId,\n        GenerationStage.COMPLETE,\n      );\n      return finalMessage;\n    },\n    [\n      componentList,\n      toolRegistry,\n      componentToolAssociations,\n      currentThread.id,\n      switchCurrentThread,\n      addThreadMessage,\n      client,\n      updateThreadStatus,\n      handleAdvanceStream,\n    ],\n  );\n\n  return (\n    <TamboThreadContext.Provider\n      value={{\n        thread: currentThread,\n        switchCurrentThread,\n        startNewThread,\n        addThreadMessage,\n        updateThreadMessage,\n        inputValue,\n        setInputValue,\n        sendThreadMessage,\n        generationStage: (currentThread?.generationStage ??\n          GenerationStage.IDLE) as GenerationStage,\n        generationStatusMessage: currentThread?.statusMessage ?? \"\",\n        isIdle: isIdleStage(\n          (currentThread?.generationStage ??\n            GenerationStage.IDLE) as GenerationStage,\n        ),\n      }}\n    >\n      {children}\n    </TamboThreadContext.Provider>\n  );\n};\n\n/**\n * The useTamboThread hook provides access to the current thread context\n * to the descendants of the TamboThreadProvider.\n * @returns All state and actions for the current thread\n */\nexport const useTamboThread = () => {\n  const context = useContext(TamboThreadContext);\n  if (context === undefined) {\n    throw new Error(\"useTamboThread must be used within a TamboThreadProvider\");\n  }\n  return context;\n};\n"]}
+{"version":3,"file":"tambo-thread-provider.js","sourceRoot":"","sources":["../../src/providers/tambo-thread-provider.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;AACb,OAAgB,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AAClE,OAAO,KAAK,EAAE,EACZ,aAAa,EAEb,WAAW,EACX,UAAU,EACV,SAAS,EACT,OAAO,EACP,QAAQ,GACT,MAAM,OAAO,CAAC;AACf,OAAO,EACL,eAAe,EACf,WAAW,GAEZ,MAAM,sCAAsC,CAAC;AAE9C,OAAO,EAAE,0BAA0B,EAAE,MAAM,4BAA4B,CAAC;AACxE,OAAO,EACL,sBAAsB,EACtB,gBAAgB,EAChB,oBAAoB,EACpB,yBAAyB,GAC1B,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AACzD,OAAO,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAyC7D;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAgB;IAC7C,EAAE,EAAE,aAAa;IACjB,QAAQ,EAAE,EAAE;IACZ,SAAS,EAAE,EAAE;IACb,SAAS,EAAE,EAAE;IACb,SAAS,EAAE,EAAE;IACb,QAAQ,EAAE,EAAE;CACb,CAAC;AAEF,MAAM,CAAC,MAAM,kBAAkB,GAAG,aAAa,CAA0B;IACvE,MAAM,EAAE,kBAAkB;IAC1B;;OAEG;IACH,mBAAmB,EAAE,GAAG,EAAE;QACxB,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;IACzD,CAAC;IACD;;OAEG;IACH,cAAc,EAAE,GAAG,EAAE;QACnB,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;IACpD,CAAC;IACD;;OAEG;IACH,gBAAgB,EAAE,GAAG,EAAE;QACrB,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;IAChE,CAAC;IACD,UAAU,EAAE,EAAE;IACd;;OAEG;IACH,aAAa,EAAE,GAAG,EAAE;QAClB,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;IACnD,CAAC;IACD;;OAEG;IACH,mBAAmB,EAAE,GAAG,EAAE;QACxB,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;IACzD,CAAC;IACD;;OAEG;IACH,iBAAiB,EAAE,GAAG,EAAE;QACtB,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC7C,CAAC;IACD,eAAe,EAAE,eAAe,CAAC,IAAI;IACrC,uBAAuB,EAAE,EAAE;IAC3B,MAAM,EAAE,IAAI;CACb,CAAC,CAAC;AAEH;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,mBAAmB,GAAgC,CAAC,EAC/D,QAAQ,GACT,EAAE,EAAE;IACH,MAAM,CAAC,SAAS,EAAE,YAAY,CAAC,GAAG,QAAQ,CAA8B;QACtE,CAAC,kBAAkB,CAAC,EAAE,CAAC,EAAE,kBAAkB;KAC5C,CAAC,CAAC;IACH,MAAM,MAAM,GAAG,cAAc,EAAE,CAAC;IAChC,MAAM,EAAE,aAAa,EAAE,YAAY,EAAE,yBAAyB,EAAE,GAC9D,gBAAgB,EAAE,CAAC;IACrB,MAAM,CAAC,UAAU,EAAE,aAAa,CAAC,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC;IAEjD,MAAM,CAAC,eAAe,EAAE,kBAAkB,CAAC,GAAG,QAAQ,CACpD,kBAAkB,CAAC,EAAE,CACtB,CAAC;IACF,MAAM,aAAa,GAA4B,SAAS,CAAC,eAAe,CAAC,CAAC;IAE1E,sFAAsF;IACtF,MAAM,mBAAmB,GAAG,OAAO,CAAC,GAAG,EAAE;QACvC,MAAM,YAAY,GAAG,IAAI,GAAG,EAA8B,CAAC;QAC3D,IAAI,aAAa,EAAE,CAAC;YAClB,KAAK,MAAM,OAAO,IAAI,aAAa,CAAC,QAAQ,EAAE,CAAC;gBAC7C,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YACxC,CAAC;QACH,CAAC;QACD,OAAO,YAAY,CAAC;IACtB,CAAC,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC;IAEpB,MAAM,WAAW,GAAG,WAAW,CAC7B,KAAK,EAAE,QAAgB,EAAE,EAAE;QACzB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAC5D,MAAM,4BAA4B,GAAG;YACnC,GAAG,MAAM;YACT,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;gBACxC,IAAI,mBAAmB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,CAAC;oBACxC,MAAM,eAAe,GAAG,mBAAmB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;oBAC5D,OAAO;wBACL,GAAG,eAAe;wBAClB,GAAG,OAAO;qBACX,CAAC;gBACJ,CAAC;gBACD,IAAI,OAAO,CAAC,SAAS,EAAE,aAAa,EAAE,CAAC;oBACrC,MAAM,oBAAoB,GAAG,0BAA0B,CACrD,OAAO,EACP,aAAa,CACd,CAAC;oBACF,OAAO,oBAAoB,CAAC;gBAC9B,CAAC;gBACD,OAAO,OAAO,CAAC;YACjB,CAAC,CAAC;SACH,CAAC;QAEF,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE;YACvB,MAAM,gBAAgB,GAAG;gBACvB,GAAG,OAAO;gBACV,CAAC,QAAQ,CAAC,EAAE,4BAA4B;aACzC,CAAC;YACF,OAAO,gBAAgB,CAAC;QAC1B,CAAC,CAAC,CAAC;IACL,CAAC,EACD,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,aAAa,EAAE,mBAAmB,CAAC,CAC1D,CAAC;IAEF,SAAS,CAAC,GAAG,EAAE;QACb,IACE,eAAe;YACf,eAAe,KAAK,kBAAkB,CAAC,EAAE;YACzC,CAAC,SAAS,CAAC,eAAe,CAAC,EAC3B,CAAC;YACD,WAAW,CAAC,eAAe,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC,EAAE,CAAC,eAAe,EAAE,WAAW,EAAE,SAAS,CAAC,CAAC,CAAC;IAE9C,MAAM,gBAAgB,GAAG,WAAW,CAClC,KAAK,EACH,OAA2B,EAC3B,YAAY,GAAG,IAAI,EACnB,YAAoB,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAC5C,EAAE;QACF,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,OAAO,CAAC,IAAI,CAAC,wDAAwD,CAAC,CAAC;YACvE,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,MAAM,WAAW,GAEb;YACF,GAAG,OAAO;YACV,iBAAiB,EACf,OAAO,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC,SAAS;YAC1D,SAAS;SACV,CAAC;QACF,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;QAClC,MAAM,SAAS,GAAG,WAAW,CAAC,EAAE,CAAC;QACjC,sDAAsD;QACtD,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE;YACvB,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,OAAO,OAAO,CAAC;YACjB,CAAC;YACD,MAAM,YAAY,GAAG,OAAO,CAAC,QAAQ,CAAC,EAAE,QAAQ,IAAI,EAAE,CAAC;YACvD,MAAM,WAAW,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,KAAK,SAAS,CAAC,CAAC;YACrE,gDAAgD;YAChD,MAAM,eAAe,GAAG,WAAW;gBACjC,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;oBACvB,IAAI,GAAG,CAAC,EAAE,KAAK,SAAS,EAAE,CAAC;wBACzB,OAAO,WAAW,CAAC;oBACrB,CAAC;oBACD,OAAO,GAAG,CAAC;gBACb,CAAC,CAAC;gBACJ,CAAC,CAAC,CAAC,GAAG,YAAY,EAAE,WAAW,CAAC,CAAC;YAEnC,MAAM,gBAAgB,GAAG;gBACvB,GAAG,OAAO;gBACV,CAAC,QAAQ,CAAC,EAAE;oBACV,GAAG,OAAO,CAAC,QAAQ,CAAC;oBACpB,QAAQ,EAAE,eAAe;iBAC1B;aACF,CAAC;YACF,OAAO,gBAAgB,CAAC;QAC1B,CAAC,CAAC,CAAC;QAEH,IAAI,YAAY,EAAE,CAAC;YACjB,gEAAgE;YAChE,MAAM,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE;gBAC1D,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,oDAAoD;aACrD,CAAC,CAAC;QACL,CAAC;QACD,OAAO,SAAS,CAAC,QAAQ,CAAC,EAAE,QAAQ,IAAI,EAAE,CAAC;IAC7C,CAAC,EACD,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,aAAa,EAAE,SAAS,CAAC,CACzD,CAAC;IAEF,MAAM,mBAAmB,GAAG,WAAW,CACrC,KAAK,EACH,EAAU,EACV,OAA2B,EAC3B,YAAY,GAAG,IAAI,EACnB,YAAoB,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAC5C,EAAE;QACF,MAAM,WAAW,GAAuB;YACtC,GAAG,OAAO;YACV,SAAS;SACV,CAAC;QAEF,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE;YACvB,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;gBACtB,OAAO,OAAO,CAAC;YACjB,CAAC;YACD,MAAM,YAAY,GAAG,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,QAAQ,IAAI,EAAE,CAAC;YAC/D,MAAM,eAAe,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;gBAC/C,IAAI,GAAG,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC;oBAClB,OAAO,WAAW,CAAC;gBACrB,CAAC;gBACD,OAAO,GAAG,CAAC;YACb,CAAC,CAAC,CAAC;YACH,OAAO;gBACL,GAAG,OAAO;gBACV,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE;oBAClB,GAAG,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC;oBAC5B,QAAQ,EAAE,eAAe;iBAC1B;aACF,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,IAAI,YAAY,EAAE,CAAC;YACjB,gEAAgE;YAChE,MAAM,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE;gBAC1D,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,oDAAoD;aACrD,CAAC,CAAC;QACL,CAAC;IACH,CAAC,EACD,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAC/B,CAAC;IAEF,MAAM,cAAc,GAAG,WAAW,CAAC,GAAG,EAAE;QACtC,kBAAkB,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC;QAC1C,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE;YACvB,OAAO;gBACL,GAAG,OAAO;gBACV,CAAC,kBAAkB,CAAC,EAAE,CAAC,EAAE,kBAAkB;aAC5C,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,mBAAmB,GAAG,WAAW,CACrC,KAAK,EAAE,QAAgB,EAAE,KAAK,GAAG,IAAI,EAAE,EAAE;QACvC,IAAI,QAAQ,KAAK,kBAAkB,CAAC,EAAE,EAAE,CAAC;YACvC,OAAO,CAAC,IAAI,CAAC,gDAAgD,CAAC,CAAC;YAC/D,OAAO;QACT,CAAC;QACD,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QAC7B,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE;YACvB,IAAI,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACtB,OAAO,OAAO,CAAC;YACjB,CAAC;YACD,yEAAyE;YACzE,MAAM,gBAAgB,GAAG;gBACvB,GAAG,OAAO;gBACV,CAAC,QAAQ,CAAC,EAAE;oBACV,GAAG,OAAO,CAAC,kBAAkB,CAAC,EAAE,CAAC;oBACjC,EAAE,EAAE,QAAQ;iBACb;aACF,CAAC;YACF,OAAO,gBAAgB,CAAC;QAC1B,CAAC,CAAC,CAAC;QACH,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,WAAW,CAAC,QAAQ,CAAC,CAAC;QAC9B,CAAC;IACH,CAAC,EACD,CAAC,WAAW,CAAC,CACd,CAAC;IAEF,MAAM,kBAAkB,GAAG,WAAW,CACpC,CAAC,QAAgB,EAAE,KAAsB,EAAE,aAAsB,EAAE,EAAE;QACnE,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE;YACvB,MAAM,gBAAgB,GAAG;gBACvB,GAAG,OAAO;gBACV,CAAC,QAAQ,CAAC,EAAE;oBACV,GAAG,OAAO,CAAC,QAAQ,CAAC;oBACpB,eAAe,EAAE,KAAK;oBACtB,aAAa,EAAE,aAAa;iBAC7B;aACF,CAAC;YACF,OAAO,gBAAgB,CAAC;QAC1B,CAAC,CAAC,CAAC;IACL,CAAC,EACD,EAAE,CACH,CAAC;IAEF,MAAM,mBAAmB,GAAG,WAAW,CACrC,KAAK,EACH,MAAiE,EACjE,MAAgD,EAChD,QAAgB,EACa,EAAE;QAC/B,IAAI,YAAsD,CAAC;QAC3D,IAAI,cAAc,GAAG,KAAK,CAAC;QAC3B,kBAAkB,CAAC,QAAQ,EAAE,eAAe,CAAC,kBAAkB,CAAC,CAAC;QAEjE,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YACjC,IAAI,KAAK,CAAC,kBAAkB,CAAC,eAAe,EAAE,CAAC;gBAC7C,kBAAkB,CAChB,KAAK,CAAC,kBAAkB,CAAC,QAAQ,EACjC,eAAe,CAAC,gBAAgB,CACjC,CAAC;gBACF,MAAM,gBAAgB,GAAG,MAAM,cAAc,CAC3C,KAAK,CAAC,kBAAkB,EACxB,YAAY,CACb,CAAC;gBACF,MAAM,sBAAsB,GAC1B,OAAO,gBAAgB,KAAK,QAAQ;oBAClC,CAAC,CAAC,gBAAgB;oBAClB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;gBACvC,MAAM,sBAAsB,GAC1B;oBACE,GAAG,MAAM;oBACT,eAAe,EAAE;wBACf,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,sBAAsB,EAAE,CAAC;wBACzD,IAAI,EAAE,MAAM;wBACZ,UAAU,EAAE,eAAe;wBAC3B,SAAS,EAAE,KAAK,CAAC,kBAAkB,CAAC,SAAS;wBAC7C,YAAY,EAAE,KAAK,CAAC,kBAAkB,CAAC,YAAY;qBACpD;iBACF,CAAC;gBACJ,kBAAkB,CAChB,KAAK,CAAC,kBAAkB,CAAC,QAAQ,EACjC,eAAe,CAAC,kBAAkB,CACnC,CAAC;gBACF,MAAM,sBAAsB,GAAG,MAAM,aAAa,CAChD,MAAM,EACN,sBAAsB,EACtB,KAAK,CAAC,kBAAkB,CAAC,QAAQ,CAClC,CAAC;gBAEF,OAAO,MAAM,mBAAmB,CAC9B,sBAAsB,EACtB,sBAAsB,EACtB,KAAK,CAAC,kBAAkB,CAAC,QAAQ,CAClC,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,IACE,CAAC,cAAc;oBACf,KAAK,CAAC,kBAAkB,CAAC,QAAQ;oBACjC,KAAK,CAAC,kBAAkB,CAAC,QAAQ,KAAK,aAAa,EAAE,EAAE,EACvD,CAAC;oBACD,cAAc,GAAG,IAAI,CAAC;oBACtB,MAAM,mBAAmB,CAAC,KAAK,CAAC,kBAAkB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;gBACtE,CAAC;gBAED,IAAI,CAAC,YAAY,EAAE,CAAC;oBAClB,YAAY,GAAG,KAAK,CAAC,kBAAkB,CAAC,SAAS,EAAE,aAAa;wBAC9D,CAAC,CAAC,0BAA0B,CACxB,KAAK,CAAC,kBAAkB,EACxB,aAAa,CACd;wBACH,CAAC,CAAC,KAAK,CAAC,kBAAkB,CAAC;oBAC7B,MAAM,gBAAgB,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC;gBAC9C,CAAC;qBAAM,CAAC;oBACN,+EAA+E;oBAC/E,MAAM,YAAY,GAChB,KAAK,CAAC,kBAAkB,CAAC,EAAE,KAAK,YAAY,CAAC,EAAE,CAAC;oBAElD,YAAY,GAAG,KAAK,CAAC,kBAAkB,CAAC,SAAS,EAAE,aAAa;wBAC9D,CAAC,CAAC,0BAA0B,CACxB,KAAK,CAAC,kBAAkB,EACxB,aAAa,CACd;wBACH,CAAC,CAAC,KAAK,CAAC,kBAAkB,CAAC;oBAE7B,IAAI,YAAY,EAAE,CAAC;wBACjB,MAAM,gBAAgB,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC;oBAC9C,CAAC;yBAAM,CAAC;wBACN,MAAM,mBAAmB,CAAC,YAAY,CAAC,EAAE,EAAE,YAAY,EAAE,KAAK,CAAC,CAAC;oBAClE,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,kBAAkB,CAChB,YAAY,EAAE,QAAQ,IAAI,QAAQ,EAClC,eAAe,CAAC,QAAQ,CACzB,CAAC;QACF,OAAO,CACL,YAAY,IAAI;YACd,QAAQ,EAAE,EAAE;YACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,yBAAyB,EAAE,CAAC;YAC5D,IAAI,EAAE,WAAW;YACjB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE;YACvB,cAAc,EAAE,EAAE;SACnB,CACF,CAAC;IACJ,CAAC,EACD;QACE,gBAAgB;QAChB,MAAM;QACN,aAAa;QACb,aAAa,EAAE,EAAE;QACjB,mBAAmB;QACnB,YAAY;QACZ,mBAAmB;QACnB,kBAAkB;KACnB,CACF,CAAC;IAEF,MAAM,iBAAiB,GAAG,WAAW,CACnC,KAAK,EACH,OAAe,EACf,UAII,EAAE,QAAQ,EAAE,kBAAkB,CAAC,EAAE,EAAE,EACV,EAAE;QAC/B,MAAM,EAAE,QAAQ,GAAG,aAAa,CAAC,EAAE,EAAE,cAAc,EAAE,GAAG,OAAO,CAAC;QAChE,kBAAkB,CAAC,QAAQ,EAAE,eAAe,CAAC,kBAAkB,CAAC,CAAC;QAEjE,gBAAgB,CACd;YACE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;YAC1C,iBAAiB,EAAE,IAAI;YACvB,IAAI,EAAE,MAAM;YACZ,QAAQ,EAAE,QAAQ;YAClB,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE;YACvB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,cAAc,EAAE,EAAE;SACnB,EACD,KAAK,CACN,CAAC;QAEF,MAAM,mBAAmB,GAAG,sBAAsB,CAChD,aAAa,EACb,YAAY,EACZ,yBAAyB,CAC1B,CAAC;QACF,MAAM,iBAAiB,GAAG,oBAAoB,CAC5C,YAAY,EACZ,yBAAyB,CAC1B,CAAC;QACF,MAAM,MAAM,GAA6C;YACvD,eAAe,EAAE;gBACf,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;gBAC1C,IAAI,EAAE,MAAM;aACb;YACD,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,mBAAmB,EAAE,mBAAmB;YACxC,WAAW,EAAE,iBAAiB,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAC1C,yBAAyB,CAAC,IAAI,CAAC,CAChC;SACF,CAAC;QAEF,IAAI,cAAc,EAAE,CAAC;YACnB,MAAM,qBAAqB,GAAG,MAAM,aAAa,CAC/C,MAAM,EACN,MAAM,EACN,QAAQ,KAAK,kBAAkB,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAC1D,CAAC;YACF,OAAO,MAAM,mBAAmB,CAC9B,qBAAqB,EACrB,MAAM,EACN,QAAQ,CACT,CAAC;QACJ,CAAC;QACD,IAAI,eAAe,GAAG,MAAM,CAAC,QAAQ,KAAK,kBAAkB,CAAC,EAAE;YAC7D,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC;YACrC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAC;QAEvD,mBAAmB;QACnB,OAAO,eAAe,CAAC,kBAAkB,CAAC,eAAe,EAAE,CAAC;YAC1D,kBAAkB,CAAC,QAAQ,EAAE,eAAe,CAAC,gBAAgB,CAAC,CAAC;YAC/D,MAAM,gBAAgB,GAAG,MAAM,cAAc,CAC3C,eAAe,CAAC,kBAAkB,EAClC,YAAY,CACb,CAAC;YACF,MAAM,kBAAkB,GACtB,OAAO,gBAAgB,KAAK,QAAQ;gBAClC,CAAC,CAAC,gBAAgB;gBAClB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YACvC,MAAM,sBAAsB,GAC1B;gBACE,GAAG,MAAM;gBACT,eAAe,EAAE;oBACf,GAAG,MAAM,CAAC,eAAe;oBACzB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC;oBACrD,IAAI,EAAE,MAAM;oBACZ,UAAU,EAAE,eAAe;oBAC3B,SAAS,EAAE,eAAe,CAAC,kBAAkB,CAAC,SAAS;oBACvD,YAAY,EAAE,eAAe,CAAC,kBAAkB,CAAC,YAAY;iBAC9D;aACF,CAAC;YACJ,kBAAkB,CAAC,QAAQ,EAAE,eAAe,CAAC,mBAAmB,CAAC,CAAC;YAClE,eAAe,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CACrD,eAAe,CAAC,kBAAkB,CAAC,QAAQ,EAC3C,sBAAsB,CACvB,CAAC;QACJ,CAAC;QAED,MAAM,YAAY,GAAG,eAAe,CAAC,kBAAkB,CAAC,SAAS;YAC/D,EAAE,aAAa;YACf,CAAC,CAAC,0BAA0B,CACxB,eAAe,CAAC,kBAAkB,EAClC,aAAa,CACd;YACH,CAAC,CAAC,eAAe,CAAC,kBAAkB,CAAC;QACvC,MAAM,mBAAmB,CAAC,eAAe,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QACvE,kBAAkB,CAChB,eAAe,CAAC,kBAAkB,CAAC,QAAQ,EAC3C,eAAe,CAAC,QAAQ,CACzB,CAAC;QACF,OAAO,YAAY,CAAC;IACtB,CAAC,EACD;QACE,aAAa;QACb,YAAY;QACZ,yBAAyB;QACzB,aAAa,CAAC,EAAE;QAChB,mBAAmB;QACnB,gBAAgB;QAChB,MAAM;QACN,kBAAkB;QAClB,mBAAmB;KACpB,CACF,CAAC;IAEF,OAAO,CACL,oBAAC,kBAAkB,CAAC,QAAQ,IAC1B,KAAK,EAAE;YACL,MAAM,EAAE,aAAa;YACrB,mBAAmB;YACnB,cAAc;YACd,gBAAgB;YAChB,mBAAmB;YACnB,UAAU;YACV,aAAa;YACb,iBAAiB;YACjB,eAAe,EAAE,CAAC,aAAa,EAAE,eAAe;gBAC9C,eAAe,CAAC,IAAI,CAAoB;YAC1C,uBAAuB,EAAE,aAAa,EAAE,aAAa,IAAI,EAAE;YAC3D,MAAM,EAAE,WAAW,CACjB,CAAC,aAAa,EAAE,eAAe;gBAC7B,eAAe,CAAC,IAAI,CAAoB,CAC3C;SACF,IAEA,QAAQ,CACmB,CAC/B,CAAC;AACJ,CAAC,CAAC;AAEF;;;;GAIG;AACH,MAAM,CAAC,MAAM,cAAc,GAAG,GAAG,EAAE;IACjC,MAAM,OAAO,GAAG,UAAU,CAAC,kBAAkB,CAAC,CAAC;IAC/C,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;QAC1B,MAAM,IAAI,KAAK,CAAC,0DAA0D,CAAC,CAAC;IAC9E,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC","sourcesContent":["\"use client\";\nimport TamboAI, { advanceStream } from \"@tambo-ai/typescript-sdk\";\nimport React, {\n  createContext,\n  PropsWithChildren,\n  useCallback,\n  useContext,\n  useEffect,\n  useMemo,\n  useState,\n} from \"react\";\nimport {\n  GenerationStage,\n  isIdleStage,\n  TamboThreadMessage,\n} from \"../model/generate-component-response\";\nimport { TamboThread } from \"../model/tambo-thread\";\nimport { renderComponentIntoMessage } from \"../util/generate-component\";\nimport {\n  getAvailableComponents,\n  getClientContext,\n  getUnassociatedTools,\n  mapTamboToolToContextTool,\n} from \"../util/registry\";\nimport { handleToolCall } from \"../util/tool-caller\";\nimport { useTamboClient } from \"./tambo-client-provider\";\nimport { useTamboRegistry } from \"./tambo-registry-provider\";\n\nexport interface TamboThreadContextProps {\n  /** The current thread */\n  thread: TamboThread;\n  /** Switch to a different thread */\n  switchCurrentThread: (threadId: string, fetch?: boolean) => void;\n  /** Start a new thread */\n  startNewThread: () => void;\n  /** Add a message to the current thread */\n  addThreadMessage: (\n    message: TamboThreadMessage,\n    sendToServer: boolean,\n  ) => Promise<TamboAI.Beta.Threads.ThreadMessage[]>;\n  /** Update a message in the current thread */\n  updateThreadMessage: (\n    id: string,\n    message: TamboThreadMessage,\n    sendToServer: boolean,\n  ) => Promise<void>;\n  /** The input value of the current thread */\n  inputValue: string;\n  /** Set the input value of the current thread */\n  setInputValue: (value: string) => void;\n  /** Send a message to the current thread */\n  sendThreadMessage: (\n    message: string,\n    options: {\n      threadId?: string;\n      streamResponse?: boolean;\n      contextKey?: string;\n    },\n  ) => Promise<TamboThreadMessage>;\n  /** The generation stage of the current thread - updated as the thread progresses */\n  generationStage: GenerationStage;\n  /** The generation status message of the current thread - updated as the thread progresses */\n  generationStatusMessage: string;\n  /** Whether the thread is idle */\n  isIdle: boolean;\n}\n\n/**\n * This is a stub entry for when the thread is not yet created, the first time\n * the user sends a message\n *\n * Note that the consumer needs to be careful never to send `PLACEHOLDER_THREAD.id` to the server,\n * as this doesn't really exist on the server side.\n */\nexport const PLACEHOLDER_THREAD: TamboThread = {\n  id: \"placeholder\",\n  messages: [],\n  createdAt: \"\",\n  projectId: \"\",\n  updatedAt: \"\",\n  metadata: {},\n};\n\nexport const TamboThreadContext = createContext<TamboThreadContextProps>({\n  thread: PLACEHOLDER_THREAD,\n  /**\n   *\n   */\n  switchCurrentThread: () => {\n    throw new Error(\"switchCurrentThread not implemented\");\n  },\n  /**\n   *\n   */\n  startNewThread: () => {\n    throw new Error(\"startNewThread not implemented\");\n  },\n  /**\n   *\n   */\n  addThreadMessage: () => {\n    throw new Error(\"updateThreadMessageHistory not implemented\");\n  },\n  inputValue: \"\",\n  /**\n   *\n   */\n  setInputValue: () => {\n    throw new Error(\"setInputValue not implemented\");\n  },\n  /**\n   *\n   */\n  updateThreadMessage: () => {\n    throw new Error(\"updateThreadMessage not implemented\");\n  },\n  /**\n   *\n   */\n  sendThreadMessage: () => {\n    throw new Error(\"advance not implemented\");\n  },\n  generationStage: GenerationStage.IDLE,\n  generationStatusMessage: \"\",\n  isIdle: true,\n});\n\n/**\n * The TamboThreadProvider is a React provider that provides a thread context\n * to the descendants of the provider.\n * @param props - The props for the TamboThreadProvider\n * @param props.children - The children to wrap\n * @returns The TamboThreadProvider component\n */\nexport const TamboThreadProvider: React.FC<PropsWithChildren> = ({\n  children,\n}) => {\n  const [threadMap, setThreadMap] = useState<Record<string, TamboThread>>({\n    [PLACEHOLDER_THREAD.id]: PLACEHOLDER_THREAD,\n  });\n  const client = useTamboClient();\n  const { componentList, toolRegistry, componentToolAssociations } =\n    useTamboRegistry();\n  const [inputValue, setInputValue] = useState(\"\");\n\n  const [currentThreadId, setCurrentThreadId] = useState<string>(\n    PLACEHOLDER_THREAD.id,\n  );\n  const currentThread: TamboThread | undefined = threadMap[currentThreadId];\n\n  // Use existing messages from the current thread to avoid re-generating any components\n  const currentMessageCache = useMemo(() => {\n    const messageCache = new Map<string, TamboThreadMessage>();\n    if (currentThread) {\n      for (const message of currentThread.messages) {\n        messageCache.set(message.id, message);\n      }\n    }\n    return messageCache;\n  }, [currentThread]);\n\n  const fetchThread = useCallback(\n    async (threadId: string) => {\n      const thread = await client.beta.threads.retrieve(threadId);\n      const threadWithRenderedComponents = {\n        ...thread,\n        messages: thread.messages.map((message) => {\n          if (currentMessageCache.has(message.id)) {\n            const renderedMessage = currentMessageCache.get(message.id);\n            return {\n              ...renderedMessage,\n              ...message,\n            };\n          }\n          if (message.component?.componentName) {\n            const messageWithComponent = renderComponentIntoMessage(\n              message,\n              componentList,\n            );\n            return messageWithComponent;\n          }\n          return message;\n        }),\n      };\n\n      setThreadMap((prevMap) => {\n        const updatedThreadMap = {\n          ...prevMap,\n          [threadId]: threadWithRenderedComponents,\n        };\n        return updatedThreadMap;\n      });\n    },\n    [client.beta.threads, componentList, currentMessageCache],\n  );\n\n  useEffect(() => {\n    if (\n      currentThreadId &&\n      currentThreadId !== PLACEHOLDER_THREAD.id &&\n      !threadMap[currentThreadId]\n    ) {\n      fetchThread(currentThreadId);\n    }\n  }, [currentThreadId, fetchThread, threadMap]);\n\n  const addThreadMessage = useCallback(\n    async (\n      message: TamboThreadMessage,\n      sendToServer = true,\n      createdAt: string = new Date().toISOString(),\n    ) => {\n      if (!currentThread) {\n        console.warn(\"Cannot add messages if we do not have a current thread\");\n        return [];\n      }\n      const chatMessage: TamboThreadMessage & {\n        additionalContext?: string;\n      } = {\n        ...message,\n        additionalContext:\n          message.role === \"user\" ? getClientContext() : undefined,\n        createdAt,\n      };\n      const threadId = message.threadId;\n      const messageId = chatMessage.id;\n      // optimistically update the thread in the local state\n      setThreadMap((prevMap) => {\n        if (!threadId) {\n          return prevMap;\n        }\n        const prevMessages = prevMap[threadId]?.messages || [];\n        const haveMessage = prevMessages.find((msg) => msg.id === messageId);\n        // Update in place if the message already exists\n        const updatedMessages = haveMessage\n          ? prevMessages.map((msg) => {\n              if (msg.id === messageId) {\n                return chatMessage;\n              }\n              return msg;\n            })\n          : [...prevMessages, chatMessage];\n\n        const updatedThreadMap = {\n          ...prevMap,\n          [threadId]: {\n            ...prevMap[threadId],\n            messages: updatedMessages,\n          },\n        };\n        return updatedThreadMap;\n      });\n\n      if (sendToServer) {\n        // TODO: if this fails, we need to revert the local state update\n        await client.beta.threads.messages.create(message.threadId, {\n          content: message.content,\n          role: message.role,\n          // additionalContext: chatMessage.additionalContext,\n        });\n      }\n      return threadMap[threadId]?.messages || [];\n    },\n    [client.beta.threads.messages, currentThread, threadMap],\n  );\n\n  const updateThreadMessage = useCallback(\n    async (\n      id: string,\n      message: TamboThreadMessage,\n      sendToServer = true,\n      createdAt: string = new Date().toISOString(),\n    ) => {\n      const chatMessage: TamboThreadMessage = {\n        ...message,\n        createdAt,\n      };\n\n      setThreadMap((prevMap) => {\n        if (!message.threadId) {\n          return prevMap;\n        }\n        const prevMessages = prevMap[message.threadId]?.messages || [];\n        const updatedMessages = prevMessages.map((msg) => {\n          if (msg.id === id) {\n            return chatMessage;\n          }\n          return msg;\n        });\n        return {\n          ...prevMap,\n          [message.threadId]: {\n            ...prevMap[message.threadId],\n            messages: updatedMessages,\n          },\n        };\n      });\n      if (sendToServer) {\n        // TODO: if this fails, we need to revert the local state update\n        await client.beta.threads.messages.create(message.threadId, {\n          content: message.content,\n          role: message.role,\n          // additionalContext: chatMessage.additionalContext,\n        });\n      }\n    },\n    [client.beta.threads.messages],\n  );\n\n  const startNewThread = useCallback(() => {\n    setCurrentThreadId(PLACEHOLDER_THREAD.id);\n    setThreadMap((prevMap) => {\n      return {\n        ...prevMap,\n        [PLACEHOLDER_THREAD.id]: PLACEHOLDER_THREAD,\n      };\n    });\n  }, []);\n\n  const switchCurrentThread = useCallback(\n    async (threadId: string, fetch = true) => {\n      if (threadId === PLACEHOLDER_THREAD.id) {\n        console.warn(\"Switching to placeholder thread, may be a bug.\");\n        return;\n      }\n      setCurrentThreadId(threadId);\n      setThreadMap((prevMap) => {\n        if (prevMap[threadId]) {\n          return prevMap;\n        }\n        // If this is a new thread, add placeholder thread messages to the thread\n        const updatedThreadMap = {\n          ...prevMap,\n          [threadId]: {\n            ...prevMap[PLACEHOLDER_THREAD.id],\n            id: threadId,\n          },\n        };\n        return updatedThreadMap;\n      });\n      if (fetch) {\n        await fetchThread(threadId);\n      }\n    },\n    [fetchThread],\n  );\n\n  const updateThreadStatus = useCallback(\n    (threadId: string, stage: GenerationStage, statusMessage?: string) => {\n      setThreadMap((prevMap) => {\n        const updatedThreadMap = {\n          ...prevMap,\n          [threadId]: {\n            ...prevMap[threadId],\n            generationStage: stage,\n            statusMessage: statusMessage,\n          },\n        };\n        return updatedThreadMap;\n      });\n    },\n    [],\n  );\n\n  const handleAdvanceStream = useCallback(\n    async (\n      stream: AsyncIterable<TamboAI.Beta.Threads.ThreadAdvanceResponse>,\n      params: TamboAI.Beta.Threads.ThreadAdvanceParams,\n      threadId: string,\n    ): Promise<TamboThreadMessage> => {\n      let finalMessage: Readonly<TamboThreadMessage> | undefined;\n      let hasSetThreadId = false;\n      updateThreadStatus(threadId, GenerationStage.STREAMING_RESPONSE);\n\n      for await (const chunk of stream) {\n        if (chunk.responseMessageDto.toolCallRequest) {\n          updateThreadStatus(\n            chunk.responseMessageDto.threadId,\n            GenerationStage.FETCHING_CONTEXT,\n          );\n          const toolCallResponse = await handleToolCall(\n            chunk.responseMessageDto,\n            toolRegistry,\n          );\n          const toolCallResponseString =\n            typeof toolCallResponse === \"string\"\n              ? toolCallResponse\n              : JSON.stringify(toolCallResponse);\n          const toolCallResponseParams: TamboAI.Beta.Threads.ThreadAdvanceParams =\n            {\n              ...params,\n              messageToAppend: {\n                content: [{ type: \"text\", text: toolCallResponseString }],\n                role: \"tool\",\n                actionType: \"tool_response\",\n                component: chunk.responseMessageDto.component,\n                tool_call_id: chunk.responseMessageDto.tool_call_id,\n              },\n            };\n          updateThreadStatus(\n            chunk.responseMessageDto.threadId,\n            GenerationStage.STREAMING_RESPONSE,\n          );\n          const toolCallResponseStream = await advanceStream(\n            client,\n            toolCallResponseParams,\n            chunk.responseMessageDto.threadId,\n          );\n\n          return await handleAdvanceStream(\n            toolCallResponseStream,\n            toolCallResponseParams,\n            chunk.responseMessageDto.threadId,\n          );\n        } else {\n          if (\n            !hasSetThreadId &&\n            chunk.responseMessageDto.threadId &&\n            chunk.responseMessageDto.threadId !== currentThread?.id\n          ) {\n            hasSetThreadId = true;\n            await switchCurrentThread(chunk.responseMessageDto.threadId, false);\n          }\n\n          if (!finalMessage) {\n            finalMessage = chunk.responseMessageDto.component?.componentName\n              ? renderComponentIntoMessage(\n                  chunk.responseMessageDto,\n                  componentList,\n                )\n              : chunk.responseMessageDto;\n            await addThreadMessage(finalMessage, false);\n          } else {\n            // if we start getting a new message mid-stream, put the previous one on screen\n            const isNewMessage =\n              chunk.responseMessageDto.id !== finalMessage.id;\n\n            finalMessage = chunk.responseMessageDto.component?.componentName\n              ? renderComponentIntoMessage(\n                  chunk.responseMessageDto,\n                  componentList,\n                )\n              : chunk.responseMessageDto;\n\n            if (isNewMessage) {\n              await addThreadMessage(finalMessage, false);\n            } else {\n              await updateThreadMessage(finalMessage.id, finalMessage, false);\n            }\n          }\n        }\n      }\n\n      updateThreadStatus(\n        finalMessage?.threadId ?? threadId,\n        GenerationStage.COMPLETE,\n      );\n      return (\n        finalMessage ?? {\n          threadId: \"\",\n          content: [{ type: \"text\", text: `Error processing stream` }],\n          role: \"assistant\",\n          createdAt: new Date().toISOString(),\n          id: crypto.randomUUID(),\n          componentState: {},\n        }\n      );\n    },\n    [\n      addThreadMessage,\n      client,\n      componentList,\n      currentThread?.id,\n      switchCurrentThread,\n      toolRegistry,\n      updateThreadMessage,\n      updateThreadStatus,\n    ],\n  );\n\n  const sendThreadMessage = useCallback(\n    async (\n      message: string,\n      options: {\n        threadId?: string;\n        streamResponse?: boolean;\n        contextKey?: string;\n      } = { threadId: PLACEHOLDER_THREAD.id },\n    ): Promise<TamboThreadMessage> => {\n      const { threadId = currentThread.id, streamResponse } = options;\n      updateThreadStatus(threadId, GenerationStage.CHOOSING_COMPONENT);\n\n      addThreadMessage(\n        {\n          content: [{ type: \"text\", text: message }],\n          renderedComponent: null,\n          role: \"user\",\n          threadId: threadId,\n          id: crypto.randomUUID(),\n          createdAt: new Date().toISOString(),\n          componentState: {},\n        },\n        false,\n      );\n\n      const availableComponents = getAvailableComponents(\n        componentList,\n        toolRegistry,\n        componentToolAssociations,\n      );\n      const unassociatedTools = getUnassociatedTools(\n        toolRegistry,\n        componentToolAssociations,\n      );\n      const params: TamboAI.Beta.Threads.ThreadAdvanceParams = {\n        messageToAppend: {\n          content: [{ type: \"text\", text: message }],\n          role: \"user\",\n        },\n        contextKey: options.contextKey,\n        availableComponents: availableComponents,\n        clientTools: unassociatedTools.map((tool) =>\n          mapTamboToolToContextTool(tool),\n        ),\n      };\n\n      if (streamResponse) {\n        const advanceStreamResponse = await advanceStream(\n          client,\n          params,\n          threadId === PLACEHOLDER_THREAD.id ? undefined : threadId,\n        );\n        return await handleAdvanceStream(\n          advanceStreamResponse,\n          params,\n          threadId,\n        );\n      }\n      let advanceResponse = await (threadId === PLACEHOLDER_THREAD.id\n        ? client.beta.threads.advance(params)\n        : client.beta.threads.advanceById(threadId, params));\n\n      //handle tool calls\n      while (advanceResponse.responseMessageDto.toolCallRequest) {\n        updateThreadStatus(threadId, GenerationStage.FETCHING_CONTEXT);\n        const toolCallResponse = await handleToolCall(\n          advanceResponse.responseMessageDto,\n          toolRegistry,\n        );\n        const toolResponseString =\n          typeof toolCallResponse === \"string\"\n            ? toolCallResponse\n            : JSON.stringify(toolCallResponse);\n        const toolCallResponseParams: TamboAI.Beta.Threads.ThreadAdvanceParams =\n          {\n            ...params,\n            messageToAppend: {\n              ...params.messageToAppend,\n              content: [{ type: \"text\", text: toolResponseString }],\n              role: \"tool\",\n              actionType: \"tool_response\",\n              component: advanceResponse.responseMessageDto.component,\n              tool_call_id: advanceResponse.responseMessageDto.tool_call_id,\n            },\n          };\n        updateThreadStatus(threadId, GenerationStage.HYDRATING_COMPONENT);\n        advanceResponse = await client.beta.threads.advanceById(\n          advanceResponse.responseMessageDto.threadId,\n          toolCallResponseParams,\n        );\n      }\n\n      const finalMessage = advanceResponse.responseMessageDto.component\n        ?.componentName\n        ? renderComponentIntoMessage(\n            advanceResponse.responseMessageDto,\n            componentList,\n          )\n        : advanceResponse.responseMessageDto;\n      await switchCurrentThread(advanceResponse.responseMessageDto.threadId);\n      updateThreadStatus(\n        advanceResponse.responseMessageDto.threadId,\n        GenerationStage.COMPLETE,\n      );\n      return finalMessage;\n    },\n    [\n      componentList,\n      toolRegistry,\n      componentToolAssociations,\n      currentThread.id,\n      switchCurrentThread,\n      addThreadMessage,\n      client,\n      updateThreadStatus,\n      handleAdvanceStream,\n    ],\n  );\n\n  return (\n    <TamboThreadContext.Provider\n      value={{\n        thread: currentThread,\n        switchCurrentThread,\n        startNewThread,\n        addThreadMessage,\n        updateThreadMessage,\n        inputValue,\n        setInputValue,\n        sendThreadMessage,\n        generationStage: (currentThread?.generationStage ??\n          GenerationStage.IDLE) as GenerationStage,\n        generationStatusMessage: currentThread?.statusMessage ?? \"\",\n        isIdle: isIdleStage(\n          (currentThread?.generationStage ??\n            GenerationStage.IDLE) as GenerationStage,\n        ),\n      }}\n    >\n      {children}\n    </TamboThreadContext.Provider>\n  );\n};\n\n/**\n * The useTamboThread hook provides access to the current thread context\n * to the descendants of the TamboThreadProvider.\n * @returns All state and actions for the current thread\n */\nexport const useTamboThread = () => {\n  const context = useContext(TamboThreadContext);\n  if (context === undefined) {\n    throw new Error(\"useTamboThread must be used within a TamboThreadProvider\");\n  }\n  return context;\n};\n"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tambo-ai/react",
-  "version": "0.21.1",
+  "version": "0.21.3",
   "description": "React client package for Tambo AI",
   "repository": {
     "type": "git",