@theokit/agents 0.22.0 → 0.23.0

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

@@ -1,5 +1,5 @@
 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, t as SkillsOptions, e as ContextWindowOptions, q as ProjectContextOptions, j as McpServersMap, b as CompactionDecoratorConfig, R as ReasoningEffort } from './skills-BzEzl0YN.js';
+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';
@@ -136,6 +136,8 @@ 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[];
@@ -420,6 +422,11 @@ interface RuntimeOverrides {
      * `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;
     /**
@@ -470,6 +477,56 @@ declare function createSdkAgentStream(compiled: CompiledAgentOptions, compiledTo
  */
 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.
  *
@@ -819,4 +876,4 @@ declare function agentsPlugin(opts: AgentsPluginOptions): {
     register(app: PluginApp): void;
 };
 
-export { createSdkAgentStream 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, buildModelSelection as U, compileAgent as V, compileContextWindow as W, compileProjectContext as X, compileSkills as Y, compileTools as Z, createAgentExecutionContext as _, type CompiledTool as a, delegate as a0, generateAgentManifest as a1, generateAgentRoutes as a2, isAgentContext as a3, isApprovalRequired as a4, isDone as a5, isError as a6, isTextDelta as a7, isToolCall as a8, isToolResult as a9, ladderReflectionStrategy as aa, loopStrategyConfigSchema as ab, noopReflectionStrategy as ac, projectContextMetadataOnlyKnobs as ad, reflectionStrategyConfigSchema as ae, resolveLoopStrategy as af, streamAgentResponse as ag, translateSdkEvent as ah, validateUniqueRoutes as ai, walkAgentMetadata as aj, 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 buildModelSelection, V as compileAgent, W as compileContextWindow, X as compileProjectContext, Y as compileSkills, Z as compileTools, _ as createAgentExecutionContext, $ as createSdkAgentStream, a0 as delegate, a1 as generateAgentManifest, a2 as generateAgentRoutes, a3 as isAgentContext, a4 as isApprovalRequired, a5 as isDone, a6 as isError, a7 as isTextDelta, a8 as isToolCall, a9 as isToolResult, ad as projectContextMetadataOnlyKnobs, ag as streamAgentResponse, ah as translateSdkEvent, ai as validateUniqueRoutes, aj as walkAgentMetadata } from './bridge-entry-CHodDWs0.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-BzEzl0YN.js';
+import './skills-CTzgfoff.js';
 import '@theokit/sdk';
 import 'zod';
 import '@theokit/sdk/retry';

package/dist/bridge.js

@@ -11,7 +11,9 @@ import {
   compileTools,
   createAgentExecutionContext,
   createSdkAgentStream,
+  createThinkTagExtractor,
   delegate,
+  extractThinkTagStream,
   generateAgentManifest,
   generateAgentRoutes,
   isAgentContext,
@@ -26,7 +28,7 @@ import {
   translateSdkEvent,
   validateUniqueRoutes,
   walkAgentMetadata
-} from "./chunk-ITKVE65Q.js";
+} from "./chunk-ZTVXZIBZ.js";
 import "./chunk-GVPUUKKE.js";
 import "./chunk-7QVYU63E.js";
 export {
@@ -42,7 +44,9 @@ export {
   compileTools,
   createAgentExecutionContext,
   createSdkAgentStream,
+  createThinkTagExtractor,
   delegate,
+  extractThinkTagStream,
   generateAgentManifest,
   generateAgentRoutes,
   isAgentContext,

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

@@ -301,6 +301,7 @@ function compileAgent(walkResult, toolboxInstances = /* @__PURE__ */ new Map())
   return {
     model: walkResult.agentConfig.model,
     reasoningEffort: walkResult.agentConfig.reasoningEffort,
+    parseThinkTags: walkResult.agentConfig.parseThinkTags,
     systemPrompt: walkResult.agentConfig.systemPrompt,
     tools,
     agents,
@@ -667,6 +668,91 @@ function buildModelSelection(modelId, 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 = {};
@@ -847,6 +933,7 @@ __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]() {
@@ -908,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) {
@@ -1352,6 +1441,7 @@ var AgentRunner = class {
     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,
@@ -1634,6 +1724,8 @@ export {
   generateAgentRoutes,
   translateSdkEvent,
   buildModelSelection,
+  createThinkTagExtractor,
+  extractThinkTagStream,
   createSdkAgentStream,
   DEFAULT_KEEP_TOKENS,
   compactionStrategyConfigSchema,
@@ -1653,4 +1745,4 @@ export {
   generateAgentManifest,
   agentsPlugin
 };
-//# sourceMappingURL=chunk-ITKVE65Q.js.map
+//# sourceMappingURL=chunk-ZTVXZIBZ.js.map