Function RemoveToolUseCall

pkg/aiusechat/gemini/gemini-convertmessage.go:461–508

(chatId string, toolCallId string)

}

func RemoveToolUseCall(chatId string, toolCallId string) error {
	chat := chatstore.DefaultChatStore.Get(chatId)
	if chat == nil {
		return fmt.Errorf("chat not found: %s", chatId)
	}

	for _, genMsg := range chat.NativeMessages {
		chatMsg, ok := genMsg.(*GeminiChatMessage)
		if !ok {
			continue
		}

		partIndex := -1
		for i, part := range chatMsg.Parts {
			if part.FunctionCall != nil && part.ToolUseData != nil && part.ToolUseData.ToolCallId == toolCallId {
				partIndex = i
				break
			}
		}

		if partIndex == -1 {
			continue
		}

		updatedMsg := &GeminiChatMessage{
			MessageId: chatMsg.MessageId,
			Role:      chatMsg.Role,
			Parts:     slices.Delete(slices.Clone(chatMsg.Parts), partIndex, partIndex+1),
			Usage:     chatMsg.Usage,
		}

		if len(updatedMsg.Parts) == 0 {
			chatstore.DefaultChatStore.RemoveMessage(chatId, chatMsg.MessageId)
		} else {
			aiOpts := &uctypes.AIOptsType{
				APIType:    chat.APIType,
				Model:      chat.Model,
				APIVersion: chat.APIVersion,
			}
			if err := chatstore.DefaultChatStore.PostMessage(chatId, aiOpts, updatedMsg); err != nil {
				return err
			}
		}
		return nil
	}

	return nil
}