@p-dsh-market/conversation-knowledge-map 0.1.1 → 0.1.2

package/lib/generation-orchestrator.js

@@ -258,6 +258,7 @@ export class KnowledgeGenerationOrchestrator {
     agentDefaultModel,
     storage,
     modelRunner,
+    sessionEventSource,
     sourceReader = readSelectedSurfaces,
     now = () => Date.now(),
     idFactory = () => randomUUID()
@@ -268,6 +269,7 @@ export class KnowledgeGenerationOrchestrator {
     this.agentDefaultModel = agentDefaultModel
     this.storage = storage
     this.modelRunner = modelRunner
+    this.sessionEventSource = sessionEventSource
     this.sourceReader = sourceReader
     this.now = now
     this.idFactory = idFactory
@@ -450,28 +452,42 @@ export class KnowledgeGenerationOrchestrator {
     const model = String(selection?.model || '').trim()
     if (!provider || !model) throw new Error('当前没有可用的默认 Provider/Model。')
     const sessionId = `knowledge-map-${this.idFactory()}`
-    const handle = await this.agents.create({
-      sessionId,
-      meta: { cwd: input.cwd, origin: 'subagent' },
-      agentOptions: { provider, model, maxTokens: input.kind === 'summary' ? 2500 : 6000 },
-      signal: input.signal,
-      setup: async (agentCtx) => {
-        agentCtx?.systemPrompt?.section?.({
-          name: 'knowledge-map:protocol',
-          order: 0,
-          text: '你是 DSH 知识视图生成器。只输出调用方要求的 JSON；不要调用外部网络、文件写入或其他 Agent 工具。'
+    const liveEvents = []
+    let unsubscribe
+    if (typeof this.sessionEventSource === 'function') {
+      try {
+        unsubscribe = this.sessionEventSource((session, event) => {
+          const eventSessionId = String(session?.id || session?.header?.id || '')
+          if (eventSessionId === sessionId && event) liveEvents.push(event)
         })
-        try {
-          agentCtx?.tools?.restrict?.({ deny: ['multi_agent_discuss', 'shell', 'filesystem', 'web_search', 'browser'] })
-        } catch { /* older runtimes may not expose tool restriction */ }
-      }
-    })
+      } catch { /* event subscription is an optimization; surface reads remain the fallback */ }
+    }
+    let handle
     try {
+      handle = await this.agents.create({
+        sessionId,
+        meta: { cwd: input.cwd, origin: 'subagent' },
+        agentOptions: { provider, model, maxTokens: input.kind === 'summary' ? 2500 : 6000 },
+        signal: input.signal,
+        setup: async (agentCtx) => {
+          agentCtx?.systemPrompt?.section?.({
+            name: 'knowledge-map:protocol',
+            order: 0,
+            text: '你是 DSH 知识视图生成器。只输出调用方要求的 JSON；不要调用外部网络、文件写入或其他 Agent 工具。'
+          })
+          try {
+            agentCtx?.tools?.restrict?.({ deny: ['multi_agent_discuss', 'shell', 'filesystem', 'web_search', 'browser'] })
+          } catch { /* older runtimes may not expose tool restriction */ }
+        }
+      })
       handle.agent.followup(makeUserMessage(input.prompt, `${sessionId}-${input.kind}`))
       await handle.agent.whenIdle()
+      const liveText = extractAgentText({ events: liveEvents })
+      if (liveText.trim()) return parseStructuredOutput(liveText)
       return parseStructuredOutput(await readAgentText(this.sessionQuery, sessionId, input.signal))
     } finally {
-      await handle.dispose?.()
+      await unsubscribe?.()
+      await handle?.dispose?.()
     }
   }
 

package/lib/index.js

@@ -111,13 +111,15 @@ export function createHost(options = {}) {
       const skills = options.skills || getService(ctx, 'skills')
       const webServer = options.webServer || getService(ctx, 'webServer')
       const storage = options.storage || new WorkspaceStorage(options.storageOptions)
+      const sessionEventSource = options.sessionEventSource || (typeof ctx?.on === 'function' ? (listener) => ctx.on('session/event', listener) : null)
       const orchestrator = options.orchestrator || new KnowledgeGenerationOrchestrator({
         sessionQuery,
         sessions,
         agents,
         agentDefaultModel,
         storage,
-        modelRunner: options.modelRunner
+        modelRunner: options.modelRunner,
+        sessionEventSource
       })
       const confirmations = new Map()
       const sseClients = new Set()

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@p-dsh-market/conversation-knowledge-map",
-  "version": "0.1.1",
+  "version": "0.1.2",
   "description": "DSH 多对话思维导图与静态知识图谱",
   "keywords": [
     "dsh",