fix: undo moving internal utils to /react/internal (#620)

.changeset/wet-pillows-hope.md

@@ -0,0 +1,10 @@
+---
+"@assistant-ui/react-syntax-highlighter": patch
+"@assistant-ui/react-playground": patch
+"@assistant-ui/react-hook-form": patch
+"@assistant-ui/react-markdown": patch
+"@assistant-ui/react-ai-sdk": patch
+"@assistant-ui/react": patch
+---
+
+fix: undo moving internal utilities to /react/internal

packages/react-ai-sdk/src/ui/use-assistant/VercelUseAssistantRuntime.tsx

@@ -1,11 +1,13 @@
-import { type ThreadMessage, ModelConfigProvider } from "@assistant-ui/react";
-import { useAssistant } from "ai/react";
 import {
-  ProxyConfigProvider,
-  BaseAssistantRuntime,
-} from "@assistant-ui/react/internal";
+  type ThreadMessage,
+  INTERNAL,
+  ModelConfigProvider,
+} from "@assistant-ui/react";
+import { useAssistant } from "ai/react";
 import { VercelUseAssistantThreadRuntime } from "./VercelUseAssistantThreadRuntime";
 
+const { ProxyConfigProvider, BaseAssistantRuntime } = INTERNAL;
+
 export const hasUpcomingMessage = (
   isRunning: boolean,
   messages: ThreadMessage[],

packages/react-ai-sdk/src/ui/use-chat/VercelUseChatRuntime.tsx

@@ -1,11 +1,9 @@
-import {
-  ProxyConfigProvider,
-  BaseAssistantRuntime,
-} from "@assistant-ui/react/internal";
-import { ModelConfigProvider } from "@assistant-ui/react";
+import { INTERNAL, ModelConfigProvider } from "@assistant-ui/react";
 import { useChat } from "ai/react";
 import { VercelUseChatThreadRuntime } from "./VercelUseChatThreadRuntime";
 
+const { ProxyConfigProvider, BaseAssistantRuntime } = INTERNAL;
+
 export class VercelUseChatRuntime extends BaseAssistantRuntime<VercelUseChatThreadRuntime> {
   private readonly _proxyConfigProvider = new ProxyConfigProvider();
 

packages/react-ai-sdk/src/ui/use-chat/VercelUseChatThreadRuntime.tsx

@@ -4,8 +4,8 @@ import {
   type AppendMessage,
   type ThreadMessage,
   AddToolResultOptions,
+  INTERNAL,
 } from "@assistant-ui/react";
-import { MessageRepository } from "@assistant-ui/react/internal";
 import type { Message } from "ai";
 import { type StoreApi, type UseBoundStore, create } from "zustand";
 import { useChat } from "ai/react";
@@ -14,6 +14,8 @@ import { sliceMessagesUntil } from "../utils/sliceMessagesUntil";
 import { useVercelAIComposerSync } from "../utils/useVercelAIComposerSync";
 import { useVercelAIThreadSync } from "../utils/useVercelAIThreadSync";
 
+const { MessageRepository } = INTERNAL;
+
 export const hasUpcomingMessage = (
   isRunning: boolean,
   messages: ThreadMessage[],

packages/react-markdown/src/primitives/MarkdownText.tsx

@@ -1,6 +1,6 @@
 "use client";
 
-import { useContentPartText } from "@assistant-ui/react";
+import { INTERNAL, useContentPartText } from "@assistant-ui/react";
 import {
   ElementRef,
   ElementType,
@@ -21,10 +21,11 @@ import {
 } from "../overrides/defaultComponents";
 import { useCallbackRef } from "@radix-ui/react-use-callback-ref";
 import { CodeOverride } from "../overrides/CodeOverride";
-import { useSmooth } from "@assistant-ui/react/internal";
 import { Primitive } from "@radix-ui/react-primitive";
 import classNames from "classnames";
 
+const { useSmooth } = INTERNAL;
+
 type MarkdownTextPrimitiveElement = ElementRef<typeof Primitive.div>;
 type PrimitiveDivProps = ComponentPropsWithoutRef<typeof Primitive.div>;
 

packages/react-markdown/src/ui/code-header.tsx

@@ -1,11 +1,12 @@
 import { FC } from "react";
 import { CheckIcon, CopyIcon } from "lucide-react";
-import { useThreadConfig } from "@assistant-ui/react";
-import { TooltipIconButton } from "@assistant-ui/react/internal";
+import { INTERNAL, useThreadConfig } from "@assistant-ui/react";
 
 import { CodeHeaderProps } from "../overrides/types";
 import { useCopyToClipboard } from "./useCopyToClipboard";
 
+const { TooltipIconButton } = INTERNAL;
+
 export const CodeHeader: FC<CodeHeaderProps> = ({ language, code }) => {
   const {
     strings: {

packages/react-markdown/src/ui/markdown-text.tsx

@@ -5,10 +5,9 @@ import {
   MarkdownTextPrimitive,
   MarkdownTextPrimitiveProps,
 } from "../primitives/MarkdownText";
-import {
-  withSmoothContextProvider,
-  useSmoothStatus,
-} from "@assistant-ui/react/internal";
+import { INTERNAL } from "@assistant-ui/react";
+
+const { withSmoothContextProvider, useSmoothStatus } = INTERNAL;
 
 export type MakeMarkdownTextProps = MarkdownTextPrimitiveProps;
 

packages/react-playground/src/lib/playground-runtime.ts

@@ -22,16 +22,14 @@ import {
   ChatModelRunResult,
   CoreMessage,
   fromCoreMessage,
+  INTERNAL,
 } from "@assistant-ui/react";
-import {
-  BaseAssistantRuntime,
-  ProxyConfigProvider,
-  generateId,
-} from "@assistant-ui/react/internal";
 import { LanguageModelV1FunctionTool } from "@ai-sdk/provider";
 import { useState } from "react";
 import { create } from "zustand";
 
+const { BaseAssistantRuntime, ProxyConfigProvider, generateId } = INTERNAL;
+
 const makeModelConfigStore = () =>
   create<ModelConfig>(() => ({
     system: "",

packages/react/package.json

@@ -52,16 +52,6 @@
         "default": "./dist/edge.js"
       }
     },
-    "./internal": {
-      "import": {
-        "types": "./dist/internal.d.mts",
-        "default": "./dist/internal.mjs"
-      },
-      "require": {
-        "types": "./dist/internal.d.ts",
-        "default": "./dist/internal.js"
-      }
-    },
     "./tailwindcss": {
       "import": {
         "types": "./dist/tailwindcss/index.d.mts",
@@ -83,7 +73,6 @@
   "files": [
     "dist",
     "edge",
-    "internal",
     "tailwindcss",
     "README.md"
   ],

packages/react/scripts/build.mts

@@ -3,23 +3,27 @@ import { copyFileSync, mkdirSync } from "node:fs";
 
 // JS
 await build({
-  entry: [
-    "src/index.ts",
-    "src/edge.ts",
-    "src/tailwindcss/index.ts",
-    "src/internal.ts",
-  ],
+  entry: ["src/index.ts", "src/edge.ts", "src/tailwindcss/index.ts"],
   format: ["cjs", "esm"],
   dts: true,
   sourcemap: true,
   clean: true,
+  splitting: true,
   esbuildOptions: (options) => {
     options.banner = {
       js: '"use client";',
     };
   },
 });
 
+// TODO find a way to bundle edge with the rest of the package
+await build({
+  entry: ["src/edge.ts"],
+  format: ["cjs", "esm"],
+  dts: true,
+  sourcemap: true,
+});
+
 // css
 
 await build({

packages/react/src/index.ts

@@ -7,7 +7,4 @@ export * from "./runtimes";
 export * from "./types";
 export * from "./ui";
 
-/**
- * @deprecated Use `@assistant-ui/react/internal` instead. This will be removed in 0.6.0.
- */
 export * as INTERNAL from "./internal";