@lobehub/chat/src/store/discover/slices/provider/action.ts

Lobe Chat - an open-source, high-performance chatbot framework that supports speech synthesis, multimodal, and extensible Function Call plugin system. Supports one-click free deployment of your private ChatGPT/LLM web application.

import useSWR, { type SWRResponse } from 'swr';
import type { StateCreator } from 'zustand/vanilla';

import { discoverService } from '@/services/discover';
import { DiscoverStore } from '@/store/discover';
import { globalHelpers } from '@/store/global/helpers';
import {
  DiscoverProviderDetail,
  IdentifiersResponse,
  ProviderListResponse,
  ProviderQueryParams,
} from '@/types/discover';

export interface ProviderAction {
  useProviderDetail: (params: {
    identifier: string;
    withReadme?: boolean;
  }) => SWRResponse<DiscoverProviderDetail | undefined>;
  useProviderIdentifiers: () => SWRResponse<IdentifiersResponse>;
  useProviderList: (params?: ProviderQueryParams) => SWRResponse<ProviderListResponse>;
}

export const createProviderSlice: StateCreator<
  DiscoverStore,
  [['zustand/devtools', never]],
  [],
  ProviderAction
> = () => ({
  useProviderDetail: (params) => {
    const locale = globalHelpers.getCurrentLanguage();
    return useSWR(
      ['provider-details', locale, params.identifier].filter(Boolean).join('-'),
      async () => discoverService.getProviderDetail(params),
      {
        revalidateOnFocus: false,
      },
    );
  },

  useProviderIdentifiers: () => {
    return useSWR('provider-identifiers', async () => discoverService.getProviderIdentifiers(), {
      revalidateOnFocus: false,
    });
  },

  useProviderList: (params = {}) => {
    const locale = globalHelpers.getCurrentLanguage();
    return useSWR(
      ['provider-list', locale, ...Object.values(params)].filter(Boolean).join('-'),
      async () =>
        discoverService.getProviderList({
          ...params,
          page: params.page ? Number(params.page) : 1,
          pageSize: params.pageSize ? Number(params.pageSize) : 21,
        }),
      {
        revalidateOnFocus: false,
      },
    );
  },
});