@fleetagent/pi-coding-agent 0.0.4 → 0.0.6

package/dist/core/agent-session.js

@@ -663,10 +663,12 @@ export class AgentSession {
         const loaderAppendSystemPrompt = this._resourceLoader.getAppendSystemPrompt();
         const appendSystemPrompt = loaderAppendSystemPrompt.length > 0 ? loaderAppendSystemPrompt.join("\n\n") : undefined;
         const loadedSkills = this._resourceLoader.getSkills().skills;
+        const loadedRules = this._resourceLoader.getRules().rules;
         const loadedContextFiles = this._resourceLoader.getAgentsFiles().agentsFiles;
         this._baseSystemPromptOptions = {
             cwd: this._cwd,
             skills: loadedSkills,
+            rules: loadedRules,
             contextFiles: loadedContextFiles,
             customPrompt: loaderSystemPrompt,
             appendSystemPrompt,
@@ -928,10 +930,11 @@ export class AgentSession {
                     currentImages = inputResult.images ?? currentImages;
                 }
             }
-            // Expand skill commands (/skill:name args) and prompt templates (/template args)
+            // Expand skill/rule commands (/skill:name or /rule:name args) and prompt templates (/template args)
             let expandedText = currentText;
             if (expandPromptTemplates) {
                 expandedText = this._expandSkillCommand(expandedText);
+                expandedText = this._expandRuleCommand(expandedText);
                 expandedText = expandPromptTemplate(expandedText, [...this.promptTemplates]);
             }
             // If streaming, queue via steer() or followUp() based on option
@@ -1084,11 +1087,35 @@ export class AgentSession {
             return text; // Return original on error
         }
     }
+    _expandRuleCommand(text) {
+        if (!text.startsWith("/rule:"))
+            return text;
+        const spaceIndex = text.indexOf(" ");
+        const ruleName = spaceIndex === -1 ? text.slice(6) : text.slice(6, spaceIndex);
+        const args = spaceIndex === -1 ? "" : text.slice(spaceIndex + 1).trim();
+        const rule = this.resourceLoader.getRules().rules.find((r) => r.name === ruleName);
+        if (!rule)
+            return text; // Unknown rule, pass through
+        try {
+            const content = readFileSync(rule.filePath, "utf-8");
+            const body = stripFrontmatter(content).trim();
+            const ruleBlock = `<rule name="${rule.name}" location="${rule.filePath}">\nReferences are relative to ${rule.baseDir}.\n\n${body}\n</rule>`;
+            return args ? `${ruleBlock}\n\n${args}` : ruleBlock;
+        }
+        catch (err) {
+            this._extensionRunner.emitError({
+                extensionPath: rule.filePath,
+                event: "rule_expansion",
+                error: err instanceof Error ? err.message : String(err),
+            });
+            return text;
+        }
+    }
     /**
      * Queue a steering message while the agent is running.
      * Delivered after the current assistant turn finishes executing its tool calls,
      * before the next LLM call.
-     * Expands skill commands and prompt templates. Errors on extension commands.
+     * Expands skill/rule commands and prompt templates. Errors on extension commands.
      * @param images Optional image attachments to include with the message
      * @throws Error if text is an extension command
      */
@@ -1097,15 +1124,16 @@ export class AgentSession {
         if (text.startsWith("/")) {
             this._throwIfExtensionCommand(text);
         }
-        // Expand skill commands and prompt templates
+        // Expand skill/rule commands and prompt templates
         let expandedText = this._expandSkillCommand(text);
+        expandedText = this._expandRuleCommand(expandedText);
         expandedText = expandPromptTemplate(expandedText, [...this.promptTemplates]);
         await this._queueSteer(expandedText, images);
     }
     /**
      * Queue a follow-up message to be processed after the agent finishes.
      * Delivered only when agent has no more tool calls or steering messages.
-     * Expands skill commands and prompt templates. Errors on extension commands.
+     * Expands skill/rule commands and prompt templates. Errors on extension commands.
      * @param images Optional image attachments to include with the message
      * @throws Error if text is an extension command
      */
@@ -1114,8 +1142,9 @@ export class AgentSession {
         if (text.startsWith("/")) {
             this._throwIfExtensionCommand(text);
         }
-        // Expand skill commands and prompt templates
+        // Expand skill/rule commands and prompt templates
         let expandedText = this._expandSkillCommand(text);
+        expandedText = this._expandRuleCommand(expandedText);
         expandedText = expandPromptTemplate(expandedText, [...this.promptTemplates]);
         await this._queueFollowUp(expandedText, images);
     }
@@ -1851,12 +1880,13 @@ export class AgentSession {
         if (!this._extensionRunner.hasHandlers("resources_discover")) {
             return;
         }
-        const { skillPaths, promptPaths, themePaths } = await this._extensionRunner.emitResourcesDiscover(this._cwd, reason);
-        if (skillPaths.length === 0 && promptPaths.length === 0 && themePaths.length === 0) {
+        const { skillPaths, rulePaths, promptPaths, themePaths } = await this._extensionRunner.emitResourcesDiscover(this._cwd, reason);
+        if (skillPaths.length === 0 && rulePaths.length === 0 && promptPaths.length === 0 && themePaths.length === 0) {
             return;
         }
         const extensionPaths = {
             skillPaths: this.buildExtensionResourcePaths(skillPaths),
+            rulePaths: this.buildExtensionResourcePaths(rulePaths),
             promptPaths: this.buildExtensionResourcePaths(promptPaths),
             themePaths: this.buildExtensionResourcePaths(themePaths),
         };
@@ -1926,7 +1956,13 @@ export class AgentSession {
                 source: "skill",
                 sourceInfo: skill.sourceInfo,
             }));
-            return [...extensionCommands, ...templates, ...skills];
+            const rules = this._resourceLoader.getRules().rules.map((rule) => ({
+                name: `rule:${rule.name}`,
+                description: rule.description,
+                source: "rule",
+                sourceInfo: rule.sourceInfo,
+            }));
+            return [...extensionCommands, ...templates, ...skills, ...rules];
         };
         runner.bindCore({
             sendMessage: (message, options) => {