@theokit/sdk 2.0.0 → 2.1.0

package/dist/a2a/index.cjs

@@ -132,6 +132,24 @@ var init_agent_builder = __esm({
   }
 });
 
+// src/internal/default-retriable.ts
+function defaultRetriableForCode(code) {
+  switch (code) {
+    case "rate_limit":
+    case "timeout":
+    case "server_error":
+    case "network":
+    case "provider_unreachable":
+      return true;
+    default:
+      return false;
+  }
+}
+var init_default_retriable = __esm({
+  "src/internal/default-retriable.ts"() {
+  }
+});
+
 // src/internal/security/redact.ts
 function readEnvOnce() {
   const raw = process.env.THEOKIT_REDACT_SECRETS;
@@ -282,7 +300,8 @@ __export(errors_exports, {
   UnsupportedBudgetOperationError: () => UnsupportedBudgetOperationError,
   UnsupportedRunOperationError: () => UnsupportedRunOperationError,
   UnsupportedTaskOperationError: () => UnsupportedTaskOperationError,
-  coerceToKnownAgentRunErrorCode: () => coerceToKnownAgentRunErrorCode
+  coerceToKnownAgentRunErrorCode: () => coerceToKnownAgentRunErrorCode,
+  isTransientError: () => isTransientError
 });
 function coerceToKnownAgentRunErrorCode(code) {
   if (code !== void 0 && KNOWN_AGENT_RUN_ERROR_CODES.has(code)) {
@@ -314,21 +333,13 @@ function safeStringify(value) {
     return String(value);
   }
 }
-function defaultRetriableForCode(code) {
-  switch (code) {
-    case "rate_limit":
-    case "timeout":
-    case "server_error":
-    case "network":
-    case "provider_unreachable":
-      return true;
-    default:
-      return false;
-  }
+function isTransientError(err) {
+  return err instanceof TheokitAgentError && err.isRetryable === true;
 }
 var KNOWN_AGENT_RUN_ERROR_CODES, TheokitAgentError, AuthenticationError, RateLimitError, ConfigurationError, IntegrationNotConnectedError, NetworkError, UnknownAgentError, AgentRunError, UnsupportedRunOperationError, CredentialPoolExhaustedError, MemoryAdapterError, InvalidTaskIdError, TaskNotFoundError, UnsupportedTaskOperationError, BudgetExceededError, AgentDisposedError, UnsupportedBudgetOperationError;
 var init_errors = __esm({
   "src/errors.ts"() {
+    init_default_retriable();
     init_redact();
     KNOWN_AGENT_RUN_ERROR_CODES = /* @__PURE__ */ new Set([
       "rate_limit",
@@ -1027,6 +1038,19 @@ function sanitizeIdentifier(input, options) {
   }
   return input.toLowerCase();
 }
+function safeFilenameForId(id, options) {
+  if (id.length === 0) {
+    throw new ConfigurationError("Filename id must be a non-empty string", {
+      code: "invalid_filename_id"
+    });
+  }
+  const maxLen = options?.maxLen;
+  const lower = id.toLowerCase();
+  if (lower.length <= maxLen && IDENTIFIER_PATTERN.test(lower)) {
+    return lower;
+  }
+  return `h-${crypto.createHash("sha256").update(id).digest("hex").slice(0, 16)}`;
+}
 var PathTraversalError, IDENTIFIER_PATTERN;
 var init_path_guard = __esm({
   "src/internal/security/path-guard.ts"() {
@@ -1043,10 +1067,10 @@ var init_path_guard = __esm({
   }
 });
 
-// src/internal/runtime/default-model.ts
+// src/internal/runtime/config/default-model.ts
 var DEFAULT_AGENTIC_MODEL_ID;
 var init_default_model = __esm({
-  "src/internal/runtime/default-model.ts"() {
+  "src/internal/runtime/config/default-model.ts"() {
     DEFAULT_AGENTIC_MODEL_ID = "google/gemini-2.0-flash-001";
   }
 });
@@ -1442,7 +1466,7 @@ var init_agent_registry = __esm({
   }
 });
 
-// src/internal/runtime/system-prompt.ts
+// src/internal/runtime/system-prompt/system-prompt.ts
 async function resolveSystemPrompt(agent, override, ctx) {
   if (override !== void 0) return override;
   if (agent === void 0) return void 0;
@@ -1460,7 +1484,7 @@ async function resolveSystemPromptForSend(agent, override, buildCtx) {
   return resolveSystemPrompt(agent, override, ctx);
 }
 var init_system_prompt = __esm({
-  "src/internal/runtime/system-prompt.ts"() {
+  "src/internal/runtime/system-prompt/system-prompt.ts"() {
   }
 });
 
@@ -3114,7 +3138,7 @@ var init_paths = __esm({
   }
 });
 
-// src/internal/runtime/yaml-frontmatter.ts
+// src/internal/runtime/context/yaml-frontmatter.ts
 function parseSimpleYaml(text) {
   const fields = {};
   for (const line of text.split(/\r?\n/)) {
@@ -3138,7 +3162,7 @@ function coerce(raw) {
   return raw;
 }
 var init_yaml_frontmatter = __esm({
-  "src/internal/runtime/yaml-frontmatter.ts"() {
+  "src/internal/runtime/context/yaml-frontmatter.ts"() {
   }
 });
 async function loadMarkdownEntities(opts) {
@@ -4142,7 +4166,7 @@ function spawnAndCollect(options) {
   });
 }
 var init_spawn_collect = __esm({
-  "src/internal/runtime/spawn-collect.ts"() {
+  "src/internal/runtime/lifecycle/spawn-collect.ts"() {
   }
 });
 
@@ -4239,41 +4263,11 @@ var init_hooks_executor = __esm({
     };
   }
 });
-
-// src/internal/runtime/memory/memory-path-selector.ts
-function shouldUsePortMemoryPath() {
-  const env = globalThis.process?.env;
-  if (env === void 0) return false;
-  const val = env[PORT_MEMORY_PATH_ENV_VAR];
-  return val === "1" || val === "true";
-}
-function resolveMemoryProviderForLoop(consumerSupplied, defaultAdapter, portPathEnabled) {
-  if (consumerSupplied !== void 0) return consumerSupplied;
-  if (portPathEnabled) return defaultAdapter;
-  return void 0;
-}
-function resolveMemoryToolsForLoop(legacyTools, portPathEnabled) {
-  if (portPathEnabled) return void 0;
-  return legacyTools;
-}
-function resolveActiveMemorySummaryForSend(legacySummary, portPathEnabled) {
-  if (portPathEnabled) return void 0;
-  return legacySummary;
-}
-var PORT_MEMORY_PATH_ENV_VAR;
-var init_memory_path_selector = __esm({
-  "src/internal/runtime/memory/memory-path-selector.ts"() {
-    PORT_MEMORY_PATH_ENV_VAR = "THEOKIT_PORT_MEMORY_PATH";
-  }
-});
 function sessionsDir(cwd) {
   return path.join(memoryDir(cwd), "sessions");
 }
 function sessionSummaryPath(cwd, runId) {
-  return path.join(sessionsDir(cwd), `${sanitizeRunId(runId)}.md`);
-}
-function sanitizeRunId(runId) {
-  return runId.replace(/[^a-zA-Z0-9_-]/g, "_").slice(0, 128);
+  return path.join(sessionsDir(cwd), `${safeFilenameForId(runId, { maxLen: 128 })}.md`);
 }
 function truncate(text) {
   if (text.length <= MAX_TURN_CHARS) return text;
@@ -4309,11 +4303,39 @@ var MAX_TURN_CHARS;
 var init_session_summary_writer = __esm({
   "src/internal/memory/storage/session-summary-writer.ts"() {
     init_atomic_write();
+    init_path_guard();
     init_types();
     init_markdown_store();
     MAX_TURN_CHARS = 2e3;
   }
 });
+
+// src/internal/runtime/memory/memory-path-selector.ts
+function shouldUsePortMemoryPath() {
+  const env = globalThis.process?.env;
+  if (env === void 0) return false;
+  const val = env[PORT_MEMORY_PATH_ENV_VAR];
+  return val === "1" || val === "true";
+}
+function resolveMemoryProviderForLoop(consumerSupplied, defaultAdapter, portPathEnabled) {
+  if (consumerSupplied !== void 0) return consumerSupplied;
+  if (portPathEnabled) return defaultAdapter;
+  return void 0;
+}
+function resolveMemoryToolsForLoop(legacyTools, portPathEnabled) {
+  if (portPathEnabled) return void 0;
+  return legacyTools;
+}
+function resolveActiveMemorySummaryForSend(legacySummary, portPathEnabled) {
+  if (portPathEnabled) return void 0;
+  return legacySummary;
+}
+var PORT_MEMORY_PATH_ENV_VAR;
+var init_memory_path_selector = __esm({
+  "src/internal/runtime/memory/memory-path-selector.ts"() {
+    PORT_MEMORY_PATH_ENV_VAR = "THEOKIT_PORT_MEMORY_PATH";
+  }
+});
 function sessionFilePath(cwd, agentId) {
   const safe2 = sanitizeIdentifier(agentId, { maxLen: 128 });
   return safePathJoin(cwd, ".theokit", "agents", safe2, "messages.jsonl");
@@ -4587,7 +4609,7 @@ var init_agent_session = __esm({
   }
 });
 
-// src/internal/runtime/post-run-lifecycle.ts
+// src/internal/runtime/lifecycle/post-run-lifecycle.ts
 async function runPostRunLifecycle(inputs) {
   const {
     run,
@@ -4648,7 +4670,7 @@ async function runPostRunLifecycle(inputs) {
   await flushSessionWrites();
 }
 var init_post_run_lifecycle = __esm({
-  "src/internal/runtime/post-run-lifecycle.ts"() {
+  "src/internal/runtime/lifecycle/post-run-lifecycle.ts"() {
     init_session_summary_writer();
     init_memory_path_selector();
     init_agent_session();
@@ -4840,7 +4862,7 @@ async function readWorkspaceDir(root, errorCode, describe) {
   }
 }
 var init_workspace_dir = __esm({
-  "src/internal/runtime/workspace-dir.ts"() {
+  "src/internal/runtime/config/workspace-dir.ts"() {
     init_errors();
   }
 });
@@ -5171,7 +5193,7 @@ var init_pipeline = __esm({
   }
 });
 
-// src/internal/runtime/validate-agent-options.ts
+// src/internal/runtime/validation/validate-agent-options.ts
 function validateAgentOptions(options) {
   rejectProgrammaticHooks(options);
   ensureRuntimeShape(options);
@@ -5364,12 +5386,82 @@ function validateMemory(options) {
 }
 var TOOL_NAME_PATTERN, RESERVED_TOOL_NAMES;
 var init_validate_agent_options = __esm({
-  "src/internal/runtime/validate-agent-options.ts"() {
+  "src/internal/runtime/validation/validate-agent-options.ts"() {
     init_errors();
     TOOL_NAME_PATTERN = /^[a-zA-Z][a-zA-Z0-9_-]{0,63}$/;
     RESERVED_TOOL_NAMES = /* @__PURE__ */ new Set(["shell", "memory_search", "memory_get"]);
   }
 });
+
+// src/internal/runtime/config/providers-manager.ts
+function providerFromModel(model) {
+  if (model === void 0) return void 0;
+  const id = model.id;
+  if (id.includes(":")) return id.split(":")[0];
+  return void 0;
+}
+function resolveRoute(route, modelProvider, plugins) {
+  if (route.capability === "chat" && modelProvider === route.provider) {
+    const modelName = extractModelName(modelProvider, route);
+    const base = {
+      capability: route.capability,
+      provider: route.provider,
+      reason: "explicit-model-provider"
+    };
+    if (modelName !== void 0) base.model = modelName;
+    return base;
+  }
+  if (plugins?.enabled !== void 0 && plugins.enabled.length > 0) {
+    return {
+      capability: route.capability,
+      provider: route.provider,
+      reason: "first-available-plugin-provider"
+    };
+  }
+  return {
+    capability: route.capability,
+    provider: route.provider,
+    reason: "explicit-route"
+  };
+}
+function extractModelName(prefix, route) {
+  if (route.model !== void 0) return route.model;
+  return prefix === route.provider ? defaultModelForProvider(prefix) : void 0;
+}
+function defaultModelForProvider(provider) {
+  if (provider === "anthropic") return "claude-3-7-sonnet";
+  return void 0;
+}
+var ProvidersManagerImpl;
+var init_providers_manager = __esm({
+  "src/internal/runtime/config/providers-manager.ts"() {
+    ProvidersManagerImpl = class {
+      constructor(model, providers, plugins) {
+        this.model = model;
+        this.providers = providers;
+        this.plugins = plugins;
+      }
+      model;
+      providers;
+      plugins;
+      routes() {
+        const resolved = [];
+        const modelProvider = providerFromModel(this.model);
+        const seen = /* @__PURE__ */ new Set();
+        if (this.providers?.routes !== void 0) {
+          for (const route of this.providers.routes) {
+            const resolvedRoute = resolveRoute(route, modelProvider, this.plugins);
+            if (!seen.has(route.capability)) {
+              seen.add(route.capability);
+              resolved.push(resolvedRoute);
+            }
+          }
+        }
+        return Promise.resolve(resolved);
+      }
+    };
+  }
+});
 function truncateWithMarker(content, max) {
   if (content.length <= max) {
     return { truncated: false, finalContent: content };
@@ -6135,8 +6227,8 @@ var init_plugins_manager = __esm({
     init_errors();
     init_markdown_config_loader();
     init_path_guard();
-    init_hooks_source();
     init_workspace_dir();
+    init_hooks_source();
     init_plugin_frontmatter();
     PluginsManager = class {
       constructor(cwd, enabled, settingSourcesIncludePlugins, cloud, localPaths) {
@@ -6219,76 +6311,6 @@ var init_plugins_manager = __esm({
   }
 });
 
-// src/internal/runtime/providers-manager.ts
-function providerFromModel(model) {
-  if (model === void 0) return void 0;
-  const id = model.id;
-  if (id.includes(":")) return id.split(":")[0];
-  return void 0;
-}
-function resolveRoute(route, modelProvider, plugins) {
-  if (route.capability === "chat" && modelProvider === route.provider) {
-    const modelName = extractModelName(modelProvider, route);
-    const base = {
-      capability: route.capability,
-      provider: route.provider,
-      reason: "explicit-model-provider"
-    };
-    if (modelName !== void 0) base.model = modelName;
-    return base;
-  }
-  if (plugins?.enabled !== void 0 && plugins.enabled.length > 0) {
-    return {
-      capability: route.capability,
-      provider: route.provider,
-      reason: "first-available-plugin-provider"
-    };
-  }
-  return {
-    capability: route.capability,
-    provider: route.provider,
-    reason: "explicit-route"
-  };
-}
-function extractModelName(prefix, route) {
-  if (route.model !== void 0) return route.model;
-  return prefix === route.provider ? defaultModelForProvider(prefix) : void 0;
-}
-function defaultModelForProvider(provider) {
-  if (provider === "anthropic") return "claude-3-7-sonnet";
-  return void 0;
-}
-var ProvidersManagerImpl;
-var init_providers_manager = __esm({
-  "src/internal/runtime/providers-manager.ts"() {
-    ProvidersManagerImpl = class {
-      constructor(model, providers, plugins) {
-        this.model = model;
-        this.providers = providers;
-        this.plugins = plugins;
-      }
-      model;
-      providers;
-      plugins;
-      routes() {
-        const resolved = [];
-        const modelProvider = providerFromModel(this.model);
-        const seen = /* @__PURE__ */ new Set();
-        if (this.providers?.routes !== void 0) {
-          for (const route of this.providers.routes) {
-            const resolvedRoute = resolveRoute(route, modelProvider, this.plugins);
-            if (!seen.has(route.capability)) {
-              seen.add(route.capability);
-              resolved.push(resolvedRoute);
-            }
-          }
-        }
-        return Promise.resolve(resolved);
-      }
-    };
-  }
-});
-
 // src/internal/runtime/skills/skill-frontmatter.ts
 function asString(v) {
   return typeof v === "string" ? v : void 0;
@@ -6495,9 +6517,9 @@ function bootstrapSubmanagers(args) {
 }
 var init_local_agent_bootstrap = __esm({
   "src/internal/runtime/local-agent/local-agent-bootstrap.ts"() {
+    init_providers_manager();
     init_context_manager();
     init_plugins_manager();
-    init_providers_manager();
     init_agent_registry();
     init_skills_manager();
   }
@@ -6671,7 +6693,7 @@ var init_budget = __esm({
   }
 });
 
-// src/internal/runtime/validate-response.ts
+// src/internal/runtime/validation/validate-response.ts
 function validateResponse(response) {
   const trimmed = response.content.trim();
   const toolCallsArr = Array.isArray(response.toolCalls) ? response.toolCalls : [];
@@ -6684,7 +6706,7 @@ function validateResponse(response) {
   return { ok: true };
 }
 var init_validate_response = __esm({
-  "src/internal/runtime/validate-response.ts"() {
+  "src/internal/runtime/validation/validate-response.ts"() {
   }
 });
 
@@ -7107,11 +7129,80 @@ function checkToolWhitelist(toolName) {
 }
 var toolWhitelistStore;
 var init_async_local_storage = __esm({
-  "src/internal/runtime/async-local-storage.ts"() {
+  "src/internal/runtime/concurrency/async-local-storage.ts"() {
     toolWhitelistStore = new async_hooks.AsyncLocalStorage();
   }
 });
 
+// src/internal/runtime/concurrency/async-semaphore.ts
+function createSemaphore(permits) {
+  if (!Number.isInteger(permits) || permits < 1) {
+    throw new ConfigurationError(
+      `async-semaphore: permits must be a positive integer, got ${permits}`,
+      { code: "invalid_concurrency" }
+    );
+  }
+  let active = 0;
+  const queue = [];
+  function tryGrant() {
+    if (active < permits && queue.length > 0) {
+      const resolve3 = queue.shift();
+      if (resolve3 !== void 0) {
+        active += 1;
+        resolve3();
+      }
+    }
+  }
+  return {
+    inFlight: () => active,
+    pending: () => queue.length + active,
+    async acquire() {
+      await new Promise((resolve3) => {
+        queue.push(resolve3);
+        tryGrant();
+      });
+      let released = false;
+      return () => {
+        if (released) return;
+        released = true;
+        active -= 1;
+        tryGrant();
+      };
+    }
+  };
+}
+var init_async_semaphore = __esm({
+  "src/internal/runtime/concurrency/async-semaphore.ts"() {
+    init_errors();
+  }
+});
+
+// src/internal/runtime/concurrency/map-with-concurrency.ts
+async function mapWithConcurrency(items, concurrency, fn, options) {
+  const semaphore = createSemaphore(concurrency);
+  const signal = NEVER_ABORT;
+  return Promise.all(
+    items.map(async (item, index) => {
+      const release = await semaphore.acquire();
+      try {
+        if (signal.aborted) {
+          throw signal.reason instanceof Error ? signal.reason : new Error("mapWithConcurrency: aborted");
+        }
+        return await fn(item, index, signal);
+      } finally {
+        release();
+      }
+    })
+  );
+}
+var NEVER_ABORT;
+var init_map_with_concurrency = __esm({
+  "src/internal/runtime/concurrency/map-with-concurrency.ts"() {
+    init_async_semaphore();
+    NEVER_ABORT = new AbortController().signal;
+  }
+});
+
 // src/internal/tool-dispatch/repair-middleware.ts
 function repairToolCall(raw, registry) {
   const repairs = [];
@@ -7190,7 +7281,7 @@ var init_repair_middleware = __esm({
   }
 });
 
-// src/internal/runtime/shell-tool.ts
+// src/internal/runtime/tools/shell-tool.ts
 async function runShell(options) {
   if (options.sandbox === true && isObviouslyUnsafe(options.command)) {
     return {
@@ -7221,7 +7312,7 @@ function isObviouslyUnsafe(command) {
   return false;
 }
 var init_shell_tool = __esm({
-  "src/internal/runtime/shell-tool.ts"() {
+  "src/internal/runtime/tools/shell-tool.ts"() {
     init_spawn_collect();
   }
 });
@@ -7301,38 +7392,12 @@ var init_tool_executors = __esm({
 // src/internal/agent-loop/tool-dispatch.ts
 async function dispatchTools(inputs, tools, toolCalls, events) {
   const maxConcurrent = inputs.maxConcurrentTools ?? 4;
-  return boundedParallel(
-    maxConcurrent,
+  return mapWithConcurrency(
     toolCalls,
+    maxConcurrent,
     (call) => dispatchSingleCall(inputs, tools, call, events)
   );
 }
-async function boundedParallel(max, items, fn) {
-  let running = 0;
-  const queue = [];
-  async function acquire() {
-    if (running < max) {
-      running++;
-      return;
-    }
-    await new Promise((resolve3) => queue.push(resolve3));
-    running++;
-  }
-  function release() {
-    running--;
-    if (queue.length > 0) queue.shift()();
-  }
-  return Promise.all(
-    items.map(async (item) => {
-      await acquire();
-      try {
-        return await fn(item);
-      } finally {
-        release();
-      }
-    })
-  );
-}
 async function dispatchSingleCall(inputs, tools, call, events) {
   const { call: workingCall, repairs } = applyRepairAndExtractCall(tools, call);
   const callId = generateCallId();
@@ -7556,6 +7621,7 @@ var init_tool_dispatch = __esm({
   "src/internal/agent-loop/tool-dispatch.ts"() {
     init_ids();
     init_async_local_storage();
+    init_map_with_concurrency();
     init_repair_middleware();
     init_tool_executors();
   }
@@ -8538,7 +8604,7 @@ var init_providers = __esm({
   }
 });
 
-// src/internal/errors/mappers/shared.ts
+// src/internal/error-mappers/shared.ts
 function parseRetryAfter(headers) {
   if (headers === void 0) return void 0;
   const raw = headers.get("retry-after");
@@ -8567,13 +8633,13 @@ function buildErrorMetadata(args) {
 }
 var RAW_MAX_BYTES;
 var init_shared = __esm({
-  "src/internal/errors/mappers/shared.ts"() {
+  "src/internal/error-mappers/shared.ts"() {
     init_security();
     RAW_MAX_BYTES = 2048;
   }
 });
 
-// src/internal/errors/mappers/anthropic.ts
+// src/internal/error-mappers/anthropic.ts
 function mapAnthropicError(args) {
   const { status, body, headers, endpoint } = args;
   const code = mapAnthropicStatusToCode(status, body);
@@ -8627,7 +8693,7 @@ function formatMessage(status, code) {
   return `Anthropic API error: ${code} (HTTP ${status})`;
 }
 var init_anthropic2 = __esm({
-  "src/internal/errors/mappers/anthropic.ts"() {
+  "src/internal/error-mappers/anthropic.ts"() {
     init_errors();
     init_shared();
   }
@@ -8934,7 +9000,7 @@ var init_anthropic3 = __esm({
   }
 });
 
-// src/internal/errors/mappers/bedrock.ts
+// src/internal/error-mappers/bedrock.ts
 function classifyBedrockError(args, awsType, message) {
   if (args.status === 429 || awsType.includes("Throttling") || awsType.includes("TooManyRequests")) {
     return "rate_limit";
@@ -8994,7 +9060,7 @@ function parseBody(body) {
   return {};
 }
 var init_bedrock2 = __esm({
-  "src/internal/errors/mappers/bedrock.ts"() {
+  "src/internal/error-mappers/bedrock.ts"() {
     init_errors();
     init_shared();
   }
@@ -9381,7 +9447,7 @@ var init_credential_pool_context = __esm({
   }
 });
 
-// src/internal/errors/mappers/openai-compatible.ts
+// src/internal/error-mappers/openai-compatible.ts
 function mapOpenAICompatibleError(args) {
   const { providerId, status, body, headers, endpoint } = args;
   const code = mapOpenAiStatusToCode(status, body);
@@ -9444,7 +9510,7 @@ function formatMessage2(providerId, status, code) {
   return `${providerId} API error: ${code} (HTTP ${status})`;
 }
 var init_openai_compatible = __esm({
-  "src/internal/errors/mappers/openai-compatible.ts"() {
+  "src/internal/error-mappers/openai-compatible.ts"() {
     init_errors();
     init_shared();
   }
@@ -9556,7 +9622,7 @@ var init_fault_injection = __esm({
   }
 });
 
-// src/internal/errors/mappers/ollama.ts
+// src/internal/error-mappers/ollama.ts
 function mapOllamaTransportError(args) {
   if (args.providerId !== "ollama") return void 0;
   const causeCode = extractCauseCode(args.cause);
@@ -9629,7 +9695,7 @@ function extractModelName2(errString) {
   return match?.[1];
 }
 var init_ollama2 = __esm({
-  "src/internal/errors/mappers/ollama.ts"() {
+  "src/internal/error-mappers/ollama.ts"() {
     init_errors();
     init_shared();
   }
@@ -10230,7 +10296,7 @@ var init_pool_aware_client = __esm({
   }
 });
 
-// src/internal/errors/mappers/vertex.ts
+// src/internal/error-mappers/vertex.ts
 function classifyVertexError(args, errStatus) {
   for (const rule of VERTEX_RULES) {
     if (rule.test(args.status, errStatus)) return rule.code;
@@ -10267,7 +10333,7 @@ function parseBody2(body) {
 }
 var VERTEX_RULES, VERTEX_ERROR_BUILDERS;
 var init_vertex2 = __esm({
-  "src/internal/errors/mappers/vertex.ts"() {
+  "src/internal/error-mappers/vertex.ts"() {
     init_errors();
     init_shared();
     VERTEX_RULES = [
@@ -10827,6 +10893,21 @@ var init_client = __esm({
   }
 });
 
+// src/internal/providers/register-plugin-providers.ts
+function registerPluginProviderProfiles(entries) {
+  let registered4 = 0;
+  for (const entry of entries) {
+    registerProvider(entry.profile);
+    registered4 += 1;
+  }
+  return registered4;
+}
+var init_register_plugin_providers = __esm({
+  "src/internal/providers/register-plugin-providers.ts"() {
+    init_registry();
+  }
+});
+
 // src/internal/tool-registry/personality-filter.ts
 function applyPersonalityFilter(exposedTools, whitelist, opts) {
   if (whitelist === void 0) return exposedTools;
@@ -10907,12 +10988,26 @@ function createRealLocalRun(options) {
   registerRun(handle);
   return handle;
 }
-function buildLoopInputs(options, runId, userText) {
+function resolveRunProvider(options) {
   registerBuiltins();
+  const profiles = options.pluginManager?.aggregated.providerProfiles ?? [];
+  const registered4 = registerPluginProviderProfiles(profiles);
+  if (registered4 > 0 && !pluginProvidersAnnounced) {
+    pluginProvidersAnnounced = true;
+    const names = profiles.map((e) => e.profile.name).join(", ");
+    process.stderr.write(
+      `[theokit-sdk] registered ${registered4} plugin provider profile(s): ${names}
+`
+    );
+  }
   const parsedModel = parseModelId(options.model?.id);
   const inferredProvider = parsedModel.provider !== void 0 && getProviderProfile(parsedModel.provider) !== void 0 ? parsedModel.provider : void 0;
   const primary = options.agentOptions.providers?.routes?.[0]?.provider ?? inferredProvider ?? detectPrimaryProvider();
   const effectiveModelId = inferredProvider !== void 0 ? parsedModel.name : options.model?.id ?? "claude-sonnet-4-6";
+  return { primary, effectiveModelId };
+}
+function buildLoopInputs(options, runId, userText) {
+  const { primary, effectiveModelId } = resolveRunProvider(options);
   const fallback = options.agentOptions.providers?.fallback;
   const apiKeys = options.agentOptions.providers?.apiKeys;
   const credentialPoolStrategy = options.agentOptions.providers?.credentialPoolStrategy;
@@ -11006,7 +11101,7 @@ function buildMcpMap(options) {
   }
   return map;
 }
-var RealLocalRun;
+var pluginProvidersAnnounced, RealLocalRun;
 var init_real_local_run = __esm({
   "src/internal/runtime/local-agent/real-local-run.ts"() {
     init_loop();
@@ -11015,10 +11110,12 @@ var init_real_local_run = __esm({
     init_router();
     init_client();
     init_providers();
+    init_register_plugin_providers();
     init_tracer();
     init_personality_filter();
     init_fixture_run_base();
     init_run_registry();
+    pluginProvidersAnnounced = false;
     RealLocalRun = class extends FixtureRunBase {
       buildInputs;
       constructor(options, buildInputs) {
@@ -11570,7 +11667,7 @@ async function embedTexts(input) {
       pending
     });
   }
-  await runBatches(input, pending, results);
+  await embedInBoundedBatches(input, pending, results);
   return results.map((v) => v ?? new Array(dimension).fill(0));
 }
 function classifyEntry(args) {
@@ -11588,45 +11685,34 @@ function classifyEntry(args) {
   args.stats.cacheMisses += 1;
   args.pending.push({ index: args.index, text: args.text, key });
 }
-async function runBatches(input, pending, results) {
+async function embedInBoundedBatches(input, pending, results) {
   const batches = [];
   for (let offset = 0; offset < pending.length; offset += MAX_BATCH) {
     batches.push(pending.slice(offset, offset + MAX_BATCH));
   }
-  let running = 0;
-  const queue = [];
-  await Promise.all(batches.map((batch) => processBatch(input, batch, results, acquire, release)));
-  async function acquire() {
-    if (running >= MAX_CONCURRENT_BATCHES) await new Promise((r) => queue.push(r));
-    running++;
-  }
-  function release() {
-    running--;
-    if (queue.length > 0) queue.shift()();
-  }
+  await mapWithConcurrency(
+    batches,
+    MAX_CONCURRENT_BATCHES,
+    (batch) => processBatch(input, batch, results)
+  );
 }
-async function processBatch(input, batch, results, acquire, release) {
-  await acquire();
-  try {
-    const vectors = await embedBatch({
-      apiKey: input.apiKey,
-      baseUrl: input.baseUrl,
-      embeddingsPath: input.embeddingsPath,
-      model: input.model,
-      inputs: batch.map((b) => b.text),
-      fetchImpl: input.fetchImpl,
-      stats: input.stats,
-      providerId: input.providerId
-    });
-    for (let j = 0; j < batch.length; j++) {
-      const slot = batch[j];
-      const vector = vectors[j];
-      if (slot === void 0 || vector === void 0) continue;
-      results[slot.index] = vector;
-      input.cache.set(slot.key, vector);
-    }
-  } finally {
-    release();
+async function processBatch(input, batch, results) {
+  const vectors = await embedBatch({
+    apiKey: input.apiKey,
+    baseUrl: input.baseUrl,
+    embeddingsPath: input.embeddingsPath,
+    model: input.model,
+    inputs: batch.map((b) => b.text),
+    fetchImpl: input.fetchImpl,
+    stats: input.stats,
+    providerId: input.providerId
+  });
+  for (let j = 0; j < batch.length; j++) {
+    const slot = batch[j];
+    const vector = vectors[j];
+    if (slot === void 0 || vector === void 0) continue;
+    results[slot.index] = vector;
+    input.cache.set(slot.key, vector);
   }
 }
 async function embedBatch(opts) {
@@ -11700,6 +11786,7 @@ var init_openai_compatible2 = __esm({
   "src/internal/memory/adapters/openai-compatible.ts"() {
     init_errors();
     init_openai_compatible();
+    init_map_with_concurrency();
     init_embedding_cache();
     MAX_BATCH = 100;
     MAX_RETRIES = 2;
@@ -12184,6 +12271,61 @@ var init_sqlite_wal = __esm({
     warnedLabels = /* @__PURE__ */ new Set();
   }
 });
+async function openSqliteResilient(options) {
+  await promises.mkdir(path.dirname(options.filePath), { recursive: true });
+  try {
+    return await openConcrete(options);
+  } catch (cause) {
+    if (options.recoverCorrupt !== false && isCorruptionError(cause)) {
+      await renameAside(options.filePath, options.label ?? "sqlite");
+      return await openConcrete(options);
+    }
+    throw cause;
+  }
+}
+async function openConcrete(options) {
+  const db = await loadDriver(options.filePath);
+  applyWalWithFallback(db, options.label ?? "sqlite");
+  await options.onOpen?.(db);
+  return db;
+}
+async function loadDriver(filePath) {
+  try {
+    const mod = await import('better-sqlite3');
+    const Ctor = mod.default ?? mod;
+    if (typeof Ctor !== "function") {
+      throw new Error(`better-sqlite3 export is not a constructor (got ${typeof Ctor})`);
+    }
+    return new Ctor(filePath);
+  } catch (cause) {
+    const message = cause instanceof Error ? cause.message : String(cause);
+    throw new ConfigurationError(
+      `Failed to load SQLite driver. Install \`better-sqlite3\` or run on Node 22.5+ for built-in \`node:sqlite\`. Cause: ${message}`,
+      { code: "sqlite_driver_unavailable", cause }
+    );
+  }
+}
+function isCorruptionError(cause) {
+  if (!(cause instanceof Error)) return false;
+  const msg = cause.message.toLowerCase();
+  return msg.includes("malformed") || msg.includes("not a database") || msg.includes("encrypted") || msg.includes("disk image is malformed");
+}
+async function renameAside(filePath, label) {
+  const asidePath = `${filePath}.corrupt-${Date.now()}`;
+  await promises.rename(filePath, asidePath).catch(() => void 0);
+  await promises.rename(`${filePath}-wal`, `${asidePath}-wal`).catch(() => void 0);
+  await promises.rename(`${filePath}-shm`, `${asidePath}-shm`).catch(() => void 0);
+  process.stderr.write(
+    `[theokit-sdk] ${label} database corrupt; renamed aside to ${asidePath} and rebuilt schema
+`
+  );
+}
+var init_sqlite_open = __esm({
+  "src/internal/persistence/sqlite-open.ts"() {
+    init_errors();
+    init_sqlite_wal();
+  }
+});
 
 // src/internal/memory/index-schema.ts
 var SCHEMA_STATEMENTS, PRAGMA_STATEMENTS;
@@ -12231,60 +12373,22 @@ var init_index_schema = __esm({
   }
 });
 async function openMemoryDb(opts) {
-  await promises.mkdir(path.dirname(opts.filePath), { recursive: true });
-  try {
-    return await openConcrete(opts.filePath);
-  } catch (cause) {
-    if (opts.recoverCorrupt !== false && isCorruptionError(cause)) {
-      await renameAside(opts.filePath);
-      return await openConcrete(opts.filePath);
+  return openSqliteResilient({
+    filePath: opts.filePath,
+    label: "memory-index",
+    recoverCorrupt: opts.recoverCorrupt,
+    onOpen: (db) => {
+      for (const pragma of PRAGMA_STATEMENTS) db.exec(pragma);
+      for (const stmt of SCHEMA_STATEMENTS) db.exec(stmt);
     }
-    throw cause;
-  }
-}
-async function openConcrete(filePath) {
-  const db = await loadDriver(filePath);
-  applyWalWithFallback(db, "memory-index");
-  for (const pragma of PRAGMA_STATEMENTS) db.exec(pragma);
-  for (const stmt of SCHEMA_STATEMENTS) db.exec(stmt);
-  return db;
-}
-async function loadDriver(filePath) {
-  try {
-    const mod = await import('better-sqlite3');
-    const Ctor = mod.default ?? mod;
-    const db = new Ctor(filePath);
-    return db;
-  } catch (cause) {
-    const message = cause instanceof Error ? cause.message : String(cause);
-    throw new ConfigurationError(
-      `Failed to load SQLite driver. Install \`better-sqlite3\` or run on Node 22.5+ for built-in \`node:sqlite\`. Cause: ${message}`,
-      { code: "sqlite_driver_unavailable", cause }
-    );
-  }
-}
-function isCorruptionError(cause) {
-  if (!(cause instanceof Error)) return false;
-  const msg = cause.message.toLowerCase();
-  return msg.includes("malformed") || msg.includes("not a database") || msg.includes("encrypted") || msg.includes("disk image is malformed");
-}
-async function renameAside(filePath) {
-  const asidePath = `${filePath}.corrupt-${Date.now()}`;
-  await promises.rename(filePath, asidePath).catch(() => void 0);
-  await promises.rename(`${filePath}-wal`, `${asidePath}-wal`).catch(() => void 0);
-  await promises.rename(`${filePath}-shm`, `${asidePath}-shm`).catch(() => void 0);
-  process.stderr.write(
-    `[theokit-sdk] memory index corrupt; renamed aside to ${asidePath} and rebuilt schema
-`
-  );
+  });
 }
 function defaultIndexPath(cwd) {
   return path.join(cwd, ".theokit", "memory", ".index", "memory.sqlite");
 }
 var init_index_db = __esm({
   "src/internal/memory/index-db.ts"() {
-    init_errors();
-    init_sqlite_wal();
+    init_sqlite_open();
     init_index_schema();
   }
 });
@@ -13836,7 +13940,7 @@ var init_local_agent_plugins = __esm({
   }
 });
 
-// src/internal/runtime/run-until.ts
+// src/internal/runtime/lifecycle/run-until.ts
 var run_until_exports = {};
 __export(run_until_exports, {
   composeContinuation: () => composeContinuation,
@@ -13929,7 +14033,7 @@ Your last response was:
 ${lastResponse.slice(0, 1e3)}`;
 }
 var init_run_until = __esm({
-  "src/internal/runtime/run-until.ts"() {
+  "src/internal/runtime/lifecycle/run-until.ts"() {
   }
 });
 
@@ -14040,16 +14144,16 @@ var init_judge_call = __esm({
 // src/internal/runtime/registry/agent-factory-registry.ts
 var agent_factory_registry_exports = {};
 __export(agent_factory_registry_exports, {
-  getAgentCreate: () => getAgentCreate,
-  setAgentCreate: () => setAgentCreate
+  getAgentFacade: () => getAgentFacade,
+  setAgentFacade: () => setAgentFacade
 });
-function setAgentCreate(fn) {
-  registered3 = fn;
+function setAgentFacade(facade) {
+  registered3 = facade;
 }
-function getAgentCreate() {
+function getAgentFacade() {
   if (registered3 === void 0) {
     throw new Error(
-      "internal: Agent.create not registered. The `agent.ts` module must be loaded before LocalAgent.runUntil/fork are invoked."
+      "internal: Agent facade not registered. The `agent.ts` module must be loaded before internal subsystems (LocalAgent.runUntil/fork, eval, scorers, cron) invoke it."
     );
   }
   return registered3;
@@ -14060,7 +14164,7 @@ var init_agent_factory_registry = __esm({
   }
 });
 
-// src/internal/runtime/fork-agent.ts
+// src/internal/runtime/lifecycle/fork-agent.ts
 var fork_agent_exports = {};
 __export(fork_agent_exports, {
   filterMemoryPlugins: () => filterMemoryPlugins,
@@ -14120,7 +14224,7 @@ function extractUsage(result) {
   };
 }
 var init_fork_agent = __esm({
-  "src/internal/runtime/fork-agent.ts"() {
+  "src/internal/runtime/lifecycle/fork-agent.ts"() {
     init_async_local_storage();
     init_local_agent_plugins();
   }
@@ -14131,8 +14235,8 @@ function localAgentRunUntil(agent, goal, options) {
   async function* wrap() {
     const { runUntilImpl: runUntilImpl2 } = await Promise.resolve().then(() => (init_run_until(), run_until_exports));
     const { judgeCallImpl: judgeCallImpl2 } = await Promise.resolve().then(() => (init_judge_call(), judge_call_exports));
-    const { getAgentCreate: getAgentCreate2 } = await Promise.resolve().then(() => (init_agent_factory_registry(), agent_factory_registry_exports));
-    const create = getAgentCreate2();
+    const { getAgentFacade: getAgentFacade2 } = await Promise.resolve().then(() => (init_agent_factory_registry(), agent_factory_registry_exports));
+    const create = getAgentFacade2().create;
     const deps = {
       judge: async (ctx, opts) => judgeCallImpl2(ctx, opts, { create })
     };
@@ -14142,9 +14246,9 @@ function localAgentRunUntil(agent, goal, options) {
 }
 async function localAgentFork(parent, options) {
   const { forkAgentImpl: forkAgentImpl2 } = await Promise.resolve().then(() => (init_fork_agent(), fork_agent_exports));
-  const { getAgentCreate: getAgentCreate2 } = await Promise.resolve().then(() => (init_agent_factory_registry(), agent_factory_registry_exports));
+  const { getAgentFacade: getAgentFacade2 } = await Promise.resolve().then(() => (init_agent_factory_registry(), agent_factory_registry_exports));
   const { withPersonalityContext: withPersonalityContext2 } = await Promise.resolve().then(() => (init_context2(), context_exports));
-  const create = getAgentCreate2();
+  const create = getAgentFacade2().create;
   return withPersonalityContext2(
     { slug: parent.personalitySlugSnapshot, isFork: true },
     () => forkAgentImpl2(parent, options, { create })
@@ -14168,7 +14272,7 @@ var init_local_agent_runtime_extensions = __esm({
   }
 });
 
-// src/internal/runtime/abort-utils.ts
+// src/internal/runtime/concurrency/abort-utils.ts
 function anySignal(signals) {
   const filtered = signals.filter((s) => s !== void 0);
   if (filtered.length === 0) {
@@ -14194,7 +14298,7 @@ function ponyfillAny(signals) {
   return ctrl.signal;
 }
 var init_abort_utils = __esm({
-  "src/internal/runtime/abort-utils.ts"() {
+  "src/internal/runtime/concurrency/abort-utils.ts"() {
   }
 });
 
@@ -14347,49 +14451,6 @@ var init_task = __esm({
   }
 });
 
-// src/internal/runtime/async-semaphore.ts
-function createSemaphore(permits) {
-  if (!Number.isInteger(permits) || permits < 1) {
-    throw new ConfigurationError(
-      `async-semaphore: permits must be a positive integer, got ${permits}`,
-      { code: "invalid_concurrency" }
-    );
-  }
-  let active = 0;
-  const queue = [];
-  function tryGrant() {
-    if (active < permits && queue.length > 0) {
-      const resolve3 = queue.shift();
-      if (resolve3 !== void 0) {
-        active += 1;
-        resolve3();
-      }
-    }
-  }
-  return {
-    inFlight: () => active,
-    pending: () => queue.length + active,
-    async acquire() {
-      await new Promise((resolve3) => {
-        queue.push(resolve3);
-        tryGrant();
-      });
-      let released = false;
-      return () => {
-        if (released) return;
-        released = true;
-        active -= 1;
-        tryGrant();
-      };
-    }
-  };
-}
-var init_async_semaphore = __esm({
-  "src/internal/runtime/async-semaphore.ts"() {
-    init_errors();
-  }
-});
-
 // src/internal/task/ring-buffer.ts
 var RingBuffer;
 var init_ring_buffer = __esm({
@@ -14952,8 +15013,8 @@ var init_local_agent = __esm({
     init_span_names();
     init_tracer();
     init_hooks_executor();
-    init_memory_path_selector();
     init_post_run_lifecycle();
+    init_memory_path_selector();
     init_agent_registry();
     init_live_agent_registry();
     init_agent_session();
@@ -16422,7 +16483,13 @@ var init_agent = __esm({
         await flushRegistrySaves();
       }
     };
-    setAgentCreate((options) => Agent.create(options));
+    setAgentFacade({
+      create: (options) => Agent.create(options),
+      prompt: (message, options) => Agent.prompt(message, options),
+      get: (agentId) => Agent.get(agentId),
+      resume: (agentId, options) => Agent.resume(agentId, options),
+      batch: (prompts, options) => Agent.batch(prompts, options)
+    });
   }
 });