Function SendToolProgress

pkg/aiusechat/aiutil/aiutil.go:286–314

SendToolProgress sends tool progress updates via SSE if the tool has a progress descriptor

(toolCallID, toolName string, jsonData []byte, chatOpts uctypes.WaveChatOpts, sseHandler *sse.SSEHandlerCh, usePartialParse bool)

// 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) {
	toolDef := chatOpts.GetToolDefinition(toolName)
	if toolDef == nil || toolDef.ToolProgressDesc == nil {
		return
	}

	var parsedJSON any
	var err error
	if usePartialParse {
		parsedJSON, err = utilfn.ParsePartialJson(jsonData)
	} else {
		err = json.Unmarshal(jsonData, &parsedJSON)
	}
	if err != nil {
		return
	}

	statusLines, err := toolDef.ToolProgressDesc(parsedJSON)
	if err != nil {
		return
	}

	progressData := &uctypes.UIMessageDataToolProgress{
		ToolCallId:  toolCallID,
		ToolName:    toolName,
		StatusLines: statusLines,
	}
	_ = sseHandler.AiMsgData("data-toolprogress", "progress-"+toolCallID, progressData)
}