Agent config: V2 for lib to get / create / update the config

dust-tt · Sep 8, 2023 · 7ddda0d · 7ddda0d
1 parent b4bb27a
commit 7ddda0d
Show file tree

Hide file tree

Showing 4 changed files with 698 additions and 62 deletions.
diff --git a/front/admin/db.ts b/front/admin/db.ts
@@ -56,11 +56,11 @@ async function main() {
   await ExtractedEvent.sync({ alter: true });
   await DocumentTrackerChangeSuggestion.sync({ alter: true });
 
-  await AgentConfiguration.sync({ alter: true });
   await AgentGenerationConfiguration.sync({ alter: true });
-
   await AgentRetrievalConfiguration.sync({ alter: true });
   await AgentDataSourceConfiguration.sync({ alter: true });
+  await AgentConfiguration.sync({ alter: true });
+
   await AgentRetrievalAction.sync({ alter: true });
   await RetrievalDocument.sync({ alter: true });
   await RetrievalDocumentChunk.sync({ alter: true });

diff --git a/front/lib/api/assistant/agent.ts b/front/lib/api/assistant/agent.ts
@@ -18,11 +18,8 @@ import { Err, Ok, Result } from "@app/lib/result";
 import { generateModelSId } from "@app/lib/utils";
 import { isRetrievalConfiguration } from "@app/types/assistant/actions/retrieval";
 import {
-  AgentActionConfigurationType,
   AgentActionSpecification,
-  AgentConfigurationStatus,
   AgentConfigurationType,
-  AgentGenerationConfigurationType,
 } from "@app/types/assistant/agent";
 import {
   AgentActionType,
@@ -31,61 +28,6 @@ import {
   UserMessageType,
 } from "@app/types/assistant/conversation";
 
-/**
- * Agent configuration.
- */
-
-export async function createAgentConfiguration(
-  auth: Authenticator,
-  {
-    name,
-    pictureUrl,
-    action,
-    generation,
-  }: {
-    name: string;
-    pictureUrl?: string;
-    action?: AgentActionConfigurationType;
-    generation?: AgentGenerationConfigurationType;
-  }
-): Promise<AgentConfigurationType> {
-  return {
-    sId: generateModelSId(),
-    name,
-    pictureUrl: pictureUrl ?? null,
-    status: "active",
-    action: action ?? null,
-    generation: generation ?? null,
-  };
-}
-
-export async function updateAgentConfiguration(
-  auth: Authenticator,
-  configurationId: string,
-  {
-    name,
-    pictureUrl,
-    status,
-    action,
-    generation,
-  }: {
-    name: string;
-    pictureUrl?: string;
-    status: AgentConfigurationStatus;
-    action?: AgentActionConfigurationType;
-    generation?: AgentGenerationConfigurationType;
-  }
-): Promise<AgentConfigurationType> {
-  return {
-    sId: generateModelSId(),
-    name,
-    pictureUrl: pictureUrl ?? null,
-    status,
-    action: action ?? null,
-    generation: generation ?? null,
-  };
-}
-
 /**
  * Action Inputs generation.
  */