| 574 | } |
| 575 | |
| 576 | void ChatModel::setMessageUsage( |
| 577 | const QString &messageId, |
| 578 | int promptTokens, |
| 579 | int completionTokens, |
| 580 | int cachedPromptTokens, |
| 581 | int reasoningTokens) |
| 582 | { |
| 583 | for (int i = 0; i < m_messages.size(); ++i) { |
| 584 | if (m_messages[i].id != messageId) |
| 585 | continue; |
| 586 | m_messages[i].promptTokens = promptTokens; |
| 587 | m_messages[i].completionTokens = completionTokens; |
| 588 | m_messages[i].cachedPromptTokens = cachedPromptTokens; |
| 589 | m_messages[i].reasoningTokens = reasoningTokens; |
| 590 | emit dataChanged( |
| 591 | index(i), |
| 592 | index(i), |
| 593 | {Roles::PromptTokens, |
| 594 | Roles::CompletionTokens, |
| 595 | Roles::CachedPromptTokens, |
| 596 | Roles::ReasoningTokens, |
| 597 | Roles::TotalTokens}); |
| 598 | emit sessionUsageChanged(); |
| 599 | return; |
| 600 | } |
| 601 | } |
| 602 | |
| 603 | int ChatModel::sessionPromptTokens() const |
| 604 | { |
no test coverage detected