From 68cbe2066453e24dfd65ca249f83a0a66aac98c8 Mon Sep 17 00:00:00 2001
From: Your Name <you@example.com>
Date: Sun, 7 Dec 2025 17:56:06 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E6=B7=BB=E5=8A=A0Kiro=E6=B8=A0?=
 =?UTF-8?q?=E9=81=93=E5=9B=BE=E7=89=87=E6=94=AF=E6=8C=81=E5=8A=9F=E8=83=BD?=
 =?UTF-8?q?=EF=BC=8C=E5=80=9F=E9=89=B4justlovemaki/AIClient-2-API=E5=AE=9E?=
 =?UTF-8?q?=E7=8E=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 internal/runtime/executor/kiro_executor.go | 43 +++++++++++++++++++++-
 1 file changed, 41 insertions(+), 2 deletions(-)

diff --git a/internal/runtime/executor/kiro_executor.go b/internal/runtime/executor/kiro_executor.go
index eb068c14..0157d68c 100644
--- a/internal/runtime/executor/kiro_executor.go
+++ b/internal/runtime/executor/kiro_executor.go
@@ -604,10 +604,22 @@ type kiroHistoryMessage struct {
 	AssistantResponseMessage *kiroAssistantResponseMessage `json:"assistantResponseMessage,omitempty"`
 }
 
+// kiroImage represents an image in Kiro API format
+type kiroImage struct {
+	Format string          `json:"format"`
+	Source kiroImageSource `json:"source"`
+}
+
+// kiroImageSource contains the image data
+type kiroImageSource struct {
+	Bytes string `json:"bytes"` // base64 encoded image data
+}
+
 type kiroUserInputMessage struct {
 	Content                 string                       `json:"content"`
 	ModelID                 string                       `json:"modelId"`
 	Origin                  string                       `json:"origin"`
+	Images                  []kiroImage                  `json:"images,omitempty"`
 	UserInputMessageContext *kiroUserInputMessageContext `json:"userInputMessageContext,omitempty"`
 }
 
@@ -824,6 +836,7 @@ func (e *KiroExecutor) buildUserMessageStruct(msg gjson.Result, modelID, origin
 	content := msg.Get("content")
 	var contentBuilder strings.Builder
 	var toolResults []kiroToolResult
+	var images []kiroImage
 
 	if content.IsArray() {
 		for _, part := range content.Array() {
@@ -831,6 +844,25 @@ func (e *KiroExecutor) buildUserMessageStruct(msg gjson.Result, modelID, origin
 			switch partType {
 			case "text":
 				contentBuilder.WriteString(part.Get("text").String())
+			case "image":
+				// Extract image data from Claude API format
+				mediaType := part.Get("source.media_type").String()
+				data := part.Get("source.data").String()
+				
+				// Extract format from media_type (e.g., "image/png" -> "png")
+				format := ""
+				if idx := strings.LastIndex(mediaType, "/"); idx != -1 {
+					format = mediaType[idx+1:]
+				}
+				
+				if format != "" && data != "" {
+					images = append(images, kiroImage{
+						Format: format,
+						Source: kiroImageSource{
+							Bytes: data,
+						},
+					})
+				}
 			case "tool_result":
 				// Extract tool result for API
 				toolUseID := part.Get("tool_use_id").String()
@@ -872,11 +904,18 @@ func (e *KiroExecutor) buildUserMessageStruct(msg gjson.Result, modelID, origin
 		contentBuilder.WriteString(content.String())
 	}
 
-	return kiroUserInputMessage{
+	userMsg := kiroUserInputMessage{
 		Content: contentBuilder.String(),
 		ModelID: modelID,
 		Origin:  origin,
-	}, toolResults
+	}
+
+	// Add images to message if present
+	if len(images) > 0 {
+		userMsg.Images = images
+	}
+
+	return userMsg, toolResults
 }
 
 // buildAssistantMessageStruct builds an assistant message with tool uses