@keystrokehq/keystroke 1.0.6 → 1.0.8

package/dist/action.cjs

@@ -1,5 +1,5 @@
 Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
-const require_dist = require("./dist-hu4oHsK8.cjs");
+const require_dist = require("./dist-DOMwPx7V.cjs");
 exports.CREDENTIAL_SCOPE_TYPES = require_dist.CREDENTIAL_SCOPE_TYPES;
 exports.actionCoreSchema = require_dist.actionCoreSchema;
 exports.createStepInvocation = require_dist.createStepInvocation;

package/dist/action.mjs

@@ -1,2 +1,2 @@
-import { S as isCredentialInput, T as toCredentialRequirement, _ as runWithinActionExecution, a as getActionCredentialRequirements, b as credentialInputSchema, c as getWorkflowRunHandle, d as isWithinActionExecution, f as registerProjectScopeGetter, g as runWithRunSignal, h as runOutsideActionExecution, i as executeAction, l as isAction, m as resolveActionTool, n as createStepInvocation, o as getRegisteredProjectScopeId, p as registerWorkflowRunGetter, r as defineAction, s as getRunSignal, t as actionCoreSchema, u as isStepInvocation, v as CREDENTIAL_SCOPE_TYPES, w as normalizeCredentialList, x as defineCredential, y as credential } from "./dist-6ug9ozqR.mjs";
+import { S as isCredentialInput, T as toCredentialRequirement, _ as runWithinActionExecution, a as getActionCredentialRequirements, b as credentialInputSchema, c as getWorkflowRunHandle, d as isWithinActionExecution, f as registerProjectScopeGetter, g as runWithRunSignal, h as runOutsideActionExecution, i as executeAction, l as isAction, m as resolveActionTool, n as createStepInvocation, o as getRegisteredProjectScopeId, p as registerWorkflowRunGetter, r as defineAction, s as getRunSignal, t as actionCoreSchema, u as isStepInvocation, v as CREDENTIAL_SCOPE_TYPES, w as normalizeCredentialList, x as defineCredential, y as credential } from "./dist-DqFdFpiB.mjs";
 export { CREDENTIAL_SCOPE_TYPES, actionCoreSchema, createStepInvocation, credential, credentialInputSchema, defineAction, defineCredential, executeAction, getActionCredentialRequirements, getRegisteredProjectScopeId, getRunSignal, getWorkflowRunHandle, isAction, isCredentialInput, isStepInvocation, isWithinActionExecution, normalizeCredentialList, registerProjectScopeGetter, registerWorkflowRunGetter, resolveActionTool, runOutsideActionExecution, runWithRunSignal, runWithinActionExecution, toCredentialRequirement };

package/dist/agent.cjs

@@ -1,7 +1,7 @@
 Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
 const require_chunk = require("./chunk-DHdmVzO2.cjs");
-const require_dist$1 = require("./dist-hu4oHsK8.cjs");
-const require_dist$2 = require("./dist-DBE-7Q9K.cjs");
+const require_dist$1 = require("./dist-DOMwPx7V.cjs");
+const require_dist$2 = require("./dist-C5otI4u3.cjs");
 const require_dist$3 = require("./dist-DO51rYHS.cjs");
 require("./env-api-keys-BFptfIIs.cjs");
 const require_event_stream = require("./event-stream-CS-ls7wC.cjs");
@@ -902,7 +902,7 @@ function loadGoogleVertexProviderModule() {
 	return googleVertexProviderModulePromise;
 }
 function loadMistralProviderModule() {
-	mistralProviderModulePromise ||= Promise.resolve().then(() => require("./mistral-WPrOS6Bw.cjs")).then((module) => {
+	mistralProviderModulePromise ||= Promise.resolve().then(() => require("./mistral-wOLLEgb9.cjs")).then((module) => {
 		const provider = module;
 		return {
 			stream: provider.streamMistral,
@@ -5325,7 +5325,7 @@ function validateToolArguments(tool, toolCall) {
 	throw new Error(errorMessage);
 }
 //#endregion
-//#region ../agent/dist/schemas-BLmL2ciI.mjs
+//#region ../agent/dist/schemas-DCBOj0XY.mjs
 const getModelLoose = require_event_stream.getModel;
 const AGENT_MODEL_ID_PATTERN = /^[a-z0-9-]+\/.+/;
 const AgentModelIdSchema = zod.z.custom((value) => typeof value === "string" && AGENT_MODEL_ID_PATTERN.test(value), "model must be vendor/model-id");
@@ -13627,6 +13627,17 @@ function ensureSessionsDir(sessionsDir) {
 function defineTool$1(tool) {
 	return tool;
 }
+function prefixLines(prefix, text) {
+	return text.split("\n").map((line, index) => index === 0 ? `${prefix} ${line}` : `  ${line}`).join("\n");
+}
+/** Unified-diff-style replacement preview for tool results and chat display. */
+function formatTextReplacement(oldText, newText) {
+	return `${prefixLines("-", oldText)}\n${prefixLines("+", newText)}`;
+}
+function truncateForToolPreview(text, limit = 4e3) {
+	if (text.length <= limit) return text;
+	return `${text.slice(0, limit)}\n\n[Truncated — ${text.length} characters total]`;
+}
 const memoryPathSchema = require_dist$3.String({ description: "Relative path under the memory dir (e.g. MEMORY.md, USER.md, archive/note.md)." });
 const searchScopeSchema = require_dist$3.Union([
 	require_dist$3.Literal("archive"),
@@ -13728,7 +13739,7 @@ function createMemoryTool(config, db) {
 					return {
 						content: [{
 							type: "text",
-							text: `Wrote ${displayPath(config, filePath)}`
+							text: `Wrote ${displayPath(config, filePath)}\n\n${truncateForToolPreview(params.content)}`
 						}],
 						details: {}
 					};
@@ -13742,7 +13753,7 @@ function createMemoryTool(config, db) {
 					return {
 						content: [{
 							type: "text",
-							text: `Edited ${displayPath(config, filePath)}`
+							text: `Edited ${displayPath(config, filePath)}\n\n${formatTextReplacement(params.oldText, params.newText)}`
 						}],
 						details: {}
 					};
@@ -13821,9 +13832,9 @@ function seedMemoryFiles(config, options) {
 	}
 	ensureMdFiles(config.memoryFile, config.userFile);
 }
-function createDefaultMemory(options = {}) {
+function createDefaultMemory(options = {}, env = process.env) {
 	return { async create({ agentId }) {
-		const paths = agentPaths(agentId);
+		const paths = agentPaths(agentId, env);
 		const config = loadMemoryConfig(paths.memoryDir, paths.sessionsDir, options);
 		seedMemoryFiles(config, options);
 		ensureArchiveDir(config.archiveDir);
@@ -13893,6 +13904,66 @@ function runWithinAgentPromptExecution(fn) {
 function isWithinAgentPromptExecution() {
 	return executingAgentPrompt.getStore() === true;
 }
+const TITLE_MODEL_ID = "deepseek/deepseek-v4-flash";
+const TITLE_SYSTEM_PROMPT = [
+	"You are an expert chat title generator.",
+	"",
+	"Given the conversation so far (initial user message), generate a short, clear, and natural title (max 6-8 words) that captures the main topic or goal.",
+	"",
+	"Rules:",
+	"- Be specific and descriptive; avoid vague titles like \"Conversation\" or \"Question\"",
+	"- Prefer action-oriented or outcome-focused phrasing when possible",
+	"- Use title case",
+	"- Keep it under 65 characters when possible",
+	"- Do not use quotes or emojis",
+	"- Return ONLY the title, nothing else",
+	"",
+	"Examples:",
+	"",
+	"User: How do I set up a new Next.js project with TypeScript?",
+	"→ Setting Up Next.js Project with TypeScript",
+	"",
+	"User: Help me debug this React hook that's causing infinite re-renders. Here's the code...",
+	"→ Debugging Infinite Re-render in React Hook",
+	"",
+	"User: Write a cold email sequence for fundraising from YC alums",
+	"→ YC Alumni Fundraising Cold Email Sequence"
+].join("\n");
+const MAX_INPUT_CHARS = 2e3;
+const MAX_TITLE_CHARS = 65;
+function normalizeTitle(raw) {
+	const trimmed = raw.replace(/\s+/g, " ").trim().replace(/^["'`]+|["'`]+$/g, "").replace(/[.!?,;:]+$/g, "").trim();
+	if (!trimmed) return null;
+	return trimmed.length > MAX_TITLE_CHARS ? `${trimmed.slice(0, MAX_TITLE_CHARS).trim()}…` : trimmed;
+}
+/** Generate a concise chat title from the user's opening message. Returns null on empty output. */
+async function generateSessionTitle(message) {
+	const input = message.trim();
+	if (!input) return null;
+	return normalizeTitle((await completeSimple(resolveAgentModel(TITLE_MODEL_ID), {
+		systemPrompt: TITLE_SYSTEM_PROMPT,
+		messages: [{
+			role: "user",
+			content: input.slice(0, MAX_INPUT_CHARS),
+			timestamp: Date.now()
+		}]
+	}, {
+		maxTokens: 32,
+		temperature: .3
+	})).content.filter((part) => part.type === "text").map((part) => part.text).join(" "));
+}
+/**
+* Generate and persist a session title from the opening message. Never throws — titling is a
+* best-effort enhancement that must not affect the prompt run.
+*/
+async function generateAndStoreSessionTitle(sessionId, message) {
+	try {
+		const title = await generateSessionTitle(message);
+		if (title) await require_dist$2.setSessionTitle(sessionId, title);
+	} catch (err) {
+		console.error("[session] title generation failed:", err);
+	}
+}
 function llmUsageFromAssistantMessage(message) {
 	const usage = message.usage;
 	if (!usage) return null;
@@ -13987,6 +14058,7 @@ async function prepareAgentSession(agentId, sessionId, options) {
 async function runAgentPrompt(agent, agentId, input, options = {}) {
 	const { sessionId } = await prepareAgentSession(agentId, input.sessionId);
 	const priorMessages = await require_dist$2.listMessageEvents(sessionId);
+	const titleTask = priorMessages.length === 0 ? generateAndStoreSessionTitle(sessionId, input.message) : void 0;
 	const runPrompt = async () => {
 		let runtime;
 		let unsubscribe;
@@ -13994,6 +14066,8 @@ async function runAgentPrompt(agent, agentId, input, options = {}) {
 		const pendingHandlers = /* @__PURE__ */ new Set();
 		const signal = require_dist$1.getRunSignal();
 		let promptDurationMs = 0;
+		let lastLiveWriteAt = 0;
+		const LIVE_MESSAGE_WRITE_INTERVAL_MS = 400;
 		try {
 			runtime = await agent.buildRuntime({
 				agentId,
@@ -14008,15 +14082,31 @@ async function runAgentPrompt(agent, agentId, input, options = {}) {
 			});
 			const activeRuntime = runtime;
 			unsubscribe = activeRuntime.pi.subscribe((event) => {
+				if (event.type === "message_start" || event.type === "message_update") {
+					const now = Date.now();
+					if (now - lastLiveWriteAt >= LIVE_MESSAGE_WRITE_INTERVAL_MS) {
+						lastLiveWriteAt = now;
+						require_dist$2.setSessionLiveMessage(sessionId, event.message).catch((err) => {
+							console.error("[session] live message persist failed:", err);
+						});
+					}
+				}
 				if (!shouldPersistAgentEvent(event)) return;
 				const tracked = (async () => {
 					await persistAgentEvent(activeRuntime.agentId, sessionId, event);
-					if (event.type === "message_end" && event.message.role === "assistant") {
-						const usage = llmUsageFromAssistantMessage(event.message);
-						if (usage) await require_dist$2.recordLlmUsageFromAssistantMessage({
-							sessionId,
-							usage
-						});
+					if (event.type === "message_end") {
+						try {
+							await require_dist$2.clearLiveMessage(sessionId);
+						} catch (err) {
+							console.error("[session] live message clear failed:", err);
+						}
+						if (event.message.role === "assistant") {
+							const usage = llmUsageFromAssistantMessage(event.message);
+							if (usage) await require_dist$2.recordLlmUsageFromAssistantMessage({
+								sessionId,
+								usage
+							});
+						}
 					}
 				})().finally(() => {
 					pendingHandlers.delete(tracked);
@@ -14066,6 +14156,12 @@ async function runAgentPrompt(agent, agentId, input, options = {}) {
 				await runtime.sandbox.dispose();
 			}
 			await require_dist$2.addAgentSessionDuration(sessionId, promptDurationMs);
+			try {
+				await require_dist$2.clearLiveMessage(sessionId);
+			} catch (err) {
+				console.error("[session] live message clear failed:", err);
+			}
+			if (titleTask) await titleTask;
 		}
 	};
 	if (options.skipAgentSessionSpan) return runWithinAgentPromptExecution(runPrompt);