@keystrokehq/keystroke 0.0.162 → 0.0.163

package/dist/{dist-DXzvN8aq.mjs → dist-BjWzektn.mjs}

@@ -1,5 +1,5 @@
 import { Bn as __require, Ln as __commonJSMin, Vn as __toESM } from "./dist-Dv44b9V1.mjs";
-import { C as normalizeCredentialList, _ as CREDENTIAL_SCOPE_LEVEL_TO_SCOPE_TYPE } from "./dist-BIbZjRd7.mjs";
+import { C as normalizeCredentialList, _ as CREDENTIAL_SCOPE_LEVEL_TO_SCOPE_TYPE } from "./dist-B_b7urZi.mjs";
 import { AsyncLocalStorage } from "node:async_hooks";
 import "node:fs";
 import { ZodFirstPartyTypeKind } from "zod/v3";
@@ -13251,7 +13251,7 @@ async function connectMcpServer(name, options) {
 }
 async function createTransport(url, transport, requestInit, fetchImpl) {
 	if (transport === "sse") {
-		const { SSEClientTransport } = await import("./sse-BBd2aLiw.mjs");
+		const { SSEClientTransport } = await import("./sse-Duxm9CEQ.mjs");
 		return new SSEClientTransport(url, {
 			requestInit,
 			fetch: fetchImpl
@@ -16447,6 +16447,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 })
@@ -16460,62 +16461,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(),
@@ -16525,7 +16478,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(),
@@ -17046,10 +16998,6 @@ const tableRegistry = {
 		pg: agentEvents,
 		sqlite: agentEventsSqlite
 	},
-	agentRuns: {
-		pg: agentRuns,
-		sqlite: agentRunsSqlite
-	},
 	usageRecords: {
 		pg: usageRecords,
 		sqlite: usageRecordsSqlite
@@ -17168,51 +17116,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({
@@ -17222,24 +17147,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) {
@@ -17342,12 +17249,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,
@@ -17359,31 +17265,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,
@@ -17394,32 +17280,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() {
@@ -18410,6 +18270,6 @@ async function resolveActionCredentials(requirements, options) {
 	return parseResolvedCredentials(requirements, await resolveCredentials(requirements, options.consumer, options.contextOverride));
 }
 //#endregion
-export { auth as A, connectMcpDefinition as C, isMcp as D, defineMcp as E, JSONRPCMessageSchema as F, createFetchWithInit as M, normalizeHeaders as N, createParser as O, zodToJsonSchema as P, withSpan as S, connectMcpStdio as T, resolveRunSourceFromTraceContext as _, resolveActionCredentials as a, getTraceContext as b, appendEvent as c, finalizeAgentRunFromPrompt as d, getAgentByRoute as f, resolveAgentRunSourceFromSession as g, recordLlmUsageFromAssistantMessage as h, isCredentialError as i, extractWWWAuthenticateParams as j, UnauthorizedError as k, createAgentRun as l, listMessageEvents as m, captureCredentialToolErrors as n, resolveMcpTools as o, getSession as p, createCredentialResolver as r, MESSAGE_EVENT_TYPE as s, buildCredentialRunContext as t, createSession as u, touchSession as v, connectMcpServer as w, logSystem as x, captureConsole as y };
+export { createFetchWithInit as A, connectMcpStdio as C, UnauthorizedError as D, createParser as E, zodToJsonSchema as M, JSONRPCMessageSchema as N, auth as O, connectMcpServer as S, isMcp as T, captureConsole as _, resolveActionCredentials as a, withSpan as b, addAgentSessionDuration as c, getAgentByRoute as d, getSession as f, touchSession as g, resolveRunSourceFromTraceContext as h, isCredentialError as i, normalizeHeaders as j, extractWWWAuthenticateParams as k, appendEvent as l, recordLlmUsageFromAssistantMessage as m, captureCredentialToolErrors as n, resolveMcpTools as o, listMessageEvents as p, createCredentialResolver as r, MESSAGE_EVENT_TYPE as s, buildCredentialRunContext as t, createSession as u, getTraceContext as v, defineMcp as w, connectMcpDefinition as x, logSystem as y };
 
-//# sourceMappingURL=dist-DXzvN8aq.mjs.map
+//# sourceMappingURL=dist-BjWzektn.mjs.map