@lobehub/chat/src/services/assistant.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 { cloneDeep, merge } from 'lodash-es';

import { DEFAULT_DISCOVER_ASSISTANT_ITEM } from '@/const/discover';
import { edgeClient } from '@/libs/trpc/client';
import { globalHelpers } from '@/store/global/helpers';
import { DiscoverAssistantItem } from '@/types/discover';

class AssistantService {
  getAssistantList = async (): Promise<DiscoverAssistantItem[]> => {
    const locale = globalHelpers.getCurrentLanguage();

    const data = await edgeClient.market.getAgentIndex.query({ locale });

    return data.agents as unknown as DiscoverAssistantItem[];
  };

  getAssistantById = async (identifier: string): Promise<DiscoverAssistantItem> => {
    const locale = globalHelpers.getCurrentLanguage();

    const assistant = await edgeClient.market.getAgent.query({ id: identifier, locale });

    return merge(cloneDeep(DEFAULT_DISCOVER_ASSISTANT_ITEM), assistant);
  };
}
export const assistantService = new AssistantService();