@tambo-ai/react 0.12.0

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

@@ -0,0 +1,91 @@
+"use client";
+import React, { createContext, useCallback, useContext, useState, } from "react";
+import { GenerationStage, isIdleStage, } from "../model/generate-component-response";
+import { generateAndHydrate } from "../util/generate-component";
+import { useTamboClient } from "./tambo-client-provider";
+import { useTamboRegistry, } from "./tambo-registry-provider";
+import { PLACEHOLDER_THREAD, useTamboThread } from "./tambo-thread-provider";
+const TamboComponentContext = createContext({
+    generateComponent: () => Promise.resolve({}),
+    generationStage: GenerationStage.IDLE,
+    isIdle: true,
+    registerComponent: () => { },
+    registerTool: () => { },
+    registerTools: () => { },
+    addToolAssociation: () => { },
+});
+export const TamboComponentProvider = ({ children, }) => {
+    const client = useTamboClient();
+    const { addThreadMessage, updateThreadMessage, switchCurrentThread, thread: currentThread, } = useTamboThread();
+    const { componentList, registerComponent, addToolAssociation, registerTool, componentToolAssociations, registerTools, toolRegistry, } = useTamboRegistry();
+    const [generationStage, setGenerationStage] = useState(GenerationStage.IDLE);
+    const isIdle = isIdleStage(generationStage);
+    const generateComponent = useCallback(async (params, options) => {
+        if (!isIdleStage(generationStage)) {
+            throw new Error(`Generation already in progress: ${generationStage}`);
+        }
+        if (params.threadId && params.threadId !== currentThread.id) {
+            switchCurrentThread(params.threadId);
+        }
+        const componentMessage = await generateAndHydrate(client, componentList, toolRegistry, componentToolAssociations, currentThread.id === PLACEHOLDER_THREAD.id
+            ? undefined
+            : currentThread.id, params.content, params, (state) => setGenerationStage(state), params.stream, options);
+        if (params.stream) {
+            let finalResponse;
+            const generationResponse = componentMessage;
+            for await (const chunk of generationResponse) {
+                if (chunk.threadId && chunk.threadId !== currentThread?.id) {
+                    switchCurrentThread(chunk.threadId);
+                }
+                if (!finalResponse) {
+                    finalResponse = chunk;
+                    addThreadMessage(finalResponse, false);
+                }
+                else {
+                    // use previous chunk's id for update, in case new chunk has a different id (from tambotion response)
+                    updateThreadMessage(finalResponse.id, chunk, false);
+                    finalResponse = chunk;
+                }
+            }
+            if (!finalResponse) {
+                throw new Error("No response from generation");
+            }
+            return finalResponse;
+        }
+        const generationResponse = componentMessage;
+        if (generationResponse.threadId &&
+            generationResponse.threadId !== currentThread?.id) {
+            switchCurrentThread(generationResponse.threadId);
+        }
+        if (generationResponse) {
+            // we do not need to send this to the server, as it is already in the thread
+            addThreadMessage(generationResponse, false);
+        }
+        return generationResponse;
+    }, [
+        generationStage,
+        switchCurrentThread,
+        currentThread,
+        client,
+        componentList,
+        addThreadMessage,
+        updateThreadMessage,
+        componentToolAssociations,
+        toolRegistry,
+    ]);
+    const value = {
+        client,
+        generateComponent,
+        registerComponent,
+        generationStage,
+        isIdle,
+        registerTool,
+        registerTools,
+        addToolAssociation,
+    };
+    return (React.createElement(TamboComponentContext.Provider, { value: value }, children));
+};
+export const useTamboComponent = () => {
+    return useContext(TamboComponentContext);
+};
+//# sourceMappingURL=tambo-component-provider.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"tambo-component-provider.js","sourceRoot":"","sources":["../../src/providers/tambo-component-provider.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;AAEb,OAAO,KAAK,EAAE,EACZ,aAAa,EAEb,WAAW,EACX,UAAU,EACV,QAAQ,GACT,MAAM,OAAO,CAAC;AAEf,OAAO,EACL,eAAe,EACf,WAAW,GAEZ,MAAM,sCAAsC,CAAC;AAC9C,OAAO,EAAE,kBAAkB,EAAE,MAAM,4BAA4B,CAAC;AAChE,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AACzD,OAAO,EAEL,gBAAgB,GACjB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAAE,kBAAkB,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AAiB7E,MAAM,qBAAqB,GAAG,aAAa,CAA6B;IACtE,iBAAiB,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,EAAwB,CAAC;IAClE,eAAe,EAAE,eAAe,CAAC,IAAI;IACrC,MAAM,EAAE,IAAI;IACZ,iBAAiB,EAAE,GAAG,EAAE,GAAE,CAAC;IAC3B,YAAY,EAAE,GAAG,EAAE,GAAE,CAAC;IACtB,aAAa,EAAE,GAAG,EAAE,GAAE,CAAC;IACvB,kBAAkB,EAAE,GAAG,EAAE,GAAE,CAAC;CAC7B,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,sBAAsB,GAAgC,CAAC,EAClE,QAAQ,GACT,EAAE,EAAE;IACH,MAAM,MAAM,GAAG,cAAc,EAAE,CAAC;IAChC,MAAM,EACJ,gBAAgB,EAChB,mBAAmB,EACnB,mBAAmB,EACnB,MAAM,EAAE,aAAa,GACtB,GAAG,cAAc,EAAE,CAAC;IACrB,MAAM,EACJ,aAAa,EACb,iBAAiB,EACjB,kBAAkB,EAClB,YAAY,EACZ,yBAAyB,EACzB,aAAa,EACb,YAAY,GACb,GAAG,gBAAgB,EAAE,CAAC;IACvB,MAAM,CAAC,eAAe,EAAE,kBAAkB,CAAC,GAAG,QAAQ,CACpD,eAAe,CAAC,IAAI,CACrB,CAAC;IAEF,MAAM,MAAM,GAAG,WAAW,CAAC,eAAe,CAAC,CAAC;IAE5C,MAAM,iBAAiB,GAAG,WAAW,CACnC,KAAK,EACH,MAEC,EACD,OAAgC,EACH,EAAE;QAC/B,IAAI,CAAC,WAAW,CAAC,eAAe,CAAC,EAAE,CAAC;YAClC,MAAM,IAAI,KAAK,CAAC,mCAAmC,eAAe,EAAE,CAAC,CAAC;QACxE,CAAC;QAED,IAAI,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,QAAQ,KAAK,aAAa,CAAC,EAAE,EAAE,CAAC;YAC5D,mBAAmB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACvC,CAAC;QACD,MAAM,gBAAgB,GAAG,MAAM,kBAAkB,CAC/C,MAAM,EACN,aAAa,EACb,YAAY,EACZ,yBAAyB,EACzB,aAAa,CAAC,EAAE,KAAK,kBAAkB,CAAC,EAAE;YACxC,CAAC,CAAC,SAAS;YACX,CAAC,CAAC,aAAa,CAAC,EAAE,EACpB,MAAM,CAAC,OAAO,EACd,MAAM,EACN,CAAC,KAAK,EAAE,EAAE,CAAC,kBAAkB,CAAC,KAAK,CAAC,EACpC,MAAM,CAAC,MAAM,EACb,OAAO,CACR,CAAC;QAEF,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAClB,IAAI,aAA6C,CAAC;YAClD,MAAM,kBAAkB,GACtB,gBAAqD,CAAC;YACxD,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,kBAAkB,EAAE,CAAC;gBAC7C,IAAI,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,QAAQ,KAAK,aAAa,EAAE,EAAE,EAAE,CAAC;oBAC3D,mBAAmB,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;gBACtC,CAAC;gBACD,IAAI,CAAC,aAAa,EAAE,CAAC;oBACnB,aAAa,GAAG,KAAK,CAAC;oBACtB,gBAAgB,CAAC,aAAa,EAAE,KAAK,CAAC,CAAC;gBACzC,CAAC;qBAAM,CAAC;oBACN,qGAAqG;oBACrG,mBAAmB,CAAC,aAAa,CAAC,EAAE,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;oBACpD,aAAa,GAAG,KAAK,CAAC;gBACxB,CAAC;YACH,CAAC;YACD,IAAI,CAAC,aAAa,EAAE,CAAC;gBACnB,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;YACjD,CAAC;YACD,OAAO,aAAa,CAAC;QACvB,CAAC;QACD,MAAM,kBAAkB,GAAG,gBAAsC,CAAC;QAClE,IACE,kBAAkB,CAAC,QAAQ;YAC3B,kBAAkB,CAAC,QAAQ,KAAK,aAAa,EAAE,EAAE,EACjD,CAAC;YACD,mBAAmB,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QACnD,CAAC;QAED,IAAI,kBAAkB,EAAE,CAAC;YACvB,4EAA4E;YAC5E,gBAAgB,CAAC,kBAAkB,EAAE,KAAK,CAAC,CAAC;QAC9C,CAAC;QAED,OAAO,kBAAkB,CAAC;IAC5B,CAAC,EACD;QACE,eAAe;QACf,mBAAmB;QACnB,aAAa;QACb,MAAM;QACN,aAAa;QACb,gBAAgB;QAChB,mBAAmB;QACnB,yBAAyB;QACzB,YAAY;KACb,CACF,CAAC;IAEF,MAAM,KAAK,GAAG;QACZ,MAAM;QACN,iBAAiB;QACjB,iBAAiB;QACjB,eAAe;QACf,MAAM;QACN,YAAY;QACZ,aAAa;QACb,kBAAkB;KACnB,CAAC;IAEF,OAAO,CACL,oBAAC,qBAAqB,CAAC,QAAQ,IAAC,KAAK,EAAE,KAAK,IACzC,QAAQ,CACsB,CAClC,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,iBAAiB,GAAG,GAAG,EAAE;IACpC,OAAO,UAAU,CAAC,qBAAqB,CAAC,CAAC;AAC3C,CAAC,CAAC","sourcesContent":["\"use client\";\nimport TamboAI from \"@tambo-ai/typescript-sdk\";\nimport React, {\n  createContext,\n  PropsWithChildren,\n  useCallback,\n  useContext,\n  useState,\n} from \"react\";\nimport { TamboTool } from \"../model/component-metadata\";\nimport {\n  GenerationStage,\n  isIdleStage,\n  TamboThreadMessage,\n} from \"../model/generate-component-response\";\nimport { generateAndHydrate } from \"../util/generate-component\";\nimport { useTamboClient } from \"./tambo-client-provider\";\nimport {\n  RegisterComponentOptions,\n  useTamboRegistry,\n} from \"./tambo-registry-provider\";\nimport { PLACEHOLDER_THREAD, useTamboThread } from \"./tambo-thread-provider\";\n\nexport interface TamboComponentContextProps {\n  registerComponent: (options: RegisterComponentOptions) => void;\n  generateComponent: (\n    params: TamboAI.Beta.Components.ComponentGenerateParams & {\n      stream?: boolean;\n    },\n    options?: TamboAI.RequestOptions,\n  ) => Promise<TamboThreadMessage>;\n  generationStage: GenerationStage;\n  isIdle: boolean;\n  registerTool: (tool: TamboTool) => void;\n  registerTools: (tools: TamboTool[]) => void;\n  addToolAssociation: (componentName: string, tool: TamboTool) => void;\n}\n\nconst TamboComponentContext = createContext<TamboComponentContextProps>({\n  generateComponent: () => Promise.resolve({} as TamboThreadMessage),\n  generationStage: GenerationStage.IDLE,\n  isIdle: true,\n  registerComponent: () => {},\n  registerTool: () => {},\n  registerTools: () => {},\n  addToolAssociation: () => {},\n});\n\nexport const TamboComponentProvider: React.FC<PropsWithChildren> = ({\n  children,\n}) => {\n  const client = useTamboClient();\n  const {\n    addThreadMessage,\n    updateThreadMessage,\n    switchCurrentThread,\n    thread: currentThread,\n  } = useTamboThread();\n  const {\n    componentList,\n    registerComponent,\n    addToolAssociation,\n    registerTool,\n    componentToolAssociations,\n    registerTools,\n    toolRegistry,\n  } = useTamboRegistry();\n  const [generationStage, setGenerationStage] = useState<GenerationStage>(\n    GenerationStage.IDLE,\n  );\n\n  const isIdle = isIdleStage(generationStage);\n\n  const generateComponent = useCallback(\n    async (\n      params: TamboAI.Beta.Components.ComponentGenerateParams & {\n        stream?: boolean;\n      },\n      options?: TamboAI.RequestOptions,\n    ): Promise<TamboThreadMessage> => {\n      if (!isIdleStage(generationStage)) {\n        throw new Error(`Generation already in progress: ${generationStage}`);\n      }\n\n      if (params.threadId && params.threadId !== currentThread.id) {\n        switchCurrentThread(params.threadId);\n      }\n      const componentMessage = await generateAndHydrate(\n        client,\n        componentList,\n        toolRegistry,\n        componentToolAssociations,\n        currentThread.id === PLACEHOLDER_THREAD.id\n          ? undefined\n          : currentThread.id,\n        params.content,\n        params,\n        (state) => setGenerationStage(state),\n        params.stream,\n        options,\n      );\n\n      if (params.stream) {\n        let finalResponse: TamboThreadMessage | undefined;\n        const generationResponse =\n          componentMessage as AsyncIterable<TamboThreadMessage>;\n        for await (const chunk of generationResponse) {\n          if (chunk.threadId && chunk.threadId !== currentThread?.id) {\n            switchCurrentThread(chunk.threadId);\n          }\n          if (!finalResponse) {\n            finalResponse = chunk;\n            addThreadMessage(finalResponse, false);\n          } else {\n            // use previous chunk's id for update, in case new chunk has a different id (from tambotion response)\n            updateThreadMessage(finalResponse.id, chunk, false);\n            finalResponse = chunk;\n          }\n        }\n        if (!finalResponse) {\n          throw new Error(\"No response from generation\");\n        }\n        return finalResponse;\n      }\n      const generationResponse = componentMessage as TamboThreadMessage;\n      if (\n        generationResponse.threadId &&\n        generationResponse.threadId !== currentThread?.id\n      ) {\n        switchCurrentThread(generationResponse.threadId);\n      }\n\n      if (generationResponse) {\n        // we do not need to send this to the server, as it is already in the thread\n        addThreadMessage(generationResponse, false);\n      }\n\n      return generationResponse;\n    },\n    [\n      generationStage,\n      switchCurrentThread,\n      currentThread,\n      client,\n      componentList,\n      addThreadMessage,\n      updateThreadMessage,\n      componentToolAssociations,\n      toolRegistry,\n    ],\n  );\n\n  const value = {\n    client,\n    generateComponent,\n    registerComponent,\n    generationStage,\n    isIdle,\n    registerTool,\n    registerTools,\n    addToolAssociation,\n  };\n\n  return (\n    <TamboComponentContext.Provider value={value}>\n      {children}\n    </TamboComponentContext.Provider>\n  );\n};\n\nexport const useTamboComponent = () => {\n  return useContext(TamboComponentContext);\n};\n"]}

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

@@ -0,0 +1,10 @@
+import React, { PropsWithChildren } from "react";
+import { TamboClientContextProps, TamboClientProviderProps } from "./tambo-client-provider";
+import { TamboComponentContextProps } from "./tambo-component-provider";
+import { TamboThreadContextProps } from "./tambo-thread-provider";
+/** TamboProvider gives full access to the TamboAI client and component registry */
+export declare const TamboProvider: React.FC<PropsWithChildren<TamboClientProviderProps>>;
+type TamboContextProps = TamboClientContextProps & TamboThreadContextProps & TamboComponentContextProps;
+export declare const TamboContext: React.Context<TamboContextProps>;
+export declare const useTambo: () => TamboContextProps;
+export {};

package/esm/providers/tambo-provider.js

@@ -0,0 +1,32 @@
+"use client";
+import React, { createContext, useContext } from "react";
+import { TamboClientProvider, useTamboClient, useTamboQueryClient, } from "./tambo-client-provider";
+import { TamboComponentProvider, useTamboComponent, } from "./tambo-component-provider";
+import { TamboRegistryProvider } from "./tambo-registry-provider";
+import { TamboThreadProvider, useTamboThread, } from "./tambo-thread-provider";
+/** TamboProvider gives full access to the TamboAI client and component registry */
+export const TamboProvider = ({ children, tamboUrl, apiKey }) => {
+    return (React.createElement(TamboClientProvider, { tamboUrl: tamboUrl, apiKey: apiKey },
+        React.createElement(TamboRegistryProvider, null,
+            React.createElement(TamboThreadProvider, null,
+                React.createElement(TamboComponentProvider, null,
+                    React.createElement(TamboCompositeProvider, null, children))))));
+};
+export const TamboContext = createContext({});
+/** TamboCompositeProvider is a provider that combines the TamboClient, TamboThread, and TamboComponent providers */
+const TamboCompositeProvider = ({ children }) => {
+    const threads = useTamboThread();
+    const client = useTamboClient();
+    const queryClient = useTamboQueryClient();
+    const componentRegistry = useTamboComponent();
+    return (React.createElement(TamboContext.Provider, { value: {
+            client,
+            queryClient,
+            ...componentRegistry,
+            ...threads,
+        } }, children));
+};
+export const useTambo = () => {
+    return useContext(TamboContext);
+};
+//# sourceMappingURL=tambo-provider.js.map

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

@@ -0,0 +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,EAAE,qBAAqB,EAAE,MAAM,2BAA2B,CAAC;AAClE,OAAO,EAEL,mBAAmB,EACnB,cAAc,GACf,MAAM,yBAAyB,CAAC;AAEjC,mFAAmF;AAEnF,MAAM,CAAC,MAAM,aAAa,GAEtB,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE;IACrC,OAAO,CACL,oBAAC,mBAAmB,IAAC,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM;QACrD,oBAAC,qBAAqB;YACpB,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,oHAAoH;AACpH,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,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 { TamboRegistryProvider } from \"./tambo-registry-provider\";\nimport {\n  TamboThreadContextProps,\n  TamboThreadProvider,\n  useTamboThread,\n} from \"./tambo-thread-provider\";\n\n/** TamboProvider gives full access to the TamboAI client and component registry */\n\nexport const TamboProvider: React.FC<\n  PropsWithChildren<TamboClientProviderProps>\n> = ({ children, tamboUrl, apiKey }) => {\n  return (\n    <TamboClientProvider tamboUrl={tamboUrl} apiKey={apiKey}>\n      <TamboRegistryProvider>\n        <TamboThreadProvider>\n          <TamboComponentProvider>\n            <TamboCompositeProvider>{children}</TamboCompositeProvider>\n          </TamboComponentProvider>\n        </TamboThreadProvider>\n      </TamboRegistryProvider>\n    </TamboClientProvider>\n  );\n};\ntype TamboContextProps = TamboClientContextProps &\n  TamboThreadContextProps &\n  TamboComponentContextProps;\n\nexport const TamboContext = createContext<TamboContextProps>(\n  {} as TamboContextProps,\n);\n\n/** TamboCompositeProvider is a provider that combines the TamboClient, TamboThread, and TamboComponent providers */\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\nexport const useTambo = () => {\n  return useContext(TamboContext);\n};\n"]}

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

@@ -0,0 +1,21 @@
+import React, { ComponentType, PropsWithChildren } from "react";
+import { ComponentRegistry, TamboTool } from "../model/component-metadata";
+export interface RegisterComponentOptions {
+    name: string;
+    description: string;
+    component: ComponentType<any>;
+    propsDefinition?: any;
+    loadingComponent?: ComponentType<any>;
+    associatedTools?: TamboTool[];
+}
+export interface TamboRegistryContext {
+    componentList: ComponentRegistry;
+    toolRegistry: Record<string, TamboTool>;
+    componentToolAssociations: Record<string, string[]>;
+    registerComponent: (options: RegisterComponentOptions) => void;
+    registerTool: (tool: TamboTool) => void;
+    registerTools: (tools: TamboTool[]) => void;
+    addToolAssociation: (componentName: string, tool: TamboTool) => void;
+}
+export declare const TamboRegistryProvider: React.FC<PropsWithChildren>;
+export declare const useTamboRegistry: () => TamboRegistryContext;

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

@@ -0,0 +1,79 @@
+"use client";
+import React, { createContext, useCallback, useContext, useState, } from "react";
+const TamboRegistryContext = createContext({
+    componentList: {},
+    toolRegistry: {},
+    componentToolAssociations: {},
+    registerComponent: () => { },
+    registerTool: () => { },
+    registerTools: () => { },
+    addToolAssociation: () => { },
+});
+export const TamboRegistryProvider = ({ children, }) => {
+    const [componentList, setComponentList] = useState({});
+    const [toolRegistry, setToolRegistry] = useState({});
+    const [componentToolAssociations, setComponentToolAssociations] = useState({});
+    const registerTool = useCallback((tool) => {
+        setToolRegistry((prev) => {
+            if (prev[tool.name]) {
+                console.warn(`Overwriting tool ${tool.name}`);
+            }
+            return {
+                ...prev,
+                [tool.name]: tool,
+            };
+        });
+    }, []);
+    const registerTools = useCallback((tools) => {
+        tools.forEach((tool) => registerTool(tool));
+    }, [registerTool]);
+    const addToolAssociation = useCallback((componentName, tool) => {
+        if (!componentList[componentName]) {
+            throw new Error(`Component ${componentName} not found in registry`);
+        }
+        setComponentToolAssociations((prev) => ({
+            ...prev,
+            [componentName]: [...(prev[componentName] || []), tool.name],
+        }));
+    }, [componentList]);
+    const registerComponent = useCallback((options) => {
+        const { name, description, component, propsDefinition = {}, loadingComponent, associatedTools, } = options;
+        setComponentList((prev) => {
+            if (prev[name]) {
+                console.warn(`overwriting component ${name}`);
+            }
+            return {
+                ...prev,
+                [name]: {
+                    component,
+                    loadingComponent,
+                    name,
+                    description,
+                    props: propsDefinition,
+                    contextTools: [],
+                },
+            };
+        });
+        if (associatedTools) {
+            registerTools(associatedTools);
+            setComponentToolAssociations((prev) => ({
+                ...prev,
+                [name]: associatedTools.map((tool) => tool.name),
+            }));
+        }
+    }, [registerTools]);
+    const value = {
+        componentList,
+        toolRegistry,
+        componentToolAssociations,
+        registerComponent,
+        registerTool,
+        registerTools,
+        addToolAssociation,
+    };
+    return (React.createElement(TamboRegistryContext.Provider, { value: value }, children));
+};
+export const useTamboRegistry = () => {
+    return useContext(TamboRegistryContext);
+};
+//# sourceMappingURL=tambo-registry-provider.js.map

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

@@ -0,0 +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,EAEZ,aAAa,EAEb,WAAW,EACX,UAAU,EACV,QAAQ,GACT,MAAM,OAAO,CAAC;AAsBf,MAAM,oBAAoB,GAAG,aAAa,CAAuB;IAC/D,aAAa,EAAE,EAAE;IACjB,YAAY,EAAE,EAAE;IAChB,yBAAyB,EAAE,EAAE;IAC7B,iBAAiB,EAAE,GAAG,EAAE,GAAE,CAAC;IAC3B,YAAY,EAAE,GAAG,EAAE,GAAE,CAAC;IACtB,aAAa,EAAE,GAAG,EAAE,GAAE,CAAC;IACvB,kBAAkB,EAAE,GAAG,EAAE,GAAE,CAAC;CAC7B,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,qBAAqB,GAAgC,CAAC,EACjE,QAAQ,GACT,EAAE,EAAE;IACH,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,CAAC,CAAC,IAAe,EAAE,EAAE;QACnD,eAAe,CAAC,CAAC,IAAI,EAAE,EAAE;YACvB,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;gBACpB,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,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,aAAa,GAAG,WAAW,CAC/B,CAAC,KAAkB,EAAE,EAAE;QACrB,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC;IAC9C,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,OAAiC,EAAE,EAAE;QACpC,MAAM,EACJ,IAAI,EACJ,WAAW,EACX,SAAS,EACT,eAAe,GAAG,EAAE,EACpB,gBAAgB,EAChB,eAAe,GAChB,GAAG,OAAO,CAAC;QAEZ,gBAAgB,CAAC,CAAC,IAAI,EAAE,EAAE;YACxB,IAAI,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;gBACf,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,EAAE,eAAe;oBACtB,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;IAEF,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,MAAM,CAAC,MAAM,gBAAgB,GAAG,GAAG,EAAE;IACnC,OAAO,UAAU,CAAC,oBAAoB,CAAC,CAAC;AAC1C,CAAC,CAAC","sourcesContent":["\"use client\";\nimport React, {\n  ComponentType,\n  createContext,\n  PropsWithChildren,\n  useCallback,\n  useContext,\n  useState,\n} from \"react\";\nimport { ComponentRegistry, TamboTool } from \"../model/component-metadata\";\n\nexport interface RegisterComponentOptions {\n  name: string;\n  description: string;\n  component: ComponentType<any>;\n  propsDefinition?: any;\n  loadingComponent?: ComponentType<any>;\n  associatedTools?: TamboTool[];\n}\n\nexport interface TamboRegistryContext {\n  componentList: ComponentRegistry;\n  toolRegistry: Record<string, TamboTool>;\n  componentToolAssociations: Record<string, string[]>;\n  registerComponent: (options: RegisterComponentOptions) => 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  registerComponent: () => {},\n  registerTool: () => {},\n  registerTools: () => {},\n  addToolAssociation: () => {},\n});\n\nexport const TamboRegistryProvider: React.FC<PropsWithChildren> = ({\n  children,\n}) => {\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((tool: TamboTool) => {\n    setToolRegistry((prev) => {\n      if (prev[tool.name]) {\n        console.warn(`Overwriting tool ${tool.name}`);\n      }\n      return {\n        ...prev,\n        [tool.name]: tool,\n      };\n    });\n  }, []);\n\n  const registerTools = useCallback(\n    (tools: TamboTool[]) => {\n      tools.forEach((tool) => registerTool(tool));\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: RegisterComponentOptions) => {\n      const {\n        name,\n        description,\n        component,\n        propsDefinition = {},\n        loadingComponent,\n        associatedTools,\n      } = options;\n\n      setComponentList((prev) => {\n        if (prev[name]) {\n          console.warn(`overwriting component ${name}`);\n        }\n        return {\n          ...prev,\n          [name]: {\n            component,\n            loadingComponent,\n            name,\n            description,\n            props: propsDefinition,\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\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\nexport const useTamboRegistry = () => {\n  return useContext(TamboRegistryContext);\n};\n"]}

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

@@ -0,0 +1,28 @@
+import TamboAI from "@tambo-ai/typescript-sdk";
+import React, { PropsWithChildren } from "react";
+import { GenerationStage, TamboThreadMessage } from "../model/generate-component-response";
+import { TamboThread } from "../model/tambo-thread";
+export interface TamboThreadContextProps {
+    thread: TamboThread;
+    switchCurrentThread: (threadId: string) => void;
+    addThreadMessage: (message: TamboThreadMessage, sendToServer: boolean) => Promise<TamboAI.Beta.Threads.ThreadMessage[]>;
+    updateThreadMessage: (id: string, message: TamboThreadMessage, sendToServer: boolean) => Promise<void>;
+    setLastThreadStatus: (status: GenerationStage) => void;
+    inputValue: string;
+    setInputValue: (value: string) => void;
+    sendThreadMessage: (message: string, options?: {
+        threadId?: string;
+        streamResponse?: boolean;
+        contextKey?: string;
+    }) => Promise<TamboThreadMessage>;
+}
+/** This is a stub entry for when the thread is not yet created, the first time
+ * the user sends a message
+ *
+ * Note that the consumer needs to be careful never to send `PLACEHOLDER_THREAD.id` to the server,
+ * as this doesn't really exist on the server side.
+ */
+export declare const PLACEHOLDER_THREAD: TamboThread;
+export declare const TamboThreadContext: React.Context<TamboThreadContextProps>;
+export declare const TamboThreadProvider: React.FC<PropsWithChildren>;
+export declare const useTamboThread: () => TamboThreadContextProps;