2 files changed, 16 insertions, 0 deletions
diff --git a/src/features/tabs/tabs-store.svelte.ts b/src/features/tabs/tabs-store.svelte.ts
index d044b1f..acf56fe 100644
--- a/src/features/tabs/tabs-store.svelte.ts
+++ b/src/features/tabs/tabs-store.svelte.ts
@@ -8,6 +8,7 @@ import {
   selectTab as reduceSelectTab,
   setModel as reduceSetModel,
   setTitle as reduceSetTitle,
+  setWorkspaceId as reduceSetWorkspaceId,
   activeTab as selectActiveTab,
 } from "./tabs";
 
@@ -28,6 +29,7 @@ export interface TabsStore {
   closeTab(conversationId: string): void;
   setModel(conversationId: string, model: string): void;
   setTitle(conversationId: string, title: string): void;
+  setWorkspaceId(conversationId: string, workspaceId: string): void;
 }
 
 export function createTabsStore(storage: TabsStorage): TabsStore {
@@ -69,5 +71,8 @@ export function createTabsStore(storage: TabsStorage): TabsStore {
     setTitle(conversationId: string, title: string): void {
       apply(reduceSetTitle(state, conversationId, title));
     },
+    setWorkspaceId(conversationId: string, workspaceId: string): void {
+      apply(reduceSetWorkspaceId(state, conversationId, workspaceId));
+    },
   };
 }
diff --git a/src/features/tabs/tabs.ts b/src/features/tabs/tabs.ts
index 63cda35..9fc9306 100644
--- a/src/features/tabs/tabs.ts
+++ b/src/features/tabs/tabs.ts
@@ -82,6 +82,17 @@ export function setTitle(state: TabsState, conversationId: string, title: string
   return { tabs, activeConversationId: state.activeConversationId };
 }
 
+export function setWorkspaceId(
+  state: TabsState,
+  conversationId: string,
+  workspaceId: string,
+): TabsState {
+  const tabs = state.tabs.map((t) =>
+    t.conversationId === conversationId ? { ...t, workspaceId } : t,
+  );
+  return { tabs, activeConversationId: state.activeConversationId };
+}
+
 export function activeTab(state: TabsState): Tab | null {
   if (state.activeConversationId === null) return null;
   return state.tabs.find((t) => t.conversationId === state.activeConversationId) ?? null;