All files / src/provider context-limit-store.ts

100% Statements 11/11
100% Branches 2/2
100% Functions 3/3
100% Lines 11/11

Press n or j to go to the next uncovered block, b, p or k for the previous block.

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            7x 167x     11x       138x 138x 133x   5x 1x 1x   4x       5x      
interface RuntimeLimitEntry {
  limit: number;
  keyFingerprint: string;
  updatedAt: number;
}
 
export class ContextLimitStore {
  private readonly entries = new Map<string, RuntimeLimitEntry>();
 
  set(modelId: string, limit: number, keyFingerprint: string): void {
    this.entries.set(modelId, { limit, keyFingerprint, updatedAt: Date.now() });
  }
 
  get(modelId: string, keyFingerprint: string): number | undefined {
    const entry = this.entries.get(modelId);
    if (!entry) {
      return undefined;
    }
    if (entry.keyFingerprint !== keyFingerprint) {
      this.entries.delete(modelId);
      return undefined;
    }
    return entry.limit;
  }
 
  clear(): void {
    this.entries.clear();
  }
}