@llmops/core 0.1.7 → 0.1.9-beta.1

package/dist/index.mjs

@@ -1,4 +1,4 @@
-import { A as object, C as zod_default, D as boolean, E as array, M as string, N as union, O as literal, P as unknown, S as workspaceSettingsSchema, T as any, _ as llmRequestsSchema, a as matchType, b as variantVersionsSchema, c as parsePartialTableData, d as validateTableData, f as SCHEMA_METADATA, g as environmentsSchema, h as environmentSecretsSchema, i as getMigrations, j as record, k as number, l as parseTableData, m as configsSchema, n as createDatabaseFromConnection, o as runAutoMigrations, p as configVariantsSchema, r as detectDatabaseType, s as logger, t as createDatabase, u as validatePartialTableData, v as schemas, w as _enum, x as variantsSchema, y as targetingRulesSchema } from "./db-D78x_Elf.mjs";
+import { A as number, C as workspaceSettingsSchema, D as array, E as any, F as unknown, M as record, N as string, O as boolean, P as union, S as variantsSchema, T as _enum, _ as environmentsSchema, a as matchType, b as targetingRulesSchema, c as logger, d as validatePartialTableData, f as validateTableData, g as environmentSecretsSchema, h as configsSchema, i as getMigrations, j as object, k as literal, l as parsePartialTableData, m as configVariantsSchema, n as createDatabaseFromConnection, o as runAutoMigrations, p as SCHEMA_METADATA, r as detectDatabaseType, s as getAuthClientOptions, t as createDatabase, u as parseTableData, v as llmRequestsSchema, w as zod_default, x as variantVersionsSchema, y as schemas } from "./db-WrnSOT3M.mjs";
 import gateway from "@llmops/gateway";
 import { sql } from "kysely";
 import * as fs from "node:fs/promises";
@@ -455,7 +455,6 @@ const llmopsConfigSchema = object({
 	auth: authSchema,
 	basePath: string().min(1, "Base path is required and cannot be empty").refine((path$1) => path$1.startsWith("/"), "Base path must start with a forward slash"),
 	providers: providersSchema,
-	autoMigrate: union([boolean(), literal("development")]).optional().default(false),
 	schema: string().optional().default("llmops")
 });
 function validateLLMOpsConfig(config) {
@@ -1592,26 +1591,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(sql`${col("createdAt")} >= ${startDate.toISOString()}`);
@@ -1624,10 +1607,7 @@ const createLLMRequestsDataLayer = (db) => {
 					baseQuery = baseQuery.where(sql`${col("tags")}->>${key} IN (${valueList})`);
 				}
 			}
-			const countQuery = baseQuery.select(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(sql`COUNT(*)`.as("total")).executeTakeFirst();
 			const total = Number(countResult?.total ?? 0);
 			return {
 				data: await baseQuery.selectAll().orderBy("createdAt", "desc").limit(limit).offset(offset).execute(),
@@ -2201,7 +2181,11 @@ const createVariantVersionsDataLayer = (db) => {
 
 //#endregion
 //#region src/datalayer/workspaceSettings.ts
-const updateWorkspaceSettings = zod_default.object({ name: zod_default.string().nullable().optional() });
+const updateWorkspaceSettings = zod_default.object({
+	name: zod_default.string().nullable().optional(),
+	setupComplete: zod_default.boolean().optional(),
+	superAdminId: zod_default.string().nullable().optional()
+});
 const createWorkspaceSettingsDataLayer = (db) => {
 	return {
 		getWorkspaceSettings: async () => {
@@ -2209,6 +2193,7 @@ const createWorkspaceSettingsDataLayer = (db) => {
 			if (!settings) settings = await db.insertInto("workspace_settings").values({
 				id: randomUUID(),
 				name: null,
+				setupComplete: false,
 				createdAt: (/* @__PURE__ */ new Date()).toISOString(),
 				updatedAt: (/* @__PURE__ */ new Date()).toISOString()
 			}).returningAll().executeTakeFirst();
@@ -2221,11 +2206,53 @@ const createWorkspaceSettingsDataLayer = (db) => {
 			if (!settings) return db.insertInto("workspace_settings").values({
 				id: 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: 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: 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();
 		}
@@ -2452,4 +2479,4 @@ function getDefaultPricingProvider() {
 }
 
 //#endregion
-export { CacheService, FileCacheBackend, MS, MemoryCacheBackend, ModelsDevPricingProvider, SCHEMA_METADATA, SupportedProviders, calculateCost, chatCompletionCreateParamsBaseSchema, configVariantsSchema, configsSchema, createDataLayer, createDatabase, createDatabaseFromConnection, createLLMRequestsDataLayer, detectDatabaseType, dollarsToMicroDollars, environmentSecretsSchema, environmentsSchema, formatCost, gateway, generateId, getDefaultPricingProvider, getMigrations, llmRequestsSchema, llmopsConfigSchema, logger, matchType, microDollarsToDollars, parsePartialTableData, parseTableData, runAutoMigrations, schemas, targetingRulesSchema, validateLLMOpsConfig, validatePartialTableData, validateTableData, variantJsonDataSchema, variantVersionsSchema, variantsSchema, workspaceSettingsSchema };
+export { CacheService, FileCacheBackend, MS, MemoryCacheBackend, ModelsDevPricingProvider, SCHEMA_METADATA, SupportedProviders, calculateCost, chatCompletionCreateParamsBaseSchema, configVariantsSchema, configsSchema, createDataLayer, createDatabase, createDatabaseFromConnection, createLLMRequestsDataLayer, detectDatabaseType, dollarsToMicroDollars, environmentSecretsSchema, environmentsSchema, formatCost, gateway, generateId, getAuthClientOptions, getDefaultPricingProvider, getMigrations, llmRequestsSchema, llmopsConfigSchema, logger, matchType, microDollarsToDollars, parsePartialTableData, parseTableData, runAutoMigrations, schemas, targetingRulesSchema, validateLLMOpsConfig, validatePartialTableData, validateTableData, variantJsonDataSchema, variantVersionsSchema, variantsSchema, workspaceSettingsSchema };

package/dist/{node-sqlite-dialect-BSbNUAzj.cjs → node-sqlite-dialect-BkZpAjVL.cjs}

@@ -1,4 +1,4 @@
-const require_db = require('./db-C_u1BuaR.cjs');
+const require_db = require('./db-CpVh76Wq.cjs');
 let kysely = require("kysely");
 
 //#region src/db/node-sqlite-dialect.ts

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@llmops/core",
-  "version": "0.1.7",
+  "version": "0.1.9-beta.1",
   "description": "Core LLMOps functionality and utilities",
   "type": "module",
   "license": "Apache-2.0",
@@ -48,10 +48,11 @@
   },
   "dependencies": {
     "@better-auth/utils": "^0.3.0",
+    "better-auth": "^1.4.10",
     "hono": "^4.10.7",
     "kysely": "^0.28.8",
     "pino": "^10.1.0",
-    "@llmops/gateway": "^0.1.7"
+    "@llmops/gateway": "^0.1.9-beta.1"
   },
   "scripts": {
     "build": "tsdown",