lobe-chat

1
import { lambdaClient } from '@/libs/trpc/client';
2
import { CreateKnowledgeBaseParams } from '@/types/knowledgeBase';
3

4
class KnowledgeBaseService {
5
  async createKnowledgeBase(params: CreateKnowledgeBaseParams) {
6
    return await lambdaClient.knowledgeBase.createKnowledgeBase.mutate(params);
7
  }
8

9
  async getKnowledgeBaseList() {
10
    return await lambdaClient.knowledgeBase.getKnowledgeBases.query();
11
  }
12

13
  async getKnowledgeBaseById(id: string) {
14
    return await lambdaClient.knowledgeBase.getKnowledgeBaseById.query({ id });
15
  }
16

17
  async updateKnowledgeBaseList(id: string, value: any) {
18
    return await lambdaClient.knowledgeBase.updateKnowledgeBase.mutate({ id, value });
19
  }
20

21
  async deleteKnowledgeBase(id: string) {
22
    return await lambdaClient.knowledgeBase.removeKnowledgeBase.mutate({ id });
23
  }
24

25
  async addFilesToKnowledgeBase(knowledgeBaseId: string, ids: string[]) {
26
    return lambdaClient.knowledgeBase.addFilesToKnowledgeBase.mutate({ ids, knowledgeBaseId });
27
  }
28

29
  async removeFilesFromKnowledgeBase(knowledgeBaseId: string, ids: string[]) {
30
    return lambdaClient.knowledgeBase.removeFilesFromKnowledgeBase.mutate({ ids, knowledgeBaseId });
31
  }
32
}
33

34
export const knowledgeBaseService = new KnowledgeBaseService();
35