@mindstudio-ai/agent 0.1.137 → 0.1.139

package/dist/index.js

@@ -3131,6 +3131,8 @@ function createFiles(call) {
 
 // src/datasources/source.ts
 import { createHash as createHash2 } from "crypto";
+var DICTIONARY_PAGE_ENTRIES = 1e4;
+var DICTIONARY_PAGE_BYTES = 4e6;
 var REMOVE_WHERE_MAX_PAGES = 100;
 var DataSource = class {
   constructor(_slug, _call) {
@@ -3251,6 +3253,120 @@ var DataSource = class {
     });
     return { chunks: chunks ?? 0, latencyMs: latencyMs ?? 0 };
   }
+  /**
+   * Which entities the passages matching a filter are about, most first.
+   *
+   * Filter on one entity (`{ entities: ['P-101A'] }`) and the answer is what
+   * appears beside it: a co-mention graph for free. Filter on a date range and
+   * it is who or what was written about then. No filter covers the corpus.
+   * `chunks` is how many matching passages carry each id, not documents: a
+   * long document's passages each count. Approximate counts, ordered most
+   * first; `limit` defaults to 20, at most 100.
+   *
+   * Shares the search rate limit (300/minute per app), and shows the same
+   * `index_warming` / `index_building` as search while a corpus loads.
+   */
+  async entities(options) {
+    const { entities, latencyMs } = await this._call("entities", {
+      slug: this._slug,
+      ...options?.filter !== void 0 ? { filter: options.filter } : {},
+      ...options?.limit !== void 0 ? { limit: options.limit } : {}
+    });
+    return { entities: entities ?? [], latencyMs: latencyMs ?? 0 };
+  }
+  /**
+   * Push this source's dictionary: every entity id you have and the names it
+   * goes by, replacing the dictionary it had. From then on every passage is
+   * tagged with the ids whose names its text mentions, beside the ones its
+   * document asserts, so `filter.entities` and {@link entities} find a tag
+   * wherever it's written. Passages indexed before are re-tagged in the
+   * background (a relink; see {@link dictionary}), and nothing is
+   * re-embedded.
+   *
+   * Push it whole after each load of the tables it comes from: an unchanged
+   * dictionary commits nothing (`unchanged: true`). Takes an array or any
+   * (async) iterable, and sends it in pages, so a register of a quarter of a
+   * million entries is fine. A repeated id merges its names.
+   */
+  async pushDictionary(entries, options) {
+    const { stageId, pageMaxEntries } = await this._call("dictionary/stage", {
+      slug: this._slug,
+      ...options?.linkerVersion !== void 0 ? { linkerVersion: options.linkerVersion } : {}
+    });
+    const perPage = Math.min(
+      DICTIONARY_PAGE_ENTRIES,
+      Number(pageMaxEntries) || DICTIONARY_PAGE_ENTRIES
+    );
+    let pages = 0;
+    let page = [];
+    let bytes = 0;
+    const send = async () => {
+      if (page.length === 0) {
+        return;
+      }
+      await this._call("dictionary/page", {
+        slug: this._slug,
+        stageId,
+        page: pages,
+        entries: page
+      });
+      pages += 1;
+      page = [];
+      bytes = 0;
+    };
+    for await (const entry of entries) {
+      const size = JSON.stringify(entry).length;
+      if (page.length >= perPage || bytes + size > DICTIONARY_PAGE_BYTES) {
+        await send();
+      }
+      page.push(entry);
+      bytes += size;
+    }
+    await send();
+    if (pages === 0) {
+      throw new Error(
+        "pushDictionary needs at least one entry. To clear what a dictionary tags, push its entries with the names you mean."
+      );
+    }
+    return this._call("dictionary/commit", {
+      slug: this._slug,
+      stageId,
+      pages
+    });
+  }
+  /**
+   * The source's current dictionary, or null when none was pushed, and the
+   * relink re-tagging its passages while one runs.
+   */
+  async dictionary() {
+    const { dictionary, relink } = await this._call("dictionary", {
+      slug: this._slug
+    });
+    return {
+      dictionary: dictionary ?? null,
+      relink: relink ? {
+        version: relink.version,
+        lap: relink.lap,
+        walked: relink.walked,
+        total: relink.total,
+        waiting: relink.waiting ?? null,
+        error: relink.error ?? null
+      } : null
+    };
+  }
+  /**
+   * What the current dictionary finds in a piece of text: each mention, in
+   * order, with the ids it tags. How to check a dictionary before relying on
+   * it. `version` is null when the source has none. Shares the search rate
+   * limit.
+   */
+  async link(text) {
+    const { version, mentions } = await this._call("link", {
+      slug: this._slug,
+      text
+    });
+    return { version: version ?? null, mentions: mentions ?? [] };
+  }
   /**
    * Every chunk of one document, exactly as it was indexed.
    *
@@ -3302,6 +3418,7 @@ var DataSource = class {
       filename: options.filename,
       ...options.contentType ? { contentType: options.contentType } : {},
       ...options.metadata !== void 0 ? { metadata: options.metadata } : {},
+      ...options.entities !== void 0 ? { entities: options.entities } : {},
       body: bytes.toString("base64")
     });
   }
@@ -4989,7 +5106,7 @@ function logTaskResult(result) {
   }
   const toolSummary = result.toolCalls.map((tc) => `${tc.name} (${tc.durationMs}ms) ${tc.success ? "\u2713" : "\u2717"}`).join(", ");
   console.log(
-    `[task] ${result.turns} turn${result.turns === 1 ? "" : "s"}` + (toolSummary ? `: ${toolSummary}` : "") + ` | ${result.parsedSuccessfully ? "output OK" : "\u26A0 output not valid JSON"} | cost: ${result.usage.totalBillingCost}`
+    `[task] ${result.turns} turn${result.turns === 1 ? "" : "s"}` + (toolSummary ? `: ${toolSummary}` : "") + ` | ${result.parsedSuccessfully ? "output OK" : "\u26A0 output not valid JSON"} | cost: ${result.usage.totalBillingCost} | trace: ${result.traceId}`
   );
 }
 async function runTaskPoll(httpConfig, body) {
@@ -5496,7 +5613,8 @@ ${structuredOutputExample}
         errors,
         turns: loopCount,
         usage: totalUsage,
-        toolCalls: toolCallLog
+        toolCalls: toolCallLog,
+        traceId
       }
     );
   };
@@ -5510,7 +5628,12 @@ ${structuredOutputExample}
           `[task] ${err.message}`,
           "task_execution_error",
           500,
-          { turns: loopCount, usage: totalUsage, toolCalls: toolCallLog }
+          {
+            turns: loopCount,
+            usage: totalUsage,
+            toolCalls: toolCallLog,
+            traceId
+          }
         );
       }
       return finish(buildResult(null, "", false, loopCount));
@@ -5519,7 +5642,8 @@ ${structuredOutputExample}
       throw new MindStudioError(
         `[task] ${err.message}`,
         err.errorCode ?? "task_turn_error",
-        err.status ?? 500
+        err.status ?? 500,
+        { traceId }
       );
     }
     throw err;
@@ -5583,9 +5707,13 @@ ${structuredOutputExample}
                 isError: true
               };
             }
+          } else if (kind === "method") {
+            result = await deps.executeMethodTool(toolCall.name, mergedInput, {
+              traceId,
+              toolCallId: toolCall.id
+            });
           } else {
-            const execute = kind === "method" ? deps.executeMethodTool : deps.executeStepTool;
-            result = await execute(toolCall.name, mergedInput);
+            result = await deps.executeStepTool(toolCall.name, mergedInput);
           }
           toolCallLog.push({
             name: toolCall.name,
@@ -6795,7 +6923,7 @@ var MindStudioAgent = class {
               };
             }
           },
-          executeMethodTool: async (methodId, input) => {
+          executeMethodTool: async (methodId, input, call) => {
             try {
               const { data } = await request(
                 { ...httpConfig, maxRetries: 0 },
@@ -6803,7 +6931,9 @@ var MindStudioAgent = class {
                 "/task/invoke-method",
                 {
                   methodId,
-                  input
+                  input,
+                  traceId: call.traceId,
+                  toolCallId: call.toolCallId
                 }
               );
               if (data.error) {
@@ -8239,7 +8369,11 @@ function documents(list) {
   return { kind: "documents", documents: Array.isArray(list) ? list : [list] };
 }
 function passthrough(options) {
-  return options?.metadata !== void 0 ? { kind: "passthrough", metadata: options.metadata } : { kind: "passthrough" };
+  return {
+    kind: "passthrough",
+    ...options?.metadata !== void 0 ? { metadata: options.metadata } : {},
+    ...options?.entities !== void 0 ? { entities: options.entities } : {}
+  };
 }
 function skip(reason) {
   return { kind: "skip", reason };
@@ -8283,6 +8417,7 @@ function objectFor(input) {
     etag: input.etag,
     lastModified: input.lastModified,
     metadata: input.metadata,
+    entities: input.entities ?? null,
     bytes,
     text: async () => new TextDecoder("utf-8").decode(await bytes()),
     json: async () => JSON.parse(new TextDecoder("utf-8").decode(await bytes()))
@@ -8372,6 +8507,128 @@ function defineMapper(source, config) {
   });
 }
 
+// src/reader/index.ts
+var DEFAULT_TIMEOUT_MS2 = 6e4;
+var DEFAULT_CONCURRENCY3 = 4;
+function errorInfo3(e) {
+  if (e instanceof Error) {
+    return { message: e.message, stack: e.stack };
+  }
+  return { message: String(e) };
+}
+function refOf(delivery) {
+  switch (delivery.kind) {
+    case "read":
+      return delivery.document.id;
+    case "removed":
+      return delivery.documentId;
+    case "cleared":
+      return delivery.sourceId;
+  }
+}
+async function withTimeout2(work, timeoutMs, ref) {
+  let timer;
+  try {
+    return await Promise.race([
+      work,
+      new Promise((_, reject) => {
+        timer = setTimeout(
+          () => reject(
+            new Error(`reader timed out after ${timeoutMs}ms on ${ref}`)
+          ),
+          timeoutMs
+        );
+      })
+    ]);
+  } finally {
+    if (timer) {
+      clearTimeout(timer);
+    }
+  }
+}
+function defineReader(source, config) {
+  const documentFor = (delivery) => {
+    let cached = delivery.chunks ? Promise.resolve(delivery.chunks) : null;
+    return {
+      ...delivery.document,
+      chunks: () => {
+        if (!cached) {
+          cached = source.chunks(delivery.document.id);
+        }
+        return cached;
+      }
+    };
+  };
+  const deliver2 = async (delivery) => {
+    switch (delivery.kind) {
+      case "read":
+        await config.read(documentFor(delivery));
+        return;
+      case "removed":
+        await config.removed(delivery.documentId);
+        return;
+      case "cleared":
+        await config.cleared?.(delivery.sourceId);
+        return;
+    }
+  };
+  const run = async (params) => {
+    const startedAt = Date.now();
+    const timeoutMs = params.timeoutMs ?? DEFAULT_TIMEOUT_MS2;
+    const concurrency = Math.max(1, params.concurrency ?? DEFAULT_CONCURRENCY3);
+    const deliveries = Array.isArray(params.deliveries) ? params.deliveries : [];
+    const outcomes = new Array(deliveries.length);
+    let next = 0;
+    const worker = async () => {
+      while (true) {
+        const index = next++;
+        if (index >= deliveries.length) {
+          return;
+        }
+        const delivery = deliveries[index];
+        const ref = refOf(delivery);
+        const t = Date.now();
+        try {
+          await withTimeout2(
+            Promise.resolve().then(() => deliver2(delivery)),
+            timeoutMs,
+            ref
+          );
+          outcomes[index] = {
+            ref,
+            kind: delivery.kind,
+            ok: true,
+            durationMs: Date.now() - t
+          };
+        } catch (e) {
+          outcomes[index] = {
+            ref,
+            kind: delivery.kind,
+            ok: false,
+            error: errorInfo3(e),
+            durationMs: Date.now() - t
+          };
+        }
+      }
+    };
+    await Promise.all(
+      Array.from({ length: Math.min(concurrency, deliveries.length) }, worker)
+    );
+    return {
+      v: 1,
+      source: source.name,
+      outcomes,
+      startedAt,
+      durationMs: Date.now() - startedAt
+    };
+  };
+  return Object.assign(run, {
+    kind: "reader",
+    source: source.name,
+    config
+  });
+}
+
 // src/generated/snippets.ts
 var monacoSnippets = {
   "activeCampaignAddNote": { fields: [["contactId", "string"], ["note", "string"]], outputKeys: [] },
@@ -8632,6 +8889,7 @@ export {
   index_default as default,
   defineJewel,
   defineMapper,
+  defineReader,
   deletes,
   documents,
   email,