elevenlabs-swift-sdk/Sources/ElevenLabs/Internal/Conversation/ChatHistoryReconciler.swift at 51f71192cdbba63e3d6a59677f478c5b34008afa · elevenlabs/elevenlabs-swift-sdk · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
struct ChatHistoryReconciler {
    private(set) var items: [any ChatHistoryItem] = []

    mutating func appendUserMessage(_ content: String) {
        items.append(Message(
            role: .user,
            content: content,
            isFinal: true
        ))
    }

    mutating func receive(_ event: TentativeUserTranscriptEvent) {
        removeTentativeUserMessages()
        items.append(Message(
            role: .user,
            content: event.transcript,
            isFinal: false,
            eventId: event.eventId
        ))
    }

    mutating func receive(_ event: UserTranscriptEvent) {
        if let index = items.firstIndex(where: {
            $0.message?.role == .user
                && $0.message?.eventId == event.eventId
        }) {
            updateMessage(at: index) {
                $0.content = event.transcript
                $0.isFinal = true
            }
            return
        }

        removeTentativeUserMessages()
        items.append(Message(
            role: .user,
            content: event.transcript,
            isFinal: true,
            eventId: event.eventId
        ))
    }

    mutating func receive(_ event: AgentChatResponsePartEvent) {
        let isEnd = event.type == .stop
        if let index = items.firstIndex(where: { $0.id == event.responseId }) {
            updateMessage(at: index) {
                guard !$0.isFinal else { return }
                $0.content += event.text
                $0.isFinal = isEnd
            }
            return
        }

        items.append(Message(
            role: .agent,
            content: event.text,
            isFinal: isEnd,
            eventId: event.eventId,
            responseId: event.responseId
        ))
    }

    mutating func receive(_ event: AgentResponseEvent) {
        upsertAgentMessage(
            responseId: event.responseId,
            eventId: event.eventId,
            content: event.response
        )
    }

    mutating func receive(_ event: AgentResponseCorrectionEvent) {
        upsertAgentMessage(
            responseId: event.responseId,
            eventId: event.eventId,
            content: event.correctedAgentResponse
        )
    }

    mutating func receive(_ event: AgentToolResponseEvent) {
        guard !items.contains(where: {
            $0.toolCall?.toolCallId == event.toolCallId
        }) else { return }
        items.append(ConversationToolCall(
            toolCallId: event.toolCallId,
            toolName: event.toolName
        ))
    }

    private mutating func upsertAgentMessage(responseId: String, eventId: Int, content: String) {
        if let index = items.firstIndex(where: { $0.id == responseId }) {
            updateMessage(at: index) {
                $0.content = content
                $0.isFinal = true
                $0.eventId = eventId
            }
        } else {
            items.append(Message(
                role: .agent,
                content: content,
                isFinal: true,
                eventId: eventId,
                responseId: responseId
            ))
        }
    }

    private mutating func updateMessage(at index: Int, update: (inout Message) -> Void) {
        guard var message = items[index] as? Message else { return }
        update(&message)
        items[index] = message
    }

    private mutating func removeTentativeUserMessages() {
        items.removeAll {
            guard let message = $0.message else { return false }
            return message.role == .user
                && !message.isFinal
        }
    }
}