@kaelio/ktx 0.5.0 → 0.6.0

package/dist/context/llm/ai-sdk-runtime.test.js

@@ -1,308 +0,0 @@
-import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
-vi.mock('ai', () => ({
-    generateText: vi.fn(),
-    stepCountIs: (n) => n,
-    tool: (def) => def,
-}));
-import { generateText } from 'ai';
-import { AiSdkKtxLlmRuntime } from './ai-sdk-runtime.js';
-describe('AiSdkKtxLlmRuntime.runAgentLoop', () => {
-    let runtime;
-    const llmProvider = {
-        getModel: vi.fn().mockReturnValue({ modelId: 'claude-sonnet-4-6', provider: 'anthropic' }),
-        getModelByName: vi.fn(),
-        cacheMarker: vi.fn(),
-        repairToolCallHandler: vi.fn(),
-        thinkingProviderOptions: vi.fn(),
-        telemetryConfig: vi.fn(),
-        promptCachingConfig: vi.fn(() => ({
-            enabled: false,
-            systemTtl: '1h',
-            toolsTtl: '1h',
-            historyTtl: '5m',
-            cacheSystem: true,
-            cacheTools: true,
-            cacheHistory: true,
-            vertexFallbackTo5m: false,
-        })),
-        activeBackend: vi.fn(() => 'anthropic'),
-    };
-    beforeEach(() => {
-        vi.clearAllMocks();
-        runtime = new AiSdkKtxLlmRuntime({ llmProvider: llmProvider });
-    });
-    afterEach(() => vi.clearAllMocks());
-    it('passes systemPrompt, userPrompt, tools, and step budget through to generateText', async () => {
-        generateText.mockResolvedValue({ text: 'ok', toolCalls: [], steps: [] });
-        const repairHandler = vi.fn();
-        llmProvider.repairToolCallHandler.mockReturnValueOnce(repairHandler);
-        const tools = { noop: { description: 'noop', inputSchema: {}, execute: vi.fn() } };
-        await runtime.runAgentLoop({
-            modelRole: 'candidateExtraction',
-            systemPrompt: 'SYS',
-            userPrompt: 'USR',
-            toolSet: tools,
-            stepBudget: 17,
-            telemetryTags: { source: 'test' },
-        });
-        const call = generateText.mock.calls[0][0];
-        expect(call.system).toEqual({ role: 'system', content: 'SYS' });
-        expect(call.messages).toEqual([{ role: 'user', content: 'USR' }]);
-        expect(call.prompt).toBeUndefined();
-        expect(call.tools.noop).toEqual(expect.objectContaining({
-            description: 'noop',
-            inputSchema: {},
-            execute: expect.any(Function),
-            toModelOutput: expect.any(Function),
-        }));
-        expect(call.stopWhen).toBe(17);
-        expect(call.temperature).toBe(0);
-        expect(call.experimental_repairToolCall).toBe(repairHandler);
-        expect(llmProvider.getModel).toHaveBeenCalledWith('candidateExtraction');
-        expect(llmProvider.repairToolCallHandler).toHaveBeenCalledWith({ source: 'ktx-agent-runner' });
-    });
-    it('returns stopReason=natural when the loop completes without error', async () => {
-        generateText.mockResolvedValue({ text: 'done', toolCalls: [], steps: [] });
-        const result = await runtime.runAgentLoop({
-            modelRole: 'candidateExtraction',
-            systemPrompt: 'system',
-            userPrompt: 'user',
-            toolSet: {},
-            stepBudget: 10,
-            telemetryTags: {},
-        });
-        expect(result.stopReason).toBe('natural');
-        expect(result.error).toBeUndefined();
-        expect(llmProvider.getModel).toHaveBeenCalledWith('candidateExtraction');
-        expect(generateText).toHaveBeenCalledWith(expect.objectContaining({
-            system: { role: 'system', content: 'system' },
-            messages: [{ role: 'user', content: 'user' }],
-        }));
-    });
-    it('returns stopReason=error with the error on generateText failure', async () => {
-        const err = new Error('LLM unavailable');
-        generateText.mockRejectedValue(err);
-        const result = await runtime.runAgentLoop({
-            modelRole: 'candidateExtraction',
-            systemPrompt: '',
-            userPrompt: '',
-            toolSet: {},
-            stepBudget: 10,
-            telemetryTags: {},
-        });
-        expect(result.stopReason).toBe('error');
-        expect(result.error).toBe(err);
-    });
-    it('invokes caller onStepFinish with incrementing stepIndex and total budget', async () => {
-        const calls = [];
-        generateText.mockImplementation(async (opts) => {
-            for (let i = 0; i < 3; i++) {
-                await opts.onStepFinish({});
-            }
-            return { text: 'ok', toolCalls: [], steps: [] };
-        });
-        await runtime.runAgentLoop({
-            modelRole: 'candidateExtraction',
-            systemPrompt: '',
-            userPrompt: '',
-            toolSet: {},
-            stepBudget: 10,
-            telemetryTags: {},
-            onStepFinish: (info) => {
-                calls.push(info);
-            },
-        });
-        expect(calls).toEqual([
-            { stepIndex: 1, stepBudget: 10 },
-            { stepIndex: 2, stepBudget: 10 },
-            { stepIndex: 3, stepBudget: 10 },
-        ]);
-    });
-    it('swallows errors thrown from caller onStepFinish without aborting the loop', async () => {
-        generateText.mockImplementation(async (opts) => {
-            await opts.onStepFinish({});
-            return { text: 'ok', toolCalls: [], steps: [] };
-        });
-        const result = await runtime.runAgentLoop({
-            modelRole: 'candidateExtraction',
-            systemPrompt: '',
-            userPrompt: '',
-            toolSet: {},
-            stepBudget: 10,
-            telemetryTags: {},
-            onStepFinish: () => {
-                throw new Error('boom');
-            },
-        });
-        expect(result.stopReason).toBe('natural');
-    });
-    it('forwards telemetryTags.source through experimental_telemetry metadata', async () => {
-        generateText.mockResolvedValue({ text: 'ok', toolCalls: [], steps: [] });
-        const telemetryConfigEnabled = {
-            isEnabled: () => true,
-            devtoolsEnabled: false,
-            appSettingsService: {
-                settings: { telemetry: { recordInputs: false, recordOutputs: false } },
-            },
-            systemConfigService: {
-                config: { instance: { name: 'test-instance' } },
-            },
-        };
-        const runtimeWithTelemetry = new AiSdkKtxLlmRuntime({
-            llmProvider: llmProvider,
-            telemetry: {
-                createTelemetry: (tags) => ({
-                    isEnabled: telemetryConfigEnabled.isEnabled(),
-                    metadata: {
-                        source: tags.source ?? 'RESEARCH',
-                        jobId: tags.jobId,
-                        unitKey: tags.unitKey,
-                    },
-                }),
-            },
-        });
-        await runtimeWithTelemetry.runAgentLoop({
-            modelRole: 'candidateExtraction',
-            systemPrompt: '',
-            userPrompt: '',
-            toolSet: {},
-            stepBudget: 10,
-            telemetryTags: { source: 'metabase', jobId: 'job-123', unitKey: 'u/1' },
-        });
-        const call = generateText.mock.calls[0][0];
-        expect(call.experimental_telemetry.metadata.source).toBe('metabase');
-    });
-    it('defaults to source=RESEARCH when telemetryTags omits source', async () => {
-        generateText.mockResolvedValue({ text: 'ok', toolCalls: [], steps: [] });
-        const telemetryConfigEnabled = {
-            isEnabled: () => true,
-            devtoolsEnabled: false,
-            appSettingsService: {
-                settings: { telemetry: { recordInputs: false, recordOutputs: false } },
-            },
-            systemConfigService: {
-                config: { instance: { name: 'test-instance' } },
-            },
-        };
-        const runtimeWithTelemetry = new AiSdkKtxLlmRuntime({
-            llmProvider: llmProvider,
-            telemetry: {
-                createTelemetry: (tags) => ({
-                    isEnabled: telemetryConfigEnabled.isEnabled(),
-                    metadata: {
-                        source: tags.source ?? 'RESEARCH',
-                        jobId: tags.jobId,
-                        unitKey: tags.unitKey,
-                    },
-                }),
-            },
-        });
-        await runtimeWithTelemetry.runAgentLoop({
-            modelRole: 'candidateExtraction',
-            systemPrompt: '',
-            userPrompt: '',
-            toolSet: {},
-            stepBudget: 10,
-            telemetryTags: { operationName: 'memory-agent-ingest' },
-        });
-        const call = generateText.mock.calls[0][0];
-        expect(call.experimental_telemetry.metadata.source).toBe('RESEARCH');
-    });
-    it('forwards jobId and unitKey through experimental_telemetry metadata', async () => {
-        generateText.mockResolvedValue({ text: 'ok', toolCalls: [], steps: [] });
-        const telemetryConfigEnabled = {
-            isEnabled: () => true,
-            devtoolsEnabled: false,
-            appSettingsService: {
-                settings: { telemetry: { recordInputs: false, recordOutputs: false } },
-            },
-            systemConfigService: {
-                config: { instance: { name: 'test-instance' } },
-            },
-        };
-        const runtimeWithTelemetry = new AiSdkKtxLlmRuntime({
-            llmProvider: llmProvider,
-            telemetry: {
-                createTelemetry: (tags) => ({
-                    isEnabled: telemetryConfigEnabled.isEnabled(),
-                    metadata: {
-                        source: tags.source ?? 'RESEARCH',
-                        jobId: tags.jobId,
-                        unitKey: tags.unitKey,
-                    },
-                }),
-            },
-        });
-        await runtimeWithTelemetry.runAgentLoop({
-            modelRole: 'candidateExtraction',
-            systemPrompt: '',
-            userPrompt: '',
-            toolSet: {},
-            stepBudget: 10,
-            telemetryTags: { source: 'metabase', jobId: 'job-777', unitKey: 'sources/users' },
-        });
-        const call = generateText.mock.calls[0][0];
-        expect(call.experimental_telemetry.metadata.jobId).toBe('job-777');
-        expect(call.experimental_telemetry.metadata.unitKey).toBe('sources/users');
-    });
-    it('records a sanitized LLM debug request when a recorder is injected', async () => {
-        generateText.mockResolvedValue({ text: 'ok', toolCalls: [], steps: [] });
-        const record = vi.fn();
-        const provider = {
-            ...llmProvider,
-            cacheMarker: vi.fn((ttl) => ({
-                anthropic: { cacheControl: { type: 'ephemeral', ttl } },
-            })),
-            promptCachingConfig: vi.fn(() => ({
-                enabled: true,
-                systemTtl: '1h',
-                toolsTtl: '1h',
-                historyTtl: '5m',
-                cacheSystem: true,
-                cacheTools: true,
-                cacheHistory: true,
-                vertexFallbackTo5m: false,
-            })),
-        };
-        const runtimeWithDebug = new AiSdkKtxLlmRuntime({
-            llmProvider: provider,
-            debugRequestRecorder: { record },
-        });
-        await runtimeWithDebug.runAgentLoop({
-            modelRole: 'candidateExtraction',
-            systemPrompt: 'SECRET SYSTEM PROMPT',
-            userPrompt: 'SECRET USER PROMPT',
-            toolSet: {
-                emit_candidate: {
-                    description: 'SECRET TOOL DESCRIPTION',
-                    inputSchema: {},
-                    execute: vi.fn(),
-                },
-            },
-            stepBudget: 10,
-            telemetryTags: { operationName: 'ingest-bundle-wu', source: 'metabase', jobId: 'job-1', unitKey: 'cards/1' },
-        });
-        expect(record).toHaveBeenCalledTimes(1);
-        expect(record).toHaveBeenCalledWith(expect.objectContaining({
-            operationName: 'ingest-bundle-wu',
-            source: 'metabase',
-            jobId: 'job-1',
-            unitKey: 'cards/1',
-            modelRole: 'candidateExtraction',
-            modelId: 'claude-sonnet-4-6',
-            messageCount: 2,
-            toolNames: ['emit_candidate'],
-        }));
-        const providerOptions = record.mock.calls[0][0].providerOptions;
-        expect(providerOptions).toEqual(expect.arrayContaining([
-            expect.objectContaining({ target: 'message', index: 0, role: 'system' }),
-            expect.objectContaining({ target: 'message-part', index: 1, role: 'user', partIndex: 0 }),
-            expect.objectContaining({ target: 'tool', name: 'emit_candidate' }),
-        ]));
-        expect(providerOptions).toHaveLength(3);
-        const serialized = JSON.stringify(record.mock.calls[0][0]);
-        expect(serialized).not.toContain('SECRET SYSTEM PROMPT');
-        expect(serialized).not.toContain('SECRET USER PROMPT');
-        expect(serialized).not.toContain('SECRET TOOL DESCRIPTION');
-    });
-});

package/dist/context/llm/claude-code-env.test.d.ts

@@ -1 +0,0 @@
-export {};

package/dist/context/llm/claude-code-env.test.js

@@ -1,17 +0,0 @@
-import { describe, expect, it } from 'vitest';
-import { CLAUDE_CODE_PROVIDER_ENV_DENYLIST, createKtxClaudeCodeEnv } from './claude-code-env.js';
-describe('createKtxClaudeCodeEnv', () => {
-    it('strips provider-routing credentials from the Claude Code child environment', () => {
-        const seeded = Object.fromEntries(CLAUDE_CODE_PROVIDER_ENV_DENYLIST.map((key) => [key, `${key}-value`]));
-        const env = createKtxClaudeCodeEnv({
-            ...seeded,
-            PATH: '/usr/bin',
-            HOME: '/Users/test',
-        });
-        for (const key of CLAUDE_CODE_PROVIDER_ENV_DENYLIST) {
-            expect(env).not.toHaveProperty(key);
-        }
-        expect(env.PATH).toBe('/usr/bin');
-        expect(env.HOME).toBe('/Users/test');
-    });
-});

package/dist/context/llm/claude-code-models.test.d.ts

@@ -1 +0,0 @@
-export {};

package/dist/context/llm/claude-code-models.test.js

@@ -1,15 +0,0 @@
-import { describe, expect, it } from 'vitest';
-import { resolveClaudeCodeModel } from './claude-code-models.js';
-describe('resolveClaudeCodeModel', () => {
-    it.each([
-        ['sonnet', 'claude-sonnet-4-6'],
-        ['opus', 'claude-opus-4-7'],
-        ['haiku', 'claude-haiku-4-5'],
-        ['claude-sonnet-4-6', 'claude-sonnet-4-6'],
-    ])('maps %s to %s', (input, expected) => {
-        expect(resolveClaudeCodeModel(input)).toBe(expected);
-    });
-    it('rejects unsupported aliases', () => {
-        expect(() => resolveClaudeCodeModel('gpt-5')).toThrow('Unsupported Claude Code model');
-    });
-});

package/dist/context/llm/claude-code-runtime.test.d.ts

@@ -1 +0,0 @@
-export {};