From 71857e35997c1f8b3d1ae0235dfb59fd9310838b Mon Sep 17 00:00:00 2001
From: marius-kilocode <marius@kilocode.ai>
Date: Thu, 25 Jun 2026 11:49:28 +0200
Subject: [PATCH] test(vscode): keep prompt stories inside providers

---
 .../src/stories/prompt-input.stories.tsx      | 20 +++++++++----------
 1 file changed, 9 insertions(+), 11 deletions(-)

diff --git a/packages/kilo-vscode/webview-ui/src/stories/prompt-input.stories.tsx b/packages/kilo-vscode/webview-ui/src/stories/prompt-input.stories.tsx
index e31a1b4df60..72d5d1ada32 100644
--- a/packages/kilo-vscode/webview-ui/src/stories/prompt-input.stories.tsx
+++ b/packages/kilo-vscode/webview-ui/src/stories/prompt-input.stories.tsx
@@ -60,12 +60,6 @@ const PromptProviders: ParentComponent<{
     workspaceDirectory: () => "/workspace",
     gitInstalled: () => true,
   }
-  const content = (
-    <div style={{ overflow: "hidden" }}>
-      <SessionContext.Provider value={session as any}>{props.children}</SessionContext.Provider>
-    </div>
-  )
-
   return (
     <StoryProviders
       noPadding
@@ -77,11 +71,15 @@ const PromptProviders: ParentComponent<{
       features={props.sandboxNetwork ? { sandboxControls: true } : undefined}
     >
       {/* overflow:hidden prevents margin-collapse so top/bottom borders are captured in screenshots */}
-      {props.sandboxNetwork ? (
-        <ServerContext.Provider value={server as any}>{content}</ServerContext.Provider>
-      ) : (
-        content
-      )}
+      <div style={{ overflow: "hidden" }}>
+        <SessionContext.Provider value={session as any}>
+          {props.sandboxNetwork ? (
+            <ServerContext.Provider value={server as any}>{props.children}</ServerContext.Provider>
+          ) : (
+            props.children
+          )}
+        </SessionContext.Provider>
+      </div>
     </StoryProviders>
   )
 }