Function CreateToolUseData

pkg/aiusechat/aiutil/aiutil.go:240–283

CreateToolUseData creates a UIMessageDataToolUse from tool call information

(toolCallID, toolName string, arguments string, chatOpts uctypes.WaveChatOpts)

// CreateToolUseData creates a UIMessageDataToolUse from tool call information
func CreateToolUseData(toolCallID, toolName string, arguments string, chatOpts uctypes.WaveChatOpts) uctypes.UIMessageDataToolUse {
	toolUseData := uctypes.UIMessageDataToolUse{
		ToolCallId: toolCallID,
		ToolName:   toolName,
		Status:     uctypes.ToolUseStatusPending,
	}

	toolDef := chatOpts.GetToolDefinition(toolName)
	if toolDef == nil {
		toolUseData.Status = uctypes.ToolUseStatusError
		toolUseData.ErrorMessage = "tool not found"
		return toolUseData
	}

	var parsedArgs any
	if err := json.Unmarshal([]byte(arguments), &parsedArgs); err != nil {
		toolUseData.Status = uctypes.ToolUseStatusError
		toolUseData.ErrorMessage = fmt.Sprintf("failed to parse tool arguments: %v", err)
		return toolUseData
	}

	if toolDef.ToolCallDesc != nil {
		toolUseData.ToolDesc = toolDef.ToolCallDesc(parsedArgs, nil, nil)
	}

	if toolDef.ToolApproval != nil {
		toolUseData.Approval = toolDef.ToolApproval(parsedArgs)
	}

	if chatOpts.TabId != "" {
		if argsMap, ok := parsedArgs.(map[string]any); ok {
			if widgetId, ok := argsMap["widget_id"].(string); ok && widgetId != "" {
				ctx, cancelFn := context.WithTimeout(context.Background(), 2*time.Second)
				defer cancelFn()
				fullBlockId, err := wcore.ResolveBlockIdFromPrefix(ctx, chatOpts.TabId, widgetId)
				if err == nil {
					toolUseData.BlockId = fullBlockId
				}
			}
		}
	}

	return toolUseData
}

// SendToolProgress sends tool progress updates via SSE if the tool has a progress descriptor
func SendToolProgress(toolCallID, toolName string, jsonData []byte, chatOpts uctypes.WaveChatOpts, sseHandler *sse.SSEHandlerCh, usePartialParse bool) {