Merge pull request #6070 from continuedev/nate/rules-as-context-providers

RulesContextProvider

Author: sestinj

core/config/profile/doLoadConfig.ts

@@ -14,12 +14,14 @@ import {
   IDE,
   IdeSettings,
   ILLMLogger,
+  RuleWithSource,
   SerializedContinueConfig,
   Tool,
 } from "../../";
 import { constructMcpSlashCommand } from "../../commands/slash/mcp";
 import { MCPManagerSingleton } from "../../context/mcp/MCPManagerSingleton";
 import MCPContextProvider from "../../context/providers/MCPContextProvider";
+import RulesContextProvider from "../../context/providers/RulesContextProvider";
 import { ControlPlaneProxyInfo } from "../../control-plane/analytics/IAnalyticsProvider.js";
 import { ControlPlaneClient } from "../../control-plane/client.js";
 import { getControlPlaneEnv } from "../../control-plane/env.js";
@@ -39,6 +41,24 @@ import { migrateJsonSharedConfig } from "../migrateSharedConfig";
 import { rectifySelectedModelsFromGlobalContext } from "../selectedModels";
 import { loadContinueConfigFromYaml } from "../yaml/loadYaml";
 
+async function loadRules(ide: IDE) {
+  const rules: RuleWithSource[] = [];
+  const errors = [];
+
+  // Add rules from .continuerules files
+  const { rules: yamlRules, errors: continueRulesErrors } =
+    await getWorkspaceContinueRuleDotFiles(ide);
+  rules.unshift(...yamlRules);
+  errors.push(...continueRulesErrors);
+
+  // Add rules from markdown files in .continue/rules
+  const { rules: markdownRules, errors: markdownRulesErrors } =
+    await loadMarkdownRules(ide);
+  rules.unshift(...markdownRules);
+  errors.push(...markdownRulesErrors);
+
+  return { rules, errors };
+}
 export default async function doLoadConfig(options: {
   ide: IDE;
   ideSettingsPromise: Promise<IdeSettings>;
@@ -125,17 +145,11 @@ export default async function doLoadConfig(options: {
   // Remove ability have undefined errors, just have an array
   errors = [...(errors ?? [])];
 
-  // Add rules from .continuerules files
-  const { rules, errors: continueRulesErrors } =
-    await getWorkspaceContinueRuleDotFiles(ide);
+  // Load rules and always include the RulesContextProvider
+  const { rules, errors: rulesErrors } = await loadRules(ide);
+  errors.push(...rulesErrors);
   newConfig.rules.unshift(...rules);
-  errors.push(...continueRulesErrors);
-
-  // Add rules from markdown files in .continue/rules
-  const { rules: markdownRules, errors: markdownRulesErrors } =
-    await loadMarkdownRules(ide);
-  newConfig.rules.unshift(...markdownRules);
-  errors.push(...markdownRulesErrors);
+  newConfig.contextProviders.push(new RulesContextProvider({}));
 
   // Rectify model selections for each role
   newConfig = rectifySelectedModelsFromGlobalContext(newConfig, profileId);

core/context/providers/RulesContextProvider.ts

@@ -0,0 +1,91 @@
+import { BaseContextProvider } from "..";
+import {
+  ContextItem,
+  ContextItemUri,
+  ContextProviderDescription,
+  ContextProviderExtras,
+  ContextSubmenuItem,
+  LoadSubmenuItemsArgs,
+  RuleWithSource,
+} from "../..";
+import { getControlPlaneEnv } from "../../control-plane/env";
+
+class RulesContextProvider extends BaseContextProvider {
+  static description: ContextProviderDescription = {
+    title: "rules",
+    displayTitle: "Rules",
+    description: "Mention rules files",
+    type: "submenu",
+    renderInlineAs: "",
+  };
+
+  // This is only used within this class. Worst case if there are exact duplicates is that one always calls the other, but this is an extreme edge case
+  // Can eventually pull in more metadata, but this is experimental
+  private getIdFromRule(rule: RuleWithSource): string {
+    return rule.ruleFile ?? rule.slug ?? rule.name ?? rule.rule;
+  }
+
+  private getNameFromRule(rule: RuleWithSource): string {
+    return rule.name ?? rule.slug ?? rule.ruleFile ?? rule.source;
+  }
+
+  private getDescriptionFromRule(rule: RuleWithSource): string {
+    return rule.description ?? rule.name ?? "";
+  }
+
+  private getUriFromRule(
+    rule: RuleWithSource,
+    appUrl: string,
+  ): ContextItemUri | undefined {
+    if (rule.ruleFile) {
+      return {
+        type: "file",
+        value: rule.ruleFile,
+      };
+    }
+
+    if (rule.slug) {
+      let url = `${appUrl}${rule.slug}`;
+      return {
+        type: "url",
+        value: url,
+      };
+    }
+
+    return undefined;
+  }
+
+  async getContextItems(
+    query: string,
+    extras: ContextProviderExtras,
+  ): Promise<ContextItem[]> {
+    const rule = extras.config.rules.find(
+      (rule) => this.getIdFromRule(rule) === query,
+    );
+    if (!rule) {
+      return [];
+    }
+
+    const env = await getControlPlaneEnv(extras.ide.getIdeSettings());
+    return [
+      {
+        name: this.getNameFromRule(rule),
+        content: rule.rule,
+        description: this.getDescriptionFromRule(rule),
+        uri: this.getUriFromRule(rule, env.APP_URL),
+      },
+    ];
+  }
+
+  async loadSubmenuItems(
+    args: LoadSubmenuItemsArgs,
+  ): Promise<ContextSubmenuItem[]> {
+    return args.config.rules.map((rule) => ({
+      id: this.getIdFromRule(rule),
+      description: this.getDescriptionFromRule(rule),
+      title: this.getNameFromRule(rule),
+    }));
+  }
+}
+
+export default RulesContextProvider;

core/context/providers/index.ts

@@ -25,6 +25,7 @@ import OSContextProvider from "./OSContextProvider";
 import PostgresContextProvider from "./PostgresContextProvider";
 import ProblemsContextProvider from "./ProblemsContextProvider";
 import RepoMapContextProvider from "./RepoMapContextProvider";
+import RulesContextProvider from "./RulesContextProvider";
 import SearchContextProvider from "./SearchContextProvider";
 import TerminalContextProvider from "./TerminalContextProvider";
 import URLContextProvider from "./URLContextProvider";
@@ -66,6 +67,7 @@ export const Providers: (typeof BaseContextProvider)[] = [
   MCPContextProvider,
   GitCommitContextProvider,
   ClipboardContextProvider,
+  RulesContextProvider,
 ];
 
 export function contextProviderClassFromName(

extensions/vscode/package-lock.json

@@ -292,6 +292,26 @@ const AtMentionDropdown = forwardRef((props: AtMentionDropdownProps, ref) => {
         },
         description: "Create a new .prompt file",
       });
+    } else if (subMenuTitle === "Mention rules files") {
+      items.push({
+        title: "Add new rule",
+        type: "action",
+        action: () => {
+          void ideMessenger.request("config/addLocalWorkspaceBlock", {
+            blockType: "rules",
+          });
+          const { tr } = props.editor.view.state;
+          const text = tr.doc.textBetween(0, tr.selection.from);
+          const start = text.lastIndexOf("@");
+          if (start !== -1) {
+            props.editor.view.dispatch(
+              tr.delete(start, tr.selection.from).scrollIntoView(),
+            );
+          }
+          props.onClose(); // Escape the mention list after creating a new rule
+        },
+        description: "Creates a rule file",
+      });
     }
     setLoadingSubmenuItem(items.find((item) => item.id === "loading"));
     setAllItems(items.filter((item) => item.id !== "loading"));

gui/src/components/mainInput/AtMentionDropdown/index.tsx

@@ -14,6 +14,7 @@ import {
   GlobeAltIcon,
   MagnifyingGlassIcon,
   PaperClipIcon,
+  PencilIcon,
   PlusIcon,
   SparklesIcon,
   TrashIcon,
@@ -42,6 +43,7 @@ export const NAMED_ICONS: { [key: string]: any } = {
   debugger: BugAntIcon,
   os: CpuChipIcon,
   tree: Bars3BottomLeftIcon,
+  rules: PencilIcon,
   "prompt-files": DocumentTextIcon,
   "repo-map": FolderIcon,
   "/clear": TrashIcon,