File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -1094,6 +1094,8 @@ public struct MTPTokenIterator: TokenIteratorProtocol {
10941094 private var pendingIndex = 0
10951095
10961096 // Internal metrics
1097+ public var acceptedDraftTokens : Int = 0
1098+ public var totalDraftTokens : Int = 0
10971099 var promptPrefillTime : TimeInterval = 0.0
10981100
10991101 /// Initialize a `MTPTokenIterator` with the given input.
@@ -1317,10 +1319,8 @@ public struct MTPTokenIterator: TokenIteratorProtocol {
13171319 processor? . didSample ( token: finalTokenOut)
13181320 pendingTokens. append ( finalTokenOut. item ( Int . self) )
13191321 }
1320-
1321- if draftTokens. count > 0 {
1322- print ( " [MTP Debug] Drafts: \( draftTokensList) , Target: \( mainTokensList) , Accepted: \( accepted) " )
1323- }
1322+ self . acceptedDraftTokens += accepted
1323+ self . totalDraftTokens += draftTokens. count
13241324
13251325 // Rewind caches for rejected tokens
13261326 let rejectedCount = draftTokens. count - accepted
You can’t perform that action at this time.
0 commit comments