@kaelio/ktx 0.1.0-rc.5 → 0.1.0

package/dist/setup.js

@@ -3,14 +3,18 @@ import { basename, join, resolve } from 'node:path';
 import { getLatestLocalIngestStatus, savedMemoryCountsForReport } from '@ktx/context/ingest';
 import { ktxLocalStateDbPath, loadKtxProject, readKtxSetupState, } from '@ktx/context/project';
 import { formatSetupNextStepLines } from './next-steps.js';
+import { runtimeInstallPolicyFromFlags } from './managed-python-command.js';
+import { readManagedPythonRuntimeStatus } from './managed-python-runtime.js';
+import { resolveProjectRuntimeRequirements } from './runtime-requirements.js';
 import { isKtxSetupExitError } from './setup-interrupt.js';
-import { readKtxAgentInstallManifest, runKtxSetupAgentsStep, } from './setup-agents.js';
+import { readKtxAgentInstallManifest, runKtxSetupAgentsStep, targetDisplayName, } from './setup-agents.js';
 import { runKtxSetupDatabasesStep, } from './setup-databases.js';
 import { runKtxSetupEmbeddingsStep } from './setup-embeddings.js';
 import { isKtxSetupLlmConfigReady, runKtxSetupAnthropicModelStep, } from './setup-models.js';
 import { runKtxSetupProjectStep } from './setup-project.js';
 import { isKtxPreAgentSetupReady, isKtxSetupReady, runKtxSetupReadyChangeMenu, } from './setup-ready-menu.js';
 import { runKtxSetupSourcesStep } from './setup-sources.js';
+import { runKtxSetupRuntimeStep, } from './setup-runtime.js';
 import { createKtxSetupPromptAdapter, createKtxSetupUiAdapter, } from './setup-prompts.js';
 import { readKtxSetupContextState, runKtxSetupContextStep, setupContextStatusFromState, } from './setup-context.js';
 const SOURCE_DRIVERS = new Set(['dbt', 'metricflow', 'metabase', 'looker', 'lookml', 'notion']);
@@ -86,7 +90,7 @@ async function readIngestContextStatus(project) {
         runId: report.runId,
     };
 }
-export async function readKtxSetupStatus(projectDir) {
+export async function readKtxSetupStatus(projectDir, options = {}) {
     const resolvedProjectDir = resolve(projectDir);
     if (!existsSync(join(resolvedProjectDir, 'ktx.yaml'))) {
         return {
@@ -95,6 +99,7 @@ export async function readKtxSetupStatus(projectDir) {
             embeddings: { ready: false },
             databases: [],
             sources: [],
+            runtime: { required: false, ready: true, features: [] },
             context: setupContextStatusFromState(await readKtxSetupContextState(resolvedProjectDir)),
             agents: [],
         };
@@ -130,6 +135,20 @@ export async function readKtxSetupStatus(projectDir) {
         });
     }
     const agents = [...agentMap.values()];
+    const runtimeRequirements = resolveProjectRuntimeRequirements(project.config, {
+        env: options.env ?? process.env,
+    });
+    let runtimeReady = runtimeRequirements.features.length === 0 || completedSteps.includes('runtime');
+    let runtimeDetail;
+    if (runtimeRequirements.features.length > 0 && options.cliVersion) {
+        const readRuntimeStatus = options.readRuntimeStatus ?? readManagedPythonRuntimeStatus;
+        const runtimeStatus = await readRuntimeStatus({ cliVersion: options.cliVersion, env: options.env ?? process.env });
+        runtimeDetail = runtimeStatus.detail;
+        runtimeReady =
+            runtimeStatus.kind === 'ready' &&
+                runtimeStatus.manifest !== undefined &&
+                runtimeRequirements.features.every((feature) => runtimeStatus.manifest?.features.includes(feature));
+    }
     return {
         project: { path: resolvedProjectDir, ready: true, name: basename(project.projectDir) || project.projectDir },
         llm,
@@ -142,6 +161,12 @@ export async function readKtxSetupStatus(projectDir) {
             ...source,
             ready: completedSteps.includes('sources'),
         })),
+        runtime: {
+            required: runtimeRequirements.features.length > 0,
+            ready: runtimeReady,
+            features: runtimeRequirements.features,
+            ...(runtimeDetail ? { detail: runtimeDetail } : {}),
+        },
         context: ingestContextStatus ?? setupContextStatus,
         agents,
     };
@@ -180,6 +205,11 @@ export function formatKtxSetupStatus(status) {
         `Embeddings ready: ${formatReady(status.embeddings.ready)}${status.embeddings.model ? ` (${status.embeddings.model})` : ''}`,
         `Databases configured: ${formatConnectionList(status.databases.map((database) => database.connectionId))}`,
         `Context sources configured: ${formatConnectionList(status.sources.map((source) => source.connectionId))}`,
+        ...(status.runtime.required
+            ? [
+                `Runtime ready: ${formatReady(status.runtime.ready)}${status.runtime.features.length > 0 ? ` (${status.runtime.features.join(', ')})` : ''}`,
+            ]
+            : []),
         `KTX context built: ${formatContextBuilt(status.context)}`,
         `Agent integration ready: ${formatReady(status.agents.some((agent) => agent.ready))}${status.agents.length > 0 ? ` (${status.agents.map((agent) => `${agent.target}:${agent.scope}`).join(', ')})` : ''}`,
     ];
@@ -188,6 +218,26 @@ export function formatKtxSetupStatus(status) {
     }
     return `${lines.join('\n')}\n`;
 }
+export function formatKtxSetupCompletionSummary(status, options = {}) {
+    const readyAgents = status.agents.filter((agent) => agent.ready).map((agent) => targetDisplayName(agent.target));
+    const lines = [
+        'Project',
+        `  ${status.project.path}`,
+        '',
+        'Context',
+        `  ${status.context.ready ? 'built' : formatContextBuilt(status.context)}`,
+        '',
+        'Agents configured',
+        `  ${readyAgents.length > 0 ? readyAgents.join(', ') : 'not installed'}`,
+    ];
+    const agentNextActions = options.agentNextActions?.trim();
+    if (agentNextActions) {
+        lines.push('', 'REQUIRED BEFORE USING AGENTS', '', ...agentNextActions.split('\n').map((line) => (line ? `  ${line}` : '')));
+    }
+    lines.push('', agentNextActions ? 'After that, try' : 'Try it');
+    lines.push('  Ask your agent: "Use KTX to show me the available tables."');
+    return lines.join('\n');
+}
 function setupStatusReady(status) {
     if (!status.project.ready) {
         return false;
@@ -198,7 +248,8 @@ function setupStatusReady(status) {
     return (status.llm.ready &&
         embeddingsReady(status.embeddings) &&
         status.databases.every((database) => database.ready) &&
-        status.sources.every((source) => source.ready));
+        status.sources.every((source) => source.ready) &&
+        status.runtime.ready);
 }
 function setupHasContextTargets(status) {
     return status.databases.length > 0 || status.sources.length > 0;
@@ -206,13 +257,14 @@ function setupHasContextTargets(status) {
 function setupContextReady(status) {
     return status.context.ready;
 }
-function writeContextNotReadyForAgents(projectDir, io) {
-    io.stderr.write('KTX context is not ready for agents.\n\n');
-    io.stderr.write(`Build context first:\n  ktx setup --project-dir ${resolve(projectDir)}\n\n`);
-    io.stderr.write(`Then install agent integration:\n  ktx setup --agents --project-dir ${resolve(projectDir)}\n`);
+function shouldPrintConciseReadySummary(status) {
+    return setupStatusReady(status) && setupContextReady(status) && status.agents.some((agent) => agent.ready);
 }
 function setupRuntimeInstallPolicy(args) {
-    return args.inputMode === 'disabled' && !args.yes ? 'never' : 'auto';
+    if (args.yes) {
+        return 'auto';
+    }
+    return runtimeInstallPolicyFromFlags({ input: args.inputMode === 'disabled' ? false : true });
 }
 async function commitSetupConfigChanges(projectDir) {
     const project = await loadKtxProject({ projectDir });
@@ -234,6 +286,7 @@ async function runKtxSetupInner(args, io, deps = {}) {
     setupUi.intro('KTX setup', io);
     let entryAction;
     let projectResult;
+    let agentNextActions;
     const canShowEntryMenu = args.showEntryMenu === true &&
         args.inputMode !== 'disabled' &&
         !args.agents &&
@@ -241,7 +294,7 @@ async function runKtxSetupInner(args, io, deps = {}) {
     setupLoop: while (true) {
         entryAction = undefined;
         if (canShowEntryMenu) {
-            const status = await readKtxSetupStatus(args.projectDir);
+            const status = await readKtxSetupStatus(args.projectDir, { cliVersion: args.cliVersion });
             entryAction = (await runKtxSetupEntryMenu(status, deps.entryMenuDeps)).action;
             if (entryAction === 'exit') {
                 (deps.entryMenuDeps?.prompts ?? createEntryMenuPromptAdapter()).cancel('Setup cancelled.');
@@ -270,7 +323,7 @@ async function runKtxSetupInner(args, io, deps = {}) {
             return projectResult.status === 'cancelled' ? 0 : 1;
         }
         const agentsRequested = args.agents || entryAction === 'agents';
-        const currentStatus = await readKtxSetupStatus(projectResult.projectDir);
+        const currentStatus = await readKtxSetupStatus(projectResult.projectDir, { cliVersion: args.cliVersion });
         let readyAction;
         if (args.inputMode !== 'disabled' && !agentsRequested) {
             if (isKtxSetupReady(currentStatus)) {
@@ -283,16 +336,18 @@ async function runKtxSetupInner(args, io, deps = {}) {
             }
         }
         const runOnly = readyAction;
+        const agentOnlySetup = agentsRequested || runOnly === 'agents';
         const shouldRunModels = !runOnly || runOnly === 'models';
         const shouldRunEmbeddings = !runOnly || runOnly === 'embeddings';
         const shouldRunDatabases = !runOnly || runOnly === 'databases';
         const shouldRunSources = !runOnly || runOnly === 'sources';
-        const shouldRunContext = agentsRequested || !runOnly || runOnly === 'context';
+        const shouldRunRuntime = !agentOnlySetup && (!runOnly || runOnly === 'runtime' || runOnly === 'context');
+        const shouldRunContext = !agentOnlySetup && (!runOnly || runOnly === 'context');
         const shouldRunAgents = agentsRequested || !runOnly || runOnly === 'agents';
         const showPromptInstructions = projectResult.confirmedCreation !== true;
-        const setupSteps = agentsRequested
-            ? ['context']
-            : ['models', 'embeddings', 'databases', 'sources', 'context'];
+        const setupSteps = agentOnlySetup
+            ? []
+            : ['models', 'embeddings', 'databases', 'sources', 'runtime', 'context'];
         if (shouldRunAgents && args.skipAgents !== true) {
             setupSteps.push('agents');
         }
@@ -306,6 +361,8 @@ async function runKtxSetupInner(args, io, deps = {}) {
                 return !args.skipDatabases && shouldRunDatabases;
             if (step === 'sources')
                 return args.skipSources !== true && shouldRunSources;
+            if (step === 'runtime')
+                return shouldRunRuntime;
             if (step === 'context')
                 return shouldRunContext;
             return shouldRunAgents && args.skipAgents !== true;
@@ -410,6 +467,16 @@ async function runKtxSetupInner(args, io, deps = {}) {
                     skipSources: args.skipSources === true || !shouldRunSources,
                 }, io);
             }
+            else if (step === 'runtime') {
+                const runtimeRunner = deps.runtime ??
+                    ((runtimeArgs, runtimeIo) => runKtxSetupRuntimeStep(runtimeArgs, runtimeIo, deps.runtimeDeps));
+                stepResult = await runtimeRunner({
+                    projectDir: projectResult.projectDir,
+                    inputMode: args.inputMode,
+                    cliVersion: args.cliVersion,
+                    runtimeInstallPolicy: setupRuntimeInstallPolicy(args),
+                }, io);
+            }
             else if (step === 'context') {
                 const contextRunner = deps.context ??
                     ((contextArgs, contextIo) => runKtxSetupContextStep(contextArgs, contextIo, deps.contextDeps));
@@ -424,7 +491,7 @@ async function runKtxSetupInner(args, io, deps = {}) {
             }
             else {
                 const agentsRunner = deps.agents ?? ((agentArgs, agentIo) => runKtxSetupAgentsStep(agentArgs, agentIo, deps.agentsDeps));
-                stepResult = await agentsRunner({
+                const agentResult = await agentsRunner({
                     projectDir: projectResult.projectDir,
                     inputMode: args.inputMode,
                     yes: args.yes,
@@ -433,7 +500,12 @@ async function runKtxSetupInner(args, io, deps = {}) {
                     scope: args.agentScope ?? 'project',
                     mode: 'mcp',
                     skipAgents: false,
+                    showNextActions: agentsRequested,
                 }, io);
+                stepResult = agentResult;
+                if (agentResult.status === 'ready') {
+                    agentNextActions = agentResult.nextActions;
+                }
             }
             if (stepResult.status === 'failed' || stepResult.status === 'missing-input') {
                 return 1;
@@ -455,10 +527,6 @@ async function runKtxSetupInner(args, io, deps = {}) {
             }
             if (step === 'context' && stepResult.status !== 'ready') {
                 if (shouldRunAgents && args.skipAgents !== true) {
-                    if (agentsRequested) {
-                        writeContextNotReadyForAgents(projectResult.projectDir, io);
-                        return args.inputMode === 'disabled' ? 1 : 0;
-                    }
                     return 0;
                 }
             }
@@ -468,18 +536,25 @@ async function runKtxSetupInner(args, io, deps = {}) {
         break;
     }
     await commitSetupConfigChanges(projectResult.projectDir);
-    const status = await readKtxSetupStatus(projectResult.projectDir);
+    const status = await readKtxSetupStatus(projectResult.projectDir, { cliVersion: args.cliVersion });
     const focusedOnAgents = args.agents || entryAction === 'agents';
     if (!focusedOnAgents) {
-        setupUi.note(formatKtxSetupStatus(status).trimEnd(), 'Project status', io, {
-            format: (line) => line,
-        });
-        setupUi.note(formatSetupNextStepLines({
-            setupReady: setupStatusReady(status),
-            hasContextTargets: setupHasContextTargets(status),
-            contextReady: setupContextReady(status),
-            agentIntegrationReady: status.agents.some((agent) => agent.ready),
-        }).join('\n'), 'What you can do next', io);
+        if (shouldPrintConciseReadySummary(status)) {
+            setupUi.note(formatKtxSetupCompletionSummary(status, { agentNextActions }), agentNextActions ? 'Finish KTX agent setup' : 'KTX project ready', io, {
+                format: (line) => line,
+            });
+        }
+        else {
+            setupUi.note(formatKtxSetupStatus(status).trimEnd(), 'Project status', io, {
+                format: (line) => line,
+            });
+            setupUi.note(formatSetupNextStepLines({
+                setupReady: setupStatusReady(status),
+                hasContextTargets: setupHasContextTargets(status),
+                contextReady: setupContextReady(status),
+                agentIntegrationReady: status.agents.some((agent) => agent.ready),
+            }).join('\n'), 'What you can do next', io);
+        }
     }
     return 0;
 }