@keystrokehq/keystroke 0.1.76 → 0.1.79

package/dist/agent.mjs

@@ -1,8 +1,8 @@
 import { A as stripPromptContext, C as mcpContentBlocksToAiSdkParts, D as parseChatAttachmentServePath, O as prependAgentUserContext, S as mcpContentBlocksFromToolOutput, T as modelSupportsMediaType, _ as defineTool, a as FilePartSchema, k as requiredDisplayTextSchema, r as DEFAULT_CLOUD_PLATFORM_ORIGIN, u as PublicModelsResponseSchema, x as isRedundantMcpJsonText, y as hasAgentTimestampContext } from "./dist-D0P3wUQe.mjs";
 import { c as getRunSignal, f as isWithinActionExecution, h as resolveActionTool, l as getWorkflowRunHandle, u as isAction } from "./dist-UT46qwZV.mjs";
-import { $ as captureConsole, A as appendEvent, B as hasSessionEventAfterSeq, C as tool, F as getAgentByRoute, G as resolveRunSourceFromTraceContext, H as latestSessionEventByTypes, K as selectActiveSkillsBySlugs, O as MESSAGE_EVENT_TYPE, P as failAgentSession, Q as touchSession, W as listSessionEvents, X as setSessionTitle, Y as setSessionStatus, _ as generateId, a as generateText, c as output_exports, d as wrapLanguageModel, et as getTraceContext, i as generateObject, j as createSession, k as addAgentSessionDuration, l as readUIMessageStream, n as convertToModelMessages, nt as withSpan, o as hasToolCall, r as extractJsonMiddleware, s as isFileUIPart, t as ToolLoopAgent, u as toUIMessageStream, y as jsonSchema, z as getSession } from "./dist-C_APGtUJ.mjs";
+import { $ as captureConsole, A as appendEvent, B as hasSessionEventAfterSeq, C as tool, F as getAgentByRoute, G as resolveRunSourceFromTraceContext, H as latestSessionEventByTypes, K as selectActiveSkillsBySlugs, O as MESSAGE_EVENT_TYPE, P as failAgentSession, Q as touchSession, W as listSessionEvents, X as setSessionTitle, Y as setSessionStatus, _ as generateId, a as generateText, c as output_exports, d as wrapLanguageModel, et as getTraceContext, i as generateObject, j as createSession, k as addAgentSessionDuration, l as readUIMessageStream, n as convertToModelMessages, nt as withSpan, o as hasToolCall, r as extractJsonMiddleware, t as ToolLoopAgent, u as toUIMessageStream, y as jsonSchema, z as getSession } from "./dist-C_APGtUJ.mjs";
 import { A as defineMcp, C as isCredentialError, D as connectMcpDefinition, E as withCredentialAssignments, O as connectMcpServer, S as enrichCredentialError, T as resolveMcpTools, b as captureCredentialToolErrors, d as isWorkflow, h as resolveWorkflowTool, j as isMcp, k as connectMcpStdio, u as isRunTimeoutError, w as resolveActionCredentials, x as createCredentialResolver, y as buildCredentialRunContext } from "./dist-DD3DTmkE.mjs";
-import { C as withLlmRunContext, S as splitAgentModelId, _ as currentLlmRunContext, a as llmUsageFromLanguageModelUsage, b as resolveModelCapabilities, c as resolveCompactionModelId, d as AgentCreateInputSchema, f as AgentModelIdSchema, g as attachmentPlaceholderText, h as ThinkingLevelSchema, i as estimateSerializedChars, l as resolveContextBudget, m as DEFAULT_THINKING_LEVEL, n as byokFromResponseHeaders, o as parseContextCompactionCheckpoint, p as DEFAULT_COMPACTION_MODEL_ID, r as compactModelMessages, s as rebuildMessagesFromCheckpoint, t as ContextBudgetTracker, u as summaryToUiMessage, v as parseAgentCreateInput, x as resolveThinkingLevel, y as resolveAgentModel } from "./context-compaction-B2veGznA-Br9-3A43.mjs";
+import { C as resolveModelCapabilities, E as withLlmRunContext, S as resolveAgentModel, T as splitAgentModelId, _ as DEFAULT_THINKING_LEVEL, a as estimateSerializedChars, b as currentLlmRunContext, c as parseContextCompactionCheckpoint, d as resolveCompactionModelId, f as resolveContextBudget, g as DEFAULT_COMPACTION_MODEL_ID, h as AgentModelIdSchema, i as estimateChatAttachmentTokens, l as placeholderChatAttachmentMessages, m as AgentCreateInputSchema, n as byokFromResponseHeaders, o as llmUsageFromLanguageModelUsage, p as summaryToUiMessage, r as compactModelMessages, s as mapMessageFileParts, t as ContextBudgetTracker, u as rebuildMessagesFromCheckpoint, v as ThinkingLevelSchema, w as resolveThinkingLevel, x as parseAgentCreateInput, y as attachmentPlaceholderText } from "./context-compaction-Bxp4WK2B-Ct4RSjlg.mjs";
 import { $ as getBoundAppRoot, c as createStubInvokeToolBridge, ct as AGENT_MOUNT, et as loadAssetManifest, ft as ensureWorkspaceDir, ht as resolveSessionRoot, i as createInvokeToolBridge, l as defaultWorkspacesRoot, m as materializeSkills, o as createSandbox, ot as resolveSandboxDefinition, pt as resolveAgentRoot, s as createSkillViewTool, t as AGENT_SKILLS_DIR, tt as materializeSandbox, u as defineSkill, v as sandboxSystemPromptInjection } from "./dist-DbUdD0Ue.mjs";
 import { z } from "zod";
 import { AsyncLocalStorage } from "node:async_hooks";
@@ -770,17 +770,6 @@ function buildMemoryFromHandle(config, db) {
 }
 //#endregion
 //#region ../agent/dist/index.mjs
-/** Map every `FileUIPart` in each message, leaving other parts and non-array messages untouched. */
-async function mapMessageFileParts(messages, mapFilePart) {
-	return Promise.all(messages.map(async (message) => {
-		if (!Array.isArray(message.parts)) return message;
-		const parts = await Promise.all(message.parts.map((part) => isFileUIPart(part) ? mapFilePart(part) : part));
-		return {
-			...message,
-			parts
-		};
-	}));
-}
 /** Replace unsupported `FileUIPart`s with text placeholders before model conversion. */
 function applyCapabilityPlaceholderToMessages(messages, capabilities) {
 	return mapMessageFileParts(messages, (part) => modelSupportsMediaType(capabilities, part.mediaType) ? part : {
@@ -1794,15 +1783,16 @@ async function runAgentPrompt(agent, agentId, input, options = {}) {
 			const budget = resolveContextBudget(activeRuntime.runConfig.resolvedModel);
 			if (budget && modelMessages.length > 0) {
 				const tracker = new ContextBudgetTracker(budget);
-				const modelMsgs = await convertToModelMessages(modelMessages);
+				const modelMsgs = await convertToModelMessages(await placeholderChatAttachmentMessages(modelMessages));
 				const estimate = tracker.estimate({
 					systemPrompt: activeRuntime.runConfig.systemPrompt,
 					messages: modelMsgs,
-					tools: activeRuntime.tools
+					tools: activeRuntime.tools,
+					extraMessageTokens: estimateChatAttachmentTokens(modelMessages)
 				});
 				if (estimate?.decision === "compact") try {
 					const priorDurable = activeCheckpoint?.durable === true ? activeCheckpoint : null;
-					const suffixModel = await convertToModelMessages(priorDurable ? sessionContext.messageEvents.filter((event) => event.seq > priorDurable.compactedThroughSeq).map((event) => event.message) : modelMessages);
+					const suffixModel = await convertToModelMessages(await placeholderChatAttachmentMessages(priorDurable ? sessionContext.messageEvents.filter((event) => event.seq > priorDurable.compactedThroughSeq).map((event) => event.message) : modelMessages));
 					const compacted = await withLlmRunContext({
 						runId: sessionId,
 						runKind: "agent"