@assistant-ui/react 0.10.23 → 0.10.25

package/src/primitives/contentPart/ContentPartText.tsx

@@ -1,37 +0,0 @@
-"use client";
-
-import { Primitive } from "@radix-ui/react-primitive";
-import {
-  type ComponentRef,
-  forwardRef,
-  ComponentPropsWithoutRef,
-  ElementType,
-} from "react";
-import { useContentPartText } from "./useContentPartText";
-import { useSmooth } from "../../utils/smooth/useSmooth";
-
-export namespace ContentPartPrimitiveText {
-  export type Element = ComponentRef<typeof Primitive.span>;
-  export type Props = Omit<
-    ComponentPropsWithoutRef<typeof Primitive.span>,
-    "children" | "asChild"
-  > & {
-    smooth?: boolean;
-    component?: ElementType;
-  };
-}
-
-export const ContentPartPrimitiveText = forwardRef<
-  ContentPartPrimitiveText.Element,
-  ContentPartPrimitiveText.Props
->(({ smooth = true, component: Component = "span", ...rest }, forwardedRef) => {
-  const { text, status } = useSmooth(useContentPartText(), smooth);
-
-  return (
-    <Component data-status={status.type} {...rest} ref={forwardedRef}>
-      {text}
-    </Component>
-  );
-});
-
-ContentPartPrimitiveText.displayName = "ContentPartPrimitive.Text";

package/src/primitives/contentPart/index.ts

@@ -1,3 +0,0 @@
-export { ContentPartPrimitiveText as Text } from "./ContentPartText";
-export { ContentPartPrimitiveImage as Image } from "./ContentPartImage";
-export { ContentPartPrimitiveInProgress as InProgress } from "./ContentPartInProgress";

package/src/primitives/contentPart/useContentPartFile.tsx

@@ -1,18 +0,0 @@
-"use client";
-
-import { ContentPartState } from "../../api/ContentPartRuntime";
-import { useContentPart } from "../../context/react/ContentPartContext";
-import { FileContentPart } from "../../types";
-
-export const useContentPartFile = () => {
-  const file = useContentPart((c) => {
-    if (c.type !== "file")
-      throw new Error(
-        "ContentPartFile can only be used inside file content parts.",
-      );
-
-    return c as ContentPartState & FileContentPart;
-  });
-
-  return file;
-};

package/src/primitives/contentPart/useContentPartImage.tsx

@@ -1,18 +0,0 @@
-"use client";
-
-import { ContentPartState } from "../../api/ContentPartRuntime";
-import { useContentPart } from "../../context/react/ContentPartContext";
-import { ImageContentPart } from "../../types";
-
-export const useContentPartImage = () => {
-  const image = useContentPart((c) => {
-    if (c.type !== "image")
-      throw new Error(
-        "ContentPartImage can only be used inside image content parts.",
-      );
-
-    return c as ContentPartState & ImageContentPart;
-  });
-
-  return image;
-};

package/src/primitives/contentPart/useContentPartReasoning.tsx

@@ -1,18 +0,0 @@
-"use client";
-
-import { ContentPartState } from "../../api/ContentPartRuntime";
-import { useContentPart } from "../../context/react/ContentPartContext";
-import { ReasoningContentPart } from "../../types";
-
-export const useContentPartReasoning = () => {
-  const text = useContentPart((c) => {
-    if (c.type !== "reasoning")
-      throw new Error(
-        "ContentPartReasoning can only be used inside reasoning content parts.",
-      );
-
-    return c as ContentPartState & ReasoningContentPart;
-  });
-
-  return text;
-};

package/src/primitives/contentPart/useContentPartSource.tsx

@@ -1,18 +0,0 @@
-"use client";
-
-import { ContentPartState } from "../../api/ContentPartRuntime";
-import { useContentPart } from "../../context/react/ContentPartContext";
-import { SourceContentPart } from "../../types";
-
-export const useContentPartSource = () => {
-  const source = useContentPart((c) => {
-    if (c.type !== "source")
-      throw new Error(
-        "ContentPartSource can only be used inside source content parts.",
-      );
-
-    return c as ContentPartState & SourceContentPart;
-  });
-
-  return source;
-};

package/src/primitives/contentPart/useContentPartText.tsx

@@ -1,18 +0,0 @@
-"use client";
-
-import { ContentPartState } from "../../api/ContentPartRuntime";
-import { useContentPart } from "../../context/react/ContentPartContext";
-import { TextContentPart, ReasoningContentPart } from "../../types";
-
-export const useContentPartText = () => {
-  const text = useContentPart((c) => {
-    if (c.type !== "text" && c.type !== "reasoning")
-      throw new Error(
-        "ContentPartText can only be used inside text or reasoning content parts.",
-      );
-
-    return c as ContentPartState & (TextContentPart | ReasoningContentPart);
-  });
-
-  return text;
-};

package/src/primitives/message/MessageContent.tsx

@@ -1,241 +0,0 @@
-"use client";
-
-import { type ComponentType, type FC, memo, useMemo } from "react";
-import {
-  TextContentPartProvider,
-  useContentPart,
-  useContentPartRuntime,
-  useToolUIs,
-} from "../../context";
-import {
-  useMessage,
-  useMessageRuntime,
-} from "../../context/react/MessageContext";
-import { ContentPartRuntimeProvider } from "../../context/providers/ContentPartRuntimeProvider";
-import { ContentPartPrimitiveText } from "../contentPart/ContentPartText";
-import { ContentPartPrimitiveImage } from "../contentPart/ContentPartImage";
-import type {
-  Unstable_AudioContentPartComponent,
-  EmptyContentPartComponent,
-  TextContentPartComponent,
-  ImageContentPartComponent,
-  SourceContentPartComponent,
-  ToolCallContentPartComponent,
-  ToolCallContentPartProps,
-  FileContentPartComponent,
-  ReasoningContentPartComponent,
-} from "../../types/ContentPartComponentTypes";
-import { ContentPartPrimitiveInProgress } from "../contentPart/ContentPartInProgress";
-import { ContentPartStatus } from "../../types/AssistantTypes";
-
-export namespace MessagePrimitiveContent {
-  export type Props = {
-    components?:
-      | {
-          Empty?: EmptyContentPartComponent | undefined;
-          Text?: TextContentPartComponent | undefined;
-          Reasoning?: ReasoningContentPartComponent | undefined;
-          Source?: SourceContentPartComponent | undefined;
-          Image?: ImageContentPartComponent | undefined;
-          File?: FileContentPartComponent | undefined;
-          Unstable_Audio?: Unstable_AudioContentPartComponent | undefined;
-          tools?:
-            | {
-                by_name?:
-                  | Record<string, ToolCallContentPartComponent | undefined>
-                  | undefined;
-                Fallback?: ComponentType<ToolCallContentPartProps> | undefined;
-              }
-            | {
-                Override: ComponentType<ToolCallContentPartProps>;
-              }
-            | undefined;
-        }
-      | undefined;
-  };
-}
-
-const ToolUIDisplay = ({
-  Fallback,
-  ...props
-}: {
-  Fallback: ToolCallContentPartComponent | undefined;
-} & ToolCallContentPartProps) => {
-  const Render = useToolUIs((s) => s.getToolUI(props.toolName)) ?? Fallback;
-  if (!Render) return null;
-  return <Render {...props} />;
-};
-
-const defaultComponents = {
-  Text: () => (
-    <p style={{ whiteSpace: "pre-line" }}>
-      <ContentPartPrimitiveText />
-      <ContentPartPrimitiveInProgress>
-        <span style={{ fontFamily: "revert" }}>{" \u25CF"}</span>
-      </ContentPartPrimitiveInProgress>
-    </p>
-  ),
-  Reasoning: () => null,
-  Source: () => null,
-  Image: () => <ContentPartPrimitiveImage />,
-  File: () => null,
-  Unstable_Audio: () => null,
-} satisfies MessagePrimitiveContent.Props["components"];
-
-type MessageContentPartComponentProps = {
-  components: MessagePrimitiveContent.Props["components"];
-};
-
-const MessageContentPartComponent: FC<MessageContentPartComponentProps> = ({
-  components: {
-    Text = defaultComponents.Text,
-    Reasoning = defaultComponents.Reasoning,
-    Image = defaultComponents.Image,
-    Source = defaultComponents.Source,
-    File = defaultComponents.File,
-    Unstable_Audio: Audio = defaultComponents.Unstable_Audio,
-    tools = {},
-  } = {},
-}) => {
-  const contentPartRuntime = useContentPartRuntime();
-
-  const part = useContentPart();
-
-  const type = part.type;
-  if (type === "tool-call") {
-    const addResult = (result: any) => contentPartRuntime.addToolResult(result);
-    if ("Override" in tools)
-      return <tools.Override {...part} addResult={addResult} />;
-    const Tool = tools.by_name?.[part.toolName] ?? tools.Fallback;
-    return <ToolUIDisplay {...part} Fallback={Tool} addResult={addResult} />;
-  }
-
-  if (part.status.type === "requires-action")
-    throw new Error("Encountered unexpected requires-action status");
-
-  switch (type) {
-    case "text":
-      return <Text {...part} />;
-
-    case "reasoning":
-      return <Reasoning {...part} />;
-
-    case "source":
-      return <Source {...part} />;
-
-    case "image":
-      // eslint-disable-next-line jsx-a11y/alt-text
-      return <Image {...part} />;
-
-    case "file":
-      return <File {...part} />;
-
-    case "audio":
-      return <Audio {...part} />;
-
-    default:
-      const unhandledType: never = type;
-      throw new Error(`Unknown content part type: ${unhandledType}`);
-  }
-};
-
-type MessageContentPartProps = {
-  partIndex: number;
-  components: MessagePrimitiveContent.Props["components"];
-};
-
-const MessageContentPartImpl: FC<MessageContentPartProps> = ({
-  partIndex,
-  components,
-}) => {
-  const messageRuntime = useMessageRuntime();
-  const runtime = useMemo(
-    () => messageRuntime.getContentPartByIndex(partIndex),
-    [messageRuntime, partIndex],
-  );
-
-  return (
-    <ContentPartRuntimeProvider runtime={runtime}>
-      <MessageContentPartComponent components={components} />
-    </ContentPartRuntimeProvider>
-  );
-};
-
-const MessageContentPart = memo(
-  MessageContentPartImpl,
-  (prev, next) =>
-    prev.partIndex === next.partIndex &&
-    prev.components?.Text === next.components?.Text &&
-    prev.components?.Reasoning === next.components?.Reasoning &&
-    prev.components?.Source === next.components?.Source &&
-    prev.components?.Image === next.components?.Image &&
-    prev.components?.File === next.components?.File &&
-    prev.components?.Unstable_Audio === next.components?.Unstable_Audio &&
-    prev.components?.tools === next.components?.tools,
-);
-
-const COMPLETE_STATUS: ContentPartStatus = Object.freeze({
-  type: "complete",
-});
-
-const EmptyContentFallback: FC<{
-  status: ContentPartStatus;
-  component: TextContentPartComponent;
-}> = ({ status, component: Component }) => {
-  return (
-    <TextContentPartProvider text="" isRunning={status.type === "running"}>
-      <Component type="text" text="" status={status} />
-    </TextContentPartProvider>
-  );
-};
-
-const EmptyContentImpl: FC<MessageContentPartComponentProps> = ({
-  components,
-}) => {
-  const status =
-    useMessage((s) => s.status as ContentPartStatus) ?? COMPLETE_STATUS;
-
-  if (components?.Empty) return <components.Empty status={status} />;
-
-  return (
-    <EmptyContentFallback
-      status={status}
-      component={components?.Text ?? defaultComponents.Text}
-    />
-  );
-};
-
-const EmptyContent = memo(
-  EmptyContentImpl,
-  (prev, next) =>
-    prev.components?.Empty === next.components?.Empty &&
-    prev.components?.Text === next.components?.Text,
-);
-
-export const MessagePrimitiveContent: FC<MessagePrimitiveContent.Props> = ({
-  components,
-}) => {
-  const contentLength = useMessage((s) => s.content.length);
-
-  if (contentLength === 0) {
-    return (
-      <>
-        <EmptyContent components={components} />
-      </>
-    );
-  }
-
-  return (
-    <>
-      {Array.from({ length: contentLength }, (_, index) => (
-        <MessageContentPart
-          key={index}
-          partIndex={index}
-          components={components}
-        />
-      ))}
-    </>
-  );
-};
-
-MessagePrimitiveContent.displayName = "MessagePrimitive.Content";