mirror of
https://github.com/Kaelio/ktx.git
synced 2026-06-28 08:49:38 +02:00
test: split cli tests from source tree (#216)
* feat(cli): define full warehouse dialect contract
* test(cli): keep dialect edge tests focused
* fix(cli): stabilize dialect contract foundation
* refactor(connectors): own read-only query preparation
* refactor(connectors): resolve dialects through registry
* refactor(connectors): keep concrete dialect classes internal
* chore(workspace): enforce dialect import boundary
* refactor(cli): resolve relationship dialect at scan boundary
* refactor(cli): use dialect display parsing for entity details
* refactor(cli): use dialect display parsing for warehouse catalog
* refactor(cli): use dialect SQL in relationship workflows
* test(cli): verify solid dialect scan workflow closure
* test: split cli tests from source tree
* refactor(cli): standardize BigQuery scope listing
* feat(sqlite): implement connector scope listing
* test(connectors): cover required table listing
* feat(cli): add warehouse driver registry
* refactor(setup): route scope discovery through driver registry
* refactor(cli): route local query execution through driver registry
* refactor(historic-sql): route dialect support through driver registry
* refactor(cli): test warehouse connections through driver registry
* fix(cli): close driver registry type export gaps
* Improve setup daemon diagnostics
* refactor(setup): centralize rail-prefixed diagnostics + query-history fallback
Extract errorMessage, writePrefixedLines, and flushPrefixedBufferedCommandOutput
into clack.ts so the setup wizard, managed daemons, and embedding/agent steps
share one rail-formatted writer. setup-databases.ts also adds a
"disable query history and retry" option when the schema-context build fails
and query history is the likely culprit, surfaced via a new
failed-query-history-unavailable status.
* fix(cli): carry catalog through the picker so BigQuery/Snowflake/SQL Server scope filters match
The setup picker's KtxTableListEntry was a 2-level { schema, name }, so
qualifiedTableId always wrote db.name into enabled_tables. When BigQuery,
Snowflake, or SQL Server later ran fast ingest, their introspect step filtered
the scope set with scopedTableNames(scope, { catalog: projectId|database, db })
— catalog was non-null on the introspect side but null in the scope refs, so
every entry was rejected, the live-database adapter staged zero table files,
and detect() failed with 'Adapter "live-database" did not recognize fetched
source output'.
Align the picker boundary with the canonical 3-level KtxTableRef:
- Add catalog: string | null to KtxTableListEntry.
- BigQuery/Snowflake/SQL Server listTables populate catalog from the
resolved projectId / database; Postgres/MySQL/ClickHouse/SQLite set null.
- qualifiedTableId emits catalog.schema.name when catalog is non-null
(resolveEnabledTables already accepts the 3-part shape) and
schemasFromEnabledTables now goes through parseDottedTableEntry so it
recovers the schema correctly from both 2-part and 3-part entries.
- Export parseDottedTableEntry from enabled-tables.ts (@internal) for picker
reuse.
Update listTables expectations in all seven connector tests and the setup /
picker test fixtures. Add a picker regression test that covers the
catalog-bearing round-trip (save + refine).
* fix(cli): allow debug telemetry under opt-out env
This commit is contained in:
Andrey Avtomonov
GitHub
parent
924868841d
commit
56985b7e09
548 changed files with 5048 additions and 2228 deletions
|
|
@ -1,106 +0,0 @@
|
|||
import { describe, expect, it, vi } from 'vitest';
|
||||
import { runKtxEmbeddingHealthCheck } from './embedding-health.js';
|
||||
|
||||
describe('KTX embedding health check', () => {
|
||||
it('runs a one-shot OpenAI embedding check through the configured provider', async () => {
|
||||
const createOpenAIClient = vi.fn(() => ({
|
||||
embeddings: {
|
||||
create: vi.fn().mockResolvedValue({
|
||||
data: [{ index: 0, embedding: [0.1, 0.2, 0.3] }],
|
||||
}),
|
||||
},
|
||||
}));
|
||||
|
||||
await expect(
|
||||
runKtxEmbeddingHealthCheck(
|
||||
{
|
||||
backend: 'openai',
|
||||
model: 'text-embedding-3-small',
|
||||
dimensions: 3,
|
||||
openai: { apiKey: 'sk-openai-test' }, // pragma: allowlist secret
|
||||
},
|
||||
{ deps: { createOpenAIClient } },
|
||||
),
|
||||
).resolves.toEqual({ ok: true });
|
||||
|
||||
expect(createOpenAIClient).toHaveBeenCalledWith({ apiKey: 'sk-openai-test', baseURL: undefined }); // pragma: allowlist secret
|
||||
});
|
||||
|
||||
it('returns failed when the provider returns the wrong dimensions', async () => {
|
||||
const createOpenAIClient = vi.fn(() => ({
|
||||
embeddings: {
|
||||
create: vi.fn().mockResolvedValue({
|
||||
data: [{ index: 0, embedding: [0.1, 0.2] }],
|
||||
}),
|
||||
},
|
||||
}));
|
||||
|
||||
await expect(
|
||||
runKtxEmbeddingHealthCheck(
|
||||
{
|
||||
backend: 'openai',
|
||||
model: 'text-embedding-3-small',
|
||||
dimensions: 3,
|
||||
openai: { apiKey: 'sk-openai-test' }, // pragma: allowlist secret
|
||||
},
|
||||
{ deps: { createOpenAIClient } },
|
||||
),
|
||||
).resolves.toEqual({
|
||||
ok: false,
|
||||
message: 'Embedding provider openai returned vector with 2 dimensions; expected 3',
|
||||
});
|
||||
});
|
||||
|
||||
it('redacts credential values from health-check failures', async () => {
|
||||
const createOpenAIClient = vi.fn(() => ({
|
||||
embeddings: {
|
||||
create: vi.fn(async () => {
|
||||
throw new Error('401 invalid api key sk-openai-secret');
|
||||
}),
|
||||
},
|
||||
}));
|
||||
|
||||
await expect(
|
||||
runKtxEmbeddingHealthCheck(
|
||||
{
|
||||
backend: 'openai',
|
||||
model: 'text-embedding-3-small',
|
||||
dimensions: 3,
|
||||
openai: { apiKey: 'sk-openai-secret' }, // pragma: allowlist secret
|
||||
},
|
||||
{ deps: { createOpenAIClient } },
|
||||
),
|
||||
).resolves.toEqual({
|
||||
ok: false,
|
||||
message: '401 invalid api key [redacted]',
|
||||
});
|
||||
});
|
||||
|
||||
it('returns failed when the health check times out', async () => {
|
||||
const createOpenAIClient = vi.fn(() => ({
|
||||
embeddings: {
|
||||
create: vi.fn(
|
||||
() =>
|
||||
new Promise<{ data: Array<{ index?: number; embedding: number[] }>; usage?: { total_tokens?: number } }>(
|
||||
() => undefined,
|
||||
),
|
||||
),
|
||||
},
|
||||
}));
|
||||
|
||||
await expect(
|
||||
runKtxEmbeddingHealthCheck(
|
||||
{
|
||||
backend: 'openai',
|
||||
model: 'text-embedding-3-small',
|
||||
dimensions: 3,
|
||||
openai: { apiKey: 'sk-openai-test' }, // pragma: allowlist secret
|
||||
},
|
||||
{ timeoutMs: 1, deps: { createOpenAIClient } },
|
||||
),
|
||||
).resolves.toEqual({
|
||||
ok: false,
|
||||
message: 'Embedding health check timed out after 1ms',
|
||||
});
|
||||
});
|
||||
});
|
||||
|
|
@ -1,3 +1,4 @@
|
|||
import { describeError } from '../error-message.js';
|
||||
import { createKtxEmbeddingProvider, type KtxEmbeddingProviderDeps } from './embedding-provider.js';
|
||||
import type { KtxEmbeddingConfig } from './types.js';
|
||||
|
||||
|
|
@ -48,7 +49,6 @@ export async function runKtxEmbeddingHealthCheck(
|
|||
}
|
||||
return { ok: true };
|
||||
} catch (error) {
|
||||
const message = error instanceof Error ? error.message : String(error);
|
||||
return { ok: false, message: redactHealthCheckMessage(message, config) };
|
||||
return { ok: false, message: redactHealthCheckMessage(describeError(error), config) };
|
||||
}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -1,137 +0,0 @@
|
|||
import { describe, expect, it, vi } from 'vitest';
|
||||
import { createKtxEmbeddingProvider } from './embedding-provider.js';
|
||||
import type { KtxEmbeddingConfig } from './types.js';
|
||||
|
||||
describe('createKtxEmbeddingProvider', () => {
|
||||
it('rejects deterministic embeddings', () => {
|
||||
const config = JSON.parse(
|
||||
JSON.stringify({
|
||||
backend: 'deterministic',
|
||||
model: 'sha256',
|
||||
dimensions: 6,
|
||||
}),
|
||||
) as KtxEmbeddingConfig;
|
||||
|
||||
expect(() => createKtxEmbeddingProvider(config)).toThrow('Unsupported KTX embedding backend: deterministic');
|
||||
});
|
||||
|
||||
it('rejects gateway embeddings', () => {
|
||||
const config = JSON.parse(
|
||||
JSON.stringify({
|
||||
backend: 'gateway',
|
||||
model: 'provider/text-embedding',
|
||||
dimensions: 2,
|
||||
gateway: { apiKey: 'gateway-key' }, // pragma: allowlist secret
|
||||
}),
|
||||
) as KtxEmbeddingConfig;
|
||||
|
||||
expect(() => createKtxEmbeddingProvider(config)).toThrow('Unsupported KTX embedding backend: gateway');
|
||||
});
|
||||
|
||||
it('uses OpenAI embeddings with configured dimensions', async () => {
|
||||
const createOpenAIClient = vi.fn(() => ({
|
||||
embeddings: {
|
||||
create: vi.fn().mockResolvedValue({
|
||||
data: [{ index: 0, embedding: [0.1, 0.2] }],
|
||||
usage: { total_tokens: 7 },
|
||||
}),
|
||||
},
|
||||
}));
|
||||
|
||||
const provider = createKtxEmbeddingProvider(
|
||||
{
|
||||
backend: 'openai',
|
||||
model: 'text-embedding-3-small',
|
||||
dimensions: 2,
|
||||
openai: { apiKey: 'openai-key', baseURL: 'https://openai.test/v1' }, // pragma: allowlist secret
|
||||
},
|
||||
{ createOpenAIClient },
|
||||
);
|
||||
|
||||
await expect(provider.embed('hello')).resolves.toEqual([0.1, 0.2]);
|
||||
expect(createOpenAIClient).toHaveBeenCalledWith({
|
||||
apiKey: 'openai-key', // pragma: allowlist secret
|
||||
baseURL: 'https://openai.test/v1',
|
||||
});
|
||||
});
|
||||
|
||||
it('supports sentence-transformers pathPrefix defaults and explicit empty prefix', async () => {
|
||||
const fetch = vi
|
||||
.fn()
|
||||
.mockResolvedValueOnce(new Response(JSON.stringify({ embedding: [0.1, 0.2] }), { status: 200 }))
|
||||
.mockResolvedValueOnce(new Response(JSON.stringify({ embedding: [0.3, 0.4] }), { status: 200 }));
|
||||
|
||||
const provider = createKtxEmbeddingProvider(
|
||||
{
|
||||
backend: 'sentence-transformers',
|
||||
model: 'all-MiniLM-L6-v2',
|
||||
dimensions: 2,
|
||||
sentenceTransformers: { baseURL: 'https://python.test/' },
|
||||
},
|
||||
{ fetch },
|
||||
);
|
||||
|
||||
await expect(provider.embed('hello')).resolves.toEqual([0.3, 0.4]);
|
||||
expect(fetch).toHaveBeenNthCalledWith(
|
||||
1,
|
||||
'https://python.test/api/embeddings/compute',
|
||||
expect.objectContaining({ method: 'POST' }),
|
||||
);
|
||||
expect(fetch).toHaveBeenNthCalledWith(
|
||||
2,
|
||||
'https://python.test/api/embeddings/compute',
|
||||
expect.objectContaining({ method: 'POST' }),
|
||||
);
|
||||
|
||||
const daemonFetch = vi
|
||||
.fn()
|
||||
.mockResolvedValueOnce(new Response(JSON.stringify({ embedding: [0.1, 0.2] }), { status: 200 }))
|
||||
.mockResolvedValueOnce(new Response(JSON.stringify({ embeddings: [[0.5, 0.6]] }), { status: 200 }));
|
||||
|
||||
const daemonProvider = createKtxEmbeddingProvider(
|
||||
{
|
||||
backend: 'sentence-transformers',
|
||||
model: 'all-MiniLM-L6-v2',
|
||||
dimensions: 2,
|
||||
sentenceTransformers: { baseURL: 'https://daemon.test/base/', pathPrefix: '' },
|
||||
},
|
||||
{ fetch: daemonFetch },
|
||||
);
|
||||
|
||||
await expect(daemonProvider.embedMany(['hello'])).resolves.toEqual([[0.5, 0.6]]);
|
||||
expect(daemonFetch).toHaveBeenNthCalledWith(
|
||||
1,
|
||||
'https://daemon.test/base/embeddings/compute',
|
||||
expect.objectContaining({ method: 'POST' }),
|
||||
);
|
||||
expect(daemonFetch).toHaveBeenNthCalledWith(
|
||||
2,
|
||||
'https://daemon.test/base/embeddings/compute-bulk',
|
||||
expect.objectContaining({ method: 'POST' }),
|
||||
);
|
||||
});
|
||||
|
||||
it('reports local HTTP daemon failures without a ktx-daemon spawn fallback cascade', async () => {
|
||||
const fetch = vi
|
||||
.fn()
|
||||
.mockResolvedValue(
|
||||
new Response('Embedding compute failed: httpx.InvalidURL: Invalid port', { status: 500 }),
|
||||
);
|
||||
|
||||
const provider = createKtxEmbeddingProvider(
|
||||
{
|
||||
backend: 'sentence-transformers',
|
||||
model: 'all-MiniLM-L6-v2',
|
||||
dimensions: 2,
|
||||
sentenceTransformers: { baseURL: 'http://127.0.0.1:8765', pathPrefix: '' },
|
||||
},
|
||||
{ fetch },
|
||||
);
|
||||
|
||||
await expect(provider.embed('hello')).rejects.toThrow(
|
||||
'Embedding provider sentence-transformers request failed with HTTP 500: Embedding compute failed: httpx.InvalidURL: Invalid port',
|
||||
);
|
||||
await expect(provider.embed('hello')).rejects.not.toThrow('ktx-daemon fallback failed');
|
||||
expect(fetch).toHaveBeenCalledTimes(1);
|
||||
});
|
||||
});
|
||||
|
|
@ -1,146 +0,0 @@
|
|||
import type { ModelMessage } from 'ai';
|
||||
import { describe, expect, it } from 'vitest';
|
||||
import { KtxMessageBuilder, splitKtxSystemMessages } from './message-builder.js';
|
||||
import { createKtxLlmProvider } from './model-provider.js';
|
||||
|
||||
function makeBuilder(overrides: Parameters<typeof createKtxLlmProvider>[0]['promptCaching'] = {}) {
|
||||
const provider = createKtxLlmProvider({
|
||||
backend: 'gateway',
|
||||
gateway: { baseURL: 'https://gateway.test' },
|
||||
modelSlots: { default: 'anthropic/claude-sonnet-4-6' },
|
||||
promptCaching: { enabled: true, ...overrides },
|
||||
});
|
||||
return new KtxMessageBuilder(provider);
|
||||
}
|
||||
|
||||
describe('KtxMessageBuilder.build', () => {
|
||||
it('caches static system, last sorted tool, and last history message', () => {
|
||||
const builder = makeBuilder();
|
||||
|
||||
const out = builder.build({
|
||||
parts: { staticSystem: 'STATIC', dynamicSystem: 'DYNAMIC' },
|
||||
history: [
|
||||
{ role: 'user', content: 'first' },
|
||||
{ role: 'assistant', content: [{ type: 'text', text: 'reply A' }, { type: 'text', text: 'reply B' }] } as ModelMessage,
|
||||
],
|
||||
currentMessage: { role: 'user', content: 'now' },
|
||||
tools: {
|
||||
zoo: { description: 'z' },
|
||||
apple: { description: 'a' },
|
||||
},
|
||||
model: 'anthropic/claude-sonnet-4-6',
|
||||
});
|
||||
|
||||
expect(out.messages[0]).toMatchObject({
|
||||
role: 'system',
|
||||
content: 'STATIC',
|
||||
providerOptions: { anthropic: { cacheControl: { type: 'ephemeral', ttl: '1h' } } },
|
||||
});
|
||||
expect(out.messages[1]).toMatchObject({ role: 'system', content: 'DYNAMIC' });
|
||||
expect((out.messages[1] as { providerOptions?: unknown }).providerOptions).toBeUndefined();
|
||||
expect((out.messages[3] as { content: Array<{ providerOptions?: unknown }> }).content[1].providerOptions).toEqual({
|
||||
anthropic: { cacheControl: { type: 'ephemeral', ttl: '5m' } },
|
||||
});
|
||||
expect(Object.keys(out.tools)).toEqual(['apple', 'zoo']);
|
||||
expect((out.tools.zoo as { providerOptions?: unknown }).providerOptions).toEqual({
|
||||
anthropic: { cacheControl: { type: 'ephemeral', ttl: '1h' } },
|
||||
});
|
||||
});
|
||||
|
||||
it('wraps leading user context onto currentMessage as a system reminder part', () => {
|
||||
const builder = makeBuilder();
|
||||
|
||||
const out = builder.build({
|
||||
parts: { staticSystem: 'STATIC', leadingUserContext: 'current_date: 2026-05-04' },
|
||||
history: [],
|
||||
currentMessage: { role: 'user', content: 'question' },
|
||||
tools: {},
|
||||
model: 'anthropic/claude-sonnet-4-6',
|
||||
});
|
||||
|
||||
expect(out.messages[out.messages.length - 1]).toMatchObject({
|
||||
role: 'user',
|
||||
content: [
|
||||
{ type: 'text', text: '<system-reminder>\ncurrent_date: 2026-05-04\n</system-reminder>' },
|
||||
{ type: 'text', text: 'question' },
|
||||
],
|
||||
});
|
||||
});
|
||||
|
||||
it('omits cache markers for non-Anthropic protocol models', () => {
|
||||
const builder = makeBuilder();
|
||||
|
||||
const out = builder.wrapSimple({
|
||||
system: 'SYS',
|
||||
messages: [{ role: 'user', content: 'q' }],
|
||||
tools: { z: {} },
|
||||
model: 'gpt-5',
|
||||
});
|
||||
|
||||
expect((out.messages[0] as { providerOptions?: unknown }).providerOptions).toBeUndefined();
|
||||
expect((out.tools.z as { providerOptions?: unknown }).providerOptions).toBeUndefined();
|
||||
});
|
||||
|
||||
it('clamps every TTL to 5m for Vertex when vertexFallbackTo5m is enabled', () => {
|
||||
const provider = createKtxLlmProvider({
|
||||
backend: 'vertex',
|
||||
vertex: { project: 'ktx-test', location: 'us-east5' },
|
||||
modelSlots: { default: 'claude-sonnet-4-6' },
|
||||
promptCaching: {
|
||||
enabled: true,
|
||||
systemTtl: '1h',
|
||||
toolsTtl: '1h',
|
||||
historyTtl: '1h',
|
||||
vertexFallbackTo5m: true,
|
||||
},
|
||||
});
|
||||
const builder = new KtxMessageBuilder(provider);
|
||||
|
||||
const out = builder.build({
|
||||
parts: { staticSystem: 'STATIC' },
|
||||
history: [{ role: 'user', content: 'history' }],
|
||||
currentMessage: { role: 'user', content: 'now' },
|
||||
tools: { z: {} },
|
||||
model: 'claude-sonnet-4-6',
|
||||
});
|
||||
|
||||
expect((out.messages[0] as { providerOptions: any }).providerOptions.anthropic.cacheControl.ttl).toBe('5m');
|
||||
expect((out.messages[1] as { content: Array<{ providerOptions: any }> }).content[0].providerOptions.anthropic.cacheControl.ttl).toBe(
|
||||
'5m',
|
||||
);
|
||||
expect((out.tools.z as { providerOptions: any }).providerOptions.anthropic.cacheControl.ttl).toBe('5m');
|
||||
});
|
||||
});
|
||||
|
||||
describe('splitKtxSystemMessages', () => {
|
||||
it('returns undefined system when no system messages are present', () => {
|
||||
const split = splitKtxSystemMessages([
|
||||
{ role: 'user', content: 'hello' },
|
||||
{ role: 'assistant', content: 'hi' },
|
||||
]);
|
||||
expect(split.system).toBeUndefined();
|
||||
expect(split.messages).toHaveLength(2);
|
||||
});
|
||||
|
||||
it('returns a single system message object when one system message is present, preserving providerOptions', () => {
|
||||
const systemMessage = {
|
||||
role: 'system' as const,
|
||||
content: 'You are helpful.',
|
||||
providerOptions: { anthropic: { cacheControl: { type: 'ephemeral' } } },
|
||||
};
|
||||
const split = splitKtxSystemMessages([systemMessage, { role: 'user', content: 'hello' }]);
|
||||
expect(split.system).toBe(systemMessage);
|
||||
expect(split.messages).toEqual([{ role: 'user', content: 'hello' }]);
|
||||
});
|
||||
|
||||
it('returns an array of system messages when multiple are present, in order', () => {
|
||||
const split = splitKtxSystemMessages([
|
||||
{ role: 'system', content: 'cached' },
|
||||
{ role: 'system', content: 'fresh' },
|
||||
{ role: 'user', content: 'hello' },
|
||||
]);
|
||||
expect(Array.isArray(split.system)).toBe(true);
|
||||
expect(split.system).toHaveLength(2);
|
||||
expect(split.messages).toEqual([{ role: 'user', content: 'hello' }]);
|
||||
});
|
||||
});
|
||||
|
|
@ -1,77 +0,0 @@
|
|||
import { wrapLanguageModel as defaultWrapLanguageModel } from 'ai';
|
||||
import { describe, expect, it, vi } from 'vitest';
|
||||
import { runKtxLlmHealthCheck } from './model-health.js';
|
||||
|
||||
const anthropicModel = { modelId: 'claude-sonnet-4-6' } as never;
|
||||
|
||||
describe('KTX LLM health check', () => {
|
||||
it('runs a minimal non-streaming model call through the configured provider', async () => {
|
||||
const generateText = vi.fn(async () => ({ text: 'ok' }));
|
||||
const createAnthropic = vi.fn(() => vi.fn(() => anthropicModel));
|
||||
const wrapLanguageModel = vi.fn(defaultWrapLanguageModel);
|
||||
|
||||
await expect(
|
||||
runKtxLlmHealthCheck(
|
||||
{
|
||||
backend: 'anthropic',
|
||||
anthropic: { apiKey: 'sk-ant-test' }, // pragma: allowlist secret
|
||||
modelSlots: { default: 'claude-sonnet-4-6' },
|
||||
},
|
||||
{ deps: { createAnthropic, generateText, devtoolsEnabled: true, wrapLanguageModel } },
|
||||
),
|
||||
).resolves.toEqual({ ok: true });
|
||||
|
||||
expect(createAnthropic).toHaveBeenCalledWith(
|
||||
expect.objectContaining({
|
||||
apiKey: 'sk-ant-test', // pragma: allowlist secret
|
||||
}),
|
||||
);
|
||||
expect(generateText).toHaveBeenCalledWith(
|
||||
expect.objectContaining({
|
||||
model: anthropicModel,
|
||||
prompt: 'Reply with exactly: ok',
|
||||
temperature: 0,
|
||||
maxOutputTokens: 8,
|
||||
}),
|
||||
);
|
||||
expect(wrapLanguageModel).not.toHaveBeenCalled();
|
||||
});
|
||||
|
||||
it('returns a failed result without exposing secret values', async () => {
|
||||
const generateText = vi.fn(async () => {
|
||||
throw new Error('401 invalid x-api-key sk-ant-secret');
|
||||
});
|
||||
|
||||
await expect(
|
||||
runKtxLlmHealthCheck(
|
||||
{
|
||||
backend: 'anthropic',
|
||||
anthropic: { apiKey: 'sk-ant-secret' }, // pragma: allowlist secret
|
||||
modelSlots: { default: 'claude-sonnet-4-6' },
|
||||
},
|
||||
{
|
||||
deps: {
|
||||
createAnthropic: vi.fn(() => vi.fn(() => anthropicModel)),
|
||||
generateText,
|
||||
},
|
||||
},
|
||||
),
|
||||
).resolves.toEqual({
|
||||
ok: false,
|
||||
message: '401 invalid x-api-key [redacted]',
|
||||
});
|
||||
});
|
||||
|
||||
it('reports claude-code as unsupported by the AI SDK health check', async () => {
|
||||
const result = await runKtxLlmHealthCheck({
|
||||
backend: 'claude-code',
|
||||
modelSlots: { default: 'sonnet' },
|
||||
promptCaching: { enabled: false },
|
||||
});
|
||||
|
||||
expect(result).toEqual({
|
||||
ok: false,
|
||||
message: expect.stringContaining('claude-code is not an AI SDK LanguageModel backend'),
|
||||
});
|
||||
});
|
||||
});
|
||||
|
|
@ -1,315 +0,0 @@
|
|||
import { devToolsMiddleware as defaultDevToolsMiddleware } from '@ai-sdk/devtools';
|
||||
import { wrapLanguageModel as defaultWrapLanguageModel, type LanguageModel } from 'ai';
|
||||
import { describe, expect, it, vi } from 'vitest';
|
||||
import { createKtxLlmProvider, type KtxLlmProviderFactoryDeps } from './model-provider.js';
|
||||
|
||||
const languageModel = (modelId: string, provider = 'test'): LanguageModel => ({ modelId, provider }) as LanguageModel;
|
||||
const devtoolsMiddleware = (): ReturnType<typeof defaultDevToolsMiddleware> => ({ specificationVersion: 'v3' });
|
||||
const wrapWith = (model: LanguageModel) =>
|
||||
vi.fn((_options: Parameters<typeof defaultWrapLanguageModel>[0]) => model as ReturnType<typeof defaultWrapLanguageModel>);
|
||||
|
||||
describe('createKtxLlmProvider', () => {
|
||||
it('wraps language models with DevTools middleware when explicitly enabled', () => {
|
||||
const anthropicModel = languageModel('claude-sonnet-4-6', 'anthropic');
|
||||
const wrappedModel = languageModel('claude-sonnet-4-6', 'anthropic-devtools');
|
||||
const middleware = devtoolsMiddleware();
|
||||
const wrapLanguageModel = wrapWith(wrappedModel);
|
||||
const devToolsMiddleware = vi.fn(devtoolsMiddleware);
|
||||
|
||||
const provider = createKtxLlmProvider(
|
||||
{
|
||||
backend: 'anthropic',
|
||||
anthropic: { apiKey: 'test-anthropic-key' }, // pragma: allowlist secret
|
||||
modelSlots: { default: 'claude-sonnet-4-6' },
|
||||
promptCaching: { enabled: false },
|
||||
},
|
||||
{
|
||||
createAnthropic: vi.fn(() => vi.fn(() => anthropicModel)),
|
||||
devtoolsEnabled: true,
|
||||
wrapLanguageModel,
|
||||
devToolsMiddleware,
|
||||
} satisfies KtxLlmProviderFactoryDeps,
|
||||
);
|
||||
|
||||
expect(provider.getModel('default')).toBe(wrappedModel);
|
||||
expect(devToolsMiddleware).toHaveBeenCalledTimes(1);
|
||||
expect(wrapLanguageModel).toHaveBeenCalledWith({
|
||||
model: anthropicModel,
|
||||
middleware,
|
||||
modelId: 'claude-sonnet-4-6',
|
||||
providerId: 'anthropic',
|
||||
});
|
||||
});
|
||||
|
||||
it('does not wrap language models by default', () => {
|
||||
const anthropicModel = languageModel('claude-sonnet-4-6', 'anthropic');
|
||||
const wrapLanguageModel = vi.fn(defaultWrapLanguageModel);
|
||||
const devToolsMiddleware = vi.fn(defaultDevToolsMiddleware);
|
||||
|
||||
const provider = createKtxLlmProvider(
|
||||
{
|
||||
backend: 'anthropic',
|
||||
anthropic: { apiKey: 'test-anthropic-key' }, // pragma: allowlist secret
|
||||
modelSlots: { default: 'claude-sonnet-4-6' },
|
||||
promptCaching: { enabled: false },
|
||||
},
|
||||
{
|
||||
createAnthropic: vi.fn(() => vi.fn(() => anthropicModel)),
|
||||
devtoolsEnabled: false,
|
||||
wrapLanguageModel,
|
||||
devToolsMiddleware,
|
||||
} satisfies KtxLlmProviderFactoryDeps,
|
||||
);
|
||||
|
||||
expect(provider.getModel('default')).toBe(anthropicModel);
|
||||
expect(wrapLanguageModel).not.toHaveBeenCalled();
|
||||
expect(devToolsMiddleware).not.toHaveBeenCalled();
|
||||
});
|
||||
|
||||
it('wraps language models when KTX_AI_DEVTOOLS_ENABLED is true', () => {
|
||||
const originalEnv = process.env.KTX_AI_DEVTOOLS_ENABLED;
|
||||
process.env.KTX_AI_DEVTOOLS_ENABLED = 'true';
|
||||
try {
|
||||
const gatewayModel = languageModel('anthropic/claude-sonnet-4-6', 'gateway');
|
||||
const wrappedModel = languageModel('anthropic/claude-sonnet-4-6', 'gateway-devtools');
|
||||
const wrapLanguageModel = wrapWith(wrappedModel);
|
||||
|
||||
const provider = createKtxLlmProvider(
|
||||
{
|
||||
backend: 'gateway',
|
||||
gateway: { baseURL: 'https://gateway.test/v1' },
|
||||
modelSlots: { default: 'anthropic/claude-sonnet-4-6' },
|
||||
promptCaching: { enabled: false },
|
||||
},
|
||||
{
|
||||
createGateway: vi.fn(() => vi.fn(() => gatewayModel)),
|
||||
wrapLanguageModel,
|
||||
devToolsMiddleware: vi.fn(devtoolsMiddleware),
|
||||
} satisfies KtxLlmProviderFactoryDeps,
|
||||
);
|
||||
|
||||
expect(provider.getModel('default')).toBe(wrappedModel);
|
||||
expect(wrapLanguageModel).toHaveBeenCalledTimes(1);
|
||||
} finally {
|
||||
if (originalEnv === undefined) {
|
||||
delete process.env.KTX_AI_DEVTOOLS_ENABLED;
|
||||
} else {
|
||||
process.env.KTX_AI_DEVTOOLS_ENABLED = originalEnv;
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
it('does not wrap language models in production even when enabled', () => {
|
||||
const originalNodeEnv = process.env.NODE_ENV;
|
||||
process.env.NODE_ENV = 'production';
|
||||
try {
|
||||
const anthropicModel = languageModel('claude-sonnet-4-6', 'anthropic');
|
||||
const wrapLanguageModel = vi.fn(defaultWrapLanguageModel);
|
||||
const devToolsMiddleware = vi.fn(defaultDevToolsMiddleware);
|
||||
|
||||
const provider = createKtxLlmProvider(
|
||||
{
|
||||
backend: 'anthropic',
|
||||
anthropic: { apiKey: 'test-anthropic-key' }, // pragma: allowlist secret
|
||||
modelSlots: { default: 'claude-sonnet-4-6' },
|
||||
promptCaching: { enabled: false },
|
||||
},
|
||||
{
|
||||
createAnthropic: vi.fn(() => vi.fn(() => anthropicModel)),
|
||||
devtoolsEnabled: true,
|
||||
wrapLanguageModel,
|
||||
devToolsMiddleware,
|
||||
} satisfies KtxLlmProviderFactoryDeps,
|
||||
);
|
||||
|
||||
expect(provider.getModel('default')).toBe(anthropicModel);
|
||||
expect(wrapLanguageModel).not.toHaveBeenCalled();
|
||||
expect(devToolsMiddleware).not.toHaveBeenCalled();
|
||||
} finally {
|
||||
if (originalNodeEnv === undefined) {
|
||||
delete process.env.NODE_ENV;
|
||||
} else {
|
||||
process.env.NODE_ENV = originalNodeEnv;
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
it('uses direct Anthropic with both beta headers', () => {
|
||||
const anthropicModel = languageModel('claude-sonnet-4-6', 'anthropic');
|
||||
const anthropic = vi.fn(() => anthropicModel);
|
||||
const createAnthropic = vi.fn(() => anthropic);
|
||||
|
||||
const provider = createKtxLlmProvider(
|
||||
{
|
||||
backend: 'anthropic',
|
||||
anthropic: { apiKey: 'test-anthropic-key', baseURL: 'https://anthropic.test' }, // pragma: allowlist secret
|
||||
modelSlots: { default: 'claude-sonnet-4-6' },
|
||||
promptCaching: { enabled: false },
|
||||
},
|
||||
{ createAnthropic, devtoolsEnabled: false },
|
||||
);
|
||||
|
||||
expect(provider.getModel('default')).toBe(anthropicModel);
|
||||
expect(createAnthropic).toHaveBeenCalledWith({
|
||||
apiKey: 'test-anthropic-key', // pragma: allowlist secret
|
||||
baseURL: 'https://anthropic.test',
|
||||
headers: {
|
||||
'anthropic-beta': 'interleaved-thinking-2025-05-14,extended-cache-ttl-2025-04-11',
|
||||
},
|
||||
});
|
||||
expect(anthropic).toHaveBeenCalledWith('claude-sonnet-4-6');
|
||||
});
|
||||
|
||||
it('uses Vertex Anthropic without the direct-Anthropic beta header', () => {
|
||||
const vertexModel = languageModel('claude-sonnet-4-6', 'vertex');
|
||||
const vertex = vi.fn(() => vertexModel);
|
||||
const createVertexAnthropic = vi.fn(() => vertex);
|
||||
|
||||
const provider = createKtxLlmProvider(
|
||||
{
|
||||
backend: 'vertex',
|
||||
vertex: { project: 'ktx-test', location: 'us-east5' },
|
||||
modelSlots: { default: 'claude-sonnet-4-6' },
|
||||
promptCaching: { enabled: false },
|
||||
},
|
||||
{ createVertexAnthropic, devtoolsEnabled: false },
|
||||
);
|
||||
|
||||
expect(provider.getModel('default')).toBe(vertexModel);
|
||||
expect(createVertexAnthropic).toHaveBeenCalledWith({ project: 'ktx-test', location: 'us-east5' });
|
||||
expect(vertex).toHaveBeenCalledWith('claude-sonnet-4-6');
|
||||
});
|
||||
|
||||
it('uses Gateway and supports role fallback to default', () => {
|
||||
const gatewayModel = languageModel('anthropic/claude-sonnet-4-6', 'gateway');
|
||||
const gateway = vi.fn(() => gatewayModel);
|
||||
const createGateway = vi.fn(() => gateway);
|
||||
|
||||
const provider = createKtxLlmProvider(
|
||||
{
|
||||
backend: 'gateway',
|
||||
gateway: { apiKey: 'gateway-key', baseURL: 'https://gateway.test/v1' }, // pragma: allowlist secret
|
||||
modelSlots: { default: 'anthropic/claude-sonnet-4-6' },
|
||||
promptCaching: { enabled: false },
|
||||
},
|
||||
{ createGateway, devtoolsEnabled: false },
|
||||
);
|
||||
|
||||
expect(provider.getModel('curator')).toBe(gatewayModel);
|
||||
expect(createGateway).toHaveBeenCalledWith({
|
||||
apiKey: 'gateway-key', // pragma: allowlist secret
|
||||
baseURL: 'https://gateway.test/v1',
|
||||
headers: {
|
||||
'anthropic-beta': 'interleaved-thinking-2025-05-14,extended-cache-ttl-2025-04-11',
|
||||
},
|
||||
});
|
||||
expect(gateway).toHaveBeenCalledWith('anthropic/claude-sonnet-4-6');
|
||||
});
|
||||
|
||||
it('uses explicit role overrides before default', () => {
|
||||
const anthropic = vi.fn((modelId: string) => languageModel(modelId, 'anthropic'));
|
||||
|
||||
const provider = createKtxLlmProvider(
|
||||
{
|
||||
backend: 'anthropic',
|
||||
anthropic: { apiKey: 'test-anthropic-key' }, // pragma: allowlist secret
|
||||
modelSlots: {
|
||||
default: 'claude-sonnet-4-6',
|
||||
triage: 'claude-haiku-4-5',
|
||||
repair: 'claude-opus-4-7',
|
||||
},
|
||||
promptCaching: { enabled: false },
|
||||
},
|
||||
{ createAnthropic: vi.fn(() => anthropic) },
|
||||
);
|
||||
|
||||
expect((provider.getModel('triage') as { modelId: string }).modelId).toBe('claude-haiku-4-5');
|
||||
expect((provider.getModel('repair') as { modelId: string }).modelId).toBe('claude-opus-4-7');
|
||||
expect((provider.getModel('reconcile') as { modelId: string }).modelId).toBe('claude-sonnet-4-6');
|
||||
});
|
||||
|
||||
it('emits cache markers only when enabled and the model speaks Anthropic protocol', () => {
|
||||
const provider = createKtxLlmProvider(
|
||||
{
|
||||
backend: 'gateway',
|
||||
gateway: { baseURL: 'https://gateway.test/v1' },
|
||||
modelSlots: { default: 'anthropic/claude-sonnet-4-6' },
|
||||
promptCaching: { enabled: true },
|
||||
},
|
||||
{ createGateway: vi.fn(() => vi.fn((modelId: string) => languageModel(modelId, 'gateway'))) },
|
||||
);
|
||||
|
||||
expect(provider.cacheMarker('1h', 'anthropic/claude-sonnet-4-6')).toEqual({
|
||||
anthropic: { cacheControl: { type: 'ephemeral', ttl: '1h' } },
|
||||
});
|
||||
expect(provider.cacheMarker('1h', 'gpt-5')).toBeUndefined();
|
||||
});
|
||||
|
||||
it('returns Anthropic thinking provider options', () => {
|
||||
const provider = createKtxLlmProvider(
|
||||
{
|
||||
backend: 'anthropic',
|
||||
anthropic: { apiKey: 'test-anthropic-key' }, // pragma: allowlist secret
|
||||
modelSlots: { default: 'claude-sonnet-4-6' },
|
||||
promptCaching: { enabled: false },
|
||||
},
|
||||
{ createAnthropic: vi.fn(() => vi.fn((modelId: string) => languageModel(modelId, 'anthropic'))) },
|
||||
);
|
||||
|
||||
expect(provider.thinkingProviderOptions('default', 12000)).toEqual({
|
||||
anthropic: {
|
||||
thinking: { type: 'enabled', budgetTokens: 12000 },
|
||||
},
|
||||
});
|
||||
});
|
||||
|
||||
it('defaults prompt caching to enabled with canonical TTLs', () => {
|
||||
const provider = createKtxLlmProvider(
|
||||
{
|
||||
backend: 'gateway',
|
||||
gateway: { baseURL: 'https://gateway.test/v1' },
|
||||
modelSlots: { default: 'anthropic/claude-sonnet-4-6' },
|
||||
},
|
||||
{ createGateway: vi.fn(() => vi.fn((modelId: string) => languageModel(modelId, 'gateway'))) },
|
||||
);
|
||||
|
||||
expect(provider.promptCachingConfig()).toEqual({
|
||||
enabled: true,
|
||||
systemTtl: '1h',
|
||||
toolsTtl: '1h',
|
||||
historyTtl: '5m',
|
||||
cacheSystem: true,
|
||||
cacheTools: true,
|
||||
cacheHistory: true,
|
||||
vertexFallbackTo5m: false,
|
||||
});
|
||||
expect(provider.cacheMarker('1h', 'anthropic/claude-sonnet-4-6')).toEqual({
|
||||
anthropic: { cacheControl: { type: 'ephemeral', ttl: '1h' } },
|
||||
});
|
||||
});
|
||||
|
||||
it('preserves explicit prompt caching opt-out', () => {
|
||||
const provider = createKtxLlmProvider(
|
||||
{
|
||||
backend: 'anthropic',
|
||||
anthropic: { apiKey: 'test-anthropic-key' }, // pragma: allowlist secret
|
||||
modelSlots: { default: 'claude-sonnet-4-6' },
|
||||
promptCaching: { enabled: false },
|
||||
},
|
||||
{ createAnthropic: vi.fn(() => vi.fn((modelId: string) => languageModel(modelId, 'anthropic'))) },
|
||||
);
|
||||
|
||||
expect(provider.promptCachingConfig().enabled).toBe(false);
|
||||
expect(provider.cacheMarker('1h', 'claude-sonnet-4-6')).toBeUndefined();
|
||||
});
|
||||
|
||||
it('throws instead of falling through when an unsupported LLM backend is passed to the AI SDK provider factory', () => {
|
||||
expect(() =>
|
||||
createKtxLlmProvider({
|
||||
backend: 'claude-code',
|
||||
modelSlots: { default: 'sonnet' },
|
||||
promptCaching: { enabled: false },
|
||||
}),
|
||||
).toThrow('claude-code is not an AI SDK LanguageModel backend');
|
||||
});
|
||||
});
|
||||
|
|
@ -1,93 +0,0 @@
|
|||
import { NoSuchToolError, type LanguageModel } from 'ai';
|
||||
import { describe, expect, it, vi } from 'vitest';
|
||||
import { createKtxToolCallRepairHandler } from './repair.js';
|
||||
|
||||
const repairModel = { modelId: 'claude-repair', provider: 'anthropic' } as LanguageModel;
|
||||
|
||||
describe('createKtxToolCallRepairHandler', () => {
|
||||
it('returns null for NoSuchToolError', async () => {
|
||||
const handler = createKtxToolCallRepairHandler({
|
||||
source: 'unit',
|
||||
getRepairModel: () => repairModel,
|
||||
generateText: vi.fn(),
|
||||
});
|
||||
|
||||
await expect(
|
||||
handler({
|
||||
system: undefined,
|
||||
messages: [],
|
||||
toolCall: { type: 'tool-call', toolName: 'missing', toolCallId: 'tc_1', input: '{}' },
|
||||
tools: {},
|
||||
inputSchema: async () => ({}),
|
||||
error: new NoSuchToolError({ toolName: 'missing' }),
|
||||
}),
|
||||
).resolves.toBeNull();
|
||||
});
|
||||
|
||||
it('repairs string input by local JSON extraction without an LLM call', async () => {
|
||||
const generateText = vi.fn();
|
||||
const handler = createKtxToolCallRepairHandler({
|
||||
source: 'unit',
|
||||
getRepairModel: () => repairModel,
|
||||
generateText,
|
||||
});
|
||||
|
||||
await expect(
|
||||
handler({
|
||||
system: undefined,
|
||||
messages: [],
|
||||
toolCall: {
|
||||
type: 'tool-call',
|
||||
toolName: 'write_source',
|
||||
toolCallId: 'tc_2',
|
||||
input: 'prefix {"path":"orders.yaml"} suffix',
|
||||
},
|
||||
tools: { write_source: {} as never },
|
||||
inputSchema: async () => ({ type: 'object' }),
|
||||
error: new Error('Invalid tool input') as never,
|
||||
}),
|
||||
).resolves.toEqual({
|
||||
type: 'tool-call',
|
||||
toolName: 'write_source',
|
||||
toolCallId: 'tc_2',
|
||||
input: '{"path":"orders.yaml"}',
|
||||
});
|
||||
expect(generateText).not.toHaveBeenCalled();
|
||||
});
|
||||
|
||||
it('falls back to the repair model when local extraction fails', async () => {
|
||||
const generateText = vi.fn().mockResolvedValue({ text: '{"path":"customers.yaml"}' });
|
||||
const handler = createKtxToolCallRepairHandler({
|
||||
source: 'unit',
|
||||
getRepairModel: () => repairModel,
|
||||
generateText,
|
||||
});
|
||||
|
||||
await expect(
|
||||
handler({
|
||||
system: undefined,
|
||||
messages: [],
|
||||
toolCall: {
|
||||
type: 'tool-call',
|
||||
toolName: 'write_source',
|
||||
toolCallId: 'tc_3',
|
||||
input: 'not json',
|
||||
},
|
||||
tools: { write_source: {} as never },
|
||||
inputSchema: async () => ({ type: 'object', properties: { path: { type: 'string' } } }),
|
||||
error: new Error('Invalid tool input') as never,
|
||||
}),
|
||||
).resolves.toEqual({
|
||||
type: 'tool-call',
|
||||
toolName: 'write_source',
|
||||
toolCallId: 'tc_3',
|
||||
input: '{"path":"customers.yaml"}',
|
||||
});
|
||||
expect(generateText).toHaveBeenCalledWith(
|
||||
expect.objectContaining({
|
||||
model: repairModel,
|
||||
prompt: expect.stringContaining('The model tried to call the tool "write_source"'),
|
||||
}),
|
||||
);
|
||||
});
|
||||
});
|
||||
Loading…
Add table
Add a link
Reference in a new issue