mirror of
https://github.com/Kaelio/ktx.git
synced 2026-06-07 07:55:13 +02:00
56985b7e09
* 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
77 lines
2.4 KiB
TypeScript
77 lines
2.4 KiB
TypeScript
import { wrapLanguageModel as defaultWrapLanguageModel } from 'ai';
|
|
import { describe, expect, it, vi } from 'vitest';
|
|
import { runKtxLlmHealthCheck } from '../../src/llm/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'),
|
|
});
|
|
});
|
|
});
|