@oh-my-pi/pi-agent-core 16.0.4 → 16.0.6

package/CHANGELOG.md

@@ -2,6 +2,30 @@
 
 ## [Unreleased]
 
+## [16.0.6] - 2026-06-18
+
+### Added
+
+- Added `transformAssistantMessage` hook to `AgentOptions` and `Agent` to allow mutating the finalized assistant message before UI emission, context appending, or tool dispatch
+
+## [16.0.5] - 2026-06-17
+
+### Breaking Changes
+
+- Changed `AgentOptions.getApiKey` and `AgentLoopConfig.getApiKey` to receive the active `Model` and return an API key or `ApiKeyResolver`, so credential routing stays model-scoped and retry context is no longer exposed through the agent-core API
+
+### Added
+
+- Added agent-loop deadline support for graceful wall-clock session stops.
+
+### Changed
+
+- Changed Gemini repetition-loop detection to live in the pi-ai stream layer instead of the agent loop. The agent no longer runs its own Gemini-gated verbatim repetition check (`detectRepetition`/`truncateRepetition`); loops now surface as a retryable transient stream error that the standard auto-retry path discards and re-samples, rather than a committed contentful error message.
+
+### Fixed
+
+- Fixed `PI_DIALECT=minimax` being ignored by the owned tool-calling env selector. ([#2759](https://github.com/can1357/oh-my-pi/issues/2759))
+
 ## [16.0.1] - 2026-06-15
 
 ### Fixed

package/README.md

@@ -151,8 +151,8 @@ const agent = new Agent({
   // Custom stream function (for proxy backends)
   streamFn: streamProxy,
 
-  // Dynamic API key resolution (for expiring OAuth tokens)
-  getApiKey: async (provider) => refreshToken(),
+  // Dynamic model-scoped API key resolution (for expiring OAuth tokens)
+  getApiKey: async (model) => tokenForModel(model),
 
   // Tool execution context (late-bound UI/session access)
   getToolContext: () => ({ /* app-defined */ }),

package/dist/types/agent.d.ts

@@ -1,4 +1,4 @@
-import { type ApiKeyResolveContext, type AssistantMessage, type AssistantMessageEvent, type Context, type CursorExecHandlers, type CursorToolResultHandler, type Effort, type ImageContent, type Message, type Model, type ProviderSessionState, type ServiceTier, type SimpleStreamOptions, type ThinkingBudgets, type ToolChoice } from "@oh-my-pi/pi-ai";
+import { type ApiKey, type AssistantMessage, type AssistantMessageEvent, type Context, type CursorExecHandlers, type CursorToolResultHandler, type Effort, type ImageContent, type Message, type Model, type ProviderSessionState, type ServiceTier, type SimpleStreamOptions, type ThinkingBudgets, type ToolChoice } from "@oh-my-pi/pi-ai";
 import type { Dialect } from "@oh-my-pi/pi-ai/dialect";
 import type { HarmonyAuditEvent } from "@oh-my-pi/pi-ai/utils/harmony-leak";
 import type { AppendOnlyContextManager } from "./append-only-context";
@@ -47,6 +47,8 @@ export interface AgentOptions {
      * Custom stream function (for proxy backends, etc.). Default uses streamSimple.
      */
     streamFn?: StreamFn;
+    /** Absolute wall-clock deadline in Unix epoch milliseconds. */
+    deadline?: number;
     /**
      * Optional session identifier forwarded to LLM providers.
      * Used by providers that support session-based caching (e.g., OpenAI Codex).
@@ -62,10 +64,10 @@ export interface AgentOptions {
      */
     providerSessionState?: Map<string, ProviderSessionState>;
     /**
-     * Resolves an API key dynamically for each LLM call.
-     * Useful for expiring tokens (e.g., GitHub Copilot OAuth).
+     * Resolves an API key or resolver dynamically for each LLM call.
+     * Useful for expiring tokens and model-scoped credential routing.
      */
-    getApiKey?: (provider: string, ctx?: ApiKeyResolveContext) => Promise<string | undefined> | string | undefined;
+    getApiKey?: (model: Model) => Promise<ApiKey | undefined> | ApiKey | undefined;
     /**
      * Inspect or replace provider payloads before they are sent.
      */
@@ -156,6 +158,12 @@ export interface AgentOptions {
      * message are emitted. See {@link AgentLoopConfig.afterToolCall} for full semantics.
      */
     afterToolCall?: AgentLoopConfig["afterToolCall"];
+    /**
+     * Called once an assistant message is finalized, before it reaches the
+     * context, the UI, or tool dispatch. May mutate the message in place (text +
+     * tool-call arguments). See {@link AgentLoopConfig.transformAssistantMessage}.
+     */
+    transformAssistantMessage?: AgentLoopConfig["transformAssistantMessage"];
     /**
      * Opt-in OpenTelemetry instrumentation. Passing `{}` enables the loop's
      * GenAI-semantic-convention spans using the global tracer provider. See
@@ -174,7 +182,7 @@ export interface AgentPromptOptions {
 export declare class Agent {
     #private;
     streamFn: StreamFn;
-    getApiKey?: (provider: string, ctx?: ApiKeyResolveContext) => Promise<string | undefined> | string | undefined;
+    getApiKey?: (model: Model) => Promise<ApiKey | undefined> | ApiKey | undefined;
     /**
      * Hook invoked after tool arguments are validated and before execution.
      * Reassign at any time to swap the implementation (e.g. on extension reload).
@@ -185,6 +193,11 @@ export declare class Agent {
      * message emission. Reassign at any time to swap the implementation.
      */
     afterToolCall?: AgentLoopConfig["afterToolCall"];
+    /**
+     * Hook invoked once an assistant message is finalized, before context append,
+     * UI emission, and tool dispatch. Reassign at any time to swap the implementation.
+     */
+    transformAssistantMessage?: AgentLoopConfig["transformAssistantMessage"];
     constructor(opts?: AgentOptions);
     /**
      * Get the current session ID used for provider caching.

package/dist/types/index.d.ts

@@ -6,5 +6,6 @@ export * from "./proxy";
 export * from "./run-collector";
 export * from "./telemetry";
 export * from "./thinking";
+export * from "./tokenizer";
 export * from "./types";
 export * from "./utils/yield";

package/dist/types/tokenizer.d.ts

@@ -0,0 +1 @@
+export declare function countTokens(text: string | string[]): number;

package/dist/types/types.d.ts

@@ -1,4 +1,4 @@
-import type { ApiKeyResolveContext, AssistantMessage, AssistantMessageEvent, AssistantMessageEventStream, Context, Effort, ImageContent, Message, Model, SimpleStreamOptions, Static, streamSimple, TextContent, Tool, ToolChoice, ToolResultMessage, TSchema } from "@oh-my-pi/pi-ai";
+import type { ApiKey, AssistantMessage, AssistantMessageEvent, AssistantMessageEventStream, Context, Effort, ImageContent, Message, Model, SimpleStreamOptions, Static, streamSimple, TextContent, Tool, ToolChoice, ToolResultMessage, TSchema } from "@oh-my-pi/pi-ai";
 import type { Dialect } from "@oh-my-pi/pi-ai/dialect";
 import type { HarmonyAuditEvent } from "@oh-my-pi/pi-ai/utils/harmony-leak";
 import type { AppendOnlyContextManager } from "./append-only-context";
@@ -29,6 +29,8 @@ export interface AgentLoopConfig extends SimpleStreamOptions {
      * Used by providers that support session-based caching (e.g., OpenAI Codex).
      */
     sessionId?: string;
+    /** Absolute wall-clock deadline in Unix epoch milliseconds. */
+    deadline?: number;
     /**
      * Optional resolver called per LLM request to produce request metadata.
      * When set, the agent loop evaluates it **after** `getApiKey` resolves the
@@ -87,12 +89,12 @@ export interface AgentLoopConfig extends SimpleStreamOptions {
      */
     transformProviderContext?: (context: Context, model: Model) => Context;
     /**
-     * Resolves an API key dynamically for each LLM call.
+     * Resolves the API key or resolver for the current model before each LLM call.
      *
-     * Useful for short-lived OAuth tokens (e.g., GitHub Copilot) that may expire
-     * during long-running tool execution phases.
+     * Returning an ApiKeyResolver lets the stream retry policy refresh or rotate
+     * the model-scoped credential after auth/usage-limit errors.
      */
-    getApiKey?: (provider: string, ctx?: ApiKeyResolveContext) => Promise<string | undefined> | string | undefined;
+    getApiKey?: (model: Model) => Promise<ApiKey | undefined> | ApiKey | undefined;
     /**
      * Returns steering messages to inject into the conversation mid-run.
      *
@@ -243,6 +245,19 @@ export interface AgentLoopConfig extends SimpleStreamOptions {
      * backlog so advice produced during the wait is injected as an aside).
      */
     onTurnEnd?: (messages: AgentMessage[], signal?: AbortSignal) => Promise<void> | void;
+    /**
+     * Called once an assistant message is finalized from the model stream, before
+     * it is appended to the context, emitted as `message_end`, or its tool calls
+     * are validated and dispatched. The hook may mutate the message in place —
+     * both its text content and its tool-call arguments — and those edits are seen
+     * by the transcript, the UI, and tool execution alike (single source of truth).
+     *
+     * Used for inline macro expansion: rewriting `@[[runtime.name(args)]]` tokens
+     * to host-computed values before anything downstream consumes the message.
+     * Runs at most once per assistant message; must not throw (a throw would abort
+     * the turn).
+     */
+    transformAssistantMessage?: (message: AssistantMessage, signal?: AbortSignal) => Promise<void> | void;
     /**
      * Called after a tool finishes executing, before `tool_execution_end` and the
      * tool-result message are emitted.

package/package.json

@@ -1,7 +1,7 @@
 {
 	"type": "module",
 	"name": "@oh-my-pi/pi-agent-core",
-	"version": "16.0.4",
+	"version": "16.0.6",
 	"description": "General-purpose agent with transport abstraction, state management, and attachment support",
 	"homepage": "https://omp.sh",
 	"author": "Can Boluk",
@@ -35,11 +35,11 @@
 		"fmt": "biome format --write ."
 	},
 	"dependencies": {
-		"@oh-my-pi/pi-ai": "16.0.4",
-		"@oh-my-pi/pi-catalog": "16.0.4",
-		"@oh-my-pi/pi-natives": "16.0.4",
-		"@oh-my-pi/pi-utils": "16.0.4",
-		"@oh-my-pi/snapcompact": "16.0.4",
+		"@oh-my-pi/pi-ai": "16.0.6",
+		"@oh-my-pi/pi-catalog": "16.0.6",
+		"@oh-my-pi/pi-natives": "16.0.6",
+		"@oh-my-pi/pi-utils": "16.0.6",
+		"@oh-my-pi/snapcompact": "16.0.6",
 		"@opentelemetry/api": "^1.9.1"
 	},
 	"devDependencies": {