@theokit/agents 0.21.2 → 0.23.0

package/dist/{bridge-entry-CEnpN6oR.d.ts → bridge-entry-DlW7p2I4.d.ts}

@@ -1,6 +1,6 @@
 import { ExecutionContext } from '@theokit/http';
-import { A as AgentOptions, T as ToolOptions, g as MainLoopMeta, a as ApprovalOptions, B as BudgetOptions, f as GatewayOptions, l as MemoryOptions, s as SkillsOptions, e as ContextWindowOptions, q as ProjectContextOptions, j as McpServersMap, b as CompactionDecoratorConfig } from './skills-CO7vBqHz.js';
-import { SystemPromptResolver, SkillsSettings, ContextSettings, PluginsSettings, Plugin, ProviderRoutingSettings, AgentDefinition, BudgetTracker, ConversationStorageAdapter, CustomTool } from '@theokit/sdk';
+import { A as AgentOptions, T as ToolOptions, g as MainLoopMeta, a as ApprovalOptions, B as BudgetOptions, f as GatewayOptions, l as MemoryOptions, t as SkillsOptions, e as ContextWindowOptions, q as ProjectContextOptions, j as McpServersMap, b as CompactionDecoratorConfig, R as ReasoningEffort } from './skills-CTzgfoff.js';
+import { SystemPromptResolver, SkillsSettings, ContextSettings, PluginsSettings, Plugin, ProviderRoutingSettings, AgentDefinition, BudgetTracker, ConversationStorageAdapter, CustomTool, ModelSelection } from '@theokit/sdk';
 import { RetryOptions } from '@theokit/sdk/retry';
 import { z } from 'zod';
 
@@ -134,6 +134,10 @@ interface CompiledSubAgent {
 /** Compiled agent options ready for SDK Agent.create(). */
 interface CompiledAgentOptions {
     model?: string;
+    /** Extended-thinking effort declared via `@Agent({ reasoningEffort })`; mapped to SDK ModelSelection.params. */
+    reasoningEffort?: ReasoningEffort;
+    /** Opt-in `<think>`-tag extraction declared via `@Agent({ parseThinkTags })` (M2); wraps the stream when true. */
+    parseThinkTags?: boolean;
     /** Static prompt OR a per-request {@link SystemPromptResolver} (V4-L.1, Axis-B). */
     systemPrompt?: string | SystemPromptResolver;
     tools: CompiledTool[];
@@ -413,6 +417,16 @@ declare function generateAgentRoutes(ctx: AgentRouteContext): AgentRoute[];
 interface RuntimeOverrides {
     /** Overrides the model for this call (`?? compiled.model ?? default`). */
     model?: string;
+    /**
+     * Per-run extended-thinking effort (`?? compiled.reasoningEffort`). Mapped to the SDK
+     * `ModelSelection.params` so the provider produces reasoning (surfaced as `thinking` StreamEvents).
+     */
+    reasoningEffort?: ReasoningEffort;
+    /**
+     * Per-run opt-in (`?? compiled.parseThinkTags`): when true, wrap the event stream with the M2
+     * `<think>`-tag extractor so inline `<think>…</think>` text becomes `thinking` StreamEvents.
+     */
+    parseThinkTags?: boolean;
     /** Per-run cwd → `Agent.create({ local: { cwd } })` → `SystemPromptContext.cwd`. */
     cwd?: string;
     /**
@@ -449,6 +463,70 @@ interface RuntimeOverrides {
  */
 declare function createSdkAgentStream(compiled: CompiledAgentOptions, compiledTools: CompiledTool[], apiKey: string, overrides?: RuntimeOverrides): (message: string, sessionId: string) => AsyncIterable<StreamEvent>;
 
+/**
+ * Model-selection mapping (M1 reasoning-visibility) — the single site that turns a
+ * provider-agnostic `ReasoningEffort` into the SDK `ModelSelection`. Kept in its own small module
+ * (not inside `sdk-adapter.ts`) so the adapter stays under the G6 500-LoC budget and the helper has
+ * a focused home (per the plan's T1.1 "or a small sibling").
+ */
+
+/**
+ * Build the SDK `ModelSelection` for a model id + optional reasoning effort. With no (or empty)
+ * effort it returns the bare `{ id }` — byte-identical to the prior behavior (backward-compat). With
+ * an effort it adds the canonical reasoning param `{ id: 'thinking', value: effort }`. Pure.
+ */
+declare function buildModelSelection(modelId: string, effort?: ReasoningEffort): ModelSelection;
+
+/**
+ * `<think>`-tag reasoning extractor (M2 reasoning-visibility) — bridge middleware.
+ *
+ * Converts inline `<think>…</think>` in the assistant TEXT stream into `thinking` segments, so
+ * models that emit reasoning as inline tags (qwen3-coder / deepseek-class) surface reasoning the
+ * same way native-reasoning providers do (M1's `reasoningEffort`). Two exports:
+ *  - `createThinkTagExtractor` — the pure incremental splitter (handles a tag straddling a chunk
+ *    boundary); the testable core.
+ *  - `extractThinkTagStream` — a StreamEvent transform that applies the extractor to `text_delta`
+ *    events and passes every other event through unchanged (Phase 2).
+ *
+ * Opt-in: wired into `createSdkAgentStream` only when `parseThinkTags` is set (Phase 3) — default
+ * off, because a code assistant can legitimately emit literal `<think>` in answer/code text.
+ *
+ * Reference patterns: Aider `reasoning_tags.py`, Vercel AI SDK `extract-reasoning-middleware.ts`
+ * (blueprint `code-assistant-reasoning-ux` ADR-3).
+ */
+
+/** A typed slice of the text stream: ordinary `text` vs extracted `thinking`. */
+interface Segment {
+    kind: 'text' | 'thinking';
+    content: string;
+}
+/**
+ * A pure, stateful, incremental `<think>` splitter. Feed it text chunks via `write`; it returns the
+ * segments it can resolve so far, holding back only a tail that is still a viable prefix of the
+ * active delimiter (so a tag split across two chunks is recognized). Call `end()` once the stream
+ * is done to flush any buffered tail (a truncated `<think>` with no close is flushed as `thinking`,
+ * so reasoning is never silently dropped).
+ *
+ * One instance per stream (per round); never shared — there is no cross-instance state.
+ */
+declare function createThinkTagExtractor(): {
+    write: (chunk: string) => Segment[];
+    end: () => Segment[];
+};
+/**
+ * Stream transform: convert inline `<think>…</think>` carried in `text_delta` events into `thinking`
+ * events, passing every other event (native `thinking`, `tool_call`, `done`, `error`, …) through
+ * unchanged. A fresh extractor per call ⇒ per-stream (per-round) state; the extractor's mode persists
+ * across interleaved non-text events (a reasoning block split by a tool call is not corrupted). On
+ * source end, the extractor is flushed so a truncated `<think>` is still surfaced (as `thinking`).
+ *
+ * Non-string `text_delta.content` is passed through untouched (defensive — never throws). The
+ * `end()` flush runs in a `finally`, so a buffered unclosed `<think>` is surfaced as `thinking`
+ * even when the source errors mid-stream (the flushed segments are delivered before the error
+ * re-propagates) — never silently dropped (Unbreakable Rule 8: fail loud, lose nothing).
+ */
+declare function extractThinkTagStream(source: AsyncIterable<StreamEvent>): AsyncGenerator<StreamEvent>;
+
 /**
  * Translates @theokit/sdk SDKMessage events → TheoKit AgentStreamEvent.
  *
@@ -798,4 +876,4 @@ declare function agentsPlugin(opts: AgentsPluginOptions): {
     register(app: PluginApp): void;
 };
 
-export { delegate as $, type AgentExecutionContext as A, BudgetExceededError as B, type CompiledAgentOptions as C, type DelegationResult as D, type ErrorEvent as E, type FileEditEvent as F, type RunStartedEvent as G, type SdkMessage as H, type IterationEvent as I, type StateUpdateEvent as J, type ThinkingEvent as K, type LoopStrategy as L, type ToolCallEvent as M, type ToolResultEvent as N, type ToolWalkResult as O, type ToolboxWalkResult as P, agentsPlugin as Q, type ReflectionStrategy as R, type StreamEvent as S, type TextDeltaEvent as T, compileAgent as U, compileContextWindow as V, compileProjectContext as W, compileSkills as X, compileTools as Y, createAgentExecutionContext as Z, createSdkAgentStream as _, type CompiledTool as a, generateAgentManifest as a0, generateAgentRoutes as a1, isAgentContext as a2, isApprovalRequired as a3, isDone as a4, isError as a5, isTextDelta as a6, isToolCall as a7, isToolResult as a8, ladderReflectionStrategy as a9, loopStrategyConfigSchema as aa, noopReflectionStrategy as ab, projectContextMetadataOnlyKnobs as ac, reflectionStrategyConfigSchema as ad, resolveLoopStrategy as ae, streamAgentResponse as af, translateSdkEvent as ag, validateUniqueRoutes as ah, walkAgentMetadata as ai, type AgentManifest as b, type AgentManifestEntry as c, type AgentManifestTool as d, type AgentRoute as e, type AgentRouteContext as f, type AgentRunInfo as g, type AgentStreamEvent as h, type AgentWalkResult as i, AgentWarningCode as j, type AgentsPluginOptions as k, type ApprovalRequiredEvent as l, type ArtifactChunkEvent as m, type ArtifactStartEvent as n, type CheckpointSavedEvent as o, type CompiledContextWindow as p, DEFAULT_MAX_ITERATIONS as q, type DelegateOptions as r, DelegationError as s, type DoneEvent as t, type LoopFinishReason as u, type LoopOutcome as v, type LoopStrategyConfig as w, type ReflectionContext as x, type ReflectionResult as y, type ReflectionStrategyConfig as z };
+export { createAgentExecutionContext as $, type AgentExecutionContext as A, BudgetExceededError as B, type CompiledAgentOptions as C, type DelegationResult as D, type ErrorEvent as E, type FileEditEvent as F, type RunStartedEvent as G, type SdkMessage as H, type IterationEvent as I, type Segment as J, type StateUpdateEvent as K, type LoopStrategy as L, type ThinkingEvent as M, type ToolCallEvent as N, type ToolResultEvent as O, type ToolWalkResult as P, type ToolboxWalkResult as Q, type ReflectionStrategy as R, type StreamEvent as S, type TextDeltaEvent as T, agentsPlugin as U, buildModelSelection as V, compileAgent as W, compileContextWindow as X, compileProjectContext as Y, compileSkills as Z, compileTools as _, type CompiledTool as a, createSdkAgentStream as a0, createThinkTagExtractor as a1, delegate as a2, extractThinkTagStream as a3, generateAgentManifest as a4, generateAgentRoutes as a5, isAgentContext as a6, isApprovalRequired as a7, isDone as a8, isError as a9, isTextDelta as aa, isToolCall as ab, isToolResult as ac, ladderReflectionStrategy as ad, loopStrategyConfigSchema as ae, noopReflectionStrategy as af, projectContextMetadataOnlyKnobs as ag, reflectionStrategyConfigSchema as ah, resolveLoopStrategy as ai, streamAgentResponse as aj, translateSdkEvent as ak, validateUniqueRoutes as al, walkAgentMetadata as am, type AgentManifest as b, type AgentManifestEntry as c, type AgentManifestTool as d, type AgentRoute as e, type AgentRouteContext as f, type AgentRunInfo as g, type AgentStreamEvent as h, type AgentWalkResult as i, AgentWarningCode as j, type AgentsPluginOptions as k, type ApprovalRequiredEvent as l, type ArtifactChunkEvent as m, type ArtifactStartEvent as n, type CheckpointSavedEvent as o, type CompiledContextWindow as p, DEFAULT_MAX_ITERATIONS as q, type DelegateOptions as r, DelegationError as s, type DoneEvent as t, type LoopFinishReason as u, type LoopOutcome as v, type LoopStrategyConfig as w, type ReflectionContext as x, type ReflectionResult as y, type ReflectionStrategyConfig as z };

package/dist/bridge.d.ts

@@ -1,6 +1,6 @@
-export { A as AgentExecutionContext, b as AgentManifest, c as AgentManifestEntry, d as AgentManifestTool, e as AgentRoute, f as AgentRouteContext, g as AgentRunInfo, h as AgentStreamEvent, i as AgentWalkResult, j as AgentWarningCode, k as AgentsPluginOptions, l as ApprovalRequiredEvent, m as ArtifactChunkEvent, n as ArtifactStartEvent, B as BudgetExceededError, o as CheckpointSavedEvent, C as CompiledAgentOptions, p as CompiledContextWindow, a as CompiledTool, r as DelegateOptions, s as DelegationError, D as DelegationResult, t as DoneEvent, E as ErrorEvent, F as FileEditEvent, I as IterationEvent, G as RunStartedEvent, H as SdkMessage, J as StateUpdateEvent, S as StreamEvent, T as TextDeltaEvent, K as ThinkingEvent, M as ToolCallEvent, N as ToolResultEvent, O as ToolWalkResult, P as ToolboxWalkResult, Q as agentsPlugin, U as compileAgent, V as compileContextWindow, W as compileProjectContext, X as compileSkills, Y as compileTools, Z as createAgentExecutionContext, _ as createSdkAgentStream, $ as delegate, a0 as generateAgentManifest, a1 as generateAgentRoutes, a2 as isAgentContext, a3 as isApprovalRequired, a4 as isDone, a5 as isError, a6 as isTextDelta, a7 as isToolCall, a8 as isToolResult, ac as projectContextMetadataOnlyKnobs, af as streamAgentResponse, ag as translateSdkEvent, ah as validateUniqueRoutes, ai as walkAgentMetadata } from './bridge-entry-CEnpN6oR.js';
+export { A as AgentExecutionContext, b as AgentManifest, c as AgentManifestEntry, d as AgentManifestTool, e as AgentRoute, f as AgentRouteContext, g as AgentRunInfo, h as AgentStreamEvent, i as AgentWalkResult, j as AgentWarningCode, k as AgentsPluginOptions, l as ApprovalRequiredEvent, m as ArtifactChunkEvent, n as ArtifactStartEvent, B as BudgetExceededError, o as CheckpointSavedEvent, C as CompiledAgentOptions, p as CompiledContextWindow, a as CompiledTool, r as DelegateOptions, s as DelegationError, D as DelegationResult, t as DoneEvent, E as ErrorEvent, F as FileEditEvent, I as IterationEvent, G as RunStartedEvent, H as SdkMessage, J as Segment, K as StateUpdateEvent, S as StreamEvent, T as TextDeltaEvent, M as ThinkingEvent, N as ToolCallEvent, O as ToolResultEvent, P as ToolWalkResult, Q as ToolboxWalkResult, U as agentsPlugin, V as buildModelSelection, W as compileAgent, X as compileContextWindow, Y as compileProjectContext, Z as compileSkills, _ as compileTools, $ as createAgentExecutionContext, a0 as createSdkAgentStream, a1 as createThinkTagExtractor, a2 as delegate, a3 as extractThinkTagStream, a4 as generateAgentManifest, a5 as generateAgentRoutes, a6 as isAgentContext, a7 as isApprovalRequired, a8 as isDone, a9 as isError, aa as isTextDelta, ab as isToolCall, ac as isToolResult, ag as projectContextMetadataOnlyKnobs, aj as streamAgentResponse, ak as translateSdkEvent, al as validateUniqueRoutes, am as walkAgentMetadata } from './bridge-entry-DlW7p2I4.js';
 import '@theokit/http';
-import './skills-CO7vBqHz.js';
+import './skills-CTzgfoff.js';
 import '@theokit/sdk';
 import 'zod';
 import '@theokit/sdk/retry';

package/dist/bridge.js

@@ -3,6 +3,7 @@ import {
   BudgetExceededError,
   DelegationError,
   agentsPlugin,
+  buildModelSelection,
   compileAgent,
   compileContextWindow,
   compileProjectContext,
@@ -10,7 +11,9 @@ import {
   compileTools,
   createAgentExecutionContext,
   createSdkAgentStream,
+  createThinkTagExtractor,
   delegate,
+  extractThinkTagStream,
   generateAgentManifest,
   generateAgentRoutes,
   isAgentContext,
@@ -25,7 +28,7 @@ import {
   translateSdkEvent,
   validateUniqueRoutes,
   walkAgentMetadata
-} from "./chunk-FRNNQVHX.js";
+} from "./chunk-ZTVXZIBZ.js";
 import "./chunk-GVPUUKKE.js";
 import "./chunk-7QVYU63E.js";
 export {
@@ -33,6 +36,7 @@ export {
   BudgetExceededError,
   DelegationError,
   agentsPlugin,
+  buildModelSelection,
   compileAgent,
   compileContextWindow,
   compileProjectContext,
@@ -40,7 +44,9 @@ export {
   compileTools,
   createAgentExecutionContext,
   createSdkAgentStream,
+  createThinkTagExtractor,
   delegate,
+  extractThinkTagStream,
   generateAgentManifest,
   generateAgentRoutes,
   isAgentContext,

package/dist/{chunk-FRNNQVHX.js → chunk-ZTVXZIBZ.js}

@@ -300,6 +300,8 @@ function compileAgent(walkResult, toolboxInstances = /* @__PURE__ */ new Map())
   const agents = compileSubAgents(walkResult.subAgentClasses);
   return {
     model: walkResult.agentConfig.model,
+    reasoningEffort: walkResult.agentConfig.reasoningEffort,
+    parseThinkTags: walkResult.agentConfig.parseThinkTags,
     systemPrompt: walkResult.agentConfig.systemPrompt,
     tools,
     agents,
@@ -649,6 +651,108 @@ function translateInteractionUpdate(update) {
 }
 __name(translateInteractionUpdate, "translateInteractionUpdate");
 
+// src/bridge/model-selection.ts
+function buildModelSelection(modelId, effort) {
+  if (!effort) return {
+    id: modelId
+  };
+  return {
+    id: modelId,
+    params: [
+      {
+        id: "thinking",
+        value: effort
+      }
+    ]
+  };
+}
+__name(buildModelSelection, "buildModelSelection");
+
+// src/bridge/think-tag-extractor.ts
+var TAG = "think";
+var OPEN = `<${TAG}>`;
+var CLOSE = `</${TAG}>`;
+function heldPrefixLength(s, delim) {
+  const max = Math.min(s.length, delim.length - 1);
+  for (let k = max; k >= 1; k--) {
+    if (s.slice(s.length - k) === delim.slice(0, k)) return k;
+  }
+  return 0;
+}
+__name(heldPrefixLength, "heldPrefixLength");
+function createThinkTagExtractor() {
+  let mode = "text";
+  let buffer = "";
+  const write = /* @__PURE__ */ __name((chunk) => {
+    buffer += chunk;
+    const out = [];
+    for (; ; ) {
+      const delim = mode === "text" ? OPEN : CLOSE;
+      const idx = buffer.indexOf(delim);
+      if (idx !== -1) {
+        const content = buffer.slice(0, idx);
+        if (content) out.push({
+          kind: mode,
+          content
+        });
+        buffer = buffer.slice(idx + delim.length);
+        mode = mode === "text" ? "thinking" : "text";
+        continue;
+      }
+      const keep = heldPrefixLength(buffer, delim);
+      const emit = buffer.slice(0, buffer.length - keep);
+      if (emit) out.push({
+        kind: mode,
+        content: emit
+      });
+      buffer = buffer.slice(buffer.length - keep);
+      break;
+    }
+    return out;
+  }, "write");
+  const end = /* @__PURE__ */ __name(() => {
+    if (!buffer) return [];
+    const seg = {
+      kind: mode,
+      content: buffer
+    };
+    buffer = "";
+    return [
+      seg
+    ];
+  }, "end");
+  return {
+    write,
+    end
+  };
+}
+__name(createThinkTagExtractor, "createThinkTagExtractor");
+function segmentToEvent(seg) {
+  return seg.kind === "thinking" ? {
+    type: "thinking",
+    content: seg.content
+  } : {
+    type: "text_delta",
+    content: seg.content
+  };
+}
+__name(segmentToEvent, "segmentToEvent");
+async function* extractThinkTagStream(source) {
+  const extractor = createThinkTagExtractor();
+  try {
+    for await (const event of source) {
+      if (event.type === "text_delta" && typeof event.content === "string") {
+        for (const seg of extractor.write(event.content)) yield segmentToEvent(seg);
+      } else {
+        yield event;
+      }
+    }
+  } finally {
+    for (const seg of extractor.end()) yield segmentToEvent(seg);
+  }
+}
+__name(extractThinkTagStream, "extractThinkTagStream");
+
 // src/bridge/sdk-adapter.ts
 function assembleM8CreateOptions(compiled) {
   const options = {};
@@ -828,6 +932,8 @@ function createDeltaSink(queue) {
 __name(createDeltaSink, "createDeltaSink");
 function createSdkAgentStream(compiled, compiledTools, apiKey, overrides = {}) {
   const model = overrides.model ?? compiled.model ?? "openai/gpt-4o-mini";
+  const reasoningEffort = overrides.reasoningEffort ?? compiled.reasoningEffort;
+  const parseThinkTags = overrides.parseThinkTags ?? compiled.parseThinkTags ?? false;
   let storage = overrides.conversationStorage;
   return (message, sessionId) => ({
     async *[Symbol.asyncIterator]() {
@@ -877,9 +983,7 @@ function createSdkAgentStream(compiled, compiledTools, apiKey, overrides = {}) {
         }
         agent = await Agent.getOrCreate(sessionId, {
           apiKey,
-          model: {
-            id: model
-          },
+          model: buildModelSelection(model, reasoningEffort),
           tools: sdkTools,
           ...m8,
           ...extra,
@@ -891,7 +995,9 @@ function createSdkAgentStream(compiled, compiledTools, apiKey, overrides = {}) {
           onDelta
         });
         const openStream = /* @__PURE__ */ __name(async () => (await sendPromise).stream(), "openStream");
-        for await (const event of mergeDeltaStream(queue, openStream, runId, state)) {
+        const merged = mergeDeltaStream(queue, openStream, runId, state);
+        const events = parseThinkTags ? extractThinkTagStream(merged) : merged;
+        for await (const event of events) {
           yield event;
         }
         if (!state.sawError) {
@@ -1334,6 +1440,8 @@ var AgentRunner = class {
     const loop = opts.maxIterations != null ? resolveLoopStrategy(this.loopStrategy.name, opts.maxIterations) : this.loopStrategy;
     const streamFactory = opts.streamFactory ?? createSdkAgentStream(this.compiled, tools, opts.apiKey, {
       model: opts.model,
+      reasoningEffort: opts.reasoningEffort,
+      parseThinkTags: opts.parseThinkTags,
       cwd: opts.cwd,
       plugins: opts.plugins,
       providers: opts.providers,
@@ -1615,6 +1723,9 @@ export {
   isApprovalRequired,
   generateAgentRoutes,
   translateSdkEvent,
+  buildModelSelection,
+  createThinkTagExtractor,
+  extractThinkTagStream,
   createSdkAgentStream,
   DEFAULT_KEEP_TOKENS,
   compactionStrategyConfigSchema,
@@ -1634,4 +1745,4 @@ export {
   generateAgentManifest,
   agentsPlugin
 };
-//# sourceMappingURL=chunk-FRNNQVHX.js.map
+//# sourceMappingURL=chunk-ZTVXZIBZ.js.map