feat: ui message

2026-04-25 07:00:48 +09:00 · 2026-04-10 16:44:44 +08:00
parent d3bf6bc90a
commit 51495ba583
19 changed files with 2141 additions and 774 deletions
@@ -0,0 +1,176 @@
+package conversation
+
+import "strings"
+
+type uiTextStreamState struct {
+	ID      int
+	Content string
+}
+
+type uiToolStreamState struct {
+	Message UIMessage
+}
+
+// UIMessageStreamConverter converts low-level stream events into complete UI messages.
+type UIMessageStreamConverter struct {
+	nextID    int
+	text      *uiTextStreamState
+	reasoning *uiTextStreamState
+	tools     map[string]*uiToolStreamState
+}
+
+// NewUIMessageStreamConverter creates a new UI stream converter.
+func NewUIMessageStreamConverter() *UIMessageStreamConverter {
+	return &UIMessageStreamConverter{
+		tools: map[string]*uiToolStreamState{},
+	}
+}
+
+// HandleEvent updates converter state and returns zero or one complete UI messages.
+func (c *UIMessageStreamConverter) HandleEvent(event UIMessageStreamEvent) []UIMessage {
+	switch strings.ToLower(strings.TrimSpace(event.Type)) {
+	case "text_start":
+		c.text = &uiTextStreamState{ID: c.nextMessageID()}
+		return nil
+
+	case "text_delta":
+		if c.text == nil {
+			c.text = &uiTextStreamState{ID: c.nextMessageID()}
+		}
+		c.text.Content += event.Delta
+		return []UIMessage{{
+			ID:      c.text.ID,
+			Type:    UIMessageText,
+			Content: c.text.Content,
+		}}
+
+	case "text_end":
+		c.text = nil
+		return nil
+
+	case "reasoning_start":
+		c.reasoning = &uiTextStreamState{ID: c.nextMessageID()}
+		return nil
+
+	case "reasoning_delta":
+		if c.reasoning == nil {
+			c.reasoning = &uiTextStreamState{ID: c.nextMessageID()}
+		}
+		c.reasoning.Content += event.Delta
+		return []UIMessage{{
+			ID:      c.reasoning.ID,
+			Type:    UIMessageReasoning,
+			Content: c.reasoning.Content,
+		}}
+
+	case "reasoning_end":
+		c.reasoning = nil
+		return nil
+
+	case "tool_call_start":
+		state := &uiToolStreamState{
+			Message: UIMessage{
+				ID:         c.nextMessageID(),
+				Type:       UIMessageTool,
+				Name:       strings.TrimSpace(event.ToolName),
+				Input:      event.Input,
+				ToolCallID: strings.TrimSpace(event.ToolCallID),
+				Running:    uiBoolPtr(true),
+			},
+		}
+		if state.Message.ToolCallID != "" {
+			c.tools[state.Message.ToolCallID] = state
+		}
+		c.text = nil
+		return []UIMessage{state.Message}
+
+	case "tool_call_progress":
+		state := c.findToolState(event.ToolCallID, event.ToolName)
+		if state == nil {
+			state = &uiToolStreamState{
+				Message: UIMessage{
+					ID:         c.nextMessageID(),
+					Type:       UIMessageTool,
+					Name:       strings.TrimSpace(event.ToolName),
+					Input:      event.Input,
+					ToolCallID: strings.TrimSpace(event.ToolCallID),
+					Running:    uiBoolPtr(true),
+				},
+			}
+			if state.Message.ToolCallID != "" {
+				c.tools[state.Message.ToolCallID] = state
+			}
+		}
+		state.Message.Progress = append(state.Message.Progress, event.Progress)
+		if event.Input != nil {
+			state.Message.Input = event.Input
+		}
+		return []UIMessage{cloneToolStreamMessage(state.Message)}
+
+	case "tool_call_end":
+		state := c.findToolState(event.ToolCallID, event.ToolName)
+		if state == nil {
+			state = &uiToolStreamState{
+				Message: UIMessage{
+					ID:         c.nextMessageID(),
+					Type:       UIMessageTool,
+					Name:       strings.TrimSpace(event.ToolName),
+					Input:      event.Input,
+					ToolCallID: strings.TrimSpace(event.ToolCallID),
+				},
+			}
+		}
+		if event.Input != nil {
+			state.Message.Input = event.Input
+		}
+		state.Message.Output = event.Output
+		state.Message.Running = uiBoolPtr(false)
+		if state.Message.ToolCallID != "" {
+			delete(c.tools, state.Message.ToolCallID)
+		}
+		return []UIMessage{cloneToolStreamMessage(state.Message)}
+
+	case "attachment_delta":
+		if len(event.Attachments) == 0 {
+			return nil
+		}
+		return []UIMessage{{
+			ID:          c.nextMessageID(),
+			Type:        UIMessageAttachments,
+			Attachments: append([]UIAttachment(nil), event.Attachments...),
+		}}
+
+	default:
+		return nil
+	}
+}
+
+func (c *UIMessageStreamConverter) nextMessageID() int {
+	id := c.nextID
+	c.nextID++
+	return id
+}
+
+func (c *UIMessageStreamConverter) findToolState(toolCallID, toolName string) *uiToolStreamState {
+	if trimmed := strings.TrimSpace(toolCallID); trimmed != "" {
+		if state, ok := c.tools[trimmed]; ok {
+			return state
+		}
+	}
+
+	normalizedName := strings.TrimSpace(toolName)
+	for _, state := range c.tools {
+		if strings.TrimSpace(state.Message.Name) == normalizedName {
+			return state
+		}
+	}
+	return nil
+}
+
+func cloneToolStreamMessage(message UIMessage) UIMessage {
+	clone := message
+	if len(message.Progress) > 0 {
+		clone.Progress = append([]any(nil), message.Progress...)
+	}
+	return clone
+}