lobe-chat

1
import { lambdaClient } from '@/libs/trpc/client';
2

3
class AgentService {
4
  async createAgentKnowledgeBase(agentId: string, knowledgeBaseId: string, enabled?: boolean) {
5
    return await lambdaClient.agent.createAgentKnowledgeBase.mutate({
6
      agentId,
7
      enabled,
8
      knowledgeBaseId,
9
    });
10
  }
11

12
  async deleteAgentKnowledgeBase(agentId: string, knowledgeBaseId: string) {
13
    return await lambdaClient.agent.deleteAgentKnowledgeBase.mutate({ agentId, knowledgeBaseId });
14
  }
15

16
  async toggleKnowledgeBase(agentId: string, knowledgeBaseId: string, enabled?: boolean) {
17
    return await lambdaClient.agent.toggleKnowledgeBase.mutate({
18
      agentId,
19
      enabled,
20
      knowledgeBaseId,
21
    });
22
  }
23

24
  async createAgentFiles(agentId: string, fileIds: string[], enabled?: boolean) {
25
    return await lambdaClient.agent.createAgentFiles.mutate({ agentId, enabled, fileIds });
26
  }
27

28
  async deleteAgentFile(agentId: string, fileId: string) {
29
    return await lambdaClient.agent.deleteAgentFile.mutate({ agentId, fileId });
30
  }
31

32
  async toggleFile(agentId: string, fileId: string, enabled?: boolean) {
33
    return await lambdaClient.agent.toggleFile.mutate({
34
      agentId,
35
      enabled,
36
      fileId,
37
    });
38
  }
39

40
  async getFilesAndKnowledgeBases(agentId: string) {
41
    return await lambdaClient.agent.getKnowledgeBasesAndFiles.query({ agentId });
42
  }
43
}
44

45
export const agentService = new AgentService();
46