@oh-my-pi/pi-agent-core 3.20.0 → 3.21.0

package/README.md

@@ -1,6 +1,6 @@
 # @oh-my-pi/pi-agent
 
-Stateful agent with tool execution and event streaming. Built on `@oh-my-pi/pi-ai`.
+Stateful agent with tool execution and event streaming. Built on `@mariozechner/pi-ai`.
 
 ## Installation
 
@@ -12,7 +12,7 @@ npm install @oh-my-pi/pi-agent
 
 ```typescript
 import { Agent } from "@oh-my-pi/pi-agent";
-import { getModel } from "@oh-my-pi/pi-ai";
+import { getModel } from "@mariozechner/pi-ai";
 
 const agent = new Agent({
 	initialState: {

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@oh-my-pi/pi-agent-core",
-	"version": "3.20.0",
+	"version": "3.21.0",
 	"description": "General-purpose agent with transport abstraction, state management, and attachment support",
 	"type": "module",
 	"main": "./src/index.ts",
@@ -13,8 +13,8 @@
 		"test": "vitest --run"
 	},
 	"dependencies": {
-		"@oh-my-pi/pi-ai": "3.20.0",
-		"@oh-my-pi/pi-tui": "3.20.0"
+		"@mariozechner/pi-ai": "^0.37.4",
+		"@oh-my-pi/pi-tui": "3.21.0"
 	},
 	"keywords": [
 		"ai",

package/src/agent-loop.ts

@@ -10,7 +10,7 @@ import {
 	streamSimple,
 	type ToolResultMessage,
 	validateToolArguments,
-} from "@oh-my-pi/pi-ai";
+} from "@mariozechner/pi-ai";
 import type {
 	AgentContext,
 	AgentEvent,

package/src/agent.ts

@@ -8,10 +8,9 @@ import {
 	type ImageContent,
 	type Message,
 	type Model,
-	type ReasoningEffort,
 	streamSimple,
 	type TextContent,
-} from "@oh-my-pi/pi-ai";
+} from "@mariozechner/pi-ai";
 import { agentLoop, agentLoopContinue } from "./agent-loop";
 import type {
 	AgentContext,
@@ -69,6 +68,12 @@ export interface AgentOptions {
 	 */
 	streamFn?: StreamFn;
 
+	/**
+	 * Optional session identifier forwarded to LLM providers.
+	 * Used by providers that support session-based caching (e.g., OpenAI Codex).
+	 */
+	sessionId?: string;
+
 	/**
 	 * Resolves an API key dynamically for each LLM call.
 	 * Useful for expiring tokens (e.g., GitHub Copilot OAuth).
@@ -105,6 +110,7 @@ export class Agent {
 	private followUpMode: "all" | "one-at-a-time";
 	private interruptMode: "immediate" | "wait";
 	public streamFn: StreamFn;
+	private _sessionId?: string;
 	public getApiKey?: (provider: string) => Promise<string | undefined> | string | undefined;
 	private getToolContext?: () => AgentToolContext | undefined;
 	private runningPrompt?: Promise<void>;
@@ -118,10 +124,26 @@ export class Agent {
 		this.followUpMode = opts.followUpMode || "one-at-a-time";
 		this.interruptMode = opts.interruptMode || "immediate";
 		this.streamFn = opts.streamFn || streamSimple;
+		this._sessionId = opts.sessionId;
 		this.getApiKey = opts.getApiKey;
 		this.getToolContext = opts.getToolContext;
 	}
 
+	/**
+	 * Get the current session ID used for provider caching.
+	 */
+	get sessionId(): string | undefined {
+		return this._sessionId;
+	}
+
+	/**
+	 * Set the session ID for provider caching.
+	 * Call this when switching sessions (new session, branch, resume).
+	 */
+	set sessionId(value: string | undefined) {
+		this._sessionId = value;
+	}
+
 	get state(): AgentState {
 		return this._state;
 	}
@@ -314,12 +336,7 @@ export class Agent {
 		this._state.streamMessage = null;
 		this._state.error = undefined;
 
-		const reasoning: ReasoningEffort | undefined =
-			this._state.thinkingLevel === "off"
-				? undefined
-				: this._state.thinkingLevel === "minimal"
-					? "low"
-					: (this._state.thinkingLevel as ReasoningEffort);
+		const reasoning = this._state.thinkingLevel === "off" ? undefined : this._state.thinkingLevel;
 
 		const context: AgentContext = {
 			systemPrompt: this._state.systemPrompt,
@@ -331,6 +348,7 @@ export class Agent {
 			model,
 			reasoning,
 			interruptMode: this.interruptMode,
+			sessionId: this._sessionId,
 			convertToLlm: this.convertToLlm,
 			transformContext: this.transformContext,
 			getApiKey: this.getApiKey,

package/src/proxy.ts

@@ -12,8 +12,8 @@ import {
 	type SimpleStreamOptions,
 	type StopReason,
 	type ToolCall,
-} from "@oh-my-pi/pi-ai";
-import { parseStreamingJson } from "@oh-my-pi/pi-ai/utils/json-parse";
+} from "@mariozechner/pi-ai";
+import { parseStreamingJson } from "@mariozechner/pi-ai/dist/utils/json-parse.js";
 
 // Create stream class matching ProxyMessageEventStream
 class ProxyMessageEventStream extends EventStream<AssistantMessageEvent, AssistantMessage> {

package/src/types.ts

@@ -8,7 +8,7 @@ import type {
 	TextContent,
 	Tool,
 	ToolResultMessage,
-} from "@oh-my-pi/pi-ai";
+} from "@mariozechner/pi-ai";
 import type { Static, TSchema } from "@sinclair/typebox";
 
 /** Stream function - can return sync or Promise for async config lookup */
@@ -29,6 +29,12 @@ export interface AgentLoopConfig extends SimpleStreamOptions {
 	 */
 	interruptMode?: "immediate" | "wait";
 
+	/**
+	 * Optional session identifier forwarded to LLM providers.
+	 * Used by providers that support session-based caching (e.g., OpenAI Codex).
+	 */
+	sessionId?: string;
+
 	/**
 	 * Converts AgentMessage[] to LLM-compatible Message[] before each LLM call.
 	 *