@mazjq/zhiro 0.0.0-stage → 0.1.2

package/dist/chunk-F6DLONN7.js

@@ -0,0 +1,328 @@
+import {
+  DomainError,
+  NotFound
+} from "./chunk-LH25QWMP.js";
+import {
+  now
+} from "./chunk-5KCN32HQ.js";
+import {
+  __name
+} from "./chunk-2SSYR7VW.js";
+
+// apps/daemon/src/service/usage.ts
+import { AsyncLocalStorage } from "node:async_hooks";
+import { ulid } from "ulid";
+var scope = new AsyncLocalStorage();
+var withPurpose = /* @__PURE__ */ __name((purpose, ref, fn) => scope.run({ purpose, ref, caller: scope.getStore()?.caller }, fn), "withPurpose");
+var currentPurpose = /* @__PURE__ */ __name(() => scope.getStore() ?? { purpose: "其他", ref: "" }, "currentPurpose");
+async function asAgentCall(via, fn) {
+  const caller = { via: via.slice(0, 32), live: true };
+  return scope.run({ purpose: "其他", ref: "", caller }, async () => {
+    try {
+      return await fn();
+    } finally {
+      caller.live = false;
+    }
+  });
+}
+__name(asAgentCall, "asAgentCall");
+var callingAgent = /* @__PURE__ */ __name(() => {
+  const c = scope.getStore()?.caller;
+  return c?.live ? c.via : null;
+}, "callingAgent");
+function recordUsage(db, msg, provider, model) {
+  const u = msg?.usage;
+  if (!u) return;
+  const s = scope.getStore() ?? { purpose: "其他", ref: "" };
+  db.prepare("insert into llm_usage(id, at, purpose, ref, provider, model, input, output, cache_read, total, cost) values (?,?,?,?,?,?,?,?,?,?,?)").run(ulid(), (/* @__PURE__ */ new Date()).toISOString(), s.purpose, s.ref, provider, model, u.input ?? 0, u.output ?? 0, u.cacheRead ?? 0, u.totalTokens ?? (u.input ?? 0) + (u.output ?? 0), u.cost?.total ?? 0);
+}
+__name(recordUsage, "recordUsage");
+var LABEL = {
+  "apply.start": "网申填写",
+  "apply.read": "网申读表",
+  "profile.import": "简历/材料抽取",
+  "profile.scan": "简历/材料抽取",
+  "document.reextract": "简历/材料抽取",
+  "resume.save": "简历/材料抽取",
+  "resume.analyze": "简历分析",
+  "inbox.pull": "收件箱分类",
+  "inbox.paste": "收件箱分类",
+  "inbox.poll": "收件箱分类",
+  "inbox.rescue": "收件箱分类",
+  "listing.capture": "简章抽取",
+  "chat": "助手",
+  "llm.test": "连通测试",
+  "company.find": "找公司入口"
+};
+function usageSummary(db) {
+  const since = /* @__PURE__ */ __name((days) => new Date(Date.now() - days * 864e5).toISOString(), "since");
+  const rows = db.prepare(`select purpose, count(*) as calls, sum(input) as input, sum(output) as output, sum(total) as total, sum(cost) as cost,
+      count(distinct nullif(ref, '')) as jobs, sum(case when at >= ? then total else 0 end) as week
+    from llm_usage group by purpose order by total desc`).all(since(7));
+  const merged = /* @__PURE__ */ new Map();
+  for (const r of rows) {
+    const label = LABEL[r.purpose] ?? r.purpose;
+    const m = merged.get(label) ?? { purpose: label, calls: 0, input: 0, output: 0, total: 0, cost: 0, jobs: 0, week: 0 };
+    for (const k of ["calls", "input", "output", "total", "cost", "jobs", "week"]) m[k] += r[k] ?? 0;
+    merged.set(label, m);
+  }
+  const items = [...merged.values()].map((m) => ({ ...m, perJob: m.jobs ? Math.round(m.total / m.jobs) : Math.round(m.total / Math.max(m.calls, 1)) }));
+  const t = db.prepare("select count(*) as calls, coalesce(sum(total), 0) as total, coalesce(sum(cost), 0) as cost, min(at) as since from llm_usage").get();
+  return { items, total: { calls: t.calls, total: t.total, cost: t.cost, since: t.since ?? null } };
+}
+__name(usageSummary, "usageSummary");
+
+// apps/daemon/src/llm.ts
+import { existsSync as existsSync2, readFileSync, writeFileSync as writeFileSync2, mkdirSync as mkdirSync2, renameSync } from "node:fs";
+import { dirname, join as join2 } from "node:path";
+import { builtinModels } from "@earendil-works/pi-ai/providers/all";
+
+// apps/daemon/src/service/model-jobs.ts
+import { createHash } from "node:crypto";
+import { existsSync, mkdirSync, writeFileSync } from "node:fs";
+import { join } from "node:path";
+import { ulid as ulid2 } from "ulid";
+var row = /* @__PURE__ */ __name((r) => ({ id: r.id, kind: r.kind, purpose: r.purpose, ref: r.ref, system: r.system, user: r.user_text, status: r.status, createdAt: r.created_at, caller: r.caller ?? "" }), "row");
+var ModelDelegated = class extends DomainError {
+  constructor(job) {
+    super("MODEL_DELEGATED", `等外部 agent 代答：llm.pending 查看、llm.reply 回答（任务 ${job.id}）`);
+    this.job = job;
+  }
+  job;
+  static {
+    __name(this, "ModelDelegated");
+  }
+};
+var EXT = { "image/png": "png", "image/jpeg": "jpg", "image/webp": "webp", "image/gif": "gif" };
+function promptOf(ctx) {
+  const parts = [], images = [];
+  for (const m of ctx.messages ?? []) {
+    for (const c of typeof m.content === "string" ? [{ type: "text", text: m.content }] : m.content ?? []) {
+      if (c.type === "text") parts.push(c.text);
+      else if (c.type === "image" && EXT[c.mimeType]) images.push({ data: c.data, mimeType: c.mimeType, sha: createHash("sha256").update(String(c.data)).digest("hex").slice(0, 16) });
+    }
+  }
+  return { text: parts.join("\n\n"), images };
+}
+__name(promptOf, "promptOf");
+function delegateComplete(db, ctx, o = {}) {
+  const system = String(ctx.systemPrompt ?? ""), p = promptOf(ctx);
+  const hash = createHash("sha256").update(system + "\0" + p.text.replace(/\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}(\.\d+)?Z/g, "") + "\0" + p.images.map((i) => i.sha).join(",")).digest("hex");
+  const r = db.prepare("select * from model_job where hash = ?").get(hash);
+  if (r && r.status !== "pending") {
+    db.prepare("update model_job set status = 'used' where id = ?").run(r.id);
+    return {
+      role: "assistant",
+      content: [{ type: "text", text: r.answer ?? "" }],
+      api: "external",
+      provider: "external",
+      model: "agent",
+      stopReason: "stop",
+      timestamp: Date.now(),
+      usage: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, totalTokens: 0, cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 } }
+    };
+  }
+  if (r) throw new ModelDelegated(row(r));
+  const { purpose, ref } = currentPurpose();
+  const id = ulid2();
+  let user = p.text;
+  if (p.images.length) {
+    if (o.imagesDir) {
+      const dir = join(o.imagesDir, id);
+      mkdirSync(dir, { recursive: true });
+      const files = p.images.map((im, i) => {
+        const f = join(dir, `image-${i + 1}.${EXT[im.mimeType]}`);
+        writeFileSync(f, Buffer.from(im.data, "base64"));
+        return f;
+      });
+      user += `
+
+（随附 ${files.length} 张图片，用你自己的读图能力打开：
+${files.join("\n")}）`;
+    } else user += `
+
+（原请求随附 ${p.images.length} 张图片未转发）`;
+  }
+  db.prepare("insert into model_job(id, kind, hash, purpose, ref, system, user_text, created_at, caller) values (?,?,?,?,?,?,?,?,?)").run(id, "complete", hash, purpose, ref, system, user, now(), o.caller ?? "");
+  throw new ModelDelegated(row(db.prepare("select * from model_job where id = ?").get(id)));
+}
+__name(delegateComplete, "delegateComplete");
+function delegateRun(db, runId, mode, prompt, schema, caller = "") {
+  db.prepare("insert into model_job(id, kind, purpose, system, user_text, created_at, caller) values (?,?,?,?,?,?,?)").run(runId, "run", mode, `自己联网调研后，调 llm.reply 提交一个符合下面 JSON Schema 的 JSON 对象（text 就是这个 JSON）：
+${JSON.stringify(schema)}`, prompt, now(), caller);
+}
+__name(delegateRun, "delegateRun");
+var pendingJobs = /* @__PURE__ */ __name((db, caller) => ({
+  items: (caller === void 0 ? db.prepare("select * from model_job where status = 'pending' order by created_at").all() : db.prepare("select * from model_job where status = 'pending' and caller in ('', ?) order by created_at").all(caller)).map(row)
+}), "pendingJobs");
+function replyJob(db, jobId, text) {
+  const r = db.prepare("select * from model_job where id = ?").get(jobId);
+  if (!r) throw new NotFound(`model job ${jobId}`);
+  if (r.kind === "run") {
+    let structured;
+    try {
+      structured = JSON.parse(text.trim().replace(/^```(?:json)?\s*|\s*```$/g, ""));
+    } catch {
+      throw new DomainError("BAD_ANSWER", "调研结果要是一个 JSON 对象（按任务里的 Schema）");
+    }
+    const run = db.prepare("select run_dir from agent_run where id = ?").get(jobId);
+    if (!run || !existsSync(run.run_dir)) throw new NotFound(`run ${jobId}`);
+    writeFileSync(join(run.run_dir, "out.json"), JSON.stringify({ structured_output: structured, result: "", session_id: "", total_cost_usd: 0 }), "utf8");
+    writeFileSync(join(run.run_dir, "exit.txt"), "0", "utf8");
+  }
+  db.prepare("update model_job set status = 'answered', answer = ?, answered_at = ? where id = ?").run(text, now(), jobId);
+  return row(db.prepare("select * from model_job where id = ?").get(jobId));
+}
+__name(replyJob, "replyJob");
+function oldestWaitingMail(db) {
+  return db.prepare("select min(ref) as r from model_job where purpose like 'inbox%' and status in ('pending', 'answered') and ref != ''").get()?.r ?? null;
+}
+__name(oldestWaitingMail, "oldestWaitingMail");
+
+// apps/daemon/src/llm.ts
+var EXTERNAL = "external";
+var isExternal = /* @__PURE__ */ __name((provider) => !provider || provider === "none" || provider === EXTERNAL, "isExternal");
+var EXTERNAL_MODEL = { id: "agent", name: "外部 agent", provider: EXTERNAL, api: "external", input: ["text"], reasoning: false };
+var FileCredentialStore = class {
+  constructor(path) {
+    this.path = path;
+  }
+  path;
+  static {
+    __name(this, "FileCredentialStore");
+  }
+  chain = Promise.resolve();
+  load() {
+    if (!existsSync2(this.path)) return {};
+    try {
+      return JSON.parse(readFileSync(this.path, "utf8"));
+    } catch {
+      return {};
+    }
+  }
+  save(all) {
+    mkdirSync2(dirname(this.path), { recursive: true });
+    const tmp = this.path + ".tmp";
+    writeFileSync2(tmp, JSON.stringify(all, null, 2), { encoding: "utf8", mode: 384 });
+    renameSync(tmp, this.path);
+  }
+  serialize(fn) {
+    const next = this.chain.then(fn, fn);
+    this.chain = next.catch(() => void 0);
+    return next;
+  }
+  async read(providerId) {
+    return this.load()[providerId];
+  }
+  async list() {
+    return Object.entries(this.load()).map(([providerId, c]) => ({ providerId, type: c.type }));
+  }
+  modify(providerId, fn) {
+    return this.serialize(async () => {
+      const all = this.load();
+      const next = await fn(all[providerId]);
+      if (next !== void 0) {
+        all[providerId] = next;
+        this.save(all);
+      }
+      return all[providerId];
+    });
+  }
+  delete(providerId) {
+    return this.serialize(async () => {
+      const all = this.load();
+      delete all[providerId];
+      this.save(all);
+    });
+  }
+};
+function credentialsPath(vault) {
+  return join2(vault, "config", "credentials.json");
+}
+__name(credentialsPath, "credentialsPath");
+function createLlm(config) {
+  const credentials = new FileCredentialStore(credentialsPath(config.vault));
+  const models = builtinModels({ credentials });
+  const llm = {};
+  const report = /* @__PURE__ */ __name((m, model) => {
+    try {
+      llm.onUsage?.(m, model);
+    } catch {
+    }
+  }, "report");
+  for (const name of ["stream", "streamSimple"]) {
+    const orig = models[name]?.bind(models);
+    if (orig) models[name] = (model, ctx, opts) => {
+      const s = orig(model, ctx, opts);
+      s.result?.().then((m) => report(m, model)).catch(() => {
+      });
+      return s;
+    };
+  }
+  const complete = models.completeSimple.bind(models);
+  models.completeSimple = async (model, ctx, opts) => {
+    const job = /* @__PURE__ */ __name(() => {
+      if (!llm.db) throw new Error("model jobs need the daemon database");
+      return delegateComplete(llm.db, ctx, { caller: callingAgent() ?? "", imagesDir: join2(config.vault, "runs", "model-jobs") });
+    }, "job");
+    if (callingAgent()) return job();
+    const to = llm.route?.() ?? (isExternal(config.llm.provider) || llm.loggedIn === false ? { kind: "waiting" } : { kind: "model" });
+    if (to.kind === "waiting") return job();
+    if (to.kind === "unavailable") throw new Error(`BACKGROUND_UNAVAILABLE: ${to.why}`);
+    if (to.kind === "harness") {
+      if (!llm.harness) throw new Error("BACKGROUND_UNAVAILABLE: 后台一次性调用没有接上");
+      const a = await llm.harness(to.name, ctx);
+      const msg = {
+        role: "assistant",
+        content: [{ type: "text", text: a.text }],
+        api: "cli",
+        provider: to.name,
+        model: a.model || to.name,
+        stopReason: "stop",
+        timestamp: Date.now(),
+        usage: { input: a.usage.input, output: a.usage.output, cacheRead: a.usage.cacheRead, cacheWrite: 0, totalTokens: a.usage.total, cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: a.costUsd ?? 0 } }
+      };
+      report(msg, { provider: to.name, id: a.model || to.name });
+      return msg;
+    }
+    return complete(model, ctx, opts);
+  };
+  const out = Object.assign(llm, {
+    models,
+    credentials,
+    model() {
+      const { provider, model } = config.llm;
+      const to = llm.route?.();
+      if (to?.kind === "harness" && !callingAgent()) return { id: to.name, name: to.name, provider: to.name, api: "cli", input: ["text", "image"], reasoning: false };
+      if (isExternal(provider)) return EXTERNAL_MODEL;
+      const m = models.getModel(provider, model);
+      if (m) return m;
+      const ids = models.getModels(provider).map((x) => x.id);
+      throw new Error(`model ${provider}/${model} not found. available for ${provider}: ${ids.slice(0, 20).join(", ") || "(unknown provider)"}`);
+    },
+    async status() {
+      const check = isExternal(config.llm.provider) ? { type: "none" } : await models.checkAuth(config.llm.provider).catch(() => void 0);
+      llm.loggedIn = !!check;
+      return { provider: config.llm.provider, model: config.llm.model, loggedIn: !!check, credentials: await credentials.list() };
+    }
+  });
+  void out.status().catch(() => {
+  });
+  return out;
+}
+__name(createLlm, "createLlm");
+
+export {
+  withPurpose,
+  currentPurpose,
+  asAgentCall,
+  callingAgent,
+  recordUsage,
+  usageSummary,
+  delegateRun,
+  pendingJobs,
+  replyJob,
+  oldestWaitingMail,
+  isExternal,
+  createLlm
+};