@theokit/sdk 5.15.2 → 5.16.0

package/dist/index.cjs

@@ -10,25 +10,25 @@ var chunk4JF52NED_cjs = require('./chunk-4JF52NED.cjs');
 require('./chunk-D3BL3EEB.cjs');
 var chunkERBKPA2N_cjs = require('./chunk-ERBKPA2N.cjs');
 require('./chunk-VTYY7XL5.cjs');
-var chunkVT3NWHFH_cjs = require('./chunk-VT3NWHFH.cjs');
-var chunkBTGYRM4J_cjs = require('./chunk-BTGYRM4J.cjs');
+var chunkTDKLSWHU_cjs = require('./chunk-TDKLSWHU.cjs');
+var chunkBJB6RDEA_cjs = require('./chunk-BJB6RDEA.cjs');
+require('./chunk-BV2MWEMV.cjs');
 var chunkER3BF23R_cjs = require('./chunk-ER3BF23R.cjs');
-require('./chunk-PDFSLD5P.cjs');
+require('./chunk-3JGRKXA7.cjs');
 require('./chunk-PNL63WDP.cjs');
+var chunkEOWIUDKG_cjs = require('./chunk-EOWIUDKG.cjs');
 require('./chunk-HI5TJWQE.cjs');
 require('./chunk-NDXZXGVP.cjs');
-require('./chunk-BV2MWEMV.cjs');
 require('./chunk-DEHPRORL.cjs');
 require('./chunk-GHX4P3V2.cjs');
 require('./chunk-EBX6XUN7.cjs');
 var chunkKC2HKUPT_cjs = require('./chunk-KC2HKUPT.cjs');
 var chunkN2ORTDSO_cjs = require('./chunk-N2ORTDSO.cjs');
-var chunk6XT5LAZG_cjs = require('./chunk-6XT5LAZG.cjs');
+var chunkBXEHUUDJ_cjs = require('./chunk-BXEHUUDJ.cjs');
 require('./chunk-MJGY3E3C.cjs');
 require('./chunk-YJNYDJ2T.cjs');
 require('./chunk-3F53FLQ7.cjs');
 require('./chunk-QOGNZMHD.cjs');
-var chunkEOWIUDKG_cjs = require('./chunk-EOWIUDKG.cjs');
 require('./chunk-2ZPEDVLM.cjs');
 require('./chunk-XR6O6G3B.cjs');
 require('./chunk-X4ZYEZPD.cjs');
@@ -61,15 +61,17 @@ require('./chunk-FZSPVIFK.cjs');
 var chunkJVH3NNJ3_cjs = require('./chunk-JVH3NNJ3.cjs');
 require('./chunk-3YCNBOX7.cjs');
 var chunkKX5QXLBS_cjs = require('./chunk-KX5QXLBS.cjs');
-require('./chunk-Q4UOCMAO.cjs');
+var chunk22HVMMXX_cjs = require('./chunk-22HVMMXX.cjs');
+require('./chunk-S5WTJ6LL.cjs');
 var chunk3EE6LVWT_cjs = require('./chunk-3EE6LVWT.cjs');
-require('./chunk-EDIYR4VR.cjs');
-var chunkZ5MB2HAN_cjs = require('./chunk-Z5MB2HAN.cjs');
+require('./chunk-SGKMTBCO.cjs');
+var chunkWVQRE7CO_cjs = require('./chunk-WVQRE7CO.cjs');
 require('./chunk-NVXXWAU7.cjs');
+require('./chunk-GA4ZRKYF.cjs');
 require('./chunk-UMCCYXFZ.cjs');
 require('./chunk-PXYINWBC.cjs');
 require('./chunk-IOFMRS6T.cjs');
-require('./chunk-EHD25JBZ.cjs');
+var chunkEHD25JBZ_cjs = require('./chunk-EHD25JBZ.cjs');
 var chunkCEGU5AVD_cjs = require('./chunk-CEGU5AVD.cjs');
 var chunkRNXXYF22_cjs = require('./chunk-RNXXYF22.cjs');
 var node_crypto = require('node:crypto');
@@ -81,8 +83,8 @@ var node_os = require('node:os');
 // src/agent-factory.ts
 function createAgentFactory(common) {
   return {
-    forSession: (agentId, overrides) => chunkBTGYRM4J_cjs.Agent.create(mergeAgentOptions(common, overrides, agentId)),
-    getOrCreate: (agentId, overrides) => chunkBTGYRM4J_cjs.Agent.getOrCreate(agentId, mergeAgentOptions(common, overrides, agentId))
+    forSession: (agentId, overrides) => chunkBJB6RDEA_cjs.Agent.create(mergeAgentOptions(common, overrides, agentId)),
+    getOrCreate: (agentId, overrides) => chunkBJB6RDEA_cjs.Agent.getOrCreate(agentId, mergeAgentOptions(common, overrides, agentId))
   };
 }
 function mergeAgentOptions(common, overrides, agentId) {
@@ -548,6 +550,94 @@ function describeCredential(input) {
     fingerprint: node_crypto.createHash("sha256").update(secret).digest("hex").slice(0, 8)
   };
 }
+var DecisionChecksumError = class extends chunkCEGU5AVD_cjs.TheokitAgentError {
+  constructor(modelPath, expected, actual) {
+    super(`decide: checksum mismatch for ${modelPath} (expected ${expected}, got ${actual})`, {
+      code: "decision_checksum_mismatch",
+      isRetryable: false
+    });
+    this.modelPath = modelPath;
+    this.expected = expected;
+    this.actual = actual;
+  }
+  modelPath;
+  expected;
+  actual;
+  name = "DecisionChecksumError";
+};
+var sessionCache = /* @__PURE__ */ new Map();
+async function defaultLoadSession(model) {
+  const name = "onnxruntime-node";
+  let ort;
+  try {
+    ort = await import(
+      /* @vite-ignore */
+      name
+    );
+  } catch {
+    throw new Error("decide: install `onnxruntime-node` to run a local classifier");
+  }
+  const session = await ort.InferenceSession.create(model);
+  const tensor = (a) => new ort.Tensor("int64", BigInt64Array.from(a, BigInt), [1, a.length]);
+  return {
+    async run(t) {
+      const feeds = {
+        input_ids: tensor(t.inputIds),
+        attention_mask: tensor(t.attentionMask ?? t.inputIds.map(() => 1))
+      };
+      const out = await session.run(feeds);
+      const first = Object.values(out)[0];
+      if (!first) throw new Error("decide: model returned no output");
+      return Array.from(first.data, Number);
+    }
+  };
+}
+function softmax(logits) {
+  const max = Math.max(...logits);
+  const exps = logits.map((x) => Math.exp(x - max));
+  const sum = exps.reduce((a, b) => a + b, 0);
+  return exps.map((e) => e / sum);
+}
+async function decide(opts) {
+  if (opts.labels.length === 0) throw new Error("decide: labels must not be empty");
+  if (new Set(opts.labels).size !== opts.labels.length) {
+    throw new Error("decide: labels must be unique");
+  }
+  if (!/^[0-9a-f]{64}$/i.test(opts.sha256)) {
+    throw new Error("decide: sha256 must be 64 hex characters");
+  }
+  const bytes = await promises.readFile(opts.modelPath);
+  const actual = node_crypto.createHash("sha256").update(bytes).digest("hex");
+  const expected = opts.sha256.toLowerCase();
+  if (actual !== expected) throw new DecisionChecksumError(opts.modelPath, expected, actual);
+  const load = () => (opts.loadSession ?? defaultLoadSession)(new Uint8Array(bytes));
+  let session;
+  if (opts.loadSession) {
+    session = await load();
+  } else {
+    let cached = sessionCache.get(actual);
+    if (!cached) {
+      cached = load();
+      sessionCache.set(actual, cached);
+      cached.catch(() => sessionCache.delete(actual));
+    }
+    session = await cached;
+  }
+  const logits = await session.run(await opts.tokenize(opts.text));
+  if (logits.length !== opts.labels.length) {
+    throw new Error(
+      `decide: model returned ${logits.length} scores for ${opts.labels.length} labels`
+    );
+  }
+  const probs = softmax(logits);
+  const scores = {};
+  let best = 0;
+  opts.labels.forEach((l, i) => {
+    scores[l] = probs[i];
+    if (probs[i] > probs[best]) best = i;
+  });
+  return { label: opts.labels[best], scores, source: "local-onnx", modelSha256: actual };
+}
 
 // src/define-provider.ts
 function defineProvider(profile, opts) {
@@ -573,8 +663,8 @@ var Provider = class _Provider {
    * `.plugins(Provider.builtins())` once, then pick a `provider/model` id. @public
    */
   static builtins() {
-    chunkZ5MB2HAN_cjs.registerBuiltins();
-    return chunkZ5MB2HAN_cjs.listProviders().map((profile) => defineProvider(profile));
+    chunkWVQRE7CO_cjs.registerBuiltins();
+    return chunkWVQRE7CO_cjs.listProviders().map((profile) => defineProvider(profile));
   }
   /**
    * The builtin serving `modelId`, or `undefined` when none does.
@@ -760,6 +850,87 @@ var EventBus = class {
   }
 };
 
+// src/generate-image.ts
+function sniffMime(b) {
+  const at = (i, ...v) => v.every((x, k) => b[i + k] === x);
+  if (at(0, 137, 80, 78, 71, 13, 10, 26, 10)) return "image/png";
+  if (at(0, 255, 216, 255)) return "image/jpeg";
+  if (at(0, 82, 73, 70, 70) && at(8, 87, 69, 66, 80)) return "image/webp";
+  return "application/octet-stream";
+}
+function parseSseLine(line) {
+  if (!line.startsWith("data:")) return void 0;
+  try {
+    return JSON.parse(line.slice(5));
+  } catch {
+    return void 0;
+  }
+}
+function imageFromSse(sse) {
+  for (const line of sse.split("\n")) {
+    const ev = parseSseLine(line);
+    const done = ev?.type === "response.output_item.done" && ev.item?.type === "image_generation_call";
+    if (done && ev?.item?.result) return ev.item.result;
+  }
+  return void 0;
+}
+async function assertOk(res) {
+  if (res.ok) return;
+  const text = await res.text().catch(() => "");
+  throw new Error(`generateImage: provider returned ${res.status} ${text.slice(0, 300)}`);
+}
+async function generateViaChatGpt(opts) {
+  const { codexFetch } = await import('./openai-chatgpt-IVAL4MW6.cjs');
+  const base = chunk22HVMMXX_cjs.trimTrailingSlashes(opts.baseUrl ?? "https://chatgpt.com/backend-api/codex");
+  const res = await (opts.fetch ?? codexFetch())(`${base}/responses`, {
+    method: "POST",
+    headers: { "content-type": "application/json", accept: "text/event-stream" },
+    body: JSON.stringify({
+      model: opts.model ?? "gpt-5.4",
+      instructions: "Generate the requested image with the image_generation tool.",
+      input: [{ role: "user", content: [{ type: "input_text", text: opts.prompt }] }],
+      tools: [{ type: "image_generation" }],
+      tool_choice: { type: "image_generation" },
+      store: false,
+      stream: true
+    }),
+    signal: opts.signal
+  });
+  await assertOk(res);
+  const b64 = imageFromSse(await res.text());
+  if (b64 === void 0) throw new Error("generateImage: provider returned no image data");
+  return b64;
+}
+async function generateImage(opts) {
+  if (opts.route === "chatgpt") {
+    const image2 = new Uint8Array(Buffer.from(await generateViaChatGpt(opts), "base64"));
+    return { image: image2, mimeType: sniffMime(image2) };
+  }
+  const apiKey = opts.apiKey ?? process.env.OPENAI_API_KEY;
+  if (!apiKey) throw new Error("generateImage: no API key (pass apiKey or set OPENAI_API_KEY)");
+  const base = chunk22HVMMXX_cjs.trimTrailingSlashes(opts.baseUrl ?? "https://api.openai.com/v1");
+  const body = {
+    model: opts.model ?? "gpt-image-1",
+    prompt: opts.prompt,
+    n: 1
+  };
+  if (opts.size !== void 0) body.size = opts.size;
+  const res = await chunkEHD25JBZ_cjs.boundFetch(opts.fetch)(`${base}/images/generations`, {
+    method: "POST",
+    headers: { "content-type": "application/json", authorization: `Bearer ${apiKey}` },
+    body: JSON.stringify(body),
+    signal: opts.signal
+  });
+  await assertOk(res);
+  const json = await res.json();
+  const b64 = json.data?.[0]?.b64_json;
+  if (typeof b64 !== "string" || b64.length === 0) {
+    throw new Error("generateImage: provider returned no image data");
+  }
+  const image = new Uint8Array(Buffer.from(b64, "base64"));
+  return { image, mimeType: sniffMime(image) };
+}
+
 // src/goal-loop.ts
 function runGoalLoop(agent, goal, options, depsOverride) {
   async function* wrap() {
@@ -1100,7 +1271,7 @@ var Memory = {
       return await peer.IndexManager.open(openArgs);
     }
     const { IndexManager } = await import('./index-manager-CT6MMG7S.cjs');
-    openArgs.embedding = await resolveEmbedding(opts.embedding, chunkBTGYRM4J_cjs.MEMORY_EMBEDDING_ADAPTERS);
+    openArgs.embedding = await resolveEmbedding(opts.embedding, chunkBJB6RDEA_cjs.MEMORY_EMBEDDING_ADAPTERS);
     return await IndexManager.open(openArgs);
   },
   /**
@@ -1111,7 +1282,7 @@ var Memory = {
    */
   async runDreamingSweep(opts) {
     const peer = await tryLoadSdkMemoryPeer();
-    const catalog = peer !== null ? peer.MEMORY_EMBEDDING_ADAPTERS : chunkBTGYRM4J_cjs.MEMORY_EMBEDDING_ADAPTERS;
+    const catalog = peer !== null ? peer.MEMORY_EMBEDDING_ADAPTERS : chunkBJB6RDEA_cjs.MEMORY_EMBEDDING_ADAPTERS;
     const sweepArgs = await buildDreamingSweepArgs(opts, catalog);
     const result = peer !== null ? (
       // biome-ignore lint/suspicious/noExplicitAny: peer is dynamically loaded — structural compat ensured by sdk-memory
@@ -1722,7 +1893,7 @@ function guardSessionDestruction(sessionId, live) {
 // src/session-messages.ts
 async function readSessionMessages2(options) {
   const cwd = options.cwd ?? process.cwd();
-  const baseDir = chunkBTGYRM4J_cjs.resolveSessionDir({ sessionDir: options.sessionDir });
+  const baseDir = chunkBJB6RDEA_cjs.resolveSessionDir({ sessionDir: options.sessionDir });
   return chunkEOWIUDKG_cjs.readSessionMessages(new chunkN2ORTDSO_cjs.FsSessionStore({ baseDir, cwd }), options.sessionId);
 }
 
@@ -1739,7 +1910,7 @@ function isBuiltAgent(m) {
   return typeof m.send === "function";
 }
 async function materialize(def, index) {
-  const { Agent: Agent2 } = await import('./agent-D6WBBNGR.cjs');
+  const { Agent: Agent2 } = await import('./agent-TM634NHQ.cjs');
   return Agent2.create({
     // `AgentDefinition.model` admits the sentinel `'inherit'`, which is not a model id. Inheriting
     // here means "declare nothing and let the default apply" — forwarding the literal would create
@@ -1881,7 +2052,7 @@ var FIXTURE_USER = {
 };
 var FIXTURE_MODELS = [
   {
-    id: chunkBTGYRM4J_cjs.DEFAULT_AGENTIC_MODEL_ID,
+    id: chunkBJB6RDEA_cjs.DEFAULT_AGENTIC_MODEL_ID,
     name: "Gemini 2.0 Flash",
     displayName: "Gemini 2.0 Flash",
     parameters: [
@@ -1957,7 +2128,7 @@ async function listLocalModelsViaOpenAiCompat(baseUrl) {
   try {
     response = await fetch(url, { method: "GET" });
   } catch (fetchErr) {
-    const mapped = chunk6XT5LAZG_cjs.mapOllamaTransportError({
+    const mapped = chunkBXEHUUDJ_cjs.mapOllamaTransportError({
       providerId: "ollama",
       cause: fetchErr,
       endpoint: "/v1/models"
@@ -1969,8 +2140,8 @@ async function listLocalModelsViaOpenAiCompat(baseUrl) {
     );
   }
   if (!response.ok) {
-    const body = await chunk6XT5LAZG_cjs.readErrorResponseBody(response);
-    const mapped = chunk6XT5LAZG_cjs.mapOllamaHttpError({
+    const body = await chunkBXEHUUDJ_cjs.readErrorResponseBody(response);
+    const mapped = chunkBXEHUUDJ_cjs.mapOllamaHttpError({
       providerId: "ollama",
       status: response.status,
       body,
@@ -2033,9 +2204,9 @@ var Theokit = class {
       });
     },
     capabilities: (providerOrModelId) => {
-      chunkZ5MB2HAN_cjs.registerBuiltins();
+      chunkWVQRE7CO_cjs.registerBuiltins();
       const providerId = providerOrModelId.includes("/") ? providerOrModelId.split("/")[0] ?? providerOrModelId : providerOrModelId;
-      return chunkZ5MB2HAN_cjs.getCatalogCapabilities(providerId);
+      return chunkWVQRE7CO_cjs.getCatalogCapabilities(providerId);
     }
   };
   /**
@@ -2074,8 +2245,8 @@ var Theokit = class {
    */
   static inspect = {
     builtinProviders: () => {
-      chunkZ5MB2HAN_cjs.registerBuiltins();
-      return chunkZ5MB2HAN_cjs.listProviders().map((p) => ({
+      chunkWVQRE7CO_cjs.registerBuiltins();
+      return chunkWVQRE7CO_cjs.listProviders().map((p) => ({
         name: p.name,
         apiMode: p.apiMode,
         authType: p.authType,
@@ -2084,7 +2255,7 @@ var Theokit = class {
         envVars: p.envVars
       }));
     },
-    embeddingAdapters: () => Object.entries(chunkBTGYRM4J_cjs.MEMORY_EMBEDDING_ADAPTERS).map(([id, adapter]) => ({
+    embeddingAdapters: () => Object.entries(chunkBJB6RDEA_cjs.MEMORY_EMBEDDING_ADAPTERS).map(([id, adapter]) => ({
       id,
       transport: adapter.transport,
       defaultModel: adapter.defaultModel
@@ -2101,9 +2272,9 @@ var Theokit = class {
   // and may be promoted onto Theokit once the agent.ts cycle is broken.
 };
 async function maybeListLocalModels(providerName) {
-  chunkZ5MB2HAN_cjs.registerBuiltins();
-  await chunk6XT5LAZG_cjs.discoverProviderPlugins();
-  const profile = chunkZ5MB2HAN_cjs.getProviderProfile(providerName);
+  chunkWVQRE7CO_cjs.registerBuiltins();
+  await chunkBXEHUUDJ_cjs.discoverProviderPlugins();
+  const profile = chunkWVQRE7CO_cjs.getProviderProfile(providerName);
   if (profile === void 0) return void 0;
   if (profile.authType !== "none") return void 0;
   const baseUrl = resolveLocalProviderBaseUrl(profile.name, profile.baseUrl);
@@ -2120,13 +2291,13 @@ async function executeCatalogRequest(request) {
   if (apiKey === void 0) {
     throw new chunkCEGU5AVD_cjs.AuthenticationError("Missing API key", { code: "missing_api_key" });
   }
-  if (chunk6XT5LAZG_cjs.shouldUseFixtureMode(apiKey)) {
+  if (chunkBXEHUUDJ_cjs.shouldUseFixtureMode(apiKey)) {
     return request.fixture;
   }
-  if (!chunk6XT5LAZG_cjs.isFixtureApiKey(apiKey) && process.env.THEOKIT_API_BASE_URL === void 0) {
+  if (!chunkBXEHUUDJ_cjs.isFixtureApiKey(apiKey) && process.env.THEOKIT_API_BASE_URL === void 0) {
     throw new chunkCEGU5AVD_cjs.AuthenticationError("Invalid API key", { code: "authentication_error" });
   }
-  return chunk6XT5LAZG_cjs.httpRequest(request.path, { apiKey });
+  return chunkBXEHUUDJ_cjs.httpRequest(request.path, { apiKey });
 }
 
 // src/tool-blast-radius.ts
@@ -2307,43 +2478,43 @@ Object.defineProperty(exports, "SqliteSessionStore", {
 });
 Object.defineProperty(exports, "Cron", {
   enumerable: true,
-  get: function () { return chunkVT3NWHFH_cjs.Cron; }
+  get: function () { return chunkTDKLSWHU_cjs.Cron; }
 });
 Object.defineProperty(exports, "Agent", {
   enumerable: true,
-  get: function () { return chunkBTGYRM4J_cjs.Agent; }
+  get: function () { return chunkBJB6RDEA_cjs.Agent; }
 });
 Object.defineProperty(exports, "AgentBuilder", {
   enumerable: true,
-  get: function () { return chunkBTGYRM4J_cjs.AgentBuilder; }
+  get: function () { return chunkBJB6RDEA_cjs.AgentBuilder; }
 });
 Object.defineProperty(exports, "Tool", {
   enumerable: true,
-  get: function () { return chunkBTGYRM4J_cjs.Tool; }
+  get: function () { return chunkBJB6RDEA_cjs.Tool; }
 });
 Object.defineProperty(exports, "ToolError", {
   enumerable: true,
-  get: function () { return chunkBTGYRM4J_cjs.ToolError; }
+  get: function () { return chunkBJB6RDEA_cjs.ToolError; }
 });
 Object.defineProperty(exports, "UsageAccumulator", {
   enumerable: true,
-  get: function () { return chunkBTGYRM4J_cjs.UsageAccumulator; }
+  get: function () { return chunkBJB6RDEA_cjs.UsageAccumulator; }
 });
 Object.defineProperty(exports, "computeCost", {
   enumerable: true,
-  get: function () { return chunkBTGYRM4J_cjs.computeCost; }
+  get: function () { return chunkBJB6RDEA_cjs.computeCost; }
 });
 Object.defineProperty(exports, "getPricingEntry", {
   enumerable: true,
-  get: function () { return chunkBTGYRM4J_cjs.getPricingEntry; }
+  get: function () { return chunkBJB6RDEA_cjs.getPricingEntry; }
 });
 Object.defineProperty(exports, "managedSettingsPathFor", {
   enumerable: true,
-  get: function () { return chunkBTGYRM4J_cjs.managedSettingsPathFor; }
+  get: function () { return chunkBJB6RDEA_cjs.managedSettingsPathFor; }
 });
 Object.defineProperty(exports, "readManagedSettings", {
   enumerable: true,
-  get: function () { return chunkBTGYRM4J_cjs.readManagedSettings; }
+  get: function () { return chunkBJB6RDEA_cjs.readManagedSettings; }
 });
 Object.defineProperty(exports, "emitRunEvent", {
   enumerable: true,
@@ -2351,11 +2522,11 @@ Object.defineProperty(exports, "emitRunEvent", {
 });
 Object.defineProperty(exports, "inferApiMode", {
   enumerable: true,
-  get: function () { return chunk6XT5LAZG_cjs.inferApiMode; }
+  get: function () { return chunkBXEHUUDJ_cjs.inferApiMode; }
 });
 Object.defineProperty(exports, "normalizeUsage", {
   enumerable: true,
-  get: function () { return chunk6XT5LAZG_cjs.normalizeUsage; }
+  get: function () { return chunkBXEHUUDJ_cjs.normalizeUsage; }
 });
 Object.defineProperty(exports, "withCwdMutex", {
   enumerable: true,
@@ -2447,6 +2618,7 @@ Object.defineProperty(exports, "setDiagnosticsSink", {
 });
 exports.AgentFactory = AgentFactory;
 exports.Budget = Budget;
+exports.DecisionChecksumError = DecisionChecksumError;
 exports.EventBus = EventBus;
 exports.JobQueue = JobQueue;
 exports.LayerOrderError = LayerOrderError;
@@ -2475,6 +2647,7 @@ exports.chargeAndCheckThresholds = chargeAndCheckThresholds;
 exports.createCounterBudgetTracker = createCounterBudgetTracker;
 exports.createTokenLimiter = createTokenLimiter;
 exports.createUnicodeNormalizer = createUnicodeNormalizer;
+exports.decide = decide;
 exports.decideApproval = decideApproval;
 exports.describeAction = describeAction;
 exports.describeCredential = describeCredential;
@@ -2482,6 +2655,7 @@ exports.effectiveToolNames = effectiveToolNames;
 exports.evaluateBlastRadius = evaluateBlastRadius;
 exports.extractRawId = extractRawId;
 exports.foldLayers = foldLayers;
+exports.generateImage = generateImage;
 exports.guardSessionDestruction = guardSessionDestruction;
 exports.isInside = isInside;
 exports.loadProjectEnv = loadProjectEnv;