RegisterCounter 注册一个模型的计数器
(modelName string, counter TokenCounter)
| 370 | |
| 371 | // RegisterCounter 注册一个模型的计数器 |
| 372 | func (m *MultiModelTokenCounter) RegisterCounter(modelName string, counter TokenCounter) { |
| 373 | m.counters[modelName] = counter |
| 374 | } |
| 375 | |
| 376 | // GetCounter 获取指定模型的计数器 |
| 377 | func (m *MultiModelTokenCounter) GetCounter(modelName string) TokenCounter { |
no outgoing calls