enh: start refactoring runAgent (#4728)

* enh: start refactoring runAgent * r --------- Co-authored-by: Henry Fontanier <[email protected]>
dust-tt · Apr 16, 2024 · 46cf694 · 46cf694
1 parent 231fc9d
commit 46cf694
Show file tree

Hide file tree

Showing 4 changed files with 226 additions and 169 deletions.
diff --git a/front/lib/api/assistant/actions/dust_app_run.ts b/front/lib/api/assistant/actions/dust_app_run.ts
@@ -224,10 +224,17 @@ export async function renderDustAppRunActionByModelId(
 // message.
 export async function* runDustApp(
   auth: Authenticator,
-  configuration: AgentConfigurationType,
-  conversation: ConversationType,
-  userMessage: UserMessageType,
-  agentMessage: AgentMessageType
+  {
+    configuration,
+    conversation,
+    userMessage,
+    agentMessage,
+  }: {
+    configuration: AgentConfigurationType;
+    conversation: ConversationType;
+    userMessage: UserMessageType;
+    agentMessage: AgentMessageType;
+  }
 ): AsyncGenerator<
   | DustAppRunParamsEvent
   | DustAppRunBlockEvent

diff --git a/front/lib/api/assistant/actions/retrieval.ts b/front/lib/api/assistant/actions/retrieval.ts
@@ -483,10 +483,17 @@ const getRefs = () => {
 // error is expected to be stored by the caller on the parent agent message.
 export async function* runRetrieval(
   auth: Authenticator,
-  configuration: AgentConfigurationType,
-  conversation: ConversationType,
-  userMessage: UserMessageType,
-  agentMessage: AgentMessageType
+  {
+    configuration,
+    conversation,
+    userMessage,
+    agentMessage,
+  }: {
+    configuration: AgentConfigurationType;
+    conversation: ConversationType;
+    userMessage: UserMessageType;
+    agentMessage: AgentMessageType;
+  }
 ): AsyncGenerator<
   RetrievalParamsEvent | RetrievalSuccessEvent | RetrievalErrorEvent,
   void

diff --git a/front/lib/api/assistant/actions/tables_query.ts b/front/lib/api/assistant/actions/tables_query.ts
@@ -112,19 +112,20 @@ export async function generateTablesQueryAppParams(
 /**
  * Run the TablesQuery app.
  */
-export async function* runTablesQuery({
-  auth,
-  configuration,
-  conversation,
-  userMessage,
-  agentMessage,
-}: {
-  auth: Authenticator;
-  configuration: AgentConfigurationType;
-  conversation: ConversationType;
-  userMessage: UserMessageType;
-  agentMessage: AgentMessageType;
-}): AsyncGenerator<
+export async function* runTablesQuery(
+  auth: Authenticator,
+  {
+    configuration,
+    conversation,
+    userMessage,
+    agentMessage,
+  }: {
+    configuration: AgentConfigurationType;
+    conversation: ConversationType;
+    userMessage: UserMessageType;
+    agentMessage: AgentMessageType;
+  }
+): AsyncGenerator<
   | TablesQueryErrorEvent
   | TablesQuerySuccessEvent
   | TablesQueryParamsEvent