Add Azure feature parity: allowedTools, canUseTool, interrupt, audio, predicted output (#4)

* Add Azure feature parity: predicted output, audio, allowedTools, canUseTool, interrupt, session tool management

New ModelSettings fields:
- prediction: predicted output for faster completions (Chat Completions)
- modalities/audio: audio output for gpt-4o-audio models (Chat Completions)
- dataSources: Azure On Your Data RAG sources (Chat Completions)
- contextManagement: server-side context compaction (Responses API)

New RunOptions:
- allowedTools: glob-style wildcard filtering (e.g. "mcp__github__*")
- canUseTool: centralized HITL permission callback (allow/deny/modify)

New StreamedRunResult:
- interrupt(): graceful stop returning partial result instead of throwing

New Session methods:
- addTools/removeTools/setTools: hot-swap tools between turns
- allowedTools/canUseTool passthrough from SessionConfig

Bug fix: canUseTool deny now short-circuits needsApproval to prevent
unnecessary interrupts when the permission callback would deny anyway.

614 unit tests + 9 integration tests + 21 battle tests, all passing.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

* Bump @usestratus/sdk to v1.2.0

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

* Fix canUseTool double-invocation for dynamic needsApproval tools

Evaluate needsApproval before canUseTool in collectPendingApprovals.
Previously, tools with a needsApproval function that returned false
would still trigger canUseTool in the approval check, then again
during execution — causing double side effects (audit logs, rate
limit decrements, latency).

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

---------

Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: tylergibbs1

packages/stratus-sdk/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@usestratus/sdk",
-  "version": "1.1.0",
+  "version": "1.2.0",
   "type": "module",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",

packages/stratus-sdk/src/azure/chat-completions-model.ts

@@ -214,6 +214,10 @@ export class AzureChatCompletionsModel implements Model {
 			if (s.user !== undefined) body.user = s.user;
 			if (s.logprobs !== undefined) body.logprobs = s.logprobs;
 			if (s.topLogprobs !== undefined) body.top_logprobs = s.topLogprobs;
+			if (s.prediction !== undefined) body.prediction = s.prediction;
+			if (s.modalities !== undefined) body.modalities = s.modalities;
+			if (s.audio !== undefined) body.audio = s.audio;
+			if (s.dataSources !== undefined) body.data_sources = s.dataSources;
 		}
 
 		return body;

packages/stratus-sdk/src/azure/responses-model.ts

@@ -386,6 +386,7 @@ export class AzureResponsesModel implements Model {
 			if (s.metadata !== undefined) body.metadata = s.metadata;
 			if (s.user !== undefined) body.user = s.user;
 			// logprobs/topLogprobs are Chat Completions-only; not supported by the Responses API.
+			if (s.contextManagement !== undefined) body.context_management = s.contextManagement;
 		}
 
 		return body;

packages/stratus-sdk/src/core/codemode/executor.ts

@@ -75,11 +75,7 @@ export class FunctionExecutor implements Executor {
 				...args: string[]
 			) => (...args: unknown[]) => Promise<unknown>;
 
-			const fn = new AsyncFunction(
-				"codemode",
-				"console",
-				`return await (${code})()`,
-			);
+			const fn = new AsyncFunction("codemode", "console", `return await (${code})()`);
 
 			const timeoutMs = this.#timeout;
 			const result = await Promise.race([

packages/stratus-sdk/src/core/codemode/index.ts

@@ -8,9 +8,9 @@
 
 import { z } from "zod";
 import { isFunctionTool } from "../hosted-tool";
+import type { AgentTool } from "../hosted-tool";
 import { tool } from "../tool";
 import type { FunctionTool } from "../tool";
-import type { AgentTool } from "../hosted-tool";
 import type { Executor } from "./executor";
 import { generateTypes, normalizeCode, sanitizeToolName } from "./types";
 

packages/stratus-sdk/src/core/codemode/types.ts

@@ -297,9 +297,7 @@ function jsonSchemaToTypeString(
 				const desc = propSchema.description;
 				const format = propSchema.format;
 				if (desc || format) {
-					const descText = desc
-						? escapeJsDoc(desc.replace(/\r?\n/g, " "))
-						: undefined;
+					const descText = desc ? escapeJsDoc(desc.replace(/\r?\n/g, " ")) : undefined;
 					const formatTag = format ? `@format ${escapeJsDoc(format)}` : undefined;
 					if (descText && formatTag) {
 						lines.push(`${indent}    /**`);

packages/stratus-sdk/src/core/index.ts

@@ -13,6 +13,10 @@ export type {
 	ToolErrorFormatter,
 	CallModelInputFilter,
 	MaxTurnsErrorHandler,
+	CanUseTool,
+	ToolPermissionAllow,
+	ToolPermissionDeny,
+	ToolPermissionResult,
 } from "./run";
 
 export { createSession, forkSession, loadSession, prompt, resumeSession, Session } from "./session";
@@ -123,6 +127,14 @@ export type {
 	ContentPart,
 	TextContentPart,
 	ImageContentPart,
+	AudioFormat,
+	AudioVoice,
+	AudioConfig,
+	Modality,
+	PredictedOutput,
+	DataSource,
+	ContextManagement,
+	ContextManagementRule,
 } from "./types";
 
 export {