@assistant-ui/react 0.12.7 → 0.12.8

package/dist/client/ChainOfThoughtClient.d.ts

@@ -0,0 +1,15 @@
+import { type ClientOutput } from "@assistant-ui/store";
+import type { ChainOfThoughtPart } from "../types/scopes/chainOfThought.js";
+import type { PartMethods } from "../types/scopes/part.js";
+export declare const ChainOfThoughtClient: (props: {
+    parts: readonly ChainOfThoughtPart[];
+    getMessagePart: (selector: {
+        index: number;
+    }) => PartMethods;
+}) => import("@assistant-ui/tap").ResourceElement<ClientOutput<"chainOfThought">, {
+    parts: readonly ChainOfThoughtPart[];
+    getMessagePart: (selector: {
+        index: number;
+    }) => PartMethods;
+}>;
+//# sourceMappingURL=ChainOfThoughtClient.d.ts.map

package/dist/client/ChainOfThoughtClient.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ChainOfThoughtClient.d.ts","sourceRoot":"","sources":["../../src/client/ChainOfThoughtClient.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,KAAK,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,KAAK,EAEV,kBAAkB,EACnB,0CAAuC;AAExC,OAAO,KAAK,EAAE,WAAW,EAAE,gCAA6B;AAMxD,eAAO,MAAM,oBAAoB;WAKtB,SAAS,kBAAkB,EAAE;oBACpB,CAAC,QAAQ,EAAE;QAAE,KAAK,EAAE,MAAM,CAAA;KAAE,KAAK,WAAW;;WADrD,SAAS,kBAAkB,EAAE;oBACpB,CAAC,QAAQ,EAAE;QAAE,KAAK,EAAE,MAAM,CAAA;KAAE,KAAK,WAAW;EAuB/D,CAAC"}

package/dist/client/ChainOfThoughtClient.js

@@ -0,0 +1,21 @@
+import { resource, tapMemo, tapState } from "@assistant-ui/tap";
+const COMPLETE_STATUS = Object.freeze({
+    type: "complete",
+});
+export const ChainOfThoughtClient = resource(({ parts, getMessagePart, }) => {
+    const [collapsed, setCollapsed] = tapState(true);
+    const status = tapMemo(() => {
+        const lastPart = parts[parts.length - 1];
+        return lastPart?.status ?? COMPLETE_STATUS;
+    }, [parts]);
+    const state = tapMemo(() => ({ parts, collapsed, status }), [parts, collapsed, status]);
+    return {
+        state,
+        methods: {
+            getState: () => state,
+            setCollapsed,
+            part: getMessagePart,
+        },
+    };
+});
+//# sourceMappingURL=ChainOfThoughtClient.js.map

package/dist/client/ChainOfThoughtClient.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ChainOfThoughtClient.js","sourceRoot":"","sources":["../../src/client/ChainOfThoughtClient.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAShE,MAAM,eAAe,GAAsB,MAAM,CAAC,MAAM,CAAC;IACvD,IAAI,EAAE,UAAU;CACjB,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,oBAAoB,GAAG,QAAQ,CAC1C,CAAC,EACC,KAAK,EACL,cAAc,GAIf,EAAkC,EAAE;IACnC,MAAM,CAAC,SAAS,EAAE,YAAY,CAAC,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;IAEjD,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,EAAE;QAC1B,MAAM,QAAQ,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACzC,OAAO,QAAQ,EAAE,MAAM,IAAI,eAAe,CAAC;IAC7C,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC;IAEZ,MAAM,KAAK,GAAG,OAAO,CACnB,GAAG,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC,EACpC,CAAC,KAAK,EAAE,SAAS,EAAE,MAAM,CAAC,CAC3B,CAAC;IAEF,OAAO;QACL,KAAK;QACL,OAAO,EAAE;YACP,QAAQ,EAAE,GAAG,EAAE,CAAC,KAAK;YACrB,YAAY;YACZ,IAAI,EAAE,cAAc;SACrB;KACF,CAAC;AACJ,CAAC,CACF,CAAC"}

package/dist/client/index.d.ts

@@ -3,4 +3,5 @@ export { Tools } from "./Tools.js";
 export { Suggestions, type SuggestionConfig } from "./Suggestions.js";
 export { ExternalThread, type ExternalThreadProps, type ExternalThreadMessage, } from "./ExternalThread.js";
 export { InMemoryThreadList, type InMemoryThreadListProps, } from "./InMemoryThreadList.js";
+export { ChainOfThoughtClient } from "./ChainOfThoughtClient.js";
 //# sourceMappingURL=index.d.ts.map

package/dist/client/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/client/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,IAAI,kBAAkB,EAAE,gCAA6B;AAC1E,OAAO,EAAE,KAAK,EAAE,mBAAgB;AAChC,OAAO,EAAE,WAAW,EAAE,KAAK,gBAAgB,EAAE,yBAAsB;AACnE,OAAO,EACL,cAAc,EACd,KAAK,mBAAmB,EACxB,KAAK,qBAAqB,GAC3B,4BAAyB;AAC1B,OAAO,EACL,kBAAkB,EAClB,KAAK,uBAAuB,GAC7B,gCAA6B"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/client/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,IAAI,kBAAkB,EAAE,gCAA6B;AAC1E,OAAO,EAAE,KAAK,EAAE,mBAAgB;AAChC,OAAO,EAAE,WAAW,EAAE,KAAK,gBAAgB,EAAE,yBAAsB;AACnE,OAAO,EACL,cAAc,EACd,KAAK,mBAAmB,EACxB,KAAK,qBAAqB,GAC3B,4BAAyB;AAC1B,OAAO,EACL,kBAAkB,EAClB,KAAK,uBAAuB,GAC7B,gCAA6B;AAC9B,OAAO,EAAE,oBAAoB,EAAE,kCAA+B"}

package/dist/client/index.js

@@ -3,4 +3,5 @@ export { Tools } from "./Tools.js";
 export { Suggestions } from "./Suggestions.js";
 export { ExternalThread, } from "./ExternalThread.js";
 export { InMemoryThreadList, } from "./InMemoryThreadList.js";
+export { ChainOfThoughtClient } from "./ChainOfThoughtClient.js";
 //# sourceMappingURL=index.js.map

package/dist/client/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/client/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,IAAI,kBAAkB,EAAE,gCAA6B;AAC1E,OAAO,EAAE,KAAK,EAAE,mBAAgB;AAChC,OAAO,EAAE,WAAW,EAAyB,yBAAsB;AACnE,OAAO,EACL,cAAc,GAGf,4BAAyB;AAC1B,OAAO,EACL,kBAAkB,GAEnB,gCAA6B"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/client/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,IAAI,kBAAkB,EAAE,gCAA6B;AAC1E,OAAO,EAAE,KAAK,EAAE,mBAAgB;AAChC,OAAO,EAAE,WAAW,EAAyB,yBAAsB;AACnE,OAAO,EACL,cAAc,GAGf,4BAAyB;AAC1B,OAAO,EACL,kBAAkB,GAEnB,gCAA6B;AAC9B,OAAO,EAAE,oBAAoB,EAAE,kCAA+B"}

package/dist/context/providers/ChainOfThoughtByIndicesProvider.d.ts

@@ -0,0 +1,6 @@
+import { type FC, type PropsWithChildren } from "react";
+export declare const ChainOfThoughtByIndicesProvider: FC<PropsWithChildren<{
+    startIndex: number;
+    endIndex: number;
+}>>;
+//# sourceMappingURL=ChainOfThoughtByIndicesProvider.d.ts.map

package/dist/context/providers/ChainOfThoughtByIndicesProvider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ChainOfThoughtByIndicesProvider.d.ts","sourceRoot":"","sources":["../../../src/context/providers/ChainOfThoughtByIndicesProvider.tsx"],"names":[],"mappings":"AAEA,OAAO,EAAE,KAAK,EAAE,EAAE,KAAK,iBAAiB,EAAE,MAAM,OAAO,CAAC;AAKxD,eAAO,MAAM,+BAA+B,EAAE,EAAE,CAC9C,iBAAiB,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;CAClB,CAAC,CAuBH,CAAC"}

package/dist/context/providers/ChainOfThoughtByIndicesProvider.js

@@ -0,0 +1,21 @@
+"use client";
+import { jsx as _jsx } from "react/jsx-runtime";
+import { useAui, useAuiState, AuiProvider } from "@assistant-ui/store";
+import { ChainOfThoughtClient } from "../../client/ChainOfThoughtClient.js";
+export const ChainOfThoughtByIndicesProvider = ({ startIndex, endIndex, children }) => {
+    const parts = useAuiState(({ message }) => message.parts.slice(startIndex, endIndex + 1));
+    const parentAui = useAui();
+    const aui = useAui({
+        chainOfThought: ChainOfThoughtClient({
+            parts,
+            getMessagePart: ({ index }) => {
+                if (index < 0 || index >= parts.length) {
+                    throw new Error(`ChainOfThought part index ${index} is out of bounds (0..${parts.length - 1})`);
+                }
+                return parentAui.message().part({ index: startIndex + index });
+            },
+        }),
+    });
+    return _jsx(AuiProvider, { value: aui, children: children });
+};
+//# sourceMappingURL=ChainOfThoughtByIndicesProvider.js.map

package/dist/context/providers/ChainOfThoughtByIndicesProvider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ChainOfThoughtByIndicesProvider.js","sourceRoot":"","sources":["../../../src/context/providers/ChainOfThoughtByIndicesProvider.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;;AAGb,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AACvE,OAAO,EAAE,oBAAoB,EAAE,6CAA0C;AAGzE,MAAM,CAAC,MAAM,+BAA+B,GAKxC,CAAC,EAAE,UAAU,EAAE,QAAQ,EAAE,QAAQ,EAAE,EAAE,EAAE;IACzC,MAAM,KAAK,GAAG,WAAW,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CACxC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,UAAU,EAAE,QAAQ,GAAG,CAAC,CAAC,CACtB,CAAC;IAE1B,MAAM,SAAS,GAAG,MAAM,EAAE,CAAC;IAE3B,MAAM,GAAG,GAAG,MAAM,CAAC;QACjB,cAAc,EAAE,oBAAoB,CAAC;YACnC,KAAK;YACL,cAAc,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE;gBAC5B,IAAI,KAAK,GAAG,CAAC,IAAI,KAAK,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;oBACvC,MAAM,IAAI,KAAK,CACb,6BAA6B,KAAK,yBAAyB,KAAK,CAAC,MAAM,GAAG,CAAC,GAAG,CAC/E,CAAC;gBACJ,CAAC;gBACD,OAAO,SAAS,CAAC,OAAO,EAAE,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,UAAU,GAAG,KAAK,EAAE,CAAC,CAAC;YACjE,CAAC;SACF,CAAC;KACH,CAAC,CAAC;IAEH,OAAO,KAAC,WAAW,IAAC,KAAK,EAAE,GAAG,YAAG,QAAQ,GAAe,CAAC;AAC3D,CAAC,CAAC"}

package/dist/context/providers/ChainOfThoughtPartByIndexProvider.d.ts

@@ -0,0 +1,5 @@
+import { type FC, type PropsWithChildren } from "react";
+export declare const ChainOfThoughtPartByIndexProvider: FC<PropsWithChildren<{
+    index: number;
+}>>;
+//# sourceMappingURL=ChainOfThoughtPartByIndexProvider.d.ts.map

package/dist/context/providers/ChainOfThoughtPartByIndexProvider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ChainOfThoughtPartByIndexProvider.d.ts","sourceRoot":"","sources":["../../../src/context/providers/ChainOfThoughtPartByIndexProvider.tsx"],"names":[],"mappings":"AAEA,OAAO,EAAE,KAAK,EAAE,EAAE,KAAK,iBAAiB,EAAE,MAAM,OAAO,CAAC;AAGxD,eAAO,MAAM,iCAAiC,EAAE,EAAE,CAChD,iBAAiB,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;CACf,CAAC,CAWH,CAAC"}

package/dist/context/providers/ChainOfThoughtPartByIndexProvider.js

@@ -0,0 +1,14 @@
+"use client";
+import { jsx as _jsx } from "react/jsx-runtime";
+import { useAui, AuiProvider, Derived } from "@assistant-ui/store";
+export const ChainOfThoughtPartByIndexProvider = ({ index, children }) => {
+    const aui = useAui({
+        part: Derived({
+            source: "chainOfThought",
+            query: { type: "index", index },
+            get: (aui) => aui.chainOfThought().part({ index }),
+        }),
+    });
+    return _jsx(AuiProvider, { value: aui, children: children });
+};
+//# sourceMappingURL=ChainOfThoughtPartByIndexProvider.js.map

package/dist/context/providers/ChainOfThoughtPartByIndexProvider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ChainOfThoughtPartByIndexProvider.js","sourceRoot":"","sources":["../../../src/context/providers/ChainOfThoughtPartByIndexProvider.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;;AAGb,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAEnE,MAAM,CAAC,MAAM,iCAAiC,GAI1C,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,EAAE;IAC1B,MAAM,GAAG,GAAG,MAAM,CAAC;QACjB,IAAI,EAAE,OAAO,CAAC;YACZ,MAAM,EAAE,gBAAgB;YACxB,KAAK,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE;YAC/B,GAAG,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,CAAC;SACnD,CAAC;KACH,CAAC,CAAC;IAEH,OAAO,KAAC,WAAW,IAAC,KAAK,EAAE,GAAG,YAAG,QAAQ,GAAe,CAAC;AAC3D,CAAC,CAAC"}

package/dist/primitives/chainOfThought/ChainOfThoughtAccordionTrigger.d.ts

@@ -0,0 +1,28 @@
+import { ActionButtonElement, ActionButtonProps } from "../../utils/createActionButton.js";
+declare const useChainOfThoughtAccordionTrigger: () => () => void;
+export declare namespace ChainOfThoughtPrimitiveAccordionTrigger {
+    type Element = ActionButtonElement;
+    /**
+     * Props for the ChainOfThoughtPrimitive.AccordionTrigger component.
+     * Inherits all button element props and action button functionality.
+     */
+    type Props = ActionButtonProps<typeof useChainOfThoughtAccordionTrigger>;
+}
+/**
+ * A button component that toggles the collapsed state of the chain of thought accordion.
+ *
+ * This component automatically handles the toggle functionality, expanding or collapsing
+ * the chain of thought parts when clicked.
+ *
+ * @example
+ * ```tsx
+ * <ChainOfThoughtPrimitive.AccordionTrigger>
+ *   Toggle Reasoning
+ * </ChainOfThoughtPrimitive.AccordionTrigger>
+ * ```
+ */
+export declare const ChainOfThoughtPrimitiveAccordionTrigger: import("react").ForwardRefExoticComponent<Omit<import("react").ClassAttributes<HTMLButtonElement> & import("react").ButtonHTMLAttributes<HTMLButtonElement> & {
+    asChild?: boolean;
+}, "ref"> & import("react").RefAttributes<HTMLButtonElement>>;
+export {};
+//# sourceMappingURL=ChainOfThoughtAccordionTrigger.d.ts.map

package/dist/primitives/chainOfThought/ChainOfThoughtAccordionTrigger.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ChainOfThoughtAccordionTrigger.d.ts","sourceRoot":"","sources":["../../../src/primitives/chainOfThought/ChainOfThoughtAccordionTrigger.tsx"],"names":[],"mappings":"AAEA,OAAO,EACL,mBAAmB,EACnB,iBAAiB,EAElB,0CAAuC;AAIxC,QAAA,MAAM,iCAAiC,kBAWtC,CAAC;AAEF,yBAAiB,uCAAuC,CAAC;IACvD,KAAY,OAAO,GAAG,mBAAmB,CAAC;IAC1C;;;OAGG;IACH,KAAY,KAAK,GAAG,iBAAiB,CACnC,OAAO,iCAAiC,CACzC,CAAC;CACH;AAED;;;;;;;;;;;;GAYG;AACH,eAAO,MAAM,uCAAuC;;6DAGnD,CAAC"}

package/dist/primitives/chainOfThought/ChainOfThoughtAccordionTrigger.js

@@ -0,0 +1,27 @@
+"use client";
+import { createActionButton, } from "../../utils/createActionButton.js";
+import { useCallback } from "react";
+import { useAuiState, useAui } from "@assistant-ui/store";
+const useChainOfThoughtAccordionTrigger = () => {
+    const aui = useAui();
+    const collapsed = useAuiState(({ chainOfThought }) => chainOfThought.collapsed);
+    const callback = useCallback(() => {
+        aui.chainOfThought().setCollapsed(!collapsed);
+    }, [aui, collapsed]);
+    return callback;
+};
+/**
+ * A button component that toggles the collapsed state of the chain of thought accordion.
+ *
+ * This component automatically handles the toggle functionality, expanding or collapsing
+ * the chain of thought parts when clicked.
+ *
+ * @example
+ * ```tsx
+ * <ChainOfThoughtPrimitive.AccordionTrigger>
+ *   Toggle Reasoning
+ * </ChainOfThoughtPrimitive.AccordionTrigger>
+ * ```
+ */
+export const ChainOfThoughtPrimitiveAccordionTrigger = createActionButton("ChainOfThoughtPrimitive.AccordionTrigger", useChainOfThoughtAccordionTrigger);
+//# sourceMappingURL=ChainOfThoughtAccordionTrigger.js.map

package/dist/primitives/chainOfThought/ChainOfThoughtAccordionTrigger.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ChainOfThoughtAccordionTrigger.js","sourceRoot":"","sources":["../../../src/primitives/chainOfThought/ChainOfThoughtAccordionTrigger.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;AAEb,OAAO,EAGL,kBAAkB,GACnB,0CAAuC;AACxC,OAAO,EAAE,WAAW,EAAE,MAAM,OAAO,CAAC;AACpC,OAAO,EAAE,WAAW,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAE1D,MAAM,iCAAiC,GAAG,GAAG,EAAE;IAC7C,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC;IACrB,MAAM,SAAS,GAAG,WAAW,CAC3B,CAAC,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC,cAAc,CAAC,SAAS,CACjD,CAAC;IAEF,MAAM,QAAQ,GAAG,WAAW,CAAC,GAAG,EAAE;QAChC,GAAG,CAAC,cAAc,EAAE,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,CAAC;IAChD,CAAC,EAAE,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC,CAAC;IAErB,OAAO,QAAQ,CAAC;AAClB,CAAC,CAAC;AAaF;;;;;;;;;;;;GAYG;AACH,MAAM,CAAC,MAAM,uCAAuC,GAAG,kBAAkB,CACvE,0CAA0C,EAC1C,iCAAiC,CAClC,CAAC"}

package/dist/primitives/chainOfThought/ChainOfThoughtParts.d.ts

@@ -0,0 +1,37 @@
+import { type FC } from "react";
+import type { ReasoningMessagePartComponent, ToolCallMessagePartComponent } from "../../types/MessagePartComponentTypes.js";
+export declare namespace ChainOfThoughtPrimitiveParts {
+    type Props = {
+        /**
+         * Component configuration for rendering chain of thought parts.
+         */
+        components?: {
+            /** Component for rendering reasoning parts */
+            Reasoning?: ReasoningMessagePartComponent | undefined;
+            /** Fallback component for tool-call parts */
+            tools?: {
+                Fallback?: ToolCallMessagePartComponent | undefined;
+            };
+        };
+    };
+}
+/**
+ * Renders the parts within a chain of thought, with support for collapsed/expanded states.
+ *
+ * When collapsed, no parts are shown. When expanded, all parts are rendered
+ * using the provided component configuration through the part scope mechanism.
+ *
+ * @example
+ * ```tsx
+ * <ChainOfThoughtPrimitive.Parts
+ *   components={{
+ *     Reasoning: ({ text }) => <p className="reasoning">{text}</p>,
+ *     tools: {
+ *       Fallback: ({ toolName }) => <div>Tool: {toolName}</div>
+ *     }
+ *   }}
+ * />
+ * ```
+ */
+export declare const ChainOfThoughtPrimitiveParts: FC<ChainOfThoughtPrimitiveParts.Props>;
+//# sourceMappingURL=ChainOfThoughtParts.d.ts.map

package/dist/primitives/chainOfThought/ChainOfThoughtParts.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ChainOfThoughtParts.d.ts","sourceRoot":"","sources":["../../../src/primitives/chainOfThought/ChainOfThoughtParts.tsx"],"names":[],"mappings":"AAEA,OAAO,EAAE,KAAK,EAAE,EAAW,MAAM,OAAO,CAAC;AAIzC,OAAO,KAAK,EACV,6BAA6B,EAC7B,4BAA4B,EAC7B,iDAA8C;AAE/C,yBAAiB,4BAA4B,CAAC;IAC5C,KAAY,KAAK,GAAG;QAClB;;WAEG;QACH,UAAU,CAAC,EAAE;YACX,8CAA8C;YAC9C,SAAS,CAAC,EAAE,6BAA6B,GAAG,SAAS,CAAC;YACtD,6CAA6C;YAC7C,KAAK,CAAC,EAAE;gBACN,QAAQ,CAAC,EAAE,4BAA4B,GAAG,SAAS,CAAC;aACrD,CAAC;SACH,CAAC;KACH,CAAC;CACH;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,eAAO,MAAM,4BAA4B,EAAE,EAAE,CAC3C,4BAA4B,CAAC,KAAK,CA8BnC,CAAC"}

package/dist/primitives/chainOfThought/ChainOfThoughtParts.js

@@ -0,0 +1,42 @@
+"use client";
+import { jsx as _jsx, Fragment as _Fragment } from "react/jsx-runtime";
+import { useMemo } from "react";
+import { useAuiState } from "@assistant-ui/store";
+import { ChainOfThoughtPartByIndexProvider } from "../../context/providers/ChainOfThoughtPartByIndexProvider.js";
+import { MessagePartComponent } from "../message/MessageParts.js";
+/**
+ * Renders the parts within a chain of thought, with support for collapsed/expanded states.
+ *
+ * When collapsed, no parts are shown. When expanded, all parts are rendered
+ * using the provided component configuration through the part scope mechanism.
+ *
+ * @example
+ * ```tsx
+ * <ChainOfThoughtPrimitive.Parts
+ *   components={{
+ *     Reasoning: ({ text }) => <p className="reasoning">{text}</p>,
+ *     tools: {
+ *       Fallback: ({ toolName }) => <div>Tool: {toolName}</div>
+ *     }
+ *   }}
+ * />
+ * ```
+ */
+export const ChainOfThoughtPrimitiveParts = ({ components }) => {
+    const collapsed = useAuiState(({ chainOfThought }) => chainOfThought.collapsed);
+    const partsLength = useAuiState(({ chainOfThought }) => chainOfThought.parts.length);
+    const messageComponents = useMemo(() => ({
+        Reasoning: components?.Reasoning,
+        tools: {
+            Fallback: components?.tools?.Fallback,
+        },
+    }), [components?.Reasoning, components?.tools?.Fallback]);
+    const elements = useMemo(() => {
+        if (collapsed)
+            return null;
+        return Array.from({ length: partsLength }, (_, index) => (_jsx(ChainOfThoughtPartByIndexProvider, { index: index, children: _jsx(MessagePartComponent, { components: messageComponents }) }, index)));
+    }, [collapsed, partsLength, messageComponents]);
+    return _jsx(_Fragment, { children: elements });
+};
+ChainOfThoughtPrimitiveParts.displayName = "ChainOfThoughtPrimitive.Parts";
+//# sourceMappingURL=ChainOfThoughtParts.js.map

package/dist/primitives/chainOfThought/ChainOfThoughtParts.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ChainOfThoughtParts.js","sourceRoot":"","sources":["../../../src/primitives/chainOfThought/ChainOfThoughtParts.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;;AAEb,OAAO,EAAW,OAAO,EAAE,MAAM,OAAO,CAAC;AACzC,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAClD,OAAO,EAAE,iCAAiC,EAAE,qEAAkE;AAC9G,OAAO,EAAE,oBAAoB,EAAE,mCAAgC;AAsB/D;;;;;;;;;;;;;;;;;GAiBG;AACH,MAAM,CAAC,MAAM,4BAA4B,GAErC,CAAC,EAAE,UAAU,EAAE,EAAE,EAAE;IACrB,MAAM,SAAS,GAAG,WAAW,CAC3B,CAAC,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC,cAAc,CAAC,SAAS,CACjD,CAAC;IACF,MAAM,WAAW,GAAG,WAAW,CAC7B,CAAC,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC,cAAc,CAAC,KAAK,CAAC,MAAM,CACpD,CAAC;IAEF,MAAM,iBAAiB,GAAG,OAAO,CAC/B,GAAG,EAAE,CAAC,CAAC;QACL,SAAS,EAAE,UAAU,EAAE,SAAS;QAChC,KAAK,EAAE;YACL,QAAQ,EAAE,UAAU,EAAE,KAAK,EAAE,QAAQ;SACtC;KACF,CAAC,EACF,CAAC,UAAU,EAAE,SAAS,EAAE,UAAU,EAAE,KAAK,EAAE,QAAQ,CAAC,CACrD,CAAC;IAEF,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,EAAE;QAC5B,IAAI,SAAS;YAAE,OAAO,IAAI,CAAC;QAE3B,OAAO,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC,CACvD,KAAC,iCAAiC,IAAa,KAAK,EAAE,KAAK,YACzD,KAAC,oBAAoB,IAAC,UAAU,EAAE,iBAAiB,GAAI,IADjB,KAAK,CAET,CACrC,CAAC,CAAC;IACL,CAAC,EAAE,CAAC,SAAS,EAAE,WAAW,EAAE,iBAAiB,CAAC,CAAC,CAAC;IAEhD,OAAO,4BAAG,QAAQ,GAAI,CAAC;AACzB,CAAC,CAAC;AAEF,4BAA4B,CAAC,WAAW,GAAG,+BAA+B,CAAC"}

package/dist/primitives/chainOfThought/ChainOfThoughtRoot.d.ts

@@ -0,0 +1,28 @@
+import { Primitive } from "@radix-ui/react-primitive";
+import { type ComponentRef, ComponentPropsWithoutRef } from "react";
+type PrimitiveDivProps = ComponentPropsWithoutRef<typeof Primitive.div>;
+export declare namespace ChainOfThoughtPrimitiveRoot {
+    type Element = ComponentRef<typeof Primitive.div>;
+    type Props = PrimitiveDivProps;
+}
+/**
+ * The root container for chain of thought components.
+ *
+ * This component provides a wrapper for chain of thought content,
+ * including reasoning and tool-call parts that can be collapsed in an accordion.
+ *
+ * @example
+ * ```tsx
+ * <ChainOfThoughtPrimitive.Root>
+ *   <ChainOfThoughtPrimitive.AccordionTrigger>
+ *     Toggle reasoning
+ *   </ChainOfThoughtPrimitive.AccordionTrigger>
+ *   <ChainOfThoughtPrimitive.Parts />
+ * </ChainOfThoughtPrimitive.Root>
+ * ```
+ */
+export declare const ChainOfThoughtPrimitiveRoot: import("react").ForwardRefExoticComponent<Omit<import("react").ClassAttributes<HTMLDivElement> & import("react").HTMLAttributes<HTMLDivElement> & {
+    asChild?: boolean;
+}, "ref"> & import("react").RefAttributes<HTMLDivElement>>;
+export {};
+//# sourceMappingURL=ChainOfThoughtRoot.d.ts.map

package/dist/primitives/chainOfThought/ChainOfThoughtRoot.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ChainOfThoughtRoot.d.ts","sourceRoot":"","sources":["../../../src/primitives/chainOfThought/ChainOfThoughtRoot.tsx"],"names":[],"mappings":"AAEA,OAAO,EAAE,SAAS,EAAE,MAAM,2BAA2B,CAAC;AACtD,OAAO,EAAE,KAAK,YAAY,EAAc,wBAAwB,EAAE,MAAM,OAAO,CAAC;AAEhF,KAAK,iBAAiB,GAAG,wBAAwB,CAAC,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC;AAExE,yBAAiB,2BAA2B,CAAC;IAC3C,KAAY,OAAO,GAAG,YAAY,CAAC,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC;IACzD,KAAY,KAAK,GAAG,iBAAiB,CAAC;CACvC;AAED;;;;;;;;;;;;;;;GAeG;AACH,eAAO,MAAM,2BAA2B;;0DAKtC,CAAC"}

package/dist/primitives/chainOfThought/ChainOfThoughtRoot.js

@@ -0,0 +1,25 @@
+"use client";
+import { jsx as _jsx } from "react/jsx-runtime";
+import { Primitive } from "@radix-ui/react-primitive";
+import { forwardRef } from "react";
+/**
+ * The root container for chain of thought components.
+ *
+ * This component provides a wrapper for chain of thought content,
+ * including reasoning and tool-call parts that can be collapsed in an accordion.
+ *
+ * @example
+ * ```tsx
+ * <ChainOfThoughtPrimitive.Root>
+ *   <ChainOfThoughtPrimitive.AccordionTrigger>
+ *     Toggle reasoning
+ *   </ChainOfThoughtPrimitive.AccordionTrigger>
+ *   <ChainOfThoughtPrimitive.Parts />
+ * </ChainOfThoughtPrimitive.Root>
+ * ```
+ */
+export const ChainOfThoughtPrimitiveRoot = forwardRef((props, ref) => {
+    return _jsx(Primitive.div, { ...props, ref: ref });
+});
+ChainOfThoughtPrimitiveRoot.displayName = "ChainOfThoughtPrimitive.Root";
+//# sourceMappingURL=ChainOfThoughtRoot.js.map

package/dist/primitives/chainOfThought/ChainOfThoughtRoot.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ChainOfThoughtRoot.js","sourceRoot":"","sources":["../../../src/primitives/chainOfThought/ChainOfThoughtRoot.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;;AAEb,OAAO,EAAE,SAAS,EAAE,MAAM,2BAA2B,CAAC;AACtD,OAAO,EAAqB,UAAU,EAA4B,MAAM,OAAO,CAAC;AAShF;;;;;;;;;;;;;;;GAeG;AACH,MAAM,CAAC,MAAM,2BAA2B,GAAG,UAAU,CAGnD,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;IACf,OAAO,KAAC,SAAS,CAAC,GAAG,OAAK,KAAK,EAAE,GAAG,EAAE,GAAG,GAAI,CAAC;AAChD,CAAC,CAAC,CAAC;AAEH,2BAA2B,CAAC,WAAW,GAAG,8BAA8B,CAAC"}

package/dist/primitives/chainOfThought/index.d.ts

@@ -0,0 +1,4 @@
+export { ChainOfThoughtPrimitiveRoot as Root, type ChainOfThoughtPrimitiveRoot as RootPrimitive, } from "./ChainOfThoughtRoot.js";
+export { ChainOfThoughtPrimitiveAccordionTrigger as AccordionTrigger, type ChainOfThoughtPrimitiveAccordionTrigger as AccordionTriggerPrimitive, } from "./ChainOfThoughtAccordionTrigger.js";
+export { ChainOfThoughtPrimitiveParts as Parts, type ChainOfThoughtPrimitiveParts as PartsPrimitive, } from "./ChainOfThoughtParts.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/primitives/chainOfThought/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/primitives/chainOfThought/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,2BAA2B,IAAI,IAAI,EACnC,KAAK,2BAA2B,IAAI,aAAa,GAClD,gCAA6B;AAE9B,OAAO,EACL,uCAAuC,IAAI,gBAAgB,EAC3D,KAAK,uCAAuC,IAAI,yBAAyB,GAC1E,4CAAyC;AAE1C,OAAO,EACL,4BAA4B,IAAI,KAAK,EACrC,KAAK,4BAA4B,IAAI,cAAc,GACpD,iCAA8B"}

package/dist/primitives/chainOfThought/index.js

@@ -0,0 +1,4 @@
+export { ChainOfThoughtPrimitiveRoot as Root, } from "./ChainOfThoughtRoot.js";
+export { ChainOfThoughtPrimitiveAccordionTrigger as AccordionTrigger, } from "./ChainOfThoughtAccordionTrigger.js";
+export { ChainOfThoughtPrimitiveParts as Parts, } from "./ChainOfThoughtParts.js";
+//# sourceMappingURL=index.js.map

package/dist/primitives/chainOfThought/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/primitives/chainOfThought/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,2BAA2B,IAAI,IAAI,GAEpC,gCAA6B;AAE9B,OAAO,EACL,uCAAuC,IAAI,gBAAgB,GAE5D,4CAAyC;AAE1C,OAAO,EACL,4BAA4B,IAAI,KAAK,GAEtC,iCAA8B"}

package/dist/primitives/index.d.ts

@@ -3,6 +3,7 @@ export * as ActionBarMorePrimitive from "./actionBarMore/index.js";
 export * as AssistantModalPrimitive from "./assistantModal/index.js";
 export * as AttachmentPrimitive from "./attachment/index.js";
 export * as BranchPickerPrimitive from "./branchPicker/index.js";
+export * as ChainOfThoughtPrimitive from "./chainOfThought/index.js";
 export * as ComposerPrimitive from "./composer/index.js";
 export * as MessagePartPrimitive from "./messagePart/index.js";
 export * as ErrorPrimitive from "./error/index.js";

package/dist/primitives/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/primitives/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,kBAAkB,6BAAoB;AAClD,OAAO,KAAK,sBAAsB,iCAAwB;AAC1D,OAAO,KAAK,uBAAuB,kCAAyB;AAC5D,OAAO,KAAK,mBAAmB,8BAAqB;AACpD,OAAO,KAAK,qBAAqB,gCAAuB;AACxD,OAAO,KAAK,iBAAiB,4BAAmB;AAChD,OAAO,KAAK,oBAAoB,+BAAsB;AACtD,OAAO,KAAK,cAAc,yBAAgB;AAC1C,OAAO,KAAK,gBAAgB,2BAAkB;AAC9C,OAAO,KAAK,eAAe,0BAAiB;AAC5C,OAAO,KAAK,mBAAmB,8BAAqB;AACpD,OAAO,KAAK,mBAAmB,8BAAqB;AACpD,OAAO,KAAK,uBAAuB,kCAAyB;AAC5D,OAAO,KAAK,2BAA2B,sCAA6B;AAEpE,OAAO,EAAE,kBAAkB,EAAE,4CAAyC;AACtE,OAAO,EAAE,uBAAuB,EAAE,iDAA8C;AAChF,OAAO,EAAE,oBAAoB,EAAE,8CAA2C;AAC1E,OAAO,EAAE,kBAAkB,EAAE,4CAAyC;AACtE,OAAO,EAAE,mBAAmB,EAAE,6CAA0C;AACxE,OAAO,EAAE,kBAAkB,EAAE,4CAAyC;AACtE,OAAO,EAAE,2BAA2B,EAAE,gDAA6C;AACnF,OAAO,EAAE,aAAa,EAAE,6BAAoB"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/primitives/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,kBAAkB,6BAAoB;AAClD,OAAO,KAAK,sBAAsB,iCAAwB;AAC1D,OAAO,KAAK,uBAAuB,kCAAyB;AAC5D,OAAO,KAAK,mBAAmB,8BAAqB;AACpD,OAAO,KAAK,qBAAqB,gCAAuB;AACxD,OAAO,KAAK,uBAAuB,kCAAyB;AAC5D,OAAO,KAAK,iBAAiB,4BAAmB;AAChD,OAAO,KAAK,oBAAoB,+BAAsB;AACtD,OAAO,KAAK,cAAc,yBAAgB;AAC1C,OAAO,KAAK,gBAAgB,2BAAkB;AAC9C,OAAO,KAAK,eAAe,0BAAiB;AAC5C,OAAO,KAAK,mBAAmB,8BAAqB;AACpD,OAAO,KAAK,mBAAmB,8BAAqB;AACpD,OAAO,KAAK,uBAAuB,kCAAyB;AAC5D,OAAO,KAAK,2BAA2B,sCAA6B;AAEpE,OAAO,EAAE,kBAAkB,EAAE,4CAAyC;AACtE,OAAO,EAAE,uBAAuB,EAAE,iDAA8C;AAChF,OAAO,EAAE,oBAAoB,EAAE,8CAA2C;AAC1E,OAAO,EAAE,kBAAkB,EAAE,4CAAyC;AACtE,OAAO,EAAE,mBAAmB,EAAE,6CAA0C;AACxE,OAAO,EAAE,kBAAkB,EAAE,4CAAyC;AACtE,OAAO,EAAE,2BAA2B,EAAE,gDAA6C;AACnF,OAAO,EAAE,aAAa,EAAE,6BAAoB"}

package/dist/primitives/index.js

@@ -3,6 +3,7 @@ export * as ActionBarMorePrimitive from "./actionBarMore/index.js";
 export * as AssistantModalPrimitive from "./assistantModal/index.js";
 export * as AttachmentPrimitive from "./attachment/index.js";
 export * as BranchPickerPrimitive from "./branchPicker/index.js";
+export * as ChainOfThoughtPrimitive from "./chainOfThought/index.js";
 export * as ComposerPrimitive from "./composer/index.js";
 export * as MessagePartPrimitive from "./messagePart/index.js";
 export * as ErrorPrimitive from "./error/index.js";

package/dist/primitives/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/primitives/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,kBAAkB,6BAAoB;AAClD,OAAO,KAAK,sBAAsB,iCAAwB;AAC1D,OAAO,KAAK,uBAAuB,kCAAyB;AAC5D,OAAO,KAAK,mBAAmB,8BAAqB;AACpD,OAAO,KAAK,qBAAqB,gCAAuB;AACxD,OAAO,KAAK,iBAAiB,4BAAmB;AAChD,OAAO,KAAK,oBAAoB,+BAAsB;AACtD,OAAO,KAAK,cAAc,yBAAgB;AAC1C,OAAO,KAAK,gBAAgB,2BAAkB;AAC9C,OAAO,KAAK,eAAe,0BAAiB;AAC5C,OAAO,KAAK,mBAAmB,8BAAqB;AACpD,OAAO,KAAK,mBAAmB,8BAAqB;AACpD,OAAO,KAAK,uBAAuB,kCAAyB;AAC5D,OAAO,KAAK,2BAA2B,sCAA6B;AAEpE,OAAO,EAAE,kBAAkB,EAAE,4CAAyC;AACtE,OAAO,EAAE,uBAAuB,EAAE,iDAA8C;AAChF,OAAO,EAAE,oBAAoB,EAAE,8CAA2C;AAC1E,OAAO,EAAE,kBAAkB,EAAE,4CAAyC;AACtE,OAAO,EAAE,mBAAmB,EAAE,6CAA0C;AACxE,OAAO,EAAE,kBAAkB,EAAE,4CAAyC;AACtE,OAAO,EAAE,2BAA2B,EAAE,gDAA6C;AACnF,OAAO,EAAE,aAAa,EAAE,6BAAoB"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/primitives/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,kBAAkB,6BAAoB;AAClD,OAAO,KAAK,sBAAsB,iCAAwB;AAC1D,OAAO,KAAK,uBAAuB,kCAAyB;AAC5D,OAAO,KAAK,mBAAmB,8BAAqB;AACpD,OAAO,KAAK,qBAAqB,gCAAuB;AACxD,OAAO,KAAK,uBAAuB,kCAAyB;AAC5D,OAAO,KAAK,iBAAiB,4BAAmB;AAChD,OAAO,KAAK,oBAAoB,+BAAsB;AACtD,OAAO,KAAK,cAAc,yBAAgB;AAC1C,OAAO,KAAK,gBAAgB,2BAAkB;AAC9C,OAAO,KAAK,eAAe,0BAAiB;AAC5C,OAAO,KAAK,mBAAmB,8BAAqB;AACpD,OAAO,KAAK,mBAAmB,8BAAqB;AACpD,OAAO,KAAK,uBAAuB,kCAAyB;AAC5D,OAAO,KAAK,2BAA2B,sCAA6B;AAEpE,OAAO,EAAE,kBAAkB,EAAE,4CAAyC;AACtE,OAAO,EAAE,uBAAuB,EAAE,iDAA8C;AAChF,OAAO,EAAE,oBAAoB,EAAE,8CAA2C;AAC1E,OAAO,EAAE,kBAAkB,EAAE,4CAAyC;AACtE,OAAO,EAAE,mBAAmB,EAAE,6CAA0C;AACxE,OAAO,EAAE,kBAAkB,EAAE,4CAAyC;AACtE,OAAO,EAAE,2BAA2B,EAAE,gDAA6C;AACnF,OAAO,EAAE,aAAa,EAAE,6BAAoB"}

package/dist/primitives/message/MessageParts.d.ts

@@ -115,6 +115,30 @@ export declare namespace MessagePrimitiveParts {
              * @param children - Rendered reasoning part components
              */
             ReasoningGroup?: ReasoningGroupComponent;
+            /**
+             * Component for rendering chain of thought (reasoning + tool-call) parts.
+             *
+             * When provided, this component takes control of rendering ALL reasoning and
+             * tool-call parts in the message. The Reasoning, tools, ReasoningGroup, and
+             * ToolGroup components are completely ignored when ChainOfThought is set.
+             *
+             * The component is automatically wrapped in a ChainOfThoughtByIndicesProvider
+             * that sets up the chainOfThought client scope with the correct parts.
+             *
+             * @example
+             * ```tsx
+             * // Chain of thought with accordion
+             * ChainOfThought: () => (
+             *   <ChainOfThoughtPrimitive.Root>
+             *     <ChainOfThoughtPrimitive.AccordionTrigger>
+             *       Toggle reasoning
+             *     </ChainOfThoughtPrimitive.AccordionTrigger>
+             *     <ChainOfThoughtPrimitive.Parts />
+             *   </ChainOfThoughtPrimitive.Root>
+             * )
+             * ```
+             */
+            ChainOfThought?: ComponentType;
         } | undefined;
         /**
          * When enabled, shows the Empty component if the last part in the message
@@ -129,6 +153,10 @@ export declare namespace MessagePrimitiveParts {
         unstable_showEmptyOnNonTextEnd?: boolean | undefined;
     };
 }
+type MessagePartComponentProps = {
+    components: MessagePrimitiveParts.Props["components"];
+};
+export declare const MessagePartComponent: FC<MessagePartComponentProps>;
 export declare namespace MessagePrimitivePartByIndex {
     type Props = {
         index: number;
@@ -178,4 +206,5 @@ export declare const MessagePrimitivePartByIndex: FC<MessagePrimitivePartByIndex
  * ```
  */
 export declare const MessagePrimitiveParts: FC<MessagePrimitiveParts.Props>;
+export {};
 //# sourceMappingURL=MessageParts.d.ts.map

package/dist/primitives/message/MessageParts.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"MessageParts.d.ts","sourceRoot":"","sources":["../../../src/primitives/message/MessageParts.tsx"],"names":[],"mappings":"AAEA,OAAO,EACL,KAAK,aAAa,EAClB,KAAK,EAAE,EAEP,iBAAiB,EAElB,MAAM,OAAO,CAAC;AAQf,OAAO,KAAK,EACV,kCAAkC,EAClC,yBAAyB,EACzB,wBAAwB,EACxB,yBAAyB,EACzB,0BAA0B,EAC1B,4BAA4B,EAC5B,wBAAwB,EACxB,wBAAwB,EACxB,6BAA6B,EAC7B,uBAAuB,EACxB,iDAA8C;AA6F/C,yBAAiB,qBAAqB,CAAC;IACrC,KAAY,KAAK,GAAG;QAClB;;;;;WAKG;QACH,UAAU,CAAC,EACP;YACE,6CAA6C;YAC7C,KAAK,CAAC,EAAE,yBAAyB,GAAG,SAAS,CAAC;YAC9C,2CAA2C;YAC3C,IAAI,CAAC,EAAE,wBAAwB,GAAG,SAAS,CAAC;YAC5C,mEAAmE;YACnE,SAAS,CAAC,EAAE,6BAA6B,GAAG,SAAS,CAAC;YACtD,6CAA6C;YAC7C,MAAM,CAAC,EAAE,0BAA0B,GAAG,SAAS,CAAC;YAChD,4CAA4C;YAC5C,KAAK,CAAC,EAAE,yBAAyB,GAAG,SAAS,CAAC;YAC9C,2CAA2C;YAC3C,IAAI,CAAC,EAAE,wBAAwB,GAAG,SAAS,CAAC;YAC5C,2DAA2D;YAC3D,cAAc,CAAC,EAAE,kCAAkC,GAAG,SAAS,CAAC;YAChE,4CAA4C;YAC5C,KAAK,CAAC,EACF;gBACE,qDAAqD;gBACrD,OAAO,CAAC,EACJ,MAAM,CAAC,MAAM,EAAE,4BAA4B,GAAG,SAAS,CAAC,GACxD,SAAS,CAAC;gBACd,gDAAgD;gBAChD,QAAQ,CAAC,EAAE,aAAa,CAAC,wBAAwB,CAAC,GAAG,SAAS,CAAC;aAChE,GACD;gBACE,qDAAqD;gBACrD,QAAQ,EAAE,aAAa,CAAC,wBAAwB,CAAC,CAAC;aACnD,GACD,SAAS,CAAC;YAEd;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;eA+CG;YACH,SAAS,CAAC,EAAE,aAAa,CACvB,iBAAiB,CAAC;gBAAE,UAAU,EAAE,MAAM,CAAC;gBAAC,QAAQ,EAAE,MAAM,CAAA;aAAE,CAAC,CAC5D,CAAC;YAEF;;;;;;;;;;;;;;;;;;;;;;;;;;;;eA4BG;YACH,cAAc,CAAC,EAAE,uBAAuB,CAAC;SAC1C,GACD,SAAS,CAAC;QACd;;;;;;;;;WASG;QACH,8BAA8B,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;KACtD,CAAC;CACH;AAqGD,yBAAiB,2BAA2B,CAAC;IAC3C,KAAY,KAAK,GAAG;QAClB,KAAK,EAAE,MAAM,CAAC;QACd,UAAU,EAAE,qBAAqB,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;KACvD,CAAC;CACH;AAED;;;;;;;;;;;;;;;;GAgBG;AACH,eAAO,MAAM,2BAA2B,EAAE,EAAE,CAAC,2BAA2B,CAAC,KAAK,CAoB3E,CAAC;AAiEJ;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,eAAO,MAAM,qBAAqB,EAAE,EAAE,CAAC,qBAAqB,CAAC,KAAK,CA6EjE,CAAC"}
+{"version":3,"file":"MessageParts.d.ts","sourceRoot":"","sources":["../../../src/primitives/message/MessageParts.tsx"],"names":[],"mappings":"AAEA,OAAO,EACL,KAAK,aAAa,EAClB,KAAK,EAAE,EAEP,iBAAiB,EAElB,MAAM,OAAO,CAAC;AASf,OAAO,KAAK,EACV,kCAAkC,EAClC,yBAAyB,EACzB,wBAAwB,EACxB,yBAAyB,EACzB,0BAA0B,EAC1B,4BAA4B,EAC5B,wBAAwB,EACxB,wBAAwB,EACxB,6BAA6B,EAC7B,uBAAuB,EACxB,iDAA8C;AAsH/C,yBAAiB,qBAAqB,CAAC;IACrC,KAAY,KAAK,GAAG;QAClB;;;;;WAKG;QACH,UAAU,CAAC,EACP;YACE,6CAA6C;YAC7C,KAAK,CAAC,EAAE,yBAAyB,GAAG,SAAS,CAAC;YAC9C,2CAA2C;YAC3C,IAAI,CAAC,EAAE,wBAAwB,GAAG,SAAS,CAAC;YAC5C,mEAAmE;YACnE,SAAS,CAAC,EAAE,6BAA6B,GAAG,SAAS,CAAC;YACtD,6CAA6C;YAC7C,MAAM,CAAC,EAAE,0BAA0B,GAAG,SAAS,CAAC;YAChD,4CAA4C;YAC5C,KAAK,CAAC,EAAE,yBAAyB,GAAG,SAAS,CAAC;YAC9C,2CAA2C;YAC3C,IAAI,CAAC,EAAE,wBAAwB,GAAG,SAAS,CAAC;YAC5C,2DAA2D;YAC3D,cAAc,CAAC,EAAE,kCAAkC,GAAG,SAAS,CAAC;YAChE,4CAA4C;YAC5C,KAAK,CAAC,EACF;gBACE,qDAAqD;gBACrD,OAAO,CAAC,EACJ,MAAM,CAAC,MAAM,EAAE,4BAA4B,GAAG,SAAS,CAAC,GACxD,SAAS,CAAC;gBACd,gDAAgD;gBAChD,QAAQ,CAAC,EAAE,aAAa,CAAC,wBAAwB,CAAC,GAAG,SAAS,CAAC;aAChE,GACD;gBACE,qDAAqD;gBACrD,QAAQ,EAAE,aAAa,CAAC,wBAAwB,CAAC,CAAC;aACnD,GACD,SAAS,CAAC;YAEd;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;eA+CG;YACH,SAAS,CAAC,EAAE,aAAa,CACvB,iBAAiB,CAAC;gBAAE,UAAU,EAAE,MAAM,CAAC;gBAAC,QAAQ,EAAE,MAAM,CAAA;aAAE,CAAC,CAC5D,CAAC;YAEF;;;;;;;;;;;;;;;;;;;;;;;;;;;;eA4BG;YACH,cAAc,CAAC,EAAE,uBAAuB,CAAC;YAEzC;;;;;;;;;;;;;;;;;;;;;;eAsBG;YACH,cAAc,CAAC,EAAE,aAAa,CAAC;SAChC,GACD,SAAS,CAAC;QACd;;;;;;;;;WASG;QACH,8BAA8B,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;KACtD,CAAC;CACH;AAmCD,KAAK,yBAAyB,GAAG;IAC/B,UAAU,EAAE,qBAAqB,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;CACvD,CAAC;AAEF,eAAO,MAAM,oBAAoB,EAAE,EAAE,CAAC,yBAAyB,CA4D9D,CAAC;AAEF,yBAAiB,2BAA2B,CAAC;IAC3C,KAAY,KAAK,GAAG;QAClB,KAAK,EAAE,MAAM,CAAC;QACd,UAAU,EAAE,qBAAqB,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;KACvD,CAAC;CACH;AAED;;;;;;;;;;;;;;;;GAgBG;AACH,eAAO,MAAM,2BAA2B,EAAE,EAAE,CAAC,2BAA2B,CAAC,KAAK,CAoB3E,CAAC;AAiEJ;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,eAAO,MAAM,qBAAqB,EAAE,EAAE,CAAC,qBAAqB,CAAC,KAAK,CA0FjE,CAAC"}

package/dist/primitives/message/MessageParts.js

@@ -3,6 +3,7 @@ import { jsx as _jsx, jsxs as _jsxs, Fragment as _Fragment } from "react/jsx-run
 import { memo, useMemo, } from "react";
 import { useAuiState, useAui } from "@assistant-ui/store";
 import { PartByIndexProvider, TextMessagePartProvider, } from "../../context/providers/index.js";
+import { ChainOfThoughtByIndicesProvider } from "../../context/providers/ChainOfThoughtByIndicesProvider.js";
 import { MessagePartPrimitiveText } from "../messagePart/MessagePartText.js";
 import { MessagePartPrimitiveImage } from "../messagePart/MessagePartImage.js";
 import { MessagePartPrimitiveInProgress } from "../messagePart/MessagePartInProgress.js";
@@ -43,39 +44,56 @@ const createGroupState = (groupType) => {
 /**
  * Groups consecutive tool-call and reasoning message parts into ranges.
  * Always groups tool calls and reasoning parts, even if there's only one.
+ * When useChainOfThought is true, groups tool-call and reasoning parts together.
  */
-const groupMessageParts = (messageTypes) => {
+const groupMessageParts = (messageTypes, useChainOfThought) => {
     const ranges = [];
-    const toolGroup = createGroupState("toolGroup");
-    const reasoningGroup = createGroupState("reasoningGroup");
-    for (let i = 0; i < messageTypes.length; i++) {
-        const type = messageTypes[i];
-        if (type === "tool-call") {
-            reasoningGroup.endGroup(i - 1, ranges);
-            toolGroup.startGroup(i);
-        }
-        else if (type === "reasoning") {
-            toolGroup.endGroup(i - 1, ranges);
-            reasoningGroup.startGroup(i);
+    if (useChainOfThought) {
+        const chainOfThoughtGroup = createGroupState("chainOfThoughtGroup");
+        for (let i = 0; i < messageTypes.length; i++) {
+            const type = messageTypes[i];
+            if (type === "tool-call" || type === "reasoning") {
+                chainOfThoughtGroup.startGroup(i);
+            }
+            else {
+                chainOfThoughtGroup.endGroup(i - 1, ranges);
+                ranges.push({ type: "single", index: i });
+            }
         }
-        else {
-            toolGroup.endGroup(i - 1, ranges);
-            reasoningGroup.endGroup(i - 1, ranges);
-            ranges.push({ type: "single", index: i });
+        chainOfThoughtGroup.finalize(messageTypes.length - 1, ranges);
+    }
+    else {
+        const toolGroup = createGroupState("toolGroup");
+        const reasoningGroup = createGroupState("reasoningGroup");
+        for (let i = 0; i < messageTypes.length; i++) {
+            const type = messageTypes[i];
+            if (type === "tool-call") {
+                reasoningGroup.endGroup(i - 1, ranges);
+                toolGroup.startGroup(i);
+            }
+            else if (type === "reasoning") {
+                toolGroup.endGroup(i - 1, ranges);
+                reasoningGroup.startGroup(i);
+            }
+            else {
+                toolGroup.endGroup(i - 1, ranges);
+                reasoningGroup.endGroup(i - 1, ranges);
+                ranges.push({ type: "single", index: i });
+            }
         }
+        toolGroup.finalize(messageTypes.length - 1, ranges);
+        reasoningGroup.finalize(messageTypes.length - 1, ranges);
     }
-    toolGroup.finalize(messageTypes.length - 1, ranges);
-    reasoningGroup.finalize(messageTypes.length - 1, ranges);
     return ranges;
 };
-const useMessagePartsGroups = () => {
+const useMessagePartsGroups = (useChainOfThought) => {
     const messageTypes = useAuiState(useShallow((s) => s.message.parts.map((c) => c.type)));
     return useMemo(() => {
         if (messageTypes.length === 0) {
             return [];
         }
-        return groupMessageParts(messageTypes);
-    }, [messageTypes]);
+        return groupMessageParts(messageTypes, useChainOfThought);
+    }, [messageTypes, useChainOfThought]);
 };
 const ToolUIDisplay = ({ Fallback, ...props }) => {
     const Render = useAuiState(({ tools }) => {
@@ -98,7 +116,7 @@ const defaultComponents = {
     ToolGroup: ({ children }) => children,
     ReasoningGroup: ({ children }) => children,
 };
-const MessagePartComponent = ({ components: { Text = defaultComponents.Text, Reasoning = defaultComponents.Reasoning, Image = defaultComponents.Image, Source = defaultComponents.Source, File = defaultComponents.File, Unstable_Audio: Audio = defaultComponents.Unstable_Audio, tools = {}, } = {}, }) => {
+export const MessagePartComponent = ({ components: { Text = defaultComponents.Text, Reasoning = defaultComponents.Reasoning, Image = defaultComponents.Image, Source = defaultComponents.Source, File = defaultComponents.File, Unstable_Audio: Audio = defaultComponents.Unstable_Audio, tools = {}, } = {}, }) => {
     const aui = useAui();
     const part = useAuiState(({ part }) => part);
     const type = part.type;
@@ -218,7 +236,8 @@ const ConditionalEmpty = memo(ConditionalEmptyImpl, (prev, next) => prev.enabled
  */
 export const MessagePrimitiveParts = ({ components, unstable_showEmptyOnNonTextEnd = true, }) => {
     const contentLength = useAuiState(({ message }) => message.parts.length);
-    const messageRanges = useMessagePartsGroups();
+    const useChainOfThought = !!components?.ChainOfThought;
+    const messageRanges = useMessagePartsGroups(useChainOfThought);
     const partsElements = useMemo(() => {
         if (contentLength === 0) {
             return _jsx(EmptyParts, { components: components });
@@ -227,6 +246,12 @@ export const MessagePrimitiveParts = ({ components, unstable_showEmptyOnNonTextE
             if (range.type === "single") {
                 return (_jsx(MessagePrimitivePartByIndex, { index: range.index, components: components }, range.index));
             }
+            else if (range.type === "chainOfThoughtGroup") {
+                const ChainOfThoughtComponent = components?.ChainOfThought;
+                if (!ChainOfThoughtComponent)
+                    return null;
+                return (_jsx(ChainOfThoughtByIndicesProvider, { startIndex: range.startIndex, endIndex: range.endIndex, children: _jsx(ChainOfThoughtComponent, {}) }, `chainOfThought-${range.startIndex}`));
+            }
             else if (range.type === "toolGroup") {
                 const ToolGroupComponent = components?.ToolGroup ?? defaultComponents.ToolGroup;
                 return (_jsx(ToolGroupComponent, { startIndex: range.startIndex, endIndex: range.endIndex, children: Array.from({ length: range.endIndex - range.startIndex + 1 }, (_, i) => (_jsx(MessagePrimitivePartByIndex, { index: range.startIndex + i, components: components }, i))) }, `tool-${range.startIndex}`));