@desplega.ai/agent-swarm 1.165.0 → 1.167.0

package/dist/cli-jqt9hrnz.js

@@ -1,168 +0,0 @@
-import {
-  cosineSimilarity,
-  deserializeEmbedding,
-  getEmbeddingProvider,
-  init_embedding,
-  serializeEmbedding
-} from "./cli-yhmwmv0g.js";
-import {
-  getDbClient,
-  init_db
-} from "./cli-qdv8kgz7.js";
-import {
-  init_secret_scrubber,
-  scrubSecrets
-} from "./cli-h1spv8v4.js";
-
-// src/be/scripts/embeddings.ts
-init_secret_scrubber();
-init_db();
-init_embedding();
-var providerOverride = null;
-function embeddingProvider() {
-  return providerOverride ?? getEmbeddingProvider();
-}
-function scriptEmbeddingText(script) {
-  return scrubSecrets([script.description, script.intent, script.signatureJson].join(`
-`));
-}
-function rowToScript(row) {
-  return {
-    id: row.id,
-    name: row.name,
-    scope: row.scope,
-    scopeId: row.scopeId ?? null,
-    source: row.source,
-    description: row.description,
-    intent: row.intent,
-    signatureJson: row.signatureJson,
-    argsJsonSchema: row.argsJsonSchema ?? null,
-    contentHash: row.contentHash,
-    version: row.version,
-    isScratch: row.isScratch === 1,
-    typeChecked: row.typeChecked === 1,
-    fsMode: row.fsMode,
-    createdByAgentId: row.createdByAgentId ?? null,
-    createdAt: row.createdAt,
-    updatedAt: row.updatedAt
-  };
-}
-async function embedScript(script) {
-  const text = scriptEmbeddingText(script);
-  const provider = embeddingProvider();
-  const embedding = await provider.embed(text);
-  if (!embedding)
-    return;
-  if (embedding.length !== provider.dimensions) {
-    console.error(`[script-embed] dimension mismatch for "${script.name}": expected=${provider.dimensions} got=${embedding.length}, skipping`);
-    return;
-  }
-  await getDbClient().run(`INSERT INTO script_embeddings (
-      scriptId, embedding, embeddingModel, embeddedText, embeddedAt
-    )
-    VALUES (?, ?, ?, ?, ?)
-    ON CONFLICT(scriptId) DO UPDATE SET
-      embedding = excluded.embedding,
-      embeddingModel = excluded.embeddingModel,
-      embeddedText = excluded.embeddedText,
-      embeddedAt = excluded.embeddedAt`, [script.id, serializeEmbedding(embedding), provider.name, text, new Date().toISOString()]);
-}
-async function candidateRows(scope, scopeId) {
-  const params = [];
-  let where = "s.isScratch = 0";
-  if (scope === "global") {
-    where += " AND s.scope = 'global' AND s.scopeId IS NULL";
-  } else if (scope === "agent") {
-    where += " AND s.scope = 'agent' AND s.scopeId = ?";
-    params.push(scopeId ?? "");
-  } else if (scopeId) {
-    where += " AND ((s.scope = 'agent' AND s.scopeId = ?) OR (s.scope = 'global' AND s.scopeId IS NULL))";
-    params.push(scopeId);
-  } else {
-    where += " AND s.scope = 'global' AND s.scopeId IS NULL";
-  }
-  return getDbClient().query(`SELECT
-      s.*,
-      e.scriptId,
-      e.embedding,
-      e.embeddingModel,
-      e.embeddedText,
-      e.embeddedAt
-    FROM script_embeddings e
-    JOIN scripts s ON s.id = e.scriptId
-    WHERE ${where}`, params);
-}
-async function scriptRows(scope, scopeId) {
-  const params = [];
-  let where = "isScratch = 0";
-  if (scope === "global") {
-    where += " AND scope = 'global' AND scopeId IS NULL";
-  } else if (scope === "agent") {
-    where += " AND scope = 'agent' AND scopeId = ?";
-    params.push(scopeId ?? "");
-  } else if (scopeId) {
-    where += " AND ((scope = 'agent' AND scopeId = ?) OR (scope = 'global' AND scopeId IS NULL))";
-    params.push(scopeId);
-  } else {
-    where += " AND scope = 'global' AND scopeId IS NULL";
-  }
-  return getDbClient().query(`SELECT *, NULL as scriptId, NULL as embedding, NULL as embeddingModel, NULL as embeddedText, NULL as embeddedAt FROM scripts WHERE ${where}`, params);
-}
-function nameMatchBonus(script, query) {
-  const trimmed = query.trim().toLowerCase();
-  if (!trimmed)
-    return 0;
-  return script.name.toLowerCase().includes(trimmed) ? 1 : 0;
-}
-async function lexicalFallback(args) {
-  const query = args.query.trim().toLowerCase();
-  return (await scriptRows(args.scope, args.scopeId)).map(rowToScript).filter((script) => {
-    if (!query)
-      return true;
-    return [script.name, script.description, script.intent].join(`
-`).toLowerCase().includes(query);
-  }).map((script) => {
-    const bonus = nameMatchBonus(script, args.query);
-    return {
-      script,
-      score: bonus || 0.5,
-      semanticScore: 0,
-      nameMatchBonus: bonus
-    };
-  }).sort((a, b) => b.score - a.score).slice(0, args.limit ?? 10);
-}
-async function searchScripts(args) {
-  const provider = embeddingProvider();
-  const queryEmbedding = await provider.embed(args.query);
-  if (!queryEmbedding)
-    return lexicalFallback(args);
-  const candidates = await candidateRows(args.scope, args.scopeId);
-  if (candidates.length === 0)
-    return lexicalFallback(args);
-  const results = [];
-  for (const row of candidates) {
-    const stored = deserializeEmbedding(row.embedding);
-    if (stored.length !== queryEmbedding.length)
-      continue;
-    const script = rowToScript(row);
-    const semanticScore = cosineSimilarity(queryEmbedding, stored);
-    const bonus = nameMatchBonus(script, args.query);
-    results.push({
-      script,
-      score: 0.7 * semanticScore + 0.3 * bonus,
-      semanticScore,
-      nameMatchBonus: bonus
-    });
-  }
-  if (results.length === 0)
-    return lexicalFallback(args);
-  return results.sort((a, b) => b.score - a.score).slice(0, args.limit ?? 10);
-}
-async function reembedAllScripts() {
-  const rows = await getDbClient().query("SELECT *, NULL as scriptId, NULL as embedding, NULL as embeddingModel, NULL as embeddedText, NULL as embeddedAt FROM scripts WHERE isScratch = 0 ORDER BY updatedAt ASC");
-  for (const row of rows) {
-    await embedScript(rowToScript(row));
-  }
-}
-
-export { embeddingProvider, embedScript, searchScripts, reembedAllScripts };