@danceiny/gotry 0.0.1-rc.18 → 0.0.1-rc.20

package/dist/src/tool-budget.js

@@ -1,136 +0,0 @@
-import { randomUUID } from 'node:crypto';
-export const TOOL_BUDGET_SOFT_CALL = 16;
-export const TOOL_BUDGET_HARD_CALL = 18;
-export const TOOL_BUDGET_EXHAUSTED = 'TOOL_BUDGET_EXHAUSTED';
-export function toolBudgetDecision(callNumber) {
-    if (callNumber > TOOL_BUDGET_HARD_CALL) {
-        return {
-            kind: 'deny',
-            reason: `${TOOL_BUDGET_EXHAUSTED}: maximum ${TOOL_BUDGET_HARD_CALL} real tool dispatches per agent turn reached; provide a final answer using existing results`
-        };
-    }
-    return {
-        kind: 'allow',
-        softSignal: callNumber === TOOL_BUDGET_SOFT_CALL,
-        exhaustsBudget: callNumber === TOOL_BUDGET_HARD_CALL
-    };
-}
-export function createToolBudgetState() {
-    let calls = 0;
-    return {
-        nextCall () {
-            calls += 1;
-            return {
-                ...toolBudgetDecision(calls),
-                callNumber: calls
-            };
-        }
-    };
-}
-function convergenceContext(code, text) {
-    return {
-        id: randomUUID(),
-        role: 'user',
-        content: [
-            {
-                type: 'text',
-                text: `${code}: ${text}`
-            }
-        ],
-        source: {
-            kind: 'plugin',
-            plugin: 'gotry-tool-budget'
-        }
-    };
-}
-function exhaustionResult(reason) {
-    return {
-        isError: true,
-        error: {
-            message: reason,
-            info: {
-                name: 'ToolBudgetError',
-                code: TOOL_BUDGET_EXHAUSTED
-            }
-        },
-        content: [
-            {
-                type: 'text',
-                text: reason
-            }
-        ],
-        additionalContexts: [
-            convergenceContext(TOOL_BUDGET_EXHAUSTED, 'No more tool dispatches are permitted in this turn. Produce the final answer from existing results.')
-        ]
-    };
-}
-export function installToolBudget(ctx) {
-    if (typeof ctx.on !== 'function') return;
-    const budgetByAgent = new Map();
-    const softExecutions = new Map();
-    const pendingFinalOnly = new Map();
-    const finalOnlyDisposers = new Map();
-    const clearAgent = (agentId)=>{
-        budgetByAgent.delete(agentId);
-        for (const [token, owner] of softExecutions){
-            if (owner === agentId) softExecutions.delete(token);
-        }
-        pendingFinalOnly.delete(agentId);
-        finalOnlyDisposers.get(agentId)?.();
-        finalOnlyDisposers.delete(agentId);
-    };
-    const enterFinalOnly = (agent)=>{
-        const agentId = String(agent.id);
-        if (finalOnlyDisposers.has(agentId)) return;
-        const globalTools = ctx.tools;
-        const scopedTools = agent.ctx?.tools;
-        if (typeof globalTools.schemas !== 'function' || typeof scopedTools?.restrict !== 'function') return;
-        const inheritedNames = globalTools.schemas().map((schema)=>schema.name).filter((name)=>name !== 'run_code');
-        if (inheritedNames.length === 0) return;
-        finalOnlyDisposers.set(agentId, scopedTools.restrict({
-            deny: inheritedNames
-        }));
-    };
-    ctx.on('session/event', (subject, event)=>{
-        const agentId = String(subject.id);
-        if (event.type === 'turn/start' || event.type === 'turn/end') {
-            clearAgent(agentId);
-            return;
-        }
-        if (event.type === 'step/end') {
-            const agent = pendingFinalOnly.get(agentId);
-            if (agent) {
-                pendingFinalOnly.delete(agentId);
-                enterFinalOnly(agent);
-            }
-        }
-    });
-    ctx.on('session/disposed', (subject)=>clearAgent(String(subject.id)));
-    ctx.on('tools/execute', (exec, next)=>{
-        if (!exec.agent) return next();
-        const agentId = String(exec.agent.id);
-        const state = budgetByAgent.get(agentId) ?? createToolBudgetState();
-        budgetByAgent.set(agentId, state);
-        const decision = state.nextCall();
-        if (decision.kind === 'deny') {
-            pendingFinalOnly.set(agentId, exec.agent);
-            return Promise.resolve(exhaustionResult(decision.reason));
-        }
-        if (decision.softSignal) softExecutions.set(exec.token, agentId);
-        if (decision.exhaustsBudget) pendingFinalOnly.set(agentId, exec.agent);
-        return next();
-    });
-    ctx.on('tools/post-execute', (exec, _result, next)=>{
-        if (!softExecutions.delete(exec.token)) return next();
-        return next().then((decision)=>({
-                ...decision,
-                additionalContexts: [
-                    ...decision.additionalContexts ?? [],
-                    convergenceContext('TOOL_BUDGET_SOFT', 'Tool call 16 reached. Converge now and produce the final answer from gathered evidence.')
-                ]
-            }));
-    });
-}
-
-
-//# sourceURL=ts/src/tool-budget.ts