@llmops/core 0.1.7 → 0.1.9-beta.1

package/dist/{index-CzGmPK4Q.d.mts → index-DLavG1bA.d.cts}

@@ -1,5 +1,5 @@
 import { ColumnType, Generated, Kysely, MssqlDialect, MysqlDialect, PostgresDialect, SqliteDialect } from "kysely";
-import * as zod0 from "zod";
+import * as zod444 from "zod";
 import { z } from "zod";
 
 //#region src/db/schema.d.ts
@@ -64,6 +64,8 @@ declare const targetingRulesSchema: z.ZodObject<{
 }, z.core.$strip>;
 declare const workspaceSettingsSchema: z.ZodObject<{
   name: z.ZodOptional<z.ZodNullable<z.ZodString>>;
+  setupComplete: z.ZodDefault<z.ZodBoolean>;
+  superAdminId: z.ZodOptional<z.ZodNullable<z.ZodString>>;
   id: z.ZodString;
   createdAt: z.ZodDate;
   updatedAt: z.ZodDate;
@@ -153,6 +155,8 @@ interface TargetingRulesTable extends BaseTable {
 }
 interface WorkspaceSettingsTable extends BaseTable {
   name: string | null;
+  setupComplete: ColumnType<boolean, boolean | undefined, boolean | undefined>;
+  superAdminId: string | null;
 }
 interface LLMRequestsTable extends BaseTable {
   requestId: string;
@@ -510,6 +514,8 @@ declare const SCHEMA_METADATA: {
       readonly order: 8;
       readonly schema: z.ZodObject<{
         name: z.ZodOptional<z.ZodNullable<z.ZodString>>;
+        setupComplete: z.ZodDefault<z.ZodBoolean>;
+        superAdminId: z.ZodOptional<z.ZodNullable<z.ZodString>>;
         id: z.ZodString;
         createdAt: z.ZodDate;
         updatedAt: z.ZodDate;
@@ -523,6 +529,14 @@ declare const SCHEMA_METADATA: {
           readonly type: "text";
           readonly nullable: true;
         };
+        readonly setupComplete: {
+          readonly type: "boolean";
+          readonly default: false;
+        };
+        readonly superAdminId: {
+          readonly type: "text";
+          readonly nullable: true;
+        };
         readonly createdAt: {
           readonly type: "timestamp";
           readonly default: "now()";
@@ -726,6 +740,8 @@ declare const schemas: {
   }, z.core.$strip>;
   readonly workspace_settings: z.ZodObject<{
     name: z.ZodOptional<z.ZodNullable<z.ZodString>>;
+    setupComplete: z.ZodDefault<z.ZodBoolean>;
+    superAdminId: z.ZodOptional<z.ZodNullable<z.ZodString>>;
     id: z.ZodString;
     createdAt: z.ZodDate;
     updatedAt: z.ZodDate;
@@ -761,7 +777,29 @@ declare const schemas: {
  * Validate data against table schema
  * Useful for runtime validation before inserting/updating
  */
-declare function validateTableData<T extends TableName>(table: T, data: unknown): zod0.ZodSafeParseSuccess<{
+declare function validateTableData<T extends TableName>(table: T, data: unknown): zod444.ZodSafeParseSuccess<{
+  slug: string;
+  id: string;
+  createdAt: Date;
+  updatedAt: Date;
+  name?: string | undefined;
+}> | zod444.ZodSafeParseError<{
+  slug: string;
+  id: string;
+  createdAt: Date;
+  updatedAt: Date;
+  name?: string | undefined;
+}> | zod444.ZodSafeParseSuccess<{
+  name: string;
+  id: string;
+  createdAt: Date;
+  updatedAt: Date;
+}> | zod444.ZodSafeParseError<{
+  name: string;
+  id: string;
+  createdAt: Date;
+  updatedAt: Date;
+}> | zod444.ZodSafeParseSuccess<{
   variantId: string;
   version: number;
   provider: string;
@@ -770,7 +808,7 @@ declare function validateTableData<T extends TableName>(table: T, data: unknown)
   id: string;
   createdAt: Date;
   updatedAt: Date;
-}> | zod0.ZodSafeParseError<{
+}> | zod444.ZodSafeParseError<{
   variantId: string;
   version: number;
   provider: string;
@@ -779,33 +817,33 @@ declare function validateTableData<T extends TableName>(table: T, data: unknown)
   id: string;
   createdAt: Date;
   updatedAt: Date;
-}> | zod0.ZodSafeParseSuccess<{
+}> | zod444.ZodSafeParseSuccess<{
   environmentId: string;
   keyName: string;
   keyValue: string;
   id: string;
   createdAt: Date;
   updatedAt: Date;
-}> | zod0.ZodSafeParseError<{
+}> | zod444.ZodSafeParseError<{
   environmentId: string;
   keyName: string;
   keyValue: string;
   id: string;
   createdAt: Date;
   updatedAt: Date;
-}> | zod0.ZodSafeParseSuccess<{
+}> | zod444.ZodSafeParseSuccess<{
   configId: string;
   variantId: string;
   id: string;
   createdAt: Date;
   updatedAt: Date;
-}> | zod0.ZodSafeParseError<{
+}> | zod444.ZodSafeParseError<{
   configId: string;
   variantId: string;
   id: string;
   createdAt: Date;
   updatedAt: Date;
-}> | zod0.ZodSafeParseSuccess<{
+}> | zod444.ZodSafeParseSuccess<{
   environmentId: string;
   configId: string;
   configVariantId: string;
@@ -817,7 +855,7 @@ declare function validateTableData<T extends TableName>(table: T, data: unknown)
   createdAt: Date;
   updatedAt: Date;
   variantVersionId?: string | null | undefined;
-}> | zod0.ZodSafeParseError<{
+}> | zod444.ZodSafeParseError<{
   environmentId: string;
   configId: string;
   configVariantId: string;
@@ -829,17 +867,21 @@ declare function validateTableData<T extends TableName>(table: T, data: unknown)
   createdAt: Date;
   updatedAt: Date;
   variantVersionId?: string | null | undefined;
-}> | zod0.ZodSafeParseSuccess<{
+}> | zod444.ZodSafeParseSuccess<{
+  setupComplete: boolean;
   id: string;
   createdAt: Date;
   updatedAt: Date;
   name?: string | null | undefined;
-}> | zod0.ZodSafeParseError<{
+  superAdminId?: string | null | undefined;
+}> | zod444.ZodSafeParseError<{
+  setupComplete: boolean;
   id: string;
   createdAt: Date;
   updatedAt: Date;
   name?: string | null | undefined;
-}> | zod0.ZodSafeParseSuccess<{
+  superAdminId?: string | null | undefined;
+}> | zod444.ZodSafeParseSuccess<{
   requestId: string;
   provider: string;
   model: string;
@@ -862,7 +904,7 @@ declare function validateTableData<T extends TableName>(table: T, data: unknown)
   variantId?: string | null | undefined;
   environmentId?: string | null | undefined;
   userId?: string | null | undefined;
-}> | zod0.ZodSafeParseError<{
+}> | zod444.ZodSafeParseError<{
   requestId: string;
   provider: string;
   model: string;
@@ -889,7 +931,17 @@ declare function validateTableData<T extends TableName>(table: T, data: unknown)
 /**
  * Validate partial data (for updates)
  */
-declare function validatePartialTableData<T extends TableName>(table: T, data: unknown): zod0.ZodSafeParseSuccess<{
+declare function validatePartialTableData<T extends TableName>(table: T, data: unknown): zod444.ZodSafeParseSuccess<{
+  name?: string | undefined;
+  id?: string | undefined;
+  createdAt?: Date | undefined;
+  updatedAt?: Date | undefined;
+}> | zod444.ZodSafeParseError<{
+  name?: string | undefined;
+  id?: string | undefined;
+  createdAt?: Date | undefined;
+  updatedAt?: Date | undefined;
+}> | zod444.ZodSafeParseSuccess<{
   variantId?: string | undefined;
   version?: number | undefined;
   provider?: string | undefined;
@@ -898,7 +950,7 @@ declare function validatePartialTableData<T extends TableName>(table: T, data: u
   id?: string | undefined;
   createdAt?: Date | undefined;
   updatedAt?: Date | undefined;
-}> | zod0.ZodSafeParseError<{
+}> | zod444.ZodSafeParseError<{
   variantId?: string | undefined;
   version?: number | undefined;
   provider?: string | undefined;
@@ -907,33 +959,33 @@ declare function validatePartialTableData<T extends TableName>(table: T, data: u
   id?: string | undefined;
   createdAt?: Date | undefined;
   updatedAt?: Date | undefined;
-}> | zod0.ZodSafeParseSuccess<{
+}> | zod444.ZodSafeParseSuccess<{
   environmentId?: string | undefined;
   keyName?: string | undefined;
   keyValue?: string | undefined;
   id?: string | undefined;
   createdAt?: Date | undefined;
   updatedAt?: Date | undefined;
-}> | zod0.ZodSafeParseError<{
+}> | zod444.ZodSafeParseError<{
   environmentId?: string | undefined;
   keyName?: string | undefined;
   keyValue?: string | undefined;
   id?: string | undefined;
   createdAt?: Date | undefined;
   updatedAt?: Date | undefined;
-}> | zod0.ZodSafeParseSuccess<{
+}> | zod444.ZodSafeParseSuccess<{
   configId?: string | undefined;
   variantId?: string | undefined;
   id?: string | undefined;
   createdAt?: Date | undefined;
   updatedAt?: Date | undefined;
-}> | zod0.ZodSafeParseError<{
+}> | zod444.ZodSafeParseError<{
   configId?: string | undefined;
   variantId?: string | undefined;
   id?: string | undefined;
   createdAt?: Date | undefined;
   updatedAt?: Date | undefined;
-}> | zod0.ZodSafeParseSuccess<{
+}> | zod444.ZodSafeParseSuccess<{
   environmentId?: string | undefined;
   configId?: string | undefined;
   configVariantId?: string | undefined;
@@ -945,7 +997,7 @@ declare function validatePartialTableData<T extends TableName>(table: T, data: u
   id?: string | undefined;
   createdAt?: Date | undefined;
   updatedAt?: Date | undefined;
-}> | zod0.ZodSafeParseError<{
+}> | zod444.ZodSafeParseError<{
   environmentId?: string | undefined;
   configId?: string | undefined;
   configVariantId?: string | undefined;
@@ -957,17 +1009,21 @@ declare function validatePartialTableData<T extends TableName>(table: T, data: u
   id?: string | undefined;
   createdAt?: Date | undefined;
   updatedAt?: Date | undefined;
-}> | zod0.ZodSafeParseSuccess<{
+}> | zod444.ZodSafeParseSuccess<{
   name?: string | null | undefined;
+  setupComplete?: boolean | undefined;
+  superAdminId?: string | null | undefined;
   id?: string | undefined;
   createdAt?: Date | undefined;
   updatedAt?: Date | undefined;
-}> | zod0.ZodSafeParseError<{
+}> | zod444.ZodSafeParseError<{
   name?: string | null | undefined;
+  setupComplete?: boolean | undefined;
+  superAdminId?: string | null | undefined;
   id?: string | undefined;
   createdAt?: Date | undefined;
   updatedAt?: Date | undefined;
-}> | zod0.ZodSafeParseSuccess<{
+}> | zod444.ZodSafeParseSuccess<{
   requestId?: string | undefined;
   configId?: string | null | undefined;
   variantId?: string | null | undefined;
@@ -990,7 +1046,7 @@ declare function validatePartialTableData<T extends TableName>(table: T, data: u
   id?: string | undefined;
   createdAt?: Date | undefined;
   updatedAt?: Date | undefined;
-}> | zod0.ZodSafeParseError<{
+}> | zod444.ZodSafeParseError<{
   requestId?: string | undefined;
   configId?: string | null | undefined;
   variantId?: string | null | undefined;
@@ -1018,6 +1074,17 @@ declare function validatePartialTableData<T extends TableName>(table: T, data: u
  * Parse and validate data, throws on error
  */
 declare function parseTableData<T extends TableName>(table: T, data: unknown): {
+  slug: string;
+  id: string;
+  createdAt: Date;
+  updatedAt: Date;
+  name?: string | undefined;
+} | {
+  name: string;
+  id: string;
+  createdAt: Date;
+  updatedAt: Date;
+} | {
   variantId: string;
   version: number;
   provider: string;
@@ -1052,10 +1119,12 @@ declare function parseTableData<T extends TableName>(table: T, data: unknown): {
   updatedAt: Date;
   variantVersionId?: string | null | undefined;
 } | {
+  setupComplete: boolean;
   id: string;
   createdAt: Date;
   updatedAt: Date;
   name?: string | null | undefined;
+  superAdminId?: string | null | undefined;
 } | {
   requestId: string;
   provider: string;
@@ -1084,6 +1153,11 @@ declare function parseTableData<T extends TableName>(table: T, data: unknown): {
  * Parse partial data, throws on error
  */
 declare function parsePartialTableData<T extends TableName>(table: T, data: unknown): {
+  name?: string | undefined;
+  id?: string | undefined;
+  createdAt?: Date | undefined;
+  updatedAt?: Date | undefined;
+} | {
   variantId?: string | undefined;
   version?: number | undefined;
   provider?: string | undefined;
@@ -1119,6 +1193,8 @@ declare function parsePartialTableData<T extends TableName>(table: T, data: unkn
   updatedAt?: Date | undefined;
 } | {
   name?: string | null | undefined;
+  setupComplete?: boolean | undefined;
+  superAdminId?: string | null | undefined;
   id?: string | undefined;
   createdAt?: Date | undefined;
   updatedAt?: Date | undefined;
@@ -1158,6 +1234,7 @@ interface MigrationOptions {
    * If provided, the schema will be created if it doesn't exist.
    */
   schema?: string;
+  rawConnection?: unknown;
 }
 declare function matchType(columnDataType: string, fieldType: string, dbType: DatabaseType$1): boolean;
 interface MigrationResult {
@@ -1178,14 +1255,13 @@ interface MigrationResult {
 }
 declare function getMigrations(db: Kysely<Database>, dbType: DatabaseType$1, options?: MigrationOptions): Promise<MigrationResult>;
 /**
- * Run migrations if needed based on autoMigrate config
+ * Run migrations if needed
  * @param db - Kysely database instance
  * @param dbType - Database type
- * @param autoMigrate - Auto-migrate configuration
  * @param options - Migration options (schema, etc.)
  * @returns true if migrations were run, false otherwise
  */
-declare function runAutoMigrations(db: Kysely<Database>, dbType: DatabaseType$1, autoMigrate: boolean | 'development', options?: MigrationOptions): Promise<{
+declare function runAutoMigrations(db: Kysely<Database>, dbType: DatabaseType$1, options?: MigrationOptions): Promise<{
   ran: boolean;
   tables: string[];
   fields: string[];

package/dist/index.cjs

@@ -1,4 +1,4 @@
-const require_db = require('./db-C_u1BuaR.cjs');
+const require_db = require('./db-CpVh76Wq.cjs');
 let __llmops_gateway = require("@llmops/gateway");
 __llmops_gateway = require_db.__toESM(__llmops_gateway);
 let kysely = require("kysely");
@@ -458,7 +458,6 @@ const llmopsConfigSchema = require_db.object({
 	auth: authSchema,
 	basePath: require_db.string().min(1, "Base path is required and cannot be empty").refine((path) => path.startsWith("/"), "Base path must start with a forward slash"),
 	providers: providersSchema,
-	autoMigrate: require_db.union([require_db.boolean(), require_db.literal("development")]).optional().default(false),
 	schema: require_db.string().optional().default("llmops")
 });
 function validateLLMOpsConfig(config) {
@@ -1595,26 +1594,10 @@ const createLLMRequestsDataLayer = (db) => {
 			const result = await listRequestsSchema.safeParseAsync(params || {});
 			if (!result.success) throw new LLMOpsError(`Invalid parameters: ${result.error.message}`);
 			const { limit, offset, configId, variantId, environmentId, provider, model, startDate, endDate, tags } = result.data;
-			console.log("[listRequests] Parsed filters:", {
-				configId,
-				variantId,
-				environmentId,
-				provider,
-				model
-			});
 			let baseQuery = db.selectFrom("llm_requests");
-			if (configId) {
-				console.log("[listRequests] Adding configId filter:", configId);
-				baseQuery = baseQuery.where("configId", "=", configId);
-			}
-			if (variantId) {
-				console.log("[listRequests] Adding variantId filter:", variantId);
-				baseQuery = baseQuery.where("variantId", "=", variantId);
-			}
-			if (environmentId) {
-				console.log("[listRequests] Adding environmentId filter:", environmentId);
-				baseQuery = baseQuery.where("environmentId", "=", environmentId);
-			}
+			if (configId) baseQuery = baseQuery.where("configId", "=", configId);
+			if (variantId) baseQuery = baseQuery.where("variantId", "=", variantId);
+			if (environmentId) baseQuery = baseQuery.where("environmentId", "=", environmentId);
 			if (provider) baseQuery = baseQuery.where("provider", "=", provider);
 			if (model) baseQuery = baseQuery.where("model", "=", model);
 			if (startDate) baseQuery = baseQuery.where(kysely.sql`${col("createdAt")} >= ${startDate.toISOString()}`);
@@ -1627,10 +1610,7 @@ const createLLMRequestsDataLayer = (db) => {
 					baseQuery = baseQuery.where(kysely.sql`${col("tags")}->>${key} IN (${valueList})`);
 				}
 			}
-			const countQuery = baseQuery.select(kysely.sql`COUNT(*)`.as("total"));
-			console.log("[listRequests] Count SQL:", countQuery.compile().sql);
-			console.log("[listRequests] Count params:", countQuery.compile().parameters);
-			const countResult = await countQuery.executeTakeFirst();
+			const countResult = await baseQuery.select(kysely.sql`COUNT(*)`.as("total")).executeTakeFirst();
 			const total = Number(countResult?.total ?? 0);
 			return {
 				data: await baseQuery.selectAll().orderBy("createdAt", "desc").limit(limit).offset(offset).execute(),
@@ -2204,7 +2184,11 @@ const createVariantVersionsDataLayer = (db) => {
 
 //#endregion
 //#region src/datalayer/workspaceSettings.ts
-const updateWorkspaceSettings = require_db.zod_default.object({ name: require_db.zod_default.string().nullable().optional() });
+const updateWorkspaceSettings = require_db.zod_default.object({
+	name: require_db.zod_default.string().nullable().optional(),
+	setupComplete: require_db.zod_default.boolean().optional(),
+	superAdminId: require_db.zod_default.string().nullable().optional()
+});
 const createWorkspaceSettingsDataLayer = (db) => {
 	return {
 		getWorkspaceSettings: async () => {
@@ -2212,6 +2196,7 @@ const createWorkspaceSettingsDataLayer = (db) => {
 			if (!settings) settings = await db.insertInto("workspace_settings").values({
 				id: (0, node_crypto.randomUUID)(),
 				name: null,
+				setupComplete: false,
 				createdAt: (/* @__PURE__ */ new Date()).toISOString(),
 				updatedAt: (/* @__PURE__ */ new Date()).toISOString()
 			}).returningAll().executeTakeFirst();
@@ -2224,11 +2209,53 @@ const createWorkspaceSettingsDataLayer = (db) => {
 			if (!settings) return db.insertInto("workspace_settings").values({
 				id: (0, node_crypto.randomUUID)(),
 				name: value.data.name ?? null,
+				setupComplete: value.data.setupComplete ?? false,
+				createdAt: (/* @__PURE__ */ new Date()).toISOString(),
+				updatedAt: (/* @__PURE__ */ new Date()).toISOString()
+			}).returningAll().executeTakeFirst();
+			const updateData = { updatedAt: (/* @__PURE__ */ new Date()).toISOString() };
+			if (value.data.name !== void 0) updateData.name = value.data.name ?? null;
+			if (value.data.setupComplete !== void 0) updateData.setupComplete = value.data.setupComplete;
+			if (value.data.superAdminId !== void 0) updateData.superAdminId = value.data.superAdminId ?? null;
+			return db.updateTable("workspace_settings").set(updateData).where("id", "=", settings.id).returningAll().executeTakeFirst();
+		},
+		getSuperAdminId: async () => {
+			return (await db.selectFrom("workspace_settings").select("superAdminId").executeTakeFirst())?.superAdminId ?? null;
+		},
+		setSuperAdminId: async (userId) => {
+			let settings = await db.selectFrom("workspace_settings").selectAll().executeTakeFirst();
+			if (settings?.superAdminId) return false;
+			if (!settings) {
+				await db.insertInto("workspace_settings").values({
+					id: (0, node_crypto.randomUUID)(),
+					name: null,
+					setupComplete: false,
+					superAdminId: userId,
+					createdAt: (/* @__PURE__ */ new Date()).toISOString(),
+					updatedAt: (/* @__PURE__ */ new Date()).toISOString()
+				}).execute();
+				return true;
+			}
+			await db.updateTable("workspace_settings").set({
+				superAdminId: userId,
+				updatedAt: (/* @__PURE__ */ new Date()).toISOString()
+			}).where("id", "=", settings.id).execute();
+			return true;
+		},
+		isSetupComplete: async () => {
+			return (await db.selectFrom("workspace_settings").select("setupComplete").executeTakeFirst())?.setupComplete ?? false;
+		},
+		markSetupComplete: async () => {
+			let settings = await db.selectFrom("workspace_settings").selectAll().executeTakeFirst();
+			if (!settings) return db.insertInto("workspace_settings").values({
+				id: (0, node_crypto.randomUUID)(),
+				name: null,
+				setupComplete: true,
 				createdAt: (/* @__PURE__ */ new Date()).toISOString(),
 				updatedAt: (/* @__PURE__ */ new Date()).toISOString()
 			}).returningAll().executeTakeFirst();
 			return db.updateTable("workspace_settings").set({
-				name: value.data.name ?? null,
+				setupComplete: true,
 				updatedAt: (/* @__PURE__ */ new Date()).toISOString()
 			}).where("id", "=", settings.id).returningAll().executeTakeFirst();
 		}
@@ -2482,6 +2509,7 @@ Object.defineProperty(exports, 'gateway', {
   }
 });
 exports.generateId = generateId;
+exports.getAuthClientOptions = require_db.getAuthClientOptions;
 exports.getDefaultPricingProvider = getDefaultPricingProvider;
 exports.getMigrations = require_db.getMigrations;
 exports.llmRequestsSchema = require_db.llmRequestsSchema;