@theokit/sdk 3.1.1 → 3.2.0

package/dist/a2a/index.js

@@ -959,10 +959,10 @@ function buildToolPrompt(prompt) {
 Respond by calling the \`output\` tool with the structured answer that matches the schema.`;
 }
 function setupStructuredOutput(schema, maxRetries) {
-  const z7 = requireZod();
+  const z8 = requireZod();
   const jsonSchema = toJsonSchema(schema, { unrepresentable: "any" });
   return {
-    z: z7,
+    z: z8,
     jsonSchema,
     maxRetries: maxRetries ?? 1,
     initialUsage: { inputTokens: 0, outputTokens: 0 }
@@ -6038,7 +6038,7 @@ async function buildSystemPromptContext(inputs, userText, memoryFacts, manager =
     memory: memoryFacts.map((fact) => ({ text: fact.text }))
   };
 }
-async function buildAssemblyContext(inputs, userText, baseSystemPrompt, memoryFacts, activeMemorySummary) {
+async function buildAssemblyContext(inputs, userText, baseSystemPrompt, memoryFacts, activeMemorySummary, contextPaths) {
   const resolved = await resolveSendSkills(inputs, userText, memoryFacts);
   const baseCtx = await buildSystemPromptContext(inputs, userText, memoryFacts, resolved.manager);
   const assemblyCtx = {
@@ -6051,19 +6051,21 @@ async function buildAssemblyContext(inputs, userText, baseSystemPrompt, memoryFa
     assemblyCtx.activeMemorySummary = activeMemorySummary;
   }
   if (inputs.context !== void 0) {
+    await inputs.context.applyScope(contextPaths);
     const internal = inputs.context.internalAssemblySnapshot();
     assemblyCtx.contextSnapshot = { sources: internal.sources };
     if (internal.maxTokens !== void 0) assemblyCtx.contextMaxTokens = internal.maxTokens;
   }
   return assemblyCtx;
 }
-async function assembleSystemPromptForSend(inputs, userText, baseSystemPrompt, memoryFacts, activeMemorySummary) {
+async function assembleSystemPromptForSend(inputs, userText, baseSystemPrompt, memoryFacts, activeMemorySummary, contextPaths) {
   const ctx = await buildAssemblyContext(
     inputs,
     userText,
     baseSystemPrompt,
     memoryFacts,
-    activeMemorySummary
+    activeMemorySummary,
+    contextPaths
   );
   return inputs.systemPromptPipeline.assemble(ctx);
 }
@@ -6770,6 +6772,14 @@ var init_context_discovery = __esm({
         followImports: false,
         priority: 40
       },
+      {
+        id: "theokit-rules",
+        pattern: ".theokit/rules/*.md",
+        scope: "globbed",
+        parser: "rules-frontmatter",
+        followImports: false,
+        priority: 45
+      },
       {
         id: "theokit-context",
         pattern: ".theokit/context/*.md",
@@ -6853,32 +6863,27 @@ var init_context_import_resolver = __esm({
     MAX_HOPS = 5;
   }
 });
-function parseMdc(content) {
-  const m = FRONTMATTER_RE.exec(content);
-  if (m === null) {
-    return { frontmatter: { alwaysApply: true }, body: content };
-  }
-  const yamlBody = m[1] ?? "";
-  const body = m[2] ?? "";
-  try {
-    const parsed = parseSimpleYaml2(yamlBody);
-    const validated = McdFrontmatterSchema.safeParse(parsed);
-    if (!validated.success) return void 0;
-    return { frontmatter: validated.data, body };
-  } catch {
-    return void 0;
-  }
+
+// src/internal/runtime/context/context-glob.ts
+function globToRegex(glob) {
+  const compiled = glob.replace(/[.+^${}()|[\]\\]/g, "\\$&").replace(/\*\*\//g, "::GLOBSTAR_SLASH::").replace(/\*\*/g, "::GLOBSTAR::").replace(/\*/g, "[^/]*").replace(/\?/g, "[^/]").replace(/::GLOBSTAR_SLASH::/g, "(?:.*/)?").replace(/::GLOBSTAR::/g, ".*");
+  return new RegExp(`^${compiled}$`);
 }
-function shouldActivate(fm, touchedFiles) {
-  if (fm.alwaysApply === true) return true;
-  if (fm.globs === void 0 || fm.globs.length === 0) return false;
-  if (touchedFiles.length === 0) return false;
-  const globRes = fm.globs.map(globToRegex);
-  return touchedFiles.some((file) => globRes.some((re) => re.test(file)));
+function anyGlobMatches(patterns, paths) {
+  if (patterns.length === 0 || paths.length === 0) return false;
+  const res = patterns.map(globToRegex);
+  return paths.some((p) => res.some((re) => re.test(p)));
 }
-function globToRegex(glob) {
-  const escaped = glob.replace(/[.+^${}()|[\]\\]/g, "\\$&").replace(/\*\*/g, "::DOUBLESTAR::").replace(/\*/g, "[^/]*").replace(/::DOUBLESTAR::/g, ".*").replace(/\?/g, ".");
-  return new RegExp(`^${escaped}$`);
+var init_context_glob = __esm({
+  "src/internal/runtime/context/context-glob.ts"() {
+  }
+});
+
+// src/internal/runtime/context/context-yaml-lite.ts
+function splitFrontmatter3(content) {
+  const m = FRONTMATTER_RE.exec(content);
+  if (m === null) return { yaml: void 0, body: content };
+  return { yaml: m[1] ?? "", body: m[2] ?? "" };
 }
 function parseSimpleYaml2(yaml) {
   const out = {};
@@ -6925,15 +6930,76 @@ function parseScalar(s) {
   if (/^-?\d+$/.test(s)) return parseInt(s, 10);
   return s.replace(/^["']|["']$/g, "");
 }
-var McdFrontmatterSchema, FRONTMATTER_RE;
+var FRONTMATTER_RE;
+var init_context_yaml_lite = __esm({
+  "src/internal/runtime/context/context-yaml-lite.ts"() {
+    FRONTMATTER_RE = /^---\s*\n([\s\S]*?)\n---\s*\n([\s\S]*)$/;
+  }
+});
+function parseMdc(content) {
+  const { yaml, body } = splitFrontmatter3(content);
+  if (yaml === void 0) {
+    return { frontmatter: { alwaysApply: true }, body: content };
+  }
+  try {
+    const parsed = parseSimpleYaml2(yaml);
+    const validated = McdFrontmatterSchema.safeParse(parsed);
+    if (!validated.success) return void 0;
+    return { frontmatter: validated.data, body };
+  } catch {
+    return void 0;
+  }
+}
+function shouldActivate(fm, touchedFiles) {
+  if (fm.alwaysApply === true) return true;
+  if (fm.globs === void 0 || fm.globs.length === 0) return false;
+  return anyGlobMatches(fm.globs, touchedFiles);
+}
+var McdFrontmatterSchema;
 var init_context_mdc_parser = __esm({
   "src/internal/runtime/context/context-mdc-parser.ts"() {
+    init_context_glob();
+    init_context_yaml_lite();
     McdFrontmatterSchema = z.object({
       description: z.string().optional(),
       globs: z.array(z.string()).optional(),
       alwaysApply: z.boolean().optional()
     });
-    FRONTMATTER_RE = /^---\s*\n([\s\S]*?)\n---\s*\n([\s\S]*)$/;
+  }
+});
+function parseRules(content) {
+  const { yaml, body } = splitFrontmatter3(content);
+  if (yaml === void 0) {
+    return { frontmatter: { alwaysApply: true }, body: content };
+  }
+  try {
+    const parsed = parseSimpleYaml2(yaml);
+    const validated = RulesFrontmatterSchema.safeParse(parsed);
+    if (!validated.success) return void 0;
+    return { frontmatter: validated.data, body };
+  } catch {
+    return void 0;
+  }
+}
+function shouldActivateRule(fm, inScopePaths) {
+  if (fm.enabled === false) return false;
+  if (fm.alwaysApply === true) return true;
+  const patterns = [...fm.paths ?? [], ...fm.globs ?? []];
+  if (patterns.length === 0) return false;
+  return anyGlobMatches(patterns, inScopePaths);
+}
+var RulesFrontmatterSchema;
+var init_context_rules_frontmatter = __esm({
+  "src/internal/runtime/context/context-rules-frontmatter.ts"() {
+    init_context_glob();
+    init_context_yaml_lite();
+    RulesFrontmatterSchema = z.object({
+      description: z.string().optional(),
+      paths: z.array(z.string()).optional(),
+      globs: z.array(z.string()).optional(),
+      alwaysApply: z.boolean().optional(),
+      enabled: z.boolean().optional()
+    });
   }
 });
 async function runDiscovery(opts) {
@@ -6969,6 +7035,9 @@ async function loadOneSource(spec, path, opts, gitRoot) {
   if (spec.parser === "mdc") {
     return loadMdcSource(spec, path, id, opts);
   }
+  if (spec.parser === "rules-frontmatter") {
+    return loadRulesSource(spec, path, id, opts);
+  }
   if (spec.parser === "frontmatter-zod") {
     return void 0;
   }
@@ -6992,6 +7061,24 @@ async function loadMdcSource(spec, path, id, opts) {
     truncated: false
   };
 }
+async function loadRulesSource(spec, path, id, opts) {
+  let raw;
+  try {
+    raw = await readFile(path, "utf8");
+  } catch {
+    return void 0;
+  }
+  const parsed = parseRules(raw);
+  if (parsed === void 0) return void 0;
+  if (!shouldActivateRule(parsed.frontmatter, opts.touchedFiles ?? [])) return void 0;
+  return {
+    id,
+    source: path,
+    content: parsed.body,
+    priority: spec.priority,
+    truncated: false
+  };
+}
 async function loadPlainMarkdownSource(spec, path, id, opts) {
   const loaded = await loadPlainMarkdown(path, { maxBytesPerFile: opts.maxBytesPerFile });
   if (loaded === void 0) return void 0;
@@ -7017,6 +7104,7 @@ var init_context_discovery_runner = __esm({
     init_context_import_resolver();
     init_context_loaders();
     init_context_mdc_parser();
+    init_context_rules_frontmatter();
   }
 });
 var ContextSourceFrontmatterSchema;
@@ -7152,6 +7240,10 @@ function matchesGlob(pattern, path) {
 function tokenizeContent(content) {
   return content.split(/\s+/).filter((token) => token.length > 0);
 }
+function sameScope(a, b) {
+  if (a.length !== b.length) return false;
+  return a.every((v, i) => v === b[i]);
+}
 var FileContextManager;
 var init_context_manager = __esm({
   "src/internal/runtime/context/context-manager.ts"() {
@@ -7172,6 +7264,13 @@ var init_context_manager = __esm({
       settings;
       settingSourcesIncludeProject;
       state;
+      /**
+       * The in-scope file set used by the most recent `refresh`. Tracked so
+       * `applyScope` only touches state when the scope actually changes — a
+       * caller that never scopes (the common case) keeps the create-time
+       * snapshot untouched, at zero per-send cost.
+       */
+      lastScope = [];
       async initialize() {
         if (!this.settingSourcesIncludeProject && this.settings.manager !== "file") {
           this.state = { config: { sources: [] }, loadedSources: [] };
@@ -7179,14 +7278,31 @@ var init_context_manager = __esm({
         }
         await this.refresh();
       }
-      async refresh() {
+      /**
+       * Re-run discovery for a specific per-send in-scope file set (T3). Path-scoped
+       * rules (`.theokit/rules/*.md`, `.cursor/rules/*.mdc` globs) activate iff a
+       * pattern matches one of `contextPaths`. Idempotent: unchanged scope is a
+       * no-op, and `undefined` scope while never previously scoped never touches
+       * state (preserves the create-time snapshot for non-users).
+       */
+      async applyScope(contextPaths) {
+        const scope = contextPaths ?? [];
+        if (scope.length === 0 && this.lastScope.length === 0) return;
+        if (sameScope(scope, this.lastScope)) return;
+        await this.refresh({ touchedFiles: scope });
+      }
+      async refresh(opts) {
+        const touchedFiles = opts?.touchedFiles ?? [];
+        this.lastScope = [...touchedFiles];
         const config = await loadContextConfig(this.cwd);
         const legacy = await loadSources(config, this.cwd);
         const maxBytesPerFile = this.settings.maxBytesPerFile ?? DEFAULT_MAX_BYTES_PER_FILE;
         const maxBytesTotal = this.settings.maxBytesTotal ?? DEFAULT_MAX_BYTES_TOTAL;
         const discovered = await runDiscovery({
           cwd: this.cwd,
-          maxBytesPerFile});
+          maxBytesPerFile,
+          touchedFiles
+        });
         const legacyAsAggregator = legacy.map((src) => ({
           id: src.name,
           source: src.path,
@@ -16963,7 +17079,8 @@ async function executeSendLocked(inputs, message, options) {
     userText,
     baseSystemPrompt,
     memoryFacts,
-    activeMemorySummary
+    activeMemorySummary,
+    options.contextPaths
   );
   const projectedSystemPrompt = await projectCurrentObjective(
     inputs.storageHandle,
@@ -17851,7 +17968,7 @@ var init_local_agent = __esm({
             runPreHook: (ut) => this.runPreHook(ut),
             // biome-ignore format: G8 budget — callbacks wire to private methods.
             resolveSystemPromptForSend: (ut, o, mf) => this.resolveSystemPrompt(ut, o, mf),
-            assembleSystemPromptForSend: (ut, bp, mf, ams) => assembleSystemPromptForSend(this.assemblyInputs(), ut, bp, mf, ams),
+            assembleSystemPromptForSend: (ut, bp, mf, ams, cp) => assembleSystemPromptForSend(this.assemblyInputs(), ut, bp, mf, ams, cp),
             dispatchRun: (msg, o, sp, mf, pm, mt, mp) => this.dispatchRun(msg, o, sp, mf, pm, mt, mp)
           },
           message,