@theokit/sdk 4.54.0 → 4.55.0

package/dist/{chunk-TPTZA6NI.js → chunk-I5BE5M5L.js}

@@ -1,11 +1,11 @@
-import { enqueueSessionWrite, hydrateSession, persistTurnToTranscript, flushSessionWrites, appendSessionMessage, getSessionMessages, discardSession, clearSession } from './chunk-DUIF54UP.js';
+import { enqueueSessionWrite, hydrateSession, persistTurnToTranscript, flushSessionWrites, appendSessionMessage, getSessionMessages, discardSession, clearSession } from './chunk-OHBIUXKM.js';
 import { IndexManager, writeSessionSummary, memoryDir, appendFact, readFacts, legacyMemoryJsonPath, memoryMdPath, appendFactToMarkdown } from './chunk-DAPSQZT4.js';
 import { loadHookConfig, warnOnce, currentPersonalityContext, warnPersonalitySwitchInsideFork } from './chunk-YMA4S2WO.js';
 import { extractCodePlugins } from './chunk-2SFBB54R.js';
 import { submit, emitRunEvent } from './chunk-SUKXXLWD.js';
 import { createOpenAiCompatibleRuntime } from './chunk-XN7NOENA.js';
 import { FsSessionStore } from './chunk-GJ6RK75E.js';
-import { discoverProviderPlugins, getConfiguredBaseUrl, httpRequest, isFixtureApiKey, presentProviderCredentialEnvVars, CircuitBreaker, buildDefaultSummarizer, shouldUseRealLocalRuntime, parseSseStream, resolveProviderChain, inferProviderFromApiKey, detectPrimaryProvider, warnProviderPrecedenceOnce, tryFirstEvent, relayStream, renderToolResultContentText } from './chunk-VDEWG5TV.js';
+import { discoverProviderPlugins, getConfiguredBaseUrl, httpRequest, isFixtureApiKey, presentProviderCredentialEnvVars, CircuitBreaker, buildDefaultSummarizer, shouldUseRealLocalRuntime, parseSseStream, resolveProviderChain, inferProviderFromApiKey, detectPrimaryProvider, warnProviderPrecedenceOnce, tryFirstEvent, relayStream, renderToolResultContentText } from './chunk-MVAPK2BX.js';
 import { resolveChildEnv } from './chunk-HKCKOAIO.js';
 import { DEFAULT_MAX_BYTES_PER_FILE, runDiscovery, truncateWithMarker } from './chunk-5PHVENFV.js';
 import { readVersionedJson, getTheokitHome, writeVersionedJson } from './chunk-UPRJR6IP.js';
@@ -26,8 +26,8 @@ import { loadSubagents } from './chunk-2QKTVKH3.js';
 import { readWorkspaceDir, parseSimpleYaml } from './chunk-AG5JPQIY.js';
 import { resolveEffectiveContextWindow, CONTEXT_WINDOW_FLOOR, CONTEXT_WINDOW_MARGIN } from './chunk-F7AQV62G.js';
 import { parseModelId } from './chunk-R7WIIPUR.js';
-import { getProviderProfile } from './chunk-SSQZA3DZ.js';
-import { getCatalogModelInfo, registerBuiltins, getProviderProfile as getProviderProfile$1, registerProvider, isPatchedModelKey } from './chunk-FXEUP75G.js';
+import { getProviderProfile } from './chunk-2BDH744Z.js';
+import { getCatalogModelInfo, registerBuiltins, getProviderProfile as getProviderProfile$1, registerProvider, isPatchedModelKey } from './chunk-Q4CMI2LP.js';
 import { AgentRunError, coerceToKnownAgentRunErrorCode, UnknownAgentError, ConfigurationError, AuthenticationError, AgentDisposedError, UnsupportedRunOperationError, NetworkError, TheokitAgentError } from './chunk-IDCKSLYH.js';
 import { resolveApiKey, API_KEY_ENV_VAR, redactSecrets } from './chunk-T7XEKOVW.js';
 import { diag } from './chunk-T7O6K6PX.js';
@@ -2694,6 +2694,9 @@ function buildContextBudgetEvent(model, resolved) {
 }
 
 // src/internal/runtime/lifecycle/post-run-lifecycle.ts
+function sessionTranscriptAllowed(memory) {
+  return memory?.enabled !== false;
+}
 function resolveWindowForRun(model, contextWindow) {
   const override = contextWindow !== void 0 && contextWindow > 0 ? contextWindow : void 0;
   return resolveEffectiveContextWindow({
@@ -2769,7 +2772,7 @@ async function runPostRunLifecycle(inputs) {
     },
     onRunEvent !== void 0 ? () => emitRunEvent(onRunEvent, { type: "compact_boundary", trigger: "auto" }) : void 0
   );
-  if (result.status === "finished" && result.result !== void 0) {
+  if (result.status === "finished" && result.result !== void 0 && sessionTranscriptAllowed(inputs.memory)) {
     const summaryArgs = {
       cwd: workspaceCwd,
       runId: result.id,
@@ -3289,11 +3292,17 @@ function validateSubagents(agents2) {
   }
 }
 var TOOL_NAME_PATTERN = /^[a-zA-Z][a-zA-Z0-9_-]{0,63}$/;
-var RESERVED_TOOL_NAMES = /* @__PURE__ */ new Set(["shell", "memory_search", "memory_get"]);
-function validateToolCatalog(tools) {
+var BUILTIN_TOOL_NAMES = ["shell", "memory_search", "memory_get"];
+var RESERVED_TOOL_NAMES = new Set(BUILTIN_TOOL_NAMES);
+function reservedNamesFor(withheld) {
+  if (withheld === void 0 || withheld.length === 0) return RESERVED_TOOL_NAMES;
+  return new Set(BUILTIN_TOOL_NAMES.filter((name) => !withheld.includes(name)));
+}
+function validateToolCatalog(tools, withheldBuiltins) {
+  const reserved = reservedNamesFor(withheldBuiltins);
   const seen = /* @__PURE__ */ new Set();
   for (const tool of tools) {
-    validateSingleTool(tool);
+    validateSingleTool(tool, reserved);
     if (seen.has(tool.name)) {
       throw new ConfigurationError(`Duplicate custom tool name "${tool.name}"`, {
         code: "duplicate_tool_name"
@@ -3311,10 +3320,10 @@ function validateCustomTools(options) {
       { code: "cloud_custom_tools_rejected" }
     );
   }
-  validateToolCatalog(tools);
+  validateToolCatalog(tools, options.withheldBuiltinTools);
 }
-function validateSingleTool(tool) {
-  validateToolName(tool);
+function validateSingleTool(tool, reserved) {
+  validateToolName(tool, reserved);
   validateToolDescription(tool);
   validateToolSchema(tool);
   if (typeof tool.handler !== "function") {
@@ -3323,7 +3332,7 @@ function validateSingleTool(tool) {
     });
   }
 }
-function validateToolName(tool) {
+function validateToolName(tool, reserved) {
   if (typeof tool.name !== "string" || tool.name.length === 0) {
     throw new ConfigurationError("Custom tool requires a non-empty name", {
       code: "tool_missing_name"
@@ -3335,7 +3344,7 @@ function validateToolName(tool) {
       { code: "tool_invalid_name" }
     );
   }
-  if (RESERVED_TOOL_NAMES.has(tool.name) || tool.name.startsWith("mcp_")) {
+  if (reserved.has(tool.name) || tool.name.startsWith("mcp_")) {
     throw new ConfigurationError(
       `Custom tool name "${tool.name}" collides with a reserved SDK tool name`,
       { code: "tool_reserved_name" }
@@ -4848,8 +4857,10 @@ async function initLoopContext(inputs) {
       memoryProviderHandle = void 0;
     }
   }
-  const tools = await collectTools(inputs.mcp, inputs.runEventSink);
+  const withheld = new Set(inputs.withheldBuiltinTools ?? []);
+  const tools = await collectTools(inputs.mcp, withheld, inputs.runEventSink);
   for (const memTool of inputs.memoryTools ?? []) {
+    if (withheld.has(memTool.name)) continue;
     tools.push({
       name: memTool.name,
       description: memTool.description,
@@ -4875,6 +4886,7 @@ async function initLoopContext(inputs) {
       providerTools = [];
     }
     for (const providerTool of providerTools) {
+      if (withheld.has(providerTool.name)) continue;
       tools.push({
         name: providerTool.name,
         description: providerTool.description,
@@ -4950,8 +4962,9 @@ async function safeListTools(client, serverName, sink) {
     return [];
   }
 }
-async function collectTools(mcp, sink) {
-  const tools = [
+function shellToolDescriptor(withheld) {
+  if (withheld.has("shell")) return [];
+  return [
     {
       name: "shell",
       description: "Run a shell command in the workspace and return stdout/stderr.",
@@ -4963,6 +4976,9 @@ async function collectTools(mcp, sink) {
       origin: "shell"
     }
   ];
+}
+async function collectTools(mcp, withheld, sink) {
+  const tools = shellToolDescriptor(withheld);
   for (const [serverName, client] of mcp.entries()) {
     const mcpTools = await safeListTools(client, serverName, sink);
     for (const tool of mcpTools) {
@@ -4981,6 +4997,13 @@ async function collectTools(mcp, sink) {
 function sanitize(name) {
   return name.replace(/[^a-zA-Z0-9-]+/g, "_").replace(/^[_-]+|[_-]+$/g, "");
 }
+var KEY_PREFIX = "theokit-";
+var KEY_HEX_LENGTH = 32;
+function derivePromptCacheKey(sessionId) {
+  if (sessionId === void 0 || sessionId.trim().length === 0) return void 0;
+  const digest = createHash("sha256").update(sessionId, "utf8").digest("hex");
+  return `${KEY_PREFIX}${digest.slice(0, KEY_HEX_LENGTH)}`;
+}
 
 // src/internal/tool-dispatch/strip-think.ts
 var THINK_PATTERN = /<think>[\s\S]*?<\/think>\s*/g;
@@ -5052,6 +5075,16 @@ async function streamLlmTurn(inputs, ctx) {
       // Step-cap force-close: forward the per-run tool gate (`"none"` on a ceiling round forces a
       // text close even though tools are advertised). Absent ⇒ provider default (auto).
       ...inputs.toolChoice !== void 0 ? { toolChoice: inputs.toolChoice } : {},
+      // usetheokit/theokit-sdk#383 — the prompt-cache key, derived HERE and from `inputs.agentId`
+      // because this is the one place that sees the session identity on every round. `agentId` is
+      // the run's session identity (`Agent.getOrCreate(sessionId)`), so the derived key is identical
+      // on round 1 and round 12 of a turn, identical on turn 2 of the same session, and different
+      // for a different session. Deriving it per round from a stable input is what makes it a cache
+      // key; minting one anywhere that runs more than once per session would not be.
+      ...(() => {
+        const key = derivePromptCacheKey(inputs.agentId);
+        return key !== void 0 ? { promptCacheKey: key } : {};
+      })(),
       messages: ctx.messages,
       tools: ctx.tools.map(toLlmTool)
     },
@@ -7022,14 +7055,20 @@ function resolveInheritedCredentials(agentOptions) {
     ...parentSandbox !== void 0 ? { sandbox: parentSandbox } : {}
   };
 }
-function buildCustomToolsInput(agentOptions, sendOptions, pluginManager, personalityToolWhitelist, agentId, personalityName, subagents, effectiveModel) {
+function withheldBuiltinsInput(agentOptions) {
+  const withheld = agentOptions.withheldBuiltinTools;
+  if (withheld === void 0 || withheld.length === 0) return {};
+  return { withheldBuiltinTools: withheld };
+}
+function buildRunToolCatalogInput(agentOptions, sendOptions, pluginManager, personalityToolWhitelist, agentId, personalityName, subagents, effectiveModel) {
+  const builtins = withheldBuiltinsInput(agentOptions);
   const baseTools = sendOptions?.tools ?? agentOptions.tools ?? [];
   const agentsForTools = subagents !== void 0 && Object.keys(subagents).length > 0 ? subagents : agentOptions.agents;
   const subagentTools = declarativeSubagentTools(agentsForTools, baseTools);
   const pluginTools = pluginManager?.aggregated.tools ?? [];
   const reasoningTools = reasoningActive(agentOptions.reasoning, effectiveModel) ? [createThinkTool()] : [];
   if (baseTools.length === 0 && subagentTools.length === 0 && pluginTools.length === 0 && reasoningTools.length === 0) {
-    return {};
+    return builtins;
   }
   const allTools = [...baseTools, ...subagentTools, ...pluginTools, ...reasoningTools];
   const merged = allTools.map((tool) => ({
@@ -7042,8 +7081,8 @@ function buildCustomToolsInput(agentOptions, sendOptions, pluginManager, persona
     agentId,
     personalityName
   });
-  if (customTools.length === 0 && personalityToolWhitelist === void 0) return {};
-  return { customTools };
+  if (customTools.length === 0 && personalityToolWhitelist === void 0) return builtins;
+  return { customTools, ...builtins };
 }
 
 // src/internal/local-agent/real-local-run.ts
@@ -7159,7 +7198,7 @@ function buildLoopInputs(options, runId, userText, userImages) {
     } : {},
     ...options.priorMessages !== void 0 ? { priorMessages: options.priorMessages } : {},
     ...options.memoryTools !== void 0 && options.memoryTools.length > 0 ? { memoryTools: options.memoryTools } : {},
-    ...buildCustomToolsInput(
+    ...buildRunToolCatalogInput(
       options.agentOptions,
       options.sendOptions,
       options.pluginManager,
@@ -9486,7 +9525,7 @@ var LocalAgent = class {
   }
   async send(message, options = {}) {
     if (options.tools !== void 0 && options.tools.length > 0) {
-      validateToolCatalog(options.tools);
+      validateToolCatalog(options.tools, this.options.withheldBuiltinTools);
     }
     const sendSpan = this._telemetry.startSpan(SPAN_NAMES.AGENT_SEND, {
       agentId: this.agentId,
@@ -9544,6 +9583,8 @@ var LocalAgent = class {
         ...options.onRunEvent !== void 0 ? { onRunEvent: options.onRunEvent } : {},
         hooksExecutor: this.hooksExecutor,
         memoryGlue: this.memoryGlue,
+        // usetheokit/theokit-sdk#382 — the transcript write reads this and nothing else.
+        memory: this.options.memory,
         ...postRunProvider !== void 0 ? { memoryProvider: postRunProvider } : {}
       });
     } finally {
@@ -10315,7 +10356,7 @@ var Agent = class _Agent {
         `No local agent "${agentId}" registered \u2014 compact targets local sessions.`
       );
     }
-    const { compactSessionTranscript, buildDefaultSummarizer: buildDefaultSummarizer2 } = await import('./compact-session-K5LXTBPJ.js');
+    const { compactSessionTranscript, buildDefaultSummarizer: buildDefaultSummarizer2 } = await import('./compact-session-FUJDVF7B.js');
     const { cwd, model, store } = await openLocalStore(reg);
     return enqueueSessionWrite(
       cwd,
@@ -10351,7 +10392,7 @@ var Agent = class _Agent {
         `No local agent "${agentId}" registered \u2014 injectSessionTurn targets local sessions.`
       );
     }
-    const { injectSessionTurn } = await import('./inject-session-DDR6X6PC.js');
+    const { injectSessionTurn } = await import('./inject-session-BJMJH5BX.js');
     const { cwd, model, store } = await openLocalStore(reg);
     await injectSessionTurn({
       store,
@@ -10426,7 +10467,7 @@ var Agent = class _Agent {
       //
       // Honest limit, stated because the alternative is an implied claim: `tools` is still the
       // DECLARED catalog. Plugin tools and the reasoning `think` tool are assembled per run by
-      // `buildCustomToolsInput`, so they are not knowable from the registry alone. Documented on
+      // `buildRunToolCatalogInput`, so they are not knowable from the registry alone. Documented on
       // `AgentDescription.tools`.
       tools: (agent.options.tools ?? []).map((tool) => ({
         name: tool.name,
@@ -10486,5 +10527,5 @@ async function openLocalStore(reg) {
 }
 
 export { Agent, AgentBuilder, DEFAULT_AGENTIC_MODEL_ID, MEMORY_EMBEDDING_ADAPTERS, Tool, ToolError, UsageAccumulator, computeCost, generateCronId, getPricingEntry, openRouterMemoryEmbeddingProviderAdapter };
-//# sourceMappingURL=chunk-TPTZA6NI.js.map
-//# sourceMappingURL=chunk-TPTZA6NI.js.map
+//# sourceMappingURL=chunk-I5BE5M5L.js.map
+//# sourceMappingURL=chunk-I5BE5M5L.js.map