dsh-omni-router 1.2.0 → 1.3.0

package/README.md

@@ -55,6 +55,8 @@
 - **状态持久化**：分类/思维模式/任务类型写入会话事件，resume 可恢复。
 - **降级保护**：Plan Mode 不可用时自动限制为只读工具。
 - **Hybrid 分类**：启发式置信度 + 低置信度时 LLM 结构化分类。
+- **Policy Engine**：`buildPolicyDecision` 统一输出完整决策对象（taskType/complexity/risk/executionMode/approval/verification/gitPolicy）。
+- **Repository Snapshot**：`buildRepositorySnapshot` 识别包管理器、测试框架、框架、入口点。
 - **Policy/State Orchestration**：用 `workflowPolicy` 状态机代替大量重复 prompt。
 - **Agent 选择**：`selectAgentForTask` 推荐 frontend/backend/db/browser/security/review agent。
 - **Router Benchmark**：内置 `benchmark/run.mjs`（399 个任务），可评估 accuracy / false-direct rate。

package/README.zh-CN.md

@@ -55,6 +55,8 @@
 - **状态持久化**：分类/思维模式/任务类型写入会话事件，resume 可恢复。
 - **降级保护**：Plan Mode 不可用时自动限制为只读工具。
 - **Hybrid 分类**：启发式置信度 + 低置信度时 LLM 结构化分类。
+- **Policy Engine**：`buildPolicyDecision` 统一输出完整决策对象（taskType/complexity/risk/executionMode/approval/verification/gitPolicy）。
+- **Repository Snapshot**：`buildRepositorySnapshot` 识别包管理器、测试框架、框架、入口点。
 - **Policy/State Orchestration**：用 `workflowPolicy` 状态机代替大量重复 prompt。
 - **Agent 选择**：`selectAgentForTask` 推荐 frontend/backend/db/browser/security/review agent。
 - **Router Benchmark**：内置 `benchmark/run.mjs`（399 个任务），可评估 accuracy / false-direct rate。

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "dsh-omni-router",
-  "version": "1.2.0",
+  "version": "1.3.0",
   "description": "Omni Router - auto-routing DeepSeek Harness preset: simple tasks run directly, complex tasks plan first with approval gate, coding tasks get TDD/delivery-gate guidance.",
   "type": "module",
   "main": "./lib/index.js",

package/src/omni-router.mjs

@@ -368,7 +368,7 @@ export function estimateRisk(text) {
   if (/(生产环境|production|prod|密钥|secret|token|drop database|drop table|rm -rf)/.test(normalized)) {
     reasons.push('production/secret/destructive')
   }
-  if (/(schema|migration|drop table|drop database|连接池|auth|权限|deploy|ci\/cd|配置|config|Webhook|验签|签名|对账|网关|删除.*(数据库|字段|表|生产|配置|auth|用户)|删掉.*(数据库|字段|表|生产|配置|auth|用户))/.test(normalized)) {
+  if (/(schema|migration|drop table|drop database|连接池|auth|权限|deploy|ci\/cd|配置|config|Webhook|验签|签名|对账|网关|退款|扣款|删除.*(数据库|字段|表|生产|配置|auth|用户)|删掉.*(数据库|字段|表|生产|配置|auth|用户))/.test(normalized)) {
     reasons.push('schema/auth/delete/deploy')
   }
   if (/(数据库|db|redis|登录|login|session|token|订单|order|支付|payment|业务逻辑|重构|refactor|api|接口|核心)/.test(normalized)) {
@@ -426,6 +426,66 @@ export function workflowPolicy(taskType, complexity, risk) {
   }
 }
 
+/**
+ * Policy Engine: unify all routing dimensions into one decision object.
+ * This is the "brain first layer" described in the Omni roadmap.
+ */
+export function buildPolicyDecision(taskText, config = {}) {
+  const complexity = classifyComplexity(taskText, config)
+  const taskType = classifyTaskType(taskText)
+  const risk = estimateRisk(taskText).level
+  const highRisk = ['high', 'critical'].includes(risk)
+  const executionMode = complexity === 'plan' || highRisk ? 'plan' : 'direct'
+  const coding = ['bugfix', 'feature', 'refactor', 'test'].includes(taskType)
+  return {
+    taskType,
+    complexity,
+    risk,
+    reasoningMode: complexity === 'plan' || highRisk ? 'max' : 'balanced',
+    contextStrategy: 'dependency-aware',
+    executionMode,
+    approvalRequired: executionMode === 'plan' || highRisk,
+    confidence: heuristicComplexity(taskText, config).confidence,
+    verification: coding ? ['unit', 'integration', 'regression'] : [],
+    gitPolicy: {
+      requireBranch: coding,
+    },
+  }
+}
+
+/**
+ * Project Brain first step: build a lightweight repository snapshot from root
+ * entries. Later this will be backed by git/ripgrep/tree-sitter.
+ */
+export function buildRepositorySnapshot(entries) {
+  const names = new Set((Array.isArray(entries) ? entries : []).map((entry) => entry.name))
+  const packageManager = names.has('pnpm-lock.yaml') ? 'pnpm'
+    : names.has('package-lock.json') ? 'npm'
+    : names.has('yarn.lock') ? 'yarn'
+    : names.has('bun.lockb') ? 'bun'
+    : null
+  const testFramework = names.has('vitest.config.ts') || names.has('vitest.config.js') ? 'vitest'
+    : names.has('jest.config.js') || names.has('jest.config.ts') ? 'jest'
+    : names.has('cypress.config.ts') || names.has('cypress.config.js') ? 'cypress'
+    : names.has('mocha.opts') ? 'mocha'
+    : null
+  const framework = names.has('next.config.js') || names.has('next.config.mjs') ? 'next'
+    : names.has('vite.config.ts') || names.has('vite.config.js') ? 'vite'
+    : names.has('angular.json') ? 'angular'
+    : names.has('nuxt.config.ts') ? 'nuxt'
+    : null
+  const entryPoints = ['src', 'lib', 'app', 'api', 'server', 'client', 'test', 'tests']
+    .filter((name) => names.has(name))
+  return {
+    packageManager,
+    testFramework,
+    framework,
+    entryPoints,
+    hasReadme: names.has('README.md'),
+    hasPackageJson: names.has('package.json'),
+  }
+}
+
 /**
  * Build a compact project-context summary from root entries and key file
  * contents. This is injected into plan mode so the model starts with context
@@ -808,6 +868,12 @@ export function apply(ctx, config = {}) {
         text: `Risk level: ${state.riskLevel}. High or critical risk requires plan approval before mutation.`,
       })
     }
+    const policyDecision = buildPolicyDecision(state.firstText || '', config)
+    sections.push({
+      name: 'omni-router:policy-decision',
+      order: 37,
+      text: `Policy decision: ${JSON.stringify(policyDecision)}`,
+    })
     const policy = workflowPolicy(taskType, state.kind || 'direct', state.riskLevel || 'low')
     sections.push({
       name: 'omni-router:policy',

package/test/omni-router.test.mjs

@@ -6,6 +6,8 @@ import {
   buildContextGraph,
   buildContextSummary,
   buildDependencyHints,
+  buildPolicyDecision,
+  buildRepositorySnapshot,
   classifyComplexity,
   classifyTaskType,
   classifyThinkingMode,
@@ -356,3 +358,28 @@ test('extractSymbolsFromText finds function/class/const names', () => {
   assert.ok(symbols.includes('TOKEN_KEY'))
   assert.ok(symbols.includes('refreshSession'))
 })
+
+test('buildPolicyDecision returns a unified policy object', () => {
+  const policy = buildPolicyDecision('修复退款偶发重复扣款', {})
+  assert.equal(policy.taskType, 'bugfix')
+  assert.equal(policy.complexity, 'plan')
+  assert.equal(policy.risk, 'high')
+  assert.equal(policy.executionMode, 'plan')
+  assert.equal(policy.approvalRequired, true)
+  assert.ok(Array.isArray(policy.verification))
+})
+
+test('buildRepositorySnapshot detects project metadata from root entries', () => {
+  const entries = [
+    { name: 'package.json', type: 'file' },
+    { name: 'pnpm-lock.yaml', type: 'file' },
+    { name: 'src', type: 'directory' },
+    { name: 'tests', type: 'directory' },
+    { name: 'vitest.config.ts', type: 'file' },
+    { name: 'README.md', type: 'file' },
+  ]
+  const snapshot = buildRepositorySnapshot(entries)
+  assert.equal(snapshot.packageManager, 'pnpm')
+  assert.equal(snapshot.testFramework, 'vitest')
+  assert.ok(snapshot.entryPoints.includes('src'))
+})