refactor: move internal utils to @assistant-ui/react/internal (#610)

.changeset/nine-experts-allow.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
+---
+
+refactor: move internal utilities to @assistant-ui/react/internal

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

@@ -1,10 +1,11 @@
-import { type ThreadMessage, INTERNAL } from "@assistant-ui/react";
-import { ModelConfigProvider } from "@assistant-ui/react";
+import { type ThreadMessage, ModelConfigProvider } from "@assistant-ui/react";
 import { useAssistant } from "ai/react";
+import {
+  ProxyConfigProvider,
+  BaseAssistantRuntime,
+} from "@assistant-ui/react/internal";
 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,10 +1,11 @@
-import { INTERNAL } from "@assistant-ui/react";
+import {
+  ProxyConfigProvider,
+  BaseAssistantRuntime,
+} from "@assistant-ui/react/internal";
 import { 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

@@ -3,18 +3,16 @@ import {
   type Unsubscribe,
   type AppendMessage,
   type ThreadMessage,
-  INTERNAL,
   AddToolResultOptions,
 } 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";
 import { getVercelAIMessage } from "../getVercelAIMessage";
 import { sliceMessagesUntil } from "../utils/sliceMessagesUntil";
 import { useVercelAIComposerSync } from "../utils/useVercelAIComposerSync";
 import { useVercelAIThreadSync } from "../utils/useVercelAIThreadSync";
-import { useChat } from "ai/react";
-
-const { MessageRepository } = INTERNAL;
 
 export const hasUpcomingMessage = (
   isRunning: boolean,

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

@@ -1,6 +1,7 @@
 "use client";
 
-import { INTERNAL, useContentPartText } from "@assistant-ui/react";
+import { useContentPartText } from "@assistant-ui/react";
+import { useSmooth } from "@assistant-ui/react/internal";
 import type { ComponentType, FC } from "react";
 import ReactMarkdown, { type Options } from "react-markdown";
 import { SyntaxHighlighterProps, CodeHeaderProps } from "../overrides/types";
@@ -14,8 +15,6 @@ import {
 import { useCallbackRef } from "@radix-ui/react-use-callback-ref";
 import { CodeOverride } from "../overrides/CodeOverride";
 
-const { useSmooth } = INTERNAL;
-
 export type MarkdownTextPrimitiveProps = Omit<
   Options,
   "components" | "children"

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

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

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

@@ -18,17 +18,19 @@ import {
   ThreadAssistantMessage,
   ChatModelAdapter,
   Unsubscribe,
-  INTERNAL,
   AssistantRuntime,
   ChatModelRunResult,
   CoreMessage,
+  fromCoreMessage,
 } 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";
-import { LanguageModelV1FunctionTool } from "@ai-sdk/provider";
-import { fromCoreMessage } from "@assistant-ui/react";
-
-const { BaseAssistantRuntime, ProxyConfigProvider, generateId } = INTERNAL;
 
 const makeModelConfigStore = () =>
   create<ModelConfig>(() => ({

packages/react/internal/README.md

@@ -0,0 +1 @@
+This directory exists to support subpath imports for TypeScript projects using --moduleResolution node.

packages/react/internal/package.json

@@ -0,0 +1,5 @@
+{
+  "main": "../dist/internal.js",
+  "module": "../dist/internal.mjs",
+  "types": "../dist/internal.d.ts"
+}

packages/react/scripts/build.mts

@@ -16,7 +16,7 @@ await build({
 });
 
 await build({
-  entry: ["src/edge.ts", "src/tailwindcss/index.ts"],
+  entry: ["src/edge.ts", "src/tailwindcss/index.ts", "src/internal.ts"],
   format: ["cjs", "esm"],
   dts: true,
   sourcemap: true,

packages/react/src/index.ts

@@ -7,4 +7,7 @@ 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";