fix(chat): update recent models when agents added to workspace from c…

…hat (Issue #2822) (#2951) Co-authored-by: Magomed-Elbi Dzhukalaev <[email protected]>
epam · Jan 17, 2025 · 245c974 · 245c974
1 parent 5d38692
commit 245c974
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/apps/chat/src/components/Chat/ChatInput/AddModelsControl.tsx b/apps/chat/src/components/Chat/ChatInput/AddModelsControl.tsx
@@ -42,6 +42,7 @@ export const AddModelsControl = ({
       ModelsActions.addInstalledModels({
         references: modelIdsToInstall,
         showSuccessToast: true,
+        updateRecentModels: true,
       }),
     );
   };

diff --git a/apps/chat/src/store/models/models.epics.ts b/apps/chat/src/store/models/models.epics.ts
@@ -358,6 +358,13 @@ const addInstalledModelsEpic: AppEpic = (action$, state$) =>
                   ),
                 );
               }
+              if (payload.updateRecentModels) {
+                actions.push(
+                  ...newInstalledModels.map(({ id }) =>
+                    of(ModelsActions.updateRecentModels({ modelId: id })),
+                  ),
+                );
+              }
 
               return concat(
                 ...actions,

diff --git a/apps/chat/src/store/models/models.reducers.ts b/apps/chat/src/store/models/models.reducers.ts
@@ -81,6 +81,7 @@ export const modelsSlice = createSlice({
       _action: PayloadAction<{
         references: string[];
         showSuccessToast?: boolean;
+        updateRecentModels?: boolean;
       }>,
     ) => state,
     removeInstalledModels: (