fix websearch action/config (#9658)

dust-tt · Dec 30, 2024 · 588f1a4 · 588f1a4
1 parent 5eb0f09
commit 588f1a4
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/front/poke/temporal/activities.ts b/front/poke/temporal/activities.ts
@@ -25,6 +25,10 @@ import {
   AgentTablesQueryConfiguration,
   AgentTablesQueryConfigurationTable,
 } from "@app/lib/models/assistant/actions/tables_query";
+import {
+  AgentWebsearchAction,
+  AgentWebsearchConfiguration,
+} from "@app/lib/models/assistant/actions/websearch";
 import {
   AgentConfiguration,
   AgentUserRelation,
@@ -297,6 +301,28 @@ export async function deleteAgentsActivity({
         transaction: t,
       });
 
+      const agentWebsearchConfigurations =
+        await AgentWebsearchConfiguration.findAll({
+          where: {
+            agentConfigurationId: agent.id,
+          },
+          transaction: t,
+        });
+      await AgentWebsearchAction.destroy({
+        where: {
+          websearchConfigurationId: {
+            [Op.in]: agentWebsearchConfigurations.map((r) => r.sId),
+          },
+        },
+        transaction: t,
+      });
+      await AgentWebsearchConfiguration.destroy({
+        where: {
+          agentConfigurationId: agent.id,
+        },
+        transaction: t,
+      });
+
       await AgentUserRelation.destroy({
         where: {
           agentConfiguration: agent.sId,