@avantf/dsh-mem 0.1.0

package/lib/types/tool_schema.js

@@ -0,0 +1,149 @@
+/**
+ * zod → DSH parameter schema derivation for the agent tools.
+ *
+ * Deliberately free of `@deepseek-ai/*` imports: everything it needs — the contract's tool
+ * specs, hence the zod schemas — comes from `@avantf/mem-contract`, so this module can be read
+ * and exercised without a harness checkout or an installed `dsh`.
+ *
+ * Two things keep this honest across a zod upgrade. First, the field-level mapping goes through
+ * zod's OWN converter (`z.toJSONSchema`, the same one the contract's MCP derivation uses)
+ * instead of reading `typeName`/`checks`/`shape` by hand, so there is one mechanism and not two
+ * hand-rolled ones. Second, only the union/object structure is read off the internals
+ * (`def.options` / `def.shape` / a literal's `def.values`) — and a wrong internal read degrades
+ * SILENTLY: every field would flatten to `{type:'string'}` and no type error would appear, which
+ * is why the derived shape is asserted by the test suite rather than eyeballed.
+ */
+import { z } from 'zod';
+import { mergedFieldDescription } from '@avantf/mem-contract';
+export function zodDef(schema) {
+    return schema?.def ?? {};
+}
+/**
+ * One field as JSON Schema, or `undefined` when `schema` is not a zod schema at all.
+ *
+ * `io: 'input'` matches how these schemas are actually used (arguments coming IN), and
+ * `unrepresentable: 'any'` turns a future unrepresentable schema into a permissive object
+ * instead of a throw at import time.
+ */
+function jsonSchemaOf(schema) {
+    if (typeof schema?.safeParse !== 'function')
+        return undefined;
+    try {
+        return z.toJSONSchema(schema, { io: 'input', unrepresentable: 'any' });
+    }
+    catch {
+        return undefined;
+    }
+}
+/** Nearest `description` on a schema (wrappers included — the converter already unwrapped them). */
+export function schemaDescription(schema) {
+    const description = jsonSchemaOf(schema)?.['description'];
+    return typeof description === 'string' && description.length > 0 ? description : undefined;
+}
+/** JSON Schema → DSH parameter spec. Recurses on `items`, which is itself JSON Schema. */
+function jsonToSpec(js) {
+    const type = js['type'];
+    let spec;
+    if (type === 'integer')
+        spec = { type: 'integer' };
+    else if (type === 'number')
+        spec = { type: 'number' };
+    else if (type === 'boolean')
+        spec = { type: 'boolean' };
+    else if (type === 'array')
+        spec = { type: 'array', items: jsonToSpec(js['items'] ?? {}) };
+    else if (type === 'object')
+        spec = { type: 'object' };
+    else if (Array.isArray(js['enum']))
+        spec = { type: 'string', enum: js['enum'].map(String) };
+    // A string literal arrives as `const`, not `enum`: keep the old shape (`enum: [value]`).
+    else if (typeof js['const'] === 'string')
+        spec = { type: 'string', enum: [js['const']] };
+    else if (typeof js['const'] === 'number')
+        spec = { type: 'number' };
+    else if (typeof js['const'] === 'boolean')
+        spec = { type: 'boolean' };
+    else
+        spec = { type: 'string' };
+    const description = js['description'];
+    return typeof description === 'string' && description.length > 0 ? { ...spec, description } : spec;
+}
+export function zodToSpec(schema) {
+    return jsonToSpec(jsonSchemaOf(schema) ?? {});
+}
+/** Materialize one zod object branch's fields, or undefined for a non-object. */
+export function branchShape(schema) {
+    return zodDef(schema).shape;
+}
+/**
+ * Flatten a tool input schema into DSH's implicit parameter map. A discriminated
+ * union contributes one required `action` enum plus the merged field set (fields
+ * stay optional at the schema level because requiredness differs per action; the
+ * per-action requirement is stated in each field's description and enforced by
+ * `spec.input` validation before dispatch). A plain object marks its own required
+ * fields directly.
+ *
+ * The merged descriptions come from the contract's {@link mergedFieldDescription}, the SAME
+ * function `toolInputJsonSchema` uses for the MCP surface. Taking each field's first branch
+ * verbatim is what made the two model-facing surfaces disagree: `mem_admin`'s `fact_id` said
+ * "action=detail 时必填" on this surface while the MCP schema said
+ * "【detail/archive/restore/pin/unpin 必填】" — so a model calling `archive` without it had to
+ * learn the requirement from a `violations` retry. Both surfaces now read the same merged
+ * descriptions, which is what keeps them in agreement field for field.
+ */
+export function flattenToolSpec(spec) {
+    const input = spec.input;
+    const def = zodDef(input);
+    const properties = {};
+    let actions = [];
+    if (Array.isArray(def.options)) {
+        // The union's fields are declared one branch at a time, so a field's real action set is the
+        // union over the branches that require it — collected here, applied below.
+        const requiredIn = new Map();
+        for (const branch of def.options) {
+            const shape = branchShape(branch);
+            if (shape === undefined)
+                continue;
+            const actionSchema = shape['action'];
+            const literal = actionSchema === undefined ? undefined : zodDef(actionSchema).values?.[0];
+            const action = typeof literal === 'string' ? literal : undefined;
+            if (action !== undefined && !actions.includes(action))
+                actions = [...actions, action];
+            for (const [key, fieldSchema] of Object.entries(shape)) {
+                if (key === 'action')
+                    continue;
+                if (properties[key] === undefined)
+                    properties[key] = zodToSpec(fieldSchema);
+                if (action === undefined)
+                    continue;
+                const optional = fieldSchema.isOptional?.() ?? false;
+                if (!optional)
+                    requiredIn.set(key, [...(requiredIn.get(key) ?? []), action]);
+            }
+        }
+        for (const [key, paramSpec] of Object.entries(properties)) {
+            const description = mergedFieldDescription(paramSpec.description, requiredIn.get(key) ?? []);
+            properties[key] = description === undefined ? paramSpec : { ...paramSpec, description };
+        }
+    }
+    else {
+        const shape = branchShape(input);
+        if (shape !== undefined) {
+            for (const [key, fieldSchema] of Object.entries(shape)) {
+                const paramSpec = zodToSpec(fieldSchema);
+                const optional = fieldSchema.isOptional?.() ?? false;
+                properties[key] = optional ? paramSpec : { ...paramSpec, required: true };
+            }
+        }
+    }
+    if (actions.length > 0) {
+        properties['action'] = {
+            type: 'string',
+            required: true,
+            enum: actions,
+            description: `要执行的操作，取值：${actions.join(' | ')}。每个操作需要哪些字段见各字段说明。`,
+        };
+    }
+    return properties;
+}
+//# sourceMappingURL=tool_schema.js.map

package/lib/types/tool_schema.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool_schema.js","sourceRoot":"","sources":["../../src/tool_schema.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AACH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AACvB,OAAO,EAAE,sBAAsB,EAAiB,MAAM,sBAAsB,CAAA;AAmB5E,MAAM,UAAU,MAAM,CAAC,MAAe;IACpC,OAAQ,MAAuC,EAAE,GAAG,IAAI,EAAE,CAAA;AAC5D,CAAC;AAED;;;;;;GAMG;AACH,SAAS,YAAY,CAAC,MAAe;IACnC,IAAI,OAAQ,MAA8C,EAAE,SAAS,KAAK,UAAU;QAAE,OAAO,SAAS,CAAA;IACtG,IAAI,CAAC;QACH,OAAO,CAAC,CAAC,YAAY,CAAC,MAAmB,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE,eAAe,EAAE,KAAK,EAAE,CAA4B,CAAA;IAChH,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAA;IAClB,CAAC;AACH,CAAC;AAED,oGAAoG;AACpG,MAAM,UAAU,iBAAiB,CAAC,MAAe;IAC/C,MAAM,WAAW,GAAG,YAAY,CAAC,MAAM,CAAC,EAAE,CAAC,aAAa,CAAC,CAAA;IACzD,OAAO,OAAO,WAAW,KAAK,QAAQ,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS,CAAA;AAC5F,CAAC;AAED,0FAA0F;AAC1F,SAAS,UAAU,CAAC,EAA2B;IAC7C,MAAM,IAAI,GAAG,EAAE,CAAC,MAAM,CAAC,CAAA;IACvB,IAAI,IAAe,CAAA;IACnB,IAAI,IAAI,KAAK,SAAS;QAAE,IAAI,GAAG,EAAE,IAAI,EAAE,SAAS,EAAE,CAAA;SAC7C,IAAI,IAAI,KAAK,QAAQ;QAAE,IAAI,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAA;SAChD,IAAI,IAAI,KAAK,SAAS;QAAE,IAAI,GAAG,EAAE,IAAI,EAAE,SAAS,EAAE,CAAA;SAClD,IAAI,IAAI,KAAK,OAAO;QAAE,IAAI,GAAG,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,CAAE,EAAE,CAAC,OAAO,CAAyC,IAAI,EAAE,CAAC,EAAE,CAAA;SAC7H,IAAI,IAAI,KAAK,QAAQ;QAAE,IAAI,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAA;SAChD,IAAI,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC;QAAE,IAAI,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAG,EAAE,CAAC,MAAM,CAAe,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAA;IAC1G,yFAAyF;SACpF,IAAI,OAAO,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ;QAAE,IAAI,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,EAAE,CAAA;SACnF,IAAI,OAAO,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ;QAAE,IAAI,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAA;SAC9D,IAAI,OAAO,EAAE,CAAC,OAAO,CAAC,KAAK,SAAS;QAAE,IAAI,GAAG,EAAE,IAAI,EAAE,SAAS,EAAE,CAAA;;QAChE,IAAI,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAA;IAC9B,MAAM,WAAW,GAAG,EAAE,CAAC,aAAa,CAAC,CAAA;IACrC,OAAO,OAAO,WAAW,KAAK,QAAQ,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,IAAI,CAAA;AACpG,CAAC;AAED,MAAM,UAAU,SAAS,CAAC,MAAe;IACvC,OAAO,UAAU,CAAC,YAAY,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAA;AAC/C,CAAC;AAED,iFAAiF;AACjF,MAAM,UAAU,WAAW,CAAC,MAAe;IACzC,OAAO,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,CAAA;AAC7B,CAAC;AAED;;;;;;;;;;;;;;;GAeG;AACH,MAAM,UAAU,eAAe,CAAC,IAAc;IAC5C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAgB,CAAA;IACnC,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,CAAC,CAAA;IACzB,MAAM,UAAU,GAA8B,EAAE,CAAA;IAChD,IAAI,OAAO,GAAa,EAAE,CAAA;IAE1B,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;QAC/B,4FAA4F;QAC5F,2EAA2E;QAC3E,MAAM,UAAU,GAAG,IAAI,GAAG,EAAoB,CAAA;QAC9C,KAAK,MAAM,MAAM,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;YACjC,MAAM,KAAK,GAAG,WAAW,CAAC,MAAM,CAAC,CAAA;YACjC,IAAI,KAAK,KAAK,SAAS;gBAAE,SAAQ;YACjC,MAAM,YAAY,GAAG,KAAK,CAAC,QAAQ,CAAC,CAAA;YACpC,MAAM,OAAO,GAAG,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAE,MAAM,CAAC,YAAY,CAAC,CAAC,MAAgC,EAAE,CAAC,CAAC,CAAC,CAAA;YACpH,MAAM,MAAM,GAAG,OAAO,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAA;YAChE,IAAI,MAAM,KAAK,SAAS,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC;gBAAE,OAAO,GAAG,CAAC,GAAG,OAAO,EAAE,MAAM,CAAC,CAAA;YACrF,KAAK,MAAM,CAAC,GAAG,EAAE,WAAW,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;gBACvD,IAAI,GAAG,KAAK,QAAQ;oBAAE,SAAQ;gBAC9B,IAAI,UAAU,CAAC,GAAG,CAAC,KAAK,SAAS;oBAAE,UAAU,CAAC,GAAG,CAAC,GAAG,SAAS,CAAC,WAAW,CAAC,CAAA;gBAC3E,IAAI,MAAM,KAAK,SAAS;oBAAE,SAAQ;gBAClC,MAAM,QAAQ,GAAI,WAA8C,CAAC,UAAU,EAAE,EAAE,IAAI,KAAK,CAAA;gBACxF,IAAI,CAAC,QAAQ;oBAAE,UAAU,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC,CAAA;YAC9E,CAAC;QACH,CAAC;QACD,KAAK,MAAM,CAAC,GAAG,EAAE,SAAS,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;YAC1D,MAAM,WAAW,GAAG,sBAAsB,CAAC,SAAS,CAAC,WAAW,EAAE,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,CAAA;YAC5F,UAAU,CAAC,GAAG,CAAC,GAAG,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,GAAG,SAAS,EAAE,WAAW,EAAE,CAAA;QACzF,CAAC;IACH,CAAC;SAAM,CAAC;QACN,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,CAAC,CAAA;QAChC,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,KAAK,MAAM,CAAC,GAAG,EAAE,WAAW,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;gBACvD,MAAM,SAAS,GAAG,SAAS,CAAC,WAAW,CAAC,CAAA;gBACxC,MAAM,QAAQ,GAAI,WAA8C,CAAC,UAAU,EAAE,EAAE,IAAI,KAAK,CAAA;gBACxF,UAAU,CAAC,GAAG,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,GAAG,SAAS,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAA;YAC3E,CAAC;QACH,CAAC;IACH,CAAC;IAED,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACvB,UAAU,CAAC,QAAQ,CAAC,GAAG;YACrB,IAAI,EAAE,QAAQ;YACd,QAAQ,EAAE,IAAI;YACd,IAAI,EAAE,OAAO;YACb,WAAW,EAAE,aAAa,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,oBAAoB;SAClE,CAAA;IACH,CAAC;IACD,OAAO,UAAU,CAAA;AACnB,CAAC"}

package/package.json

@@ -0,0 +1,109 @@
+{
+  "name": "@avantf/dsh-mem",
+  "version": "0.1.0",
+  "description": "avantf-mem DSH native Cordis plugin: host service + tools + RPC + client tabs 记忆/知识",
+  "type": "module",
+  "main": "lib/index.js",
+  "types": "lib/types/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./lib/types/index.d.ts",
+      "default": "./lib/index.js"
+    },
+    "./client": {
+      "types": "./lib/types/client/index.d.ts",
+      "default": "./lib/client.js"
+    },
+    "./package.json": "./package.json"
+  },
+  "dsh": {
+    "client": {
+      "inject": [
+        "@deepseek-ai/dsh-client-ui-conversation",
+        "@deepseek-ai/dsh-api-remotes"
+      ],
+      "platform": "web"
+    }
+  },
+  "files": [
+    "lib",
+    "README.md"
+  ],
+  "peerDependencies": {
+    "@avantf/dsh-envinit": "^0.1.3",
+    "@deepseek-ai/cordis": "^4.0.0",
+    "@deepseek-ai/dsh-system-prompt": "^0.1.5-rc.2",
+    "@deepseek-ai/dsh-tools": "^0.1.5-rc.2",
+    "@deepseek-ai/dsh-typert-protocol": "^0.1.5-rc.2",
+    "@deepseek-ai/schemastery": "^3.18.0"
+  },
+  "peerDependenciesMeta": {
+    "@avantf/dsh-envinit": {
+      "optional": true
+    },
+    "@deepseek-ai/cordis": {
+      "optional": true
+    },
+    "@deepseek-ai/dsh-system-prompt": {
+      "optional": true
+    },
+    "@deepseek-ai/dsh-tools": {
+      "optional": true
+    },
+    "@deepseek-ai/dsh-typert-protocol": {
+      "optional": true
+    },
+    "@deepseek-ai/schemastery": {
+      "optional": true
+    }
+  },
+  "dependencies": {
+    "better-sqlite3": "^11.3.0",
+    "exceljs": "^4.4.0",
+    "unpdf": "^1.8.1",
+    "yaml": "^2.5.0",
+    "zod": "4.4.3"
+  },
+  "optionalDependencies": {
+    "@huggingface/transformers": "^4.2.0",
+    "hnswlib-node": "^2.1.1",
+    "nodejieba": "^2.6.0"
+  },
+  "devDependencies": {
+    "@avantf/dsh-compat": "^0.1.0",
+    "@avantf/dsh-envinit": "^0.1.3",
+    "@avantf/mem": "0.1.0",
+    "@avantf/mem-contract": "0.1.0",
+    "@avantf/mem-convert": "0.1.0",
+    "@avantf/mem-provision": "0.1.0",
+    "@types/node": "^22.0.0",
+    "@types/react": "^18.3.0",
+    "@types/react-dom": "^18.3.0",
+    "lightningcss": "^1.32.0",
+    "react": "^18.2.0",
+    "react-dom": "^18.2.0",
+    "tsdown": "0.22.2",
+    "typescript": "^5.5.0"
+  },
+  "engines": {
+    "node": ">=22"
+  },
+  "license": "MIT",
+  "publishConfig": {
+    "access": "public"
+  },
+  "prepublishOnly": "pnpm build",
+  "repository": {
+    "type": "git",
+    "url": "git@gitee.com:ffeng86/avantf-mem.git",
+    "directory": "packages/plugin"
+  },
+  "author": "qunqi.ff",
+  "scripts": {
+    "build": "tsc -p tsconfig.json && node ../../scripts/copy-envinit-bootstrap.mjs && tsdown",
+    "bundle": "tsdown",
+    "watch": "tsdown --watch",
+    "typecheck": "tsc -p tsconfig.json --noEmit",
+    "test:dsh": "vitest run --config vitest.dsh.config.ts"
+  }
+}