@assistant-ui/react/src/client/types/ToolUI.ts

TypeScript/React library for AI Chat

import { Unsubscribe } from "@assistant-ui/tap";
import { ToolCallMessagePartComponent } from "../../types";

export type ToolUIState = Record<string, ToolCallMessagePartComponent[]>;

export type ToolUIApi = {
  getState(): ToolUIState;

  setToolUI(
    toolName: string,
    render: ToolCallMessagePartComponent,
  ): Unsubscribe;
};

export type ToolUIMeta = {
  source: "root";
  query: Record<string, never>;
};