@keystrokehq/keystroke 0.0.161 → 0.0.163

package/dist/{dist-DiqTL3ZF.cjs → dist-DqmPOnX5.cjs}

@@ -1,5 +1,5 @@
 const require_dist$1 = require("./dist-DnGkWV6x.cjs");
-const require_dist$2 = require("./dist-lWDTuftA.cjs");
+const require_dist$2 = require("./dist-DK1mbPAf.cjs");
 let node_async_hooks = require("node:async_hooks");
 require("node:fs");
 let zod_v3 = require("zod/v3");
@@ -13258,7 +13258,7 @@ async function connectMcpServer(name, options) {
 }
 async function createTransport(url, transport, requestInit, fetchImpl) {
 	if (transport === "sse") {
-		const { SSEClientTransport } = await Promise.resolve().then(() => require("./sse-DCsg_ELF.cjs"));
+		const { SSEClientTransport } = await Promise.resolve().then(() => require("./sse-DowizN6O.cjs"));
 		return new SSEClientTransport(url, {
 			requestInit,
 			fetch: fetchImpl
@@ -16454,6 +16454,7 @@ const agentSessions = pgTable("agent_sessions", {
 	parentSpanId: text$1("parent_span_id"),
 	ranByUserId: text$1("ran_by_user_id"),
 	canceledAt: timestamp("canceled_at", { withTimezone: true }),
+	totalDurationMs: integer$1("total_duration_ms").notNull().default(0),
 	createdAt: timestamp("created_at", { withTimezone: true }).notNull(),
 	updatedAt: timestamp("updated_at", { withTimezone: true }).notNull(),
 	deletedAt: timestamp("deleted_at", { withTimezone: true })
@@ -16467,62 +16468,14 @@ const agentSessionsSqlite = sqliteTable("agent_sessions", {
 	parentSpanId: text("parent_span_id"),
 	ranByUserId: text("ran_by_user_id"),
 	canceledAt: integer("canceled_at", { mode: "timestamp_ms" }),
+	totalDurationMs: integer("total_duration_ms").notNull().default(0),
 	createdAt: integer("created_at", { mode: "timestamp_ms" }).notNull(),
 	updatedAt: integer("updated_at", { mode: "timestamp_ms" }).notNull(),
 	deletedAt: integer("deleted_at", { mode: "timestamp_ms" })
 }, (table) => [index("agent_sessions_agent_updated_idx").on(table.agentId, table.updatedAt)]);
-const agentRuns = pgTable("agent_runs", {
-	id: text$1("id").primaryKey(),
-	sessionId: text$1("session_id").notNull().references(() => agentSessions.id),
-	projectId: text$1("project_id").notNull(),
-	agentId: text$1("agent_id").notNull().references(() => agents.id),
-	runIndex: integer$1("run_index").notNull(),
-	sourceKind: text$1("source_kind").$type().notNull().default("api"),
-	sourceId: text$1("source_id"),
-	ranByUserId: text$1("ran_by_user_id"),
-	status: text$1("status").$type().notNull(),
-	error: jsonb("error"),
-	startedAt: timestamp("started_at", { withTimezone: true }).notNull(),
-	finishedAt: timestamp("finished_at", { withTimezone: true }),
-	totalCostUsd: doublePrecision("total_cost_usd"),
-	totalTokens: integer$1("total_tokens"),
-	modelsUsed: jsonb("models_used").$type(),
-	createdAt: timestamp("created_at", { withTimezone: true }).notNull(),
-	updatedAt: timestamp("updated_at", { withTimezone: true }).notNull(),
-	deletedAt: timestamp("deleted_at", { withTimezone: true })
-}, (table) => [
-	index$1("agent_runs_session_run_index_idx").on(table.sessionId, table.runIndex),
-	index$1("agent_runs_agent_started_idx").on(table.agentId, table.startedAt),
-	index$1("agent_runs_project_started_idx").on(table.projectId, table.startedAt)
-]);
-const agentRunsSqlite = sqliteTable("agent_runs", {
-	id: text("id").primaryKey(),
-	sessionId: text("session_id").notNull().references(() => agentSessionsSqlite.id),
-	projectId: text("project_id").notNull(),
-	agentId: text("agent_id").notNull().references(() => agentsSqlite.id),
-	runIndex: integer("run_index").notNull(),
-	sourceKind: text("source_kind").$type().notNull().default("api"),
-	sourceId: text("source_id"),
-	ranByUserId: text("ran_by_user_id"),
-	status: text("status").$type().notNull(),
-	error: text("error", { mode: "json" }),
-	startedAt: integer("started_at", { mode: "timestamp_ms" }).notNull(),
-	finishedAt: integer("finished_at", { mode: "timestamp_ms" }),
-	totalCostUsd: real("total_cost_usd"),
-	totalTokens: integer("total_tokens"),
-	modelsUsed: text("models_used", { mode: "json" }).$type(),
-	createdAt: integer("created_at", { mode: "timestamp_ms" }).notNull(),
-	updatedAt: integer("updated_at", { mode: "timestamp_ms" }).notNull(),
-	deletedAt: integer("deleted_at", { mode: "timestamp_ms" })
-}, (table) => [
-	index("agent_runs_session_run_index_idx").on(table.sessionId, table.runIndex),
-	index("agent_runs_agent_started_idx").on(table.agentId, table.startedAt),
-	index("agent_runs_project_started_idx").on(table.projectId, table.startedAt)
-]);
 const agentEvents = pgTable("agent_events", {
 	id: text$1("id").primaryKey(),
 	sessionId: text$1("session_id").notNull().references(() => agentSessions.id),
-	runId: text$1("run_id").references(() => agentRuns.id),
 	seq: integer$1("seq").notNull(),
 	eventType: text$1("event_type").notNull(),
 	payload: jsonb("payload").notNull(),
@@ -16532,7 +16485,6 @@ const agentEvents = pgTable("agent_events", {
 const agentEventsSqlite = sqliteTable("agent_events", {
 	id: text("id").primaryKey(),
 	sessionId: text("session_id").notNull().references(() => agentSessionsSqlite.id),
-	runId: text("run_id").references(() => agentRunsSqlite.id),
 	seq: integer("seq").notNull(),
 	eventType: text("event_type").notNull(),
 	payload: text("payload", { mode: "json" }).notNull(),
@@ -17053,10 +17005,6 @@ const tableRegistry = {
 		pg: agentEvents,
 		sqlite: agentEventsSqlite
 	},
-	agentRuns: {
-		pg: agentRuns,
-		sqlite: agentRunsSqlite
-	},
 	usageRecords: {
 		pg: usageRecords,
 		sqlite: usageRecordsSqlite
@@ -17175,51 +17123,28 @@ async function insertAgentSession(values) {
 async function selectActiveAgentSession(sessionId) {
 	return (await getDb$1().select().from(agentSessions).where(scopedAnd$1(agentSessions, eq(agentSessions.id, sessionId), isNull(agentSessions.deletedAt))).limit(1))[0];
 }
-async function selectAgentSessionById(sessionId) {
-	return selectActiveAgentSession(sessionId);
-}
 async function touchAgentSession(sessionId, updatedAt) {
 	await getDb$1().update(agentSessions).set({ updatedAt }).where(scopedAnd$1(agentSessions, eq(agentSessions.id, sessionId), isNull(agentSessions.deletedAt)));
 }
+async function addAgentSessionDuration(sessionId, deltaMs) {
+	if (deltaMs <= 0) return;
+	await getDb$1().update(agentSessions).set({ totalDurationMs: sql`${agentSessions.totalDurationMs} + ${deltaMs}` }).where(scopedAnd$1(agentSessions, eq(agentSessions.id, sessionId), isNull(agentSessions.deletedAt)));
+}
 async function maxSeqForSession(sessionId) {
 	return (await getDb$1().select({ value: max(agentEvents.seq) }).from(agentEvents).innerJoin(agentSessions, eq(agentSessions.id, agentEvents.sessionId)).where(scopedAnd$1(agentSessions, eq(agentEvents.sessionId, sessionId), isNull(agentEvents.deletedAt))))[0]?.value ?? 0;
 }
 async function insertAgentEvent(values) {
-	await getDb$1().insert(agentEvents).values({
-		...values,
-		runId: values.runId ?? null
-	});
+	await getDb$1().insert(agentEvents).values(values);
 }
 async function listMessagePayloads(sessionId, messageEventType) {
-	return (await getDb$1().select({ payload: agentEvents.payload }).from(agentEvents).innerJoin(agentSessions, eq(agentSessions.id, agentEvents.sessionId)).where(scopedAnd$1(agentSessions, eq(agentEvents.sessionId, sessionId), eq(agentEvents.eventType, messageEventType), isNull(agentEvents.deletedAt))).orderBy(asc(agentEvents.seq))).map((row) => row.payload);
+	return (await listMessageEventsForSession(sessionId, messageEventType)).map((row) => row.payload);
 }
-async function nextRunIndex(sessionId) {
-	return ((await getDb$1().select({ value: max(agentRuns.runIndex) }).from(agentRuns).where(scopedAnd$1(agentRuns, eq(agentRuns.sessionId, sessionId), isNull(agentRuns.deletedAt))))[0]?.value ?? -1) + 1;
-}
-async function insertAgentRun(values) {
-	await getDb$1().insert(agentRuns).values({
-		...values,
-		projectId: projectScopeId(),
-		sourceKind: values.sourceKind ?? "api",
-		sourceId: values.sourceId ?? null,
-		ranByUserId: values.ranByUserId ?? null,
-		error: null,
-		finishedAt: null,
-		totalCostUsd: null,
-		totalTokens: null,
-		modelsUsed: null
-	});
-}
-async function finalizeAgentRun(runId, values) {
-	await getDb$1().update(agentRuns).set({
-		status: values.status,
-		finishedAt: values.finishedAt,
-		error: values.error ?? null,
-		totalCostUsd: values.totalCostUsd,
-		totalTokens: values.totalTokens,
-		modelsUsed: values.modelsUsed,
-		updatedAt: values.updatedAt
-	}).where(scopedAnd$1(agentRuns, eq(agentRuns.id, runId), isNull(agentRuns.deletedAt)));
+async function listMessageEventsForSession(sessionId, messageEventType) {
+	return await getDb$1().select({
+		payload: agentEvents.payload,
+		createdAt: agentEvents.createdAt,
+		seq: agentEvents.seq
+	}).from(agentEvents).innerJoin(agentSessions, eq(agentSessions.id, agentEvents.sessionId)).where(scopedAnd$1(agentSessions, eq(agentEvents.sessionId, sessionId), eq(agentEvents.eventType, messageEventType), isNull(agentEvents.deletedAt))).orderBy(asc(agentEvents.seq));
 }
 async function insertUsageRecord(values) {
 	await getDb$1().insert(usageRecords).values({
@@ -17229,24 +17154,6 @@ async function insertUsageRecord(values) {
 		metadata: values.metadata ?? null
 	});
 }
-async function sumUsageForRun(runId, runKind) {
-	const records = await getDb$1().select().from(usageRecords).where(and(eq(usageRecords.runId, runId), eq(usageRecords.runKind, runKind)));
-	let totalCostUsd = 0;
-	let totalTokens = 0;
-	const modelsUsed = /* @__PURE__ */ new Set();
-	for (const record of records) {
-		totalCostUsd += record.amount;
-		if (record.kind === "llm") {
-			if (record.unit === "tokens" && record.quantity != null) totalTokens += record.quantity;
-			modelsUsed.add(record.label);
-		}
-	}
-	return {
-		totalCostUsd,
-		totalTokens,
-		modelsUsed: [...modelsUsed]
-	};
-}
 const MESSAGE_EVENT_TYPE = "message";
 /** Org/user credentials live once in the shared org database; project scope stays per-project. */
 function credentialScopeAnd(scopeType, ...conditions) {
@@ -17349,12 +17256,11 @@ async function touchSession(sessionId) {
 async function nextSeq(sessionId) {
 	return await maxSeqForSession(sessionId) + 1;
 }
-async function appendEvent(sessionId, eventType, payload, options) {
+async function appendEvent(sessionId, eventType, payload) {
 	const seq = await nextSeq(sessionId);
 	await insertAgentEvent({
 		id: crypto.randomUUID(),
 		sessionId,
-		runId: options?.runId ?? null,
 		seq,
 		eventType,
 		payload,
@@ -17366,31 +17272,11 @@ async function appendEvent(sessionId, eventType, payload, options) {
 async function listMessageEvents(sessionId) {
 	return listMessagePayloads(sessionId, MESSAGE_EVENT_TYPE);
 }
-async function createAgentRun(options) {
-	const now = /* @__PURE__ */ new Date();
-	const runId = crypto.randomUUID();
-	const runIndex = await nextRunIndex(options.sessionId);
-	await insertAgentRun({
-		id: runId,
-		sessionId: options.sessionId,
-		agentId: options.agentId,
-		runIndex,
-		sourceKind: options.sourceKind,
-		sourceId: options.sourceId,
-		ranByUserId: options.ranByUserId,
-		status: "running",
-		startedAt: now,
-		createdAt: now,
-		updatedAt: now,
-		deletedAt: null
-	});
-	return runId;
-}
 async function recordLlmUsageFromAssistantMessage(options) {
 	const { usage } = options;
 	await insertUsageRecord({
 		id: crypto.randomUUID(),
-		runId: options.runId,
+		runId: options.sessionId,
 		runKind: "agent",
 		kind: "llm",
 		label: usage.label,
@@ -17401,32 +17287,6 @@ async function recordLlmUsageFromAssistantMessage(options) {
 		createdAt: /* @__PURE__ */ new Date()
 	});
 }
-async function finalizeAgentRunFromPrompt(options) {
-	const finishedAt = options.finishedAt ?? /* @__PURE__ */ new Date();
-	const rollups = await sumUsageForRun(options.runId, "agent");
-	await finalizeAgentRun(options.runId, {
-		status: options.status,
-		finishedAt,
-		error: options.error ?? null,
-		totalCostUsd: rollups.totalCostUsd,
-		totalTokens: rollups.totalTokens,
-		modelsUsed: rollups.modelsUsed,
-		updatedAt: finishedAt
-	});
-}
-async function resolveAgentRunSourceFromSession(sessionId) {
-	const session = await selectAgentSessionById(sessionId);
-	if (!session) return {
-		sourceKind: "api",
-		sourceId: null,
-		ranByUserId: null
-	};
-	return {
-		sourceKind: session.sourceKind,
-		sourceId: session.sourceId,
-		ranByUserId: session.ranByUserId
-	};
-}
 const ALGORITHM$1 = "aes-256-gcm";
 const IV_LENGTH$1 = 12;
 function resolveEncryptionKey$1() {
@@ -18435,6 +18295,12 @@ Object.defineProperty(exports, "UnauthorizedError", {
 		return UnauthorizedError;
 	}
 });
+Object.defineProperty(exports, "addAgentSessionDuration", {
+	enumerable: true,
+	get: function() {
+		return addAgentSessionDuration;
+	}
+});
 Object.defineProperty(exports, "appendEvent", {
 	enumerable: true,
 	get: function() {
@@ -18483,12 +18349,6 @@ Object.defineProperty(exports, "connectMcpStdio", {
 		return connectMcpStdio;
 	}
 });
-Object.defineProperty(exports, "createAgentRun", {
-	enumerable: true,
-	get: function() {
-		return createAgentRun;
-	}
-});
 Object.defineProperty(exports, "createCredentialResolver", {
 	enumerable: true,
 	get: function() {
@@ -18525,12 +18385,6 @@ Object.defineProperty(exports, "extractWWWAuthenticateParams", {
 		return extractWWWAuthenticateParams;
 	}
 });
-Object.defineProperty(exports, "finalizeAgentRunFromPrompt", {
-	enumerable: true,
-	get: function() {
-		return finalizeAgentRunFromPrompt;
-	}
-});
 Object.defineProperty(exports, "getAgentByRoute", {
 	enumerable: true,
 	get: function() {
@@ -18591,12 +18445,6 @@ Object.defineProperty(exports, "resolveActionCredentials", {
 		return resolveActionCredentials;
 	}
 });
-Object.defineProperty(exports, "resolveAgentRunSourceFromSession", {
-	enumerable: true,
-	get: function() {
-		return resolveAgentRunSourceFromSession;
-	}
-});
 Object.defineProperty(exports, "resolveMcpTools", {
 	enumerable: true,
 	get: function() {
@@ -18628,4 +18476,4 @@ Object.defineProperty(exports, "zodToJsonSchema", {
 	}
 });
 
-//# sourceMappingURL=dist-DiqTL3ZF.cjs.map
+//# sourceMappingURL=dist-DqmPOnX5.cjs.map