@llmops/cli 1.0.0-beta.3 → 1.0.0-beta.4

package/dist/index.mjs

@@ -1,10 +1,8 @@
 #!/usr/bin/env node
 import { command, run, string } from "@drizzle-team/brocli";
 import { logger } from "@llmops/core";
-import { getMigrations } from "@llmops/core/db";
 import { existsSync } from "node:fs";
 import yoctoSpinner from "yocto-spinner";
-import chalk from "chalk";
 import prompts from "prompts";
 import { loadConfig } from "c12";
 import path from "node:path";
@@ -21,17 +19,6 @@ const getConfig = async ({ cwd, configPath }) => {
 
 //#endregion
 //#region src/commands/migrate.ts
-/**
-* @fileoverview This file defines the 'migrate' command for the CLI application.
-* Steps:
-* 1. Look for package.json and @llmops/sdk in the current directory.
-* 2. Check if the @llmops/sdk version works with the current CLI version.
-* 3. If compatible, check for the config file passed as an argument.
-* 4. If the config file exists, read and parse it.
-* 5. If not passed, look for default config file locations.
-* 6. Use zod to validate the existing configuration schema.
-* 7. If valid, get the telemetry store from the config.
-*/
 const migrateCommand = command({
 	name: "migrate",
 	desc: "Run database migrations based on LLMOps configuration",
@@ -61,36 +48,33 @@ const migrateCommand = command({
 		}
 		const telemetry = config.telemetry;
 		const store = Array.isArray(telemetry) ? telemetry[0] : telemetry;
-		if (!store || !store._db) {
+		if (!store || !store._pool) {
 			logger.error("No telemetry store with database found. Configure pgStore in your config.");
 			process.exit(1);
 		}
-		const db = store._db;
-		const spinner = yoctoSpinner({ text: "preparing migration..." }).start();
-		const { toBeAdded, toBeCreated, runMigrations } = await getMigrations(db, "postgres", { schema: "llmops" });
-		if (!toBeAdded.length && !toBeCreated.length) {
-			spinner.stop();
-			console.log("🚀 No migrations needed.");
-			process.exit(0);
-		}
-		spinner.stop();
-		console.log(`🔑 The migration will affect the following:`);
-		for (const table of [...toBeCreated, ...toBeAdded]) console.log("->", chalk.magenta(Object.keys(table.fields).join(", ")), chalk.white("fields on"), chalk.yellow(`${table.table}`), chalk.white("table."));
 		let migrate = opts.yes;
 		if (!opts.yes) migrate = (await prompts({
 			type: "confirm",
 			name: "migrate",
-			message: "Do you want to proceed with the migration?",
+			message: "Do you want to run pending migrations?",
 			initial: false
 		})).migrate;
 		if (!migrate) {
 			console.log("Migration cancelled.");
 			process.exit(0);
 		}
-		spinner.start("migrating...");
-		await runMigrations();
-		spinner.stop();
-		console.log("✅ Migration completed successfully.");
+		const spinner = yoctoSpinner({ text: "running migrations..." }).start();
+		try {
+			const { runMigrations } = await import("@llmops/sdk/store/pg");
+			const { applied } = await runMigrations(store._pool, store._schema ?? "llmops");
+			spinner.stop();
+			if (applied.length === 0) console.log("🚀 No pending migrations.");
+			else console.log(`✅ Applied ${applied.length} migration(s): ${applied.join(", ")}`);
+		} catch (error) {
+			spinner.stop();
+			logger.error(`Migration failed: ${error}`);
+			process.exit(1);
+		}
 		process.exit(0);
 	}
 });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@llmops/cli",
-  "version": "1.0.0-beta.3",
+  "version": "1.0.0-beta.4",
   "type": "module",
   "description": "LLMOps CLI - A pluggable LLMOps toolkit for TypeScript teams",
   "license": "Apache-2.0",
@@ -45,8 +45,8 @@
     "kysely": "^0.28.8",
     "prompts": "^2.4.2",
     "yocto-spinner": "^1.0.0",
-    "@llmops/sdk": "^1.0.0-beta.3",
-    "@llmops/core": "^1.0.0-beta.3"
+    "@llmops/core": "^1.0.0-beta.4",
+    "@llmops/sdk": "^1.0.0-beta.4"
   },
   "devDependencies": {
     "@types/pg": "^8.15.6",