MediaMetz/gemini-cli
1
0
Fork 0
mirror of https://github.com/google-gemini/gemini-cli.git synced 2026-07-22 15:51:18 -07:00
Code Issues Packages Projects Releases Wiki Activity

token calculation service

Browse Source
This commit is contained in:
Your Name
2026-04-06 19:54:09 +00:00
parent c1b06fec0d
commit 1774abebe9
18 changed files with 128 additions and 109 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/core/src/context/ir/mapper.test.ts
+6 -3
View File
@@ -6,6 +6,7 @@
import { describe, it, expect } from 'vitest';
import { IrMapper } from './mapper.js';
import { ContextTokenCalculator } from '../utils/contextTokenCalculator.js';
import type { Content } from '@google/genai';
import type { UserPrompt, ToolExecution } from './types.js';
@@ -80,7 +81,8 @@ describe('IrMapper', () => {
{ role: 'model', parts: [{ text: 'Everything is done!' }] },
];
const episodes = IrMapper.toIr(rawHistory);
const tokenCalculator = new ContextTokenCalculator(4);
const episodes = IrMapper.toIr(rawHistory, tokenCalculator);
expect(episodes).toHaveLength(1);
const ep = episodes[0];
@@ -135,7 +137,8 @@ describe('IrMapper', () => {
{ role: 'model', parts: [{ text: 'Hi there' }] },
];
const initialIr = IrMapper.toIr(history);
const tokenCalculator = new ContextTokenCalculator(4);
const initialIr = IrMapper.toIr(history, tokenCalculator);
expect(initialIr).toHaveLength(1);
// Save the uniquely generated deterministic ID for the first episode
@@ -146,7 +149,7 @@ describe('IrMapper', () => {
history.push({ role: 'user', parts: [{ text: 'How are you?' }] });
history.push({ role: 'model', parts: [{ text: 'I am an AI.' }] });
const updatedIr = IrMapper.toIr(history);
const updatedIr = IrMapper.toIr(history, tokenCalculator);
expect(updatedIr).toHaveLength(2);
// 3. Verify ID Stability