From ed91948fe4e0a12ddac53b77e02e2471e23d1afb Mon Sep 17 00:00:00 2001 From: Christian Rocha Date: Thu, 14 May 2026 09:45:18 -0400 Subject: [PATCH] fix(ui): regression in tool output in client/server mode (#2878) --- internal/proto/message.go | 2 ++ internal/server/events.go | 3 +++ internal/workspace/client_workspace.go | 3 +++ 3 files changed, 8 insertions(+) diff --git a/internal/proto/message.go b/internal/proto/message.go index 0ec3fb91254029eba197fd91a59ec8e92326598b..f1ae259cf2cd3238b522c31cd46387abc4fd6173 100644 --- a/internal/proto/message.go +++ b/internal/proto/message.go @@ -153,6 +153,8 @@ type ToolResult struct { ToolCallID string `json:"tool_call_id"` Name string `json:"name"` Content string `json:"content"` + Data string `json:"data,omitempty"` + MIMEType string `json:"mime_type,omitempty"` Metadata string `json:"metadata"` IsError bool `json:"is_error"` } diff --git a/internal/server/events.go b/internal/server/events.go index 752311666bb6fcc2b1efde4d037711eaafaa0162..57536044bcdc9999d92a6fb5431297c19ef2a761 100644 --- a/internal/server/events.go +++ b/internal/server/events.go @@ -186,6 +186,9 @@ func messageToProto(m message.Message) proto.Message { ToolCallID: v.ToolCallID, Name: v.Name, Content: v.Content, + Data: v.Data, + MIMEType: v.MIMEType, + Metadata: v.Metadata, IsError: v.IsError, }) case message.Finish: diff --git a/internal/workspace/client_workspace.go b/internal/workspace/client_workspace.go index 7c4e1408882cc70859ea2ab05981461d262513e9..aba4951504c3a7848398980a0ddd870d84d458cb 100644 --- a/internal/workspace/client_workspace.go +++ b/internal/workspace/client_workspace.go @@ -722,6 +722,9 @@ func protoToMessage(m proto.Message) message.Message { ToolCallID: v.ToolCallID, Name: v.Name, Content: v.Content, + Data: v.Data, + MIMEType: v.MIMEType, + Metadata: v.Metadata, IsError: v.IsError, }) case proto.Finish: