dsh-plugin-workbench 0.0.11 → 0.0.13

package/src/dsh.d.ts

@@ -25,6 +25,18 @@ declare module '@deepseek-ai/cordis' {
   interface Context {
     /** Register a lifecycle effect; the callback may return a disposer. */
     effect(callback: () => void | (() => void), name?: string): void
+    /** Subscribe to an event; returns the disposer. */
+    on(event: string, listener: (...args: unknown[]) => void): () => void
+    /** Resolve scoped services and run a callback in that scope. */
+    inject(deps: string[], callback: (scope: Context) => void): { dispose(): unknown }
+    /** Agent registry (host side; used to teach each agent the mention grammar). */
+    agents: {
+      list(): Array<{ ctx: Context }>
+    }
+    systemPrompt: {
+      /** Register one ordered system-prompt section; returns the disposer. */
+      section(section: { name: string; order: number; text: string | ((context: unknown) => string) }): () => void
+    }
     connection: {
       rpc: {
         /** Browser-side unary call over a registered logical channel. */

package/src/index.ts

@@ -36,7 +36,7 @@ import { homedir } from 'node:os'
 import type { Context } from '@deepseek-ai/cordis'
 
 export const name = 'dsh-plugin-workbench'
-export const inject = ['fs', 'connection', 'webServer']
+export const inject = ['fs', 'connection', 'webServer', 'systemPrompt', 'agents']
 
 /** Loopback-only logical RPC channel. */
 export const CHANNEL = '/dsh-plugin-files'
@@ -330,6 +330,40 @@ function ensureLayoutPatch(): void {
   }
 }
 
+/**
+ * Stable system-prompt section teaching the model the workbench `@.\` mention
+ * grammar. The core already contributes a generic "paths prefixed with @ are
+ * files referenced by the user" section (dsh-file-reference-local, order 99);
+ * this one documents the actual syntax the GUI inserts (the `.\` workspace
+ * marker, quoted form for paths with spaces) so the model resolves mentions
+ * against the session workspace root instead of treating `@.\…` as noise.
+ * Static text only — the string is identical on every assembly (KV-cache safe).
+ */
+const FILE_MENTION_PROMPT =
+  'Workspace file mentions written by the file panel use the form `@.<relative-path>` (a `.\` or `./` prefix marks a path relative to the current session workspace root; both `\\` and `/` separators are accepted, and paths with spaces use the quoted form `@"<relative-path>"`, for example `@"\\.\\my plan.md"`). Treat any such mention as a file the user wants you to read or edit: resolve the path against the workspace root and act on it with the file tools (read, glob, grep, edit, write); never claim to have inspected a file you did not actually open.'
+
+/**
+ * File-mention grammar taught to the model (see the FILE_MENTION_PROMPT doc).
+ * One section per agent fiber: existing agents at apply time plus every agent
+ * created later (agent/created). The core already contributes a generic "paths
+ * prefixed with @ are files" section (dsh-file-reference-local, order 99);
+ * order 100 puts this more specific grammar right after it.
+ */
+function installMentionPrompt(ctx: Context): void {
+  const teach = (agent: { ctx: Context }): void => {
+    agent.ctx.systemPrompt.section({
+      name: 'dsh-plugin-workbench:file-mention',
+      order: 100,
+      text: FILE_MENTION_PROMPT,
+    })
+  }
+  for (const agent of ctx.agents.list()) teach(agent)
+  ctx.on('agent/created', (payload: unknown) => {
+    const { agent } = payload as { agent: { ctx: Context } }
+    teach(agent)
+  })
+}
+
 /**
  * One filesystem-backed RPC endpoint pair. Reads never mutate; `signal`
  * cancels the underlying fs call (or aborts between steps).
@@ -337,6 +371,8 @@ function ensureLayoutPatch(): void {
 export function apply(ctx: Context): void {
   // Re-apply the ui-layout explorer-column patch when a dsh upgrade reverted it.
   ensureLayoutPatch()
+  // Teach every agent the workbench `@.\` mention grammar (per-agent fiber).
+  installMentionPrompt(ctx)
   // Per-apply watch state: created here (not module-level) so disable/reload
   // cycles never leak watchers or SSE clients across applies.
   const watchState: WatchState = {