From d839e235e5c9175ac4b48802ee7547f4fd18cc94 Mon Sep 17 00:00:00 2001
From: Lucas Oliveira <lucas.edu.oli@hotmail.com>
Date: Tue, 6 Jan 2026 13:38:14 -0300
Subject: [PATCH] changed onboarrding mutation to change the onboarding
 settings

---
 .../api/mutations/useOnboardingMutation.ts    | 21 ++++++++++++-------
 .../_components/onboarding-card.tsx           | 12 -----------
 2 files changed, 14 insertions(+), 19 deletions(-)

diff --git a/frontend/app/api/mutations/useOnboardingMutation.ts b/frontend/app/api/mutations/useOnboardingMutation.ts
index 42b95236..911e196c 100644
--- a/frontend/app/api/mutations/useOnboardingMutation.ts
+++ b/frontend/app/api/mutations/useOnboardingMutation.ts
@@ -3,7 +3,7 @@ import {
   useMutation,
   useQueryClient,
 } from "@tanstack/react-query";
-import { ONBOARDING_OPENRAG_DOCS_FILTER_ID_KEY } from "@/lib/constants";
+import { useUpdateOnboardingStateMutation } from "./useUpdateOnboardingStateMutation";
 
 export interface OnboardingVariables {
   // Provider selection
@@ -36,12 +36,14 @@ export const useOnboardingMutation = (
   options?: Omit<
     UseMutationOptions<OnboardingResponse, Error, OnboardingVariables>,
     "mutationFn"
-  >,
+  >
 ) => {
   const queryClient = useQueryClient();
 
+  const updateOnboardingMutation = useUpdateOnboardingStateMutation();
+
   async function submitOnboarding(
-    variables: OnboardingVariables,
+    variables: OnboardingVariables
   ): Promise<OnboardingResponse> {
     const response = await fetch("/api/onboarding", {
       method: "POST",
@@ -62,10 +64,15 @@ export const useOnboardingMutation = (
   return useMutation({
     mutationFn: submitOnboarding,
     onSuccess: (data) => {
-      // Store OpenRAG Docs filter ID if returned
-      if (data.openrag_docs_filter_id && typeof window !== "undefined") {
-        localStorage.setItem(
-          ONBOARDING_OPENRAG_DOCS_FILTER_ID_KEY,
+      // Save OpenRAG docs filter ID if sample data was ingested
+      if (data.openrag_docs_filter_id) {
+        // Save to backend
+        updateOnboardingMutation.mutateAsync({
+          openrag_docs_filter_id: data.openrag_docs_filter_id,
+        });
+
+        console.log(
+          "Saved OpenRAG docs filter ID:",
           data.openrag_docs_filter_id
         );
       }
diff --git a/frontend/app/onboarding/_components/onboarding-card.tsx b/frontend/app/onboarding/_components/onboarding-card.tsx
index 5ea6218a..3d7d3e3f 100644
--- a/frontend/app/onboarding/_components/onboarding-card.tsx
+++ b/frontend/app/onboarding/_components/onboarding-card.tsx
@@ -173,8 +173,6 @@ const OnboardingCard = ({
 
   // Track which tasks we've already handled to prevent infinite loops
   const handledFailedTasksRef = useRef<Set<string>>(new Set());
-  
-  const updateOnboardingMutation = useUpdateOnboardingStateMutation();
 
   // Query tasks to track completion
   const { data: tasks } = useGetTasksQuery({
@@ -308,16 +306,6 @@ const OnboardingCard = ({
     onSuccess: (data) => {
       console.log("Onboarding completed successfully", data);
 
-      // Save OpenRAG docs filter ID if sample data was ingested
-      if (data.openrag_docs_filter_id) {
-        // Save to backend
-        updateOnboardingMutation.mutateAsync({
-          openrag_docs_filter_id: data.openrag_docs_filter_id,
-        });
-        
-        console.log("Saved OpenRAG docs filter ID:", data.openrag_docs_filter_id);
-      }
-
       // Update provider health cache to healthy since backend just validated
       const provider =
         (isEmbedding ? settings.embedding_provider : settings.llm_provider) ||