@beeuni/sdk 0.1.0 → 0.3.0

package/dist/core/turns/classify.d.ts

@@ -1,6 +1,7 @@
 import type { MessageWithParts } from '../../transcript.js';
 import type { Part, ToolPart } from '../runtime/client.js';
 import { type GatewayAttemptFailure } from './errors.js';
+import type { ToolCallBlock } from '../session-log/types.js';
 import { type ToolKind } from './tool-kind.js';
 import type { TokenUsageLike } from './types.js';
 export type ToolStatus = 'pending' | 'running' | 'done' | 'error';
@@ -30,7 +31,7 @@ export interface ToolView {
     answers?: string[][];
     childSessionId?: string;
 }
-export declare function toToolView(part: Pick<ToolPart, 'tool' | 'state'>): ToolView;
+export declare function toToolView(call: Pick<ToolPart, 'tool' | 'state'> | ToolCallBlock): ToolView;
 export interface ClassifiedTextPart {
     kind: 'text';
     id: string;

package/dist/core/turns/classify.js

@@ -1,5 +1,7 @@
 import { extractGatewayErrorDetails, unwrapError } from './errors.js';
 import { getChildSessionId } from './parts.js';
+import { isToolCallBlock, toolPartOf } from '../session-log/tool-call.js';
+import { editBlocks } from './edit-blocks.js';
 import { inputPath, toolKind } from './tool-kind.js';
 import { toolInfo } from './tool-registry.js';
 const MAX_PARSEABLE_OUTPUT_LENGTH = 256 * 1024;
@@ -91,11 +93,12 @@ function toolFiles(kind, input, metadata) {
     const filediff = record(metadata.filediff);
     const patch = filediff ? undefined : text(metadata.patch);
     const counts = patch ? patchLineCounts(patch) : undefined;
+    const blocks = filediff || patch ? [] : editBlocks(input);
     return [
         defined({
             path,
-            before: text(filediff?.before) ?? text(input.oldString),
-            after: text(filediff?.after) ?? text(input.newString),
+            before: text(filediff?.before) ?? (blocks.length ? blocks.map(([before]) => before).join('\n') : undefined),
+            after: text(filediff?.after) ?? (blocks.length ? blocks.map(([, after]) => after).join('\n') : text(input.new_source)),
             patch,
             additions: count(filediff?.additions) ?? counts?.additions,
             deletions: count(filediff?.deletions) ?? counts?.deletions,
@@ -153,7 +156,8 @@ function classifyToolState(tool, state) {
         childSessionId: getChildSessionId({ tool, state }),
     };
 }
-export function toToolView(part) {
+export function toToolView(call) {
+    const part = isToolCallBlock(call) ? toolPartOf(call) : call;
     return classifyToolState(part.tool, part.state);
 }
 function classifyText(part) {

package/dist/core/turns/edit-blocks.d.ts

@@ -0,0 +1 @@
+export declare function editBlocks(input: Record<string, unknown> | undefined): Array<[string, string]>;

package/dist/core/turns/edit-blocks.js

@@ -0,0 +1,15 @@
+export function editBlocks(input) {
+    const one = pair(input);
+    if (one)
+        return [one];
+    const edits = Array.isArray(input?.edits) ? input.edits : [];
+    return edits.flatMap((edit) => {
+        const block = edit && typeof edit === 'object' ? pair(edit) : undefined;
+        return block ? [block] : [];
+    });
+}
+function pair(input) {
+    const before = input?.oldString ?? input?.old_string ?? input?.oldText;
+    const after = input?.newString ?? input?.new_string ?? input?.newText;
+    return typeof before === 'string' && typeof after === 'string' ? [before, after] : undefined;
+}

package/dist/core/turns/parts.d.ts

@@ -1,3 +1,4 @@
+import type { ToolCallBlock } from '../session-log/types.js';
 import type { Diagnostic, MessageWithPartsLike, PartLike, RequestWithToolLike, ToolInfo, ToolPartLike } from './types.js';
 export declare function isTextPart<P extends PartLike>(part: P): part is P & {
     type: 'text';
@@ -42,8 +43,8 @@ export interface HiddenToolRef {
 }
 export declare function getHiddenToolParts(permission: RequestWithToolLike | undefined, question: RequestWithToolLike | undefined): HiddenToolRef[];
 export declare function isToolPartHidden(part: Pick<ToolPartLike, 'callID'>, messageId: string, hidden: HiddenToolRef[]): boolean;
-export declare function shouldShowToolPart(part: Pick<ToolPartLike, 'tool'>): boolean;
-export declare function getChildSessionId(part: Pick<ToolPartLike, 'tool' | 'state'>): string | undefined;
+export declare function shouldShowToolPart(part: Pick<ToolPartLike, 'tool'> | ToolCallBlock): boolean;
+export declare function getChildSessionId(call: Pick<ToolPartLike, 'tool' | 'state'> | ToolCallBlock): string | undefined;
 export declare function getChildSessionError(childMessages: readonly MessageWithPartsLike[] | undefined): string | undefined;
 export declare function getChildSessionToolParts<M extends MessageWithPartsLike>(childMessages: readonly M[]): Array<M['parts'][number] & {
     type: 'tool';

package/dist/core/turns/parts.js

@@ -1,4 +1,5 @@
 import { unwrapError } from './errors.js';
+import { isToolCallBlock, toolPartOf } from '../session-log/tool-call.js';
 import { inputPath, toolKind } from './tool-kind.js';
 import { normalizeName } from './tools/tool-meta.js';
 export function isTextPart(part) {
@@ -67,7 +68,7 @@ export function isToolPartHidden(part, messageId, hidden) {
 }
 const HIDDEN_TOOLS = new Set(['todoread', 'context_info']);
 export function shouldShowToolPart(part) {
-    return !HIDDEN_TOOLS.has(part.tool);
+    return !HIDDEN_TOOLS.has(isToolCallBlock(part) ? part.name : part.tool);
 }
 const CHILD_SESSION_TOOLS = new Set([
     'task',
@@ -82,7 +83,8 @@ const CHILD_SESSION_TOOLS = new Set([
     'task_update',
     'task_message',
 ]);
-export function getChildSessionId(part) {
+export function getChildSessionId(call) {
+    const part = isToolCallBlock(call) ? toolPartOf(call) : call;
     const tool = part.tool || '';
     if (tool.startsWith('oc-'))
         return undefined;

package/dist/core/turns/segments/burst-summary.d.ts

@@ -7,3 +7,4 @@ export interface BurstSummary {
 }
 export declare function burstSummary(parts: ReadonlyArray<Part>): BurstSummary;
 export declare function burstSummaryLabel(summary: BurstSummary, running: boolean): string;
+export declare function burstWorkLabel(parts: ReadonlyArray<Part>): string;

package/dist/core/turns/segments/burst-summary.js

@@ -1,5 +1,6 @@
 import { isReasoningPart, isToolPart } from '../parts.js';
 import { partOutcome } from '../tools/tool-outcome.js';
+import { familyForTool, narrateStep } from './narration.js';
 import { stepLabel } from './step-label.js';
 export function burstSummary(parts) {
     let total = 0;
@@ -43,3 +44,26 @@ export function burstSummaryLabel(summary, running) {
         return `${steps(total)} failed`;
     return `Completed ${completed} of ${total} steps · ${failed} failed`;
 }
+export function burstWorkLabel(parts) {
+    const families = new Map();
+    let failed = 0;
+    for (const part of parts) {
+        if (!isToolPart(part) || stepLabel(part).tier === 'plumbing')
+            continue;
+        const family = familyForTool(part.tool);
+        if (family === 'hidden')
+            continue;
+        families.set(family, [...(families.get(family) ?? []), part]);
+        if (partOutcome(part) !== 'ok')
+            failed += 1;
+    }
+    if (families.size === 0)
+        return burstSummaryLabel(burstSummary(parts), false);
+    const label = [...families]
+        .map(([family, calls], i) => {
+        const sentence = narrateStep(family, calls);
+        return i === 0 ? sentence : sentence.charAt(0).toLowerCase() + sentence.slice(1);
+    })
+        .join(', ');
+    return failed > 0 ? `${label} · ${failed} failed` : label;
+}

package/dist/core/turns/segments/group-steps.d.ts

@@ -1,11 +1,12 @@
 import type { ToolPart } from '../../runtime/client.js';
+import type { ToolCallBlock } from '../../session-log/types.js';
 import { type StepFamily } from './narration.js';
-export interface Step {
+export interface Step<P extends ToolPart | ToolCallBlock = ToolPart> {
     id: string;
     family: StepFamily;
     label: string;
-    parts: ToolPart[];
+    parts: P[];
     status: 'running' | 'error' | 'done';
     durationMs?: number;
 }
-export declare function groupSteps(parts: ToolPart[]): Step[];
+export declare function groupSteps<P extends ToolPart | ToolCallBlock = ToolPart>(given: P[]): Step<P>[];

package/dist/core/turns/segments/group-steps.js

@@ -1,3 +1,4 @@
+import { asToolPart } from '../../session-log/tool-call.js';
 import { isEmptyShowPart, isNoGroupActivityTool } from './session-activity-groups.js';
 import { familyForTool, narrateFailedStep, narrateStep } from './narration.js';
 function statusOf(parts) {
@@ -22,18 +23,19 @@ function durationOf(parts) {
         return undefined;
     return end - start;
 }
-function finalize(family, parts) {
+function finalize(family, given) {
+    const parts = given.map(asToolPart);
     const status = statusOf(parts);
     return {
         id: parts[0].callID,
         family,
         label: status === 'error' ? narrateFailedStep(family, parts) : narrateStep(family, parts),
-        parts,
+        parts: given,
         status,
         durationMs: status === 'running' ? undefined : durationOf(parts),
     };
 }
-export function groupSteps(parts) {
+export function groupSteps(given) {
     const steps = [];
     let family = null;
     let buffer = [];
@@ -43,7 +45,8 @@ export function groupSteps(parts) {
         family = null;
         buffer = [];
     };
-    for (const part of parts) {
+    for (const item of given) {
+        const part = asToolPart(item);
         const f = familyForTool(part.tool);
         if (f === 'hidden')
             continue;
@@ -51,14 +54,14 @@ export function groupSteps(parts) {
             continue;
         if (isNoGroupActivityTool(part.tool)) {
             flush();
-            steps.push(finalize(f, [part]));
+            steps.push(finalize(f, [item]));
             continue;
         }
         if (f !== family) {
             flush();
             family = f;
         }
-        buffer.push(part);
+        buffer.push(item);
     }
     flush();
     return steps;

package/dist/core/turns/segments/index.d.ts

@@ -10,3 +10,4 @@ export * from './show-group.js';
 export * from './step-label.js';
 export * from './working-turn.js';
 export * from './turn-busy-visibility.js';
+export * from './work-split.js';

package/dist/core/turns/segments/index.js

@@ -10,3 +10,4 @@ export * from './show-group.js';
 export * from './step-label.js';
 export * from './working-turn.js';
 export * from './turn-busy-visibility.js';
+export * from './work-split.js';

package/dist/core/turns/segments/narration.d.ts

@@ -1,9 +1,10 @@
 import type { ToolPart } from '../../runtime/client.js';
-export type StepFamily = 'explore' | 'edit' | 'run' | 'web' | 'create' | 'plan' | 'delegate' | 'sessions' | 'memory' | 'connectors' | 'automations' | 'projects' | 'skills' | 'ask' | 'retired' | 'other';
+import type { ToolCallBlock } from '../../session-log/types.js';
+export type StepFamily = 'explore' | 'edit' | 'run' | 'script' | 'web' | 'create' | 'plan' | 'delegate' | 'sessions' | 'memory' | 'connectors' | 'automations' | 'projects' | 'skills' | 'ask' | 'retired' | 'other';
 export declare function familyForTool(toolName: string): StepFamily | 'hidden';
 export declare function narrationToolName(toolName: string): string;
 export declare function contextLabelForTool(toolName: string): string;
 export type CreateArtifactKind = 'image' | 'video' | 'presentation';
-export declare function createArtifactKind(part: ToolPart): CreateArtifactKind | null;
-export declare function narrateStep(family: StepFamily, parts: ToolPart[]): string;
-export declare function narrateFailedStep(family: StepFamily, parts: ToolPart[]): string;
+export declare function createArtifactKind(call: ToolPart | ToolCallBlock): CreateArtifactKind | null;
+export declare function narrateStep(family: StepFamily, calls: ReadonlyArray<ToolPart | ToolCallBlock>): string;
+export declare function narrateFailedStep(family: StepFamily, calls: ReadonlyArray<ToolPart | ToolCallBlock>): string;

package/dist/core/turns/segments/narration.js

@@ -1,3 +1,4 @@
+import { asToolPart } from '../../session-log/tool-call.js';
 import { inputPath, toolKind } from '../tool-kind.js';
 import { getToolPrimaryArg, normalizeName } from '../tools/tool-meta.js';
 import { parseWebSearchOutput, wsDomain } from '../tools/web-helpers.js';
@@ -23,6 +24,7 @@ const FAMILY_OF_KIND = {
     apply_patch: 'edit',
     bash: 'run',
     pty: 'run',
+    script: 'script',
     web_search: 'web',
     webfetch: 'web',
     media: 'create',
@@ -286,7 +288,8 @@ function createPartKey(part) {
     return 'shown:result';
 }
 const PRESENTATION_GEN_ARTIFACT_ACTIONS = new Set(['create_slide', 'export_pdf', 'export_pptx']);
-export function createArtifactKind(part) {
+export function createArtifactKind(call) {
+    const part = asToolPart(call);
     const t = normalizeName(part.tool);
     if (t === 'image_gen')
         return 'image';
@@ -337,7 +340,8 @@ function mixedCreateSentence(keys) {
     });
     return `Worked on ${joinWithAnd(segments)}`;
 }
-export function narrateStep(family, parts) {
+export function narrateStep(family, calls) {
+    const parts = calls.map(asToolPart);
     const n = parts.length;
     const arg = firstArg(parts);
     switch (family) {
@@ -358,6 +362,8 @@ export function narrateStep(family, parts) {
         }
         case 'run':
             return n === 1 ? 'Ran a command' : `Ran ${n} commands`;
+        case 'script':
+            return n === 1 ? 'Ran a script' : `Ran ${n} scripts`;
         case 'web': {
             const isSearch = (p) => {
                 const t = normalizeName(p.tool);
@@ -525,7 +531,8 @@ export function narrateStep(family, parts) {
         }
     }
 }
-export function narrateFailedStep(family, parts) {
+export function narrateFailedStep(family, calls) {
+    const parts = calls.map(asToolPart);
     const arg = firstArg(parts);
     switch (family) {
         case 'explore':
@@ -539,6 +546,8 @@ export function narrateFailedStep(family, parts) {
         }
         case 'run':
             return 'A command failed';
+        case 'script':
+            return 'A script failed';
         case 'web':
             return "Couldn't finish searching the web";
         case 'create':

package/dist/core/turns/segments/session-activity-groups.d.ts

@@ -1,3 +1,4 @@
+import type { ToolCallBlock } from '../../session-log/types.js';
 export declare function normalizeActivityToolName(toolName: string | undefined): string;
 export declare function isShellActivityTool(toolName: string | undefined): boolean;
 export declare function isQuestionTool(toolName: string | undefined): boolean;
@@ -15,6 +16,6 @@ export interface ActivityPartLike {
         input?: Record<string, unknown> | null;
     } | null;
 }
-export declare function isEmptyShowPart(part: ActivityPartLike): boolean;
+export declare function isEmptyShowPart(call: ActivityPartLike | ToolCallBlock): boolean;
 export declare const STANDALONE_TOOLS: ReadonlySet<string>;
 export declare function isStandaloneActivityTool(toolName: string | undefined): boolean;

package/dist/core/turns/segments/session-activity-groups.js

@@ -1,4 +1,5 @@
 import { isShowPayloadEmpty } from '../tools/show-availability.js';
+import { isToolCallBlock, toolPartOf } from '../../session-log/tool-call.js';
 export function normalizeActivityToolName(toolName) {
     return (toolName ?? '').replace(/^oc-/, '').replace(/-/g, '_');
 }
@@ -34,7 +35,8 @@ export function isInvisibleActivityPart(part) {
         return true;
     return false;
 }
-export function isEmptyShowPart(part) {
+export function isEmptyShowPart(call) {
+    const part = isToolCallBlock(call) ? toolPartOf(call) : call;
     const name = normalizeActivityToolName(part.tool);
     if (name !== 'show' && name !== 'show_user')
         return false;

package/dist/core/turns/segments/step-label.d.ts

@@ -1,4 +1,5 @@
 import type { Part } from '../../runtime/client.js';
+import type { SessionLogBlock, ToolCallBlock } from '../../session-log/types.js';
 export type StepTier = 'primary' | 'reasoning' | 'plumbing';
 export interface StepLabel {
     verb: string;
@@ -7,4 +8,4 @@ export interface StepLabel {
     tier: StepTier;
 }
 export declare const PLUMBING_TOOLS: ReadonlySet<string>;
-export declare function stepLabel(part: Part): StepLabel;
+export declare function stepLabel(call: Part | ToolCallBlock | SessionLogBlock): StepLabel;

package/dist/core/turns/segments/step-label.js

@@ -1,6 +1,7 @@
 import { isReasoningPart, isToolPart } from '../parts.js';
 import { humanizeSearchQuery } from '../tools/search-query.js';
 import { normalizeActivityToolName } from './session-activity-groups.js';
+import { isToolCallBlock, toolPartOf } from '../../session-log/tool-call.js';
 export const PLUMBING_TOOLS = new Set([
     'get_mem',
     'memory_search',
@@ -63,7 +64,8 @@ function readObject(input, keys) {
     }
     return undefined;
 }
-export function stepLabel(part) {
+export function stepLabel(call) {
+    const part = (isToolCallBlock(call) ? toolPartOf(call) : call);
     if (isReasoningPart(part)) {
         return { verb: 'Thought', running: 'Thinking', object: undefined, tier: 'reasoning' };
     }

package/dist/core/turns/segments/work-split.d.ts

@@ -0,0 +1,10 @@
+import type { ShowGroupedSegment } from './show-group.js';
+export interface TurnWorkSplit<S extends ShowGroupedSegment = ShowGroupedSegment> {
+    work: S[];
+    hoisted: S[];
+    final: S[];
+}
+export declare function splitTurnWork<S extends ShowGroupedSegment = ShowGroupedSegment>(segments: ReadonlyArray<S>, opts: {
+    working: boolean;
+}): TurnWorkSplit<S> | null;
+export declare function workedForLabel(durationMs: number | null): string;

package/dist/core/turns/segments/work-split.js

@@ -0,0 +1,40 @@
+import { formatDuration } from '../state.js';
+import { normalizeActivityToolName } from './session-activity-groups.js';
+function isDeliverableSegment(segment) {
+    if (segment.kind === 'show-group')
+        return true;
+    if (segment.kind !== 'standalone')
+        return false;
+    const name = normalizeActivityToolName(segment.part.tool);
+    return name === 'show' || name === 'show_user';
+}
+function isBlankText(segment) {
+    return segment.kind === 'text' && !segment.part.text?.trim();
+}
+export function splitTurnWork(segments, opts) {
+    if (opts.working)
+        return null;
+    let finalIndex = -1;
+    for (let i = segments.length - 1; i >= 0; i--) {
+        const segment = segments[i];
+        if (isDeliverableSegment(segment) || isBlankText(segment))
+            continue;
+        if (segment.kind === 'text')
+            finalIndex = i;
+        break;
+    }
+    const split = { work: [], hoisted: [], final: [] };
+    segments.forEach((segment, i) => {
+        if (i === finalIndex)
+            split.final.push(segment);
+        else if (isDeliverableSegment(segment))
+            split.hoisted.push(segment);
+        else
+            split.work.push(segment);
+    });
+    return split.work.length > 0 ? split : null;
+}
+export function workedForLabel(durationMs) {
+    const duration = durationMs == null ? '' : formatDuration(durationMs);
+    return duration ? `Worked for ${duration}` : 'Worked';
+}

package/dist/core/turns/segments/working-turn.d.ts

@@ -1,4 +1,5 @@
 import type { SessionPrompt } from '../../rest/projects-client/sessions.js';
+import type { SessionLogTurn } from '../../session-log/turns/turn.js';
 interface TurnLike {
     userMessage: {
         info: {
@@ -14,6 +15,7 @@ interface TurnLike {
         };
     }>;
 }
+type BlockTurnLike = Pick<SessionLogTurn, 'id' | 'assistant'>;
 export interface WorkingTurnResolution {
     workingTurnId: string | null;
     pendingTurnIds: string[];
@@ -38,7 +40,7 @@ interface DeliveringTurnInput {
     };
 }
 export declare function fallbackBusyRowAfterTurnId(input: QueuedTurnInput & DeliveringTurnInput & {
-    turns: ReadonlyArray<TurnLike>;
+    turns: ReadonlyArray<TurnLike | BlockTurnLike>;
 }): string | null;
 export declare function turnIsConfirmedActive(input: {
     isTurnWorking: boolean;
@@ -61,7 +63,7 @@ export declare function workingTurnDrawsBusyRow(input: {
     isRetrying: boolean;
     awaitingUser?: boolean;
 }): boolean;
-export declare function freshSendHint(turns: ReadonlyArray<TurnLike>, isSentTurn: (userMessageId: string) => boolean): string | null;
+export declare function freshSendHint(turns: ReadonlyArray<TurnLike | BlockTurnLike>, isSentTurn: (userMessageId: string) => boolean): string | null;
 export declare function resolveWorkingTurn(input: {
     turns: ReadonlyArray<TurnLike>;
     hintMessageId: string | null | undefined;

package/dist/core/turns/segments/working-turn.js

@@ -1,4 +1,6 @@
 import { showTurnBusyIndicator } from './turn-busy-visibility.js';
+const turnIdOf = (turn) => ('id' in turn ? turn.id : turn.userMessage.info.id);
+const replyCountOf = (turn) => ('assistant' in turn ? turn.assistant.length : turn.assistantMessages.length);
 const completedAt = (info) => info.time?.completed;
 export function shouldSuppressWorkingTurnBusy(input) {
     if (!input.hasPendingTurns || !input.newestAssistantCompleted)
@@ -8,18 +10,17 @@ export function shouldSuppressWorkingTurnBusy(input) {
         (input.activeTurnId !== null && input.activeTurnId !== input.workingTurnId));
 }
 function isQueuedTurn(turn, input) {
-    const id = turn.userMessage.info.id;
-    return (input.pendingTurnIds.has(id) ||
-        (turn.assistantMessages.length === 0 && input.pendingPromptIds.has(id)));
+    const id = turnIdOf(turn);
+    return input.pendingTurnIds.has(id) || (replyCountOf(turn) === 0 && input.pendingPromptIds.has(id));
 }
 export function fallbackBusyRowAfterTurnId(input) {
-    const delivering = input.turns.find((turn) => input.deliveringPromptIds.has(turn.userMessage.info.id));
+    const delivering = input.turns.find((turn) => input.deliveringPromptIds.has(turnIdOf(turn)));
     if (delivering)
-        return delivering.userMessage.info.id;
+        return turnIdOf(delivering);
     const firstQueued = input.turns.findIndex((turn) => isQueuedTurn(turn, input));
     if (firstQueued <= 0)
         return null;
-    return input.turns[firstQueued - 1].userMessage.info.id;
+    return turnIdOf(input.turns[firstQueued - 1]);
 }
 export function turnIsConfirmedActive(input) {
     return input.isTurnWorking && !input.pendingDelivery && input.activeTurnId === input.turnId;
@@ -47,10 +48,10 @@ export function workingTurnDrawsBusyRow(input) {
 }
 export function freshSendHint(turns, isSentTurn) {
     for (let i = turns.length - 1; i >= 0; i--) {
-        const id = turns[i].userMessage.info.id;
+        const id = turnIdOf(turns[i]);
         if (!isSentTurn(id))
             continue;
-        return turns[i].assistantMessages.length === 0 ? id : null;
+        return replyCountOf(turns[i]) === 0 ? id : null;
     }
     return null;
 }

package/dist/core/turns/state.d.ts

@@ -1,5 +1,5 @@
 import { extractGatewayErrorDetails } from './errors.js';
-import type { MessageInfoLike, MessageWithPartsLike, ModelPricingLookup, PartLike, PartWithMessage, RetryInfo, SessionStatusLike, TurnCostInfo, TurnLike } from './types.js';
+import type { MessageWithPartsLike, PartLike, PartWithMessage, RetryInfo, SessionStatusLike, TurnCostInfo, TurnLike } from './types.js';
 export declare function isLastUserMessage(messageId: string, allMessages: readonly MessageWithPartsLike[]): boolean;
 export declare function getWorkingState(sessionStatus: SessionStatusLike | undefined, isLast: boolean): boolean;
 export declare function shouldHideResponsePart(working: boolean, responsePartId: string | undefined): boolean;
@@ -11,12 +11,7 @@ export declare function getTurnStatus(parts: ReadonlyArray<{
     part: PartLike;
 }>, childMessages?: readonly MessageWithPartsLike[]): string;
 export declare function formatDuration(ms: number): string;
-export declare const COST_MARKUP = 1.2;
-export declare function getTurnCost(parts: ReadonlyArray<PartWithMessage>, lookup?: ModelPricingLookup): TurnCostInfo | undefined;
-export declare function getSessionCost(messages: ReadonlyArray<{
-    info?: MessageInfoLike;
-    parts: PartLike[];
-}>, lookup?: ModelPricingLookup): number;
+export declare function getTurnTokens(parts: ReadonlyArray<PartWithMessage>): TurnCostInfo['tokens'] | undefined;
 export declare function formatCost(cost: number): string;
 export declare function formatTokens(tokens: number): string;
 export declare function getRetryInfo(status: SessionStatusLike | undefined): RetryInfo | undefined;