@vibe-forge/workspace-assets 0.11.1 → 0.11.2

package/__tests__/__snapshots__/workspace-assets-rich.snapshot.json

@@ -467,6 +467,7 @@
       ]
     },
     "options": {
+      "systemPrompt": "<system-prompt>\nThe project system rules are:\n# review\n\n> 必须检查发布改动的回归风险。\n\n# demo/security\n\n> Use when: 插件安全规则\n> Rule file path: node_modules/@vibe-forge/plugin-demo/rules/security.md\n> Only read this rule file when the task matches the scenario above.\n</system-prompt>\n\n\n<system-prompt>\nThe following skill modules are loaded for the project:\n# research\n\n> Skill description: 检索资料\n> Skill file path: .ai/skills/research/SKILL.md\n> Resolve relative paths in the resource content relative to the directory containing this skill file.\n\n<skill-content>\n先阅读 README.md，再补充结论。\n</skill-content>\n</system-prompt>\n\n\n<system-prompt>\nThe project includes the following entities:\n  - architect: 负责拆解方案的实体\n\nWhen solving user problems, you may specify entities through `VibeForge.StartTasks` as needed and have them coordinate multiple entity types to complete the work; use `VibeForge.GetTaskInfo` and `wait` to track progress.\n</system-prompt>\n\n\n<system-prompt>\nYou are a professional project execution manager who can skillfully direct other entities to work toward your goal. Expectations:\n\n- Never complete code development work alone\n- You must coordinate other developers to complete tasks\n- You must keep them aligned with the goal and verify that their completion reports meet the requirements\n\nChoose the appropriate workflow based on the user's needs and the actual development goal, and use the workflow identifier to locate and load the corresponding definition.\n- Pass the identifier based on the actual need. This is not a path; use the standard workflow loading capability to resolve it.\n- Decide how to pass parameters based on their descriptions and actual usage scenarios.\nThe project includes the following workflows:\n- Workflow name: release\n  - Description: 正式发布流程\n  - Identifier: release\n  - Parameters:\n    - None\n\n- Workflow name: demo/release\n  - Description: 插件发布流程\n  - Identifier: demo/release\n  - Parameters:\n    - None\n\n</system-prompt>\n\n\n执行正式发布，并整理变更摘要。",
       "systemPrompt": "<system-prompt>\nThe project system rules are:\n# review\n\n> 必须检查发布改动的回归风险。\n\n# demo/security\n\n> Use when: 插件安全规则\n> Rule file path: node_modules/@vibe-forge/plugin-demo/rules/security.md\n> Only read this rule file when the task matches the scenario above.\n</system-prompt>\n\n\n<system-prompt>\nThe following skill modules are loaded for the project:\n# research\n\n> Skill description: 检索资料\n> Skill file path: .ai/skills/research/SKILL.md\n> Resolve relative paths in the resource content relative to the directory containing this skill file.\n\n<skill-content>\n先阅读 README.md，再补充结论。\n</skill-content>\n</system-prompt>\n\n\n<system-prompt>\nThe project includes the following entities:\n  - architect: 负责拆解方案的实体\n\nWhen solving user problems, you may specify entities through `VibeForge.StartTasks` as needed and have them coordinate multiple entity types to complete the work; use `VibeForge.GetTaskInfo` and `wait` to track progress.\n</system-prompt>\n\n\n<system-prompt>\nYou are a professional project execution manager who can skillfully direct other entities to work toward your goal. Expectations:\n\n- Never complete code development work alone\n- You must coordinate other developers to complete tasks\n- You must keep them aligned with the goal and verify that their completion reports meet the requirements\n\nChoose the appropriate workflow based on the user's needs and the actual development goal, and use the workflow identifier to locate and load the corresponding definition.\n- Pass the identifier based on the actual need. This is not a path; use the standard workflow loading capability to resolve it.\n- Decide how to pass parameters based on their descriptions and actual usage scenarios.\nThe project includes the following workflows:\n- Workflow name: release\n  - Description: 正式发布流程\n  - Identifier: release\n  - Parameters:\n    - None\n\n- Workflow name: demo/release\n  - Description: 插件发布流程\n  - Identifier: demo/release\n  - Parameters:\n    - None\n\n</system-prompt>\n\n\n执行正式发布，并整理变更摘要。",
       "tools": {
         "include": [

package/__tests__/bundle.spec.ts

@@ -68,6 +68,57 @@ describe('resolveWorkspaceAssetBundle', () => {
     ]))
   })
 
+  it('loads workspace assets from the env-configured ai base dir', async () => {
+    const workspace = await createWorkspace()
+    const previousBaseDir = process.env.__VF_PROJECT_AI_BASE_DIR__
+
+    try {
+      process.env.__VF_PROJECT_AI_BASE_DIR__ = '.vf'
+      await writeDocument(join(workspace, '.vf/rules/review.md'), '---\ndescription: 评审规则\n---\n必须检查风险')
+      await writeDocument(join(workspace, '.vf/skills/research/SKILL.md'), '---\ndescription: 检索资料\n---\n阅读 README.md')
+
+      const bundle = await resolveWorkspaceAssetBundle({
+        cwd: workspace,
+        configs: [undefined, undefined],
+        useDefaultVibeForgeMcpServer: false
+      })
+
+      expect(bundle.rules.map(asset => asset.displayName)).toEqual(['review'])
+      expect(bundle.skills.map(asset => asset.displayName)).toEqual(['research'])
+    } finally {
+      if (previousBaseDir == null) {
+        delete process.env.__VF_PROJECT_AI_BASE_DIR__
+      } else {
+        process.env.__VF_PROJECT_AI_BASE_DIR__ = previousBaseDir
+      }
+    }
+  })
+
+  it('loads workspace entities from the env-configured entities dir', async () => {
+    const workspace = await createWorkspace()
+    const previousEntitiesDir = process.env.__VF_PROJECT_AI_ENTITIES_DIR__
+
+    try {
+      process.env.__VF_PROJECT_AI_ENTITIES_DIR__ = 'agents'
+      await writeDocument(join(workspace, '.ai/agents/reviewer/README.md'), '---\ndescription: 负责代码评审\n---\n检查风险')
+
+      const bundle = await resolveWorkspaceAssetBundle({
+        cwd: workspace,
+        configs: [undefined, undefined],
+        useDefaultVibeForgeMcpServer: false
+      })
+
+      expect(bundle.entities.map(asset => asset.displayName)).toEqual(['reviewer'])
+      expect(bundle.entities[0]?.sourcePath).toContain('/.ai/agents/reviewer/README.md')
+    } finally {
+      if (previousEntitiesDir == null) {
+        delete process.env.__VF_PROJECT_AI_ENTITIES_DIR__
+      } else {
+        process.env.__VF_PROJECT_AI_ENTITIES_DIR__ = previousEntitiesDir
+      }
+    }
+  })
+
   it('auto-loads managed Claude plugins from .ai/plugins as directory plugins', async () => {
     const workspace = await createWorkspace()
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@vibe-forge/workspace-assets",
-  "version": "0.11.1",
+  "version": "0.11.2",
   "description": "Workspace asset resolution and adapter asset planning for Vibe Forge",
   "imports": {
     "#~/*.js": {
@@ -32,7 +32,7 @@
     "@vibe-forge/config": "^0.11.1",
     "@vibe-forge/definition-core": "^0.11.0",
     "@vibe-forge/types": "^0.11.2",
-    "@vibe-forge/utils": "^0.11.1"
+    "@vibe-forge/utils": "^0.11.2"
   },
   "devDependencies": {
     "@types/js-yaml": "^4.0.9"

package/src/bundle-internal.ts

@@ -9,7 +9,7 @@ import {
   resolveDefaultVibeForgeMcpServerConfig
 } from '@vibe-forge/config'
 import type { Config, Definition, Entity, PluginConfig, WorkspaceAsset, WorkspaceAssetKind } from '@vibe-forge/types'
-import { resolveRelativePath } from '@vibe-forge/utils'
+import { resolveProjectAiBaseDir, resolveProjectAiEntitiesDir, resolveRelativePath } from '@vibe-forge/utils'
 import { listManagedPluginInstalls, toManagedPluginConfig } from '@vibe-forge/utils/managed-plugin'
 import {
   flattenPluginInstances,
@@ -217,13 +217,15 @@ const createOpenCodeOverlayAsset = <TKind extends OpenCodeOverlayKind>(params: {
 } as OpenCodeOverlayAsset<TKind>)
 
 const scanWorkspaceDocuments = async (cwd: string) => {
+  const aiBaseDir = resolveProjectAiBaseDir(cwd, process.env)
+  const entitiesDir = resolveProjectAiEntitiesDir(cwd, process.env)
   const [rulePaths, skillPaths, specPaths, entityDocPaths, entityJsonPaths, mcpPaths] = await Promise.all([
-    glob(['.ai/rules/*.md'], { cwd, absolute: true }),
-    glob(['.ai/skills/*/SKILL.md'], { cwd, absolute: true }),
-    glob(['.ai/specs/*.md', '.ai/specs/*/index.md'], { cwd, absolute: true }),
-    glob(['.ai/entities/*.md', '.ai/entities/*/README.md'], { cwd, absolute: true }),
-    glob(['.ai/entities/*/index.json'], { cwd, absolute: true }),
-    glob(['.ai/mcp/*.json', '.ai/mcp/*.yaml', '.ai/mcp/*.yml'], { cwd, absolute: true })
+    glob(['rules/*.md'], { cwd: aiBaseDir, absolute: true }),
+    glob(['skills/*/SKILL.md'], { cwd: aiBaseDir, absolute: true }),
+    glob(['specs/*.md', 'specs/*/index.md'], { cwd: aiBaseDir, absolute: true }),
+    glob(['*.md', '*/README.md'], { cwd: entitiesDir, absolute: true }),
+    glob(['*/index.json'], { cwd: entitiesDir, absolute: true }),
+    glob(['mcp/*.json', 'mcp/*.yaml', 'mcp/*.yml'], { cwd: aiBaseDir, absolute: true })
   ])
 
   return {
@@ -441,7 +443,7 @@ export async function collectWorkspaceAssets(params: {
         name: DEFAULT_VIBE_FORGE_MCP_SERVER_NAME,
         config: defaultVibeForgeMcpServer,
         origin: 'workspace',
-        sourcePath: resolve(params.cwd, '.ai')
+        sourcePath: resolveProjectAiBaseDir(params.cwd, process.env)
       }))
     }
   }