Function createChatStore

packages/react-headless/src/store/createChatStore.ts:17–246

(config: CreateChatStoreConfig)

  );

export const createChatStore = (config: CreateChatStoreConfig) => {
  const { storage, llm } = config;
  const { thread: threadStorage } = storage;

  const store = createStore<ChatStore>()(
    subscribeWithSelector((set, get) => ({
      // Thread List State
      threads: [],
      isLoadingThreads: false,
      threadListError: null,
      selectedThreadId: null,
      hasMoreThreads: false,
      _nextCursor: undefined,

      // Thread State
      messages: [],
      isRunning: false,
      isLoadingMessages: false,
      threadError: null,
      executingToolCallIds: new Set<string>(),
      _abortController: null,

      // ── Thread List Actions ──

      loadThreads: () => {
        set({ isLoadingThreads: true, threadListError: null });
        threadStorage
          .listThreads(undefined)
          .then(({ threads = [], nextCursor }) => {
            set({
              threads,
              isLoadingThreads: false,
              _nextCursor: nextCursor,
              hasMoreThreads: nextCursor !== undefined,
            });
          })
          .catch((e) => {
            set({ isLoadingThreads: false, threadListError: e });
          });
      },

      loadMoreThreads: () => {
        const cursor = get()._nextCursor;
        if (cursor === undefined) return;
        threadStorage
          .listThreads(cursor)
          .then(({ threads = [], nextCursor }) => {
            set((s) => ({
              threads: mergeThreadList(s.threads, threads),
              _nextCursor: nextCursor,
              hasMoreThreads: nextCursor !== undefined,
            }));
          })
          .catch((e) => {
            set({ threadListError: e });
          });
      },