@assistant-ui/react/src/legacy-runtime/client/ThreadListItemRuntimeClient.ts

TypeScript/React library for AI Chat

import { resource, tapEffect } from "@assistant-ui/tap";
import {
  ThreadListItemEventType,
  ThreadListItemRuntime,
} from "../runtime/ThreadListItemRuntime";
import { Unsubscribe } from "../../types";
import { tapApi } from "../../utils/tap-store";
import { tapSubscribable } from "../util-hooks/tapSubscribable";
import { tapEvents } from "../../client/EventContext";
import { ThreadListItemClientApi } from "../../client/types/ThreadListItem";

export const ThreadListItemClient = resource(
  ({ runtime }: { runtime: ThreadListItemRuntime }) => {
    const runtimeState = tapSubscribable(runtime);
    const events = tapEvents();

    // Bind thread list item events to event manager
    tapEffect(() => {
      const unsubscribers: Unsubscribe[] = [];

      // Subscribe to thread list item events
      const threadListItemEvents: ThreadListItemEventType[] = [
        "switched-to",
        "switched-away",
      ];

      for (const event of threadListItemEvents) {
        const unsubscribe = runtime.unstable_on(event, () => {
          events.emit(`thread-list-item.${event}`, {
            threadId: runtime.getState()!.id,
          });
        });
        unsubscribers.push(unsubscribe);
      }

      return () => {
        for (const unsub of unsubscribers) unsub();
      };
    }, [runtime, events]);

    return tapApi<ThreadListItemClientApi>(
      {
        getState: () => runtimeState,
        switchTo: runtime.switchTo,
        rename: runtime.rename,
        archive: runtime.archive,
        unarchive: runtime.unarchive,
        delete: runtime.delete,
        generateTitle: runtime.generateTitle,
        initialize: runtime.initialize,
        detach: runtime.detach,
        __internal_getRuntime: () => runtime,
      },
      {
        key: runtimeState.id,
      },
    );
  },
);