Merge pull request #429 from Kilo-Org/feat/vscode-worktree-image-attachments

feat(vscode): forward image attachments through worktree session creation

Author: iscekic

packages/kilo-vscode/src/agent-manager/AgentManagerProvider.ts

@@ -196,11 +196,17 @@ export class AgentManagerProvider implements vscode.Disposable {
     })
 
     // Step 4: Send the first message directly via httpClient
-    // TODO: When image/file attachments are supported, build the parts array
-    // from msg.files the same way KiloProvider.handleSendMessage does.
-    if (text) {
+    const fileParts = ((msg.files ?? []) as Array<{ mime: string; url: string }>).map((f) => ({
+      type: "file" as const,
+      mime: f.mime,
+      url: f.url,
+    }))
+    if (text || fileParts.length > 0) {
       try {
-        const parts: Array<{ type: "text"; text: string }> = [{ type: "text", text }]
+        const parts: Array<{ type: "text"; text: string } | { type: "file"; mime: string; url: string }> = [
+          ...fileParts,
+          ...(text ? [{ type: "text" as const, text }] : []),
+        ]
         await client.sendMessage(session.id, parts, worktree.path, {
           providerID: msg.providerID as string | undefined,
           modelID: msg.modelID as string | undefined,

packages/kilo-vscode/webview-ui/src/components/chat/PromptInput.tsx

@@ -285,6 +285,7 @@ export const PromptInput: Component = () => {
         providerID: sel?.providerID,
         modelID: sel?.modelID,
         agent: session.selectedAgent(),
+        files: attachments,
       })
     } else {
       session.sendMessage(message, sel?.providerID, sel?.modelID, attachments)

packages/kilo-vscode/webview-ui/src/types/messages.ts

@@ -736,6 +736,7 @@ export interface CreateWorktreeSessionRequest {
   providerID?: string
   modelID?: string
   agent?: string
+  files?: FileAttachment[]
 }
 
 export type WebviewMessage =