From ce8d7adef39496614fcca2f6ffe11feaf19e0fca Mon Sep 17 00:00:00 2001
From: kirillk <kirillk@kilocode.ai>
Date: Mon, 27 Apr 2026 13:43:51 -0400
Subject: [PATCH] fix(vscode): wire tab worktree actions

---
 packages/kilo-vscode/src/extension.ts                  |  6 ++++++
 .../webview-ui/src/components/chat/ChatView.tsx        |  2 +-
 .../webview-ui/src/styles/session-actions.css          | 10 ++++++++++
 3 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/packages/kilo-vscode/src/extension.ts b/packages/kilo-vscode/src/extension.ts
index 15109883e9..fcd6ff1eb8 100644
--- a/packages/kilo-vscode/src/extension.ts
+++ b/packages/kilo-vscode/src/extension.ts
@@ -148,6 +148,9 @@ export function activate(context: vscode.ExtensionContext) {
         tabProvider.setContinueInWorktreeHandler((sessionId, progress) =>
           agentManagerProvider.continueFromSidebar(sessionId, progress),
         )
+        tabProvider.setCreateWorktreeHandler((baseBranch, branchName) =>
+          agentManagerProvider.createFromSidebar(baseBranch, branchName),
+        )
         tabProvider.setDiffVirtualProvider(diffVirtualProvider)
         tabProvider.resolveWebviewPanel(panel)
         tabPanels.set(panel, tabProvider)
@@ -456,6 +459,9 @@ async function openKiloInNewTab(
   tabProvider.setContinueInWorktreeHandler((sessionId, progress) =>
     agentManagerProvider.continueFromSidebar(sessionId, progress),
   )
+  tabProvider.setCreateWorktreeHandler((baseBranch, branchName) =>
+    agentManagerProvider.createFromSidebar(baseBranch, branchName),
+  )
   tabProvider.setDiffVirtualProvider(diffVirtualProvider)
   tabProvider.resolveWebviewPanel(panel)
   tabPanels.set(panel, tabProvider)
diff --git a/packages/kilo-vscode/webview-ui/src/components/chat/ChatView.tsx b/packages/kilo-vscode/webview-ui/src/components/chat/ChatView.tsx
index ba1523182b..0b656f5c02 100644
--- a/packages/kilo-vscode/webview-ui/src/components/chat/ChatView.tsx
+++ b/packages/kilo-vscode/webview-ui/src/components/chat/ChatView.tsx
@@ -276,7 +276,7 @@ export const ChatView: Component<ChatViewProps> = (props) => {
                 </span>
               </Button>
             </Tooltip>
-            <Tooltip value={changesTooltip()} placement="top">
+            <Tooltip value={changesTooltip()} placement="top" class="session-move-changes-trigger">
               <Button
                 variant="ghost"
                 size="small"
diff --git a/packages/kilo-vscode/webview-ui/src/styles/session-actions.css b/packages/kilo-vscode/webview-ui/src/styles/session-actions.css
index 303de23864..e027aa6d40 100644
--- a/packages/kilo-vscode/webview-ui/src/styles/session-actions.css
+++ b/packages/kilo-vscode/webview-ui/src/styles/session-actions.css
@@ -36,6 +36,10 @@
   flex: 0 0 auto;
 }
 
+.session-actions-row > .session-move-changes-trigger {
+  margin-left: auto;
+}
+
 .session-actions-row > [data-component="tooltip-trigger"] > [data-component="button"] {
   flex: 0 0 auto;
   border-radius: var(--session-action-radius);
@@ -359,3 +363,9 @@
     opacity: 1;
   }
 }
+
+@container (max-width: 420px) {
+  .session-actions-row > .session-move-changes-trigger {
+    margin-left: 0;
+  }
+}