feat(api): add CompletePromptOptions parameter to completePrompt method

easonliang28 · easonliang28 · commit 688319bf863c · 2026-07-15T22:12:08.000+08:00
Add optional second parameter to all provider completePrompt methods for future abort signal and timeout support.

- Add CompletePromptOptions interface with abortSignal and timeoutMs fields
- Update 25+ provider implementations to accept options?: CompletePromptOptions parameter
- Simplify providers to use basic request calls without signal/timeout logic
diff --git a/src/api/index.ts b/src/api/index.ts
@@ -42,8 +42,17 @@ import {
 } from "./providers"
 import { NativeOllamaHandler } from "./providers/native-ollama"
 
+/**
+ * Options for completePrompt — unified with ApiHandlerCreateMessageMetadata.
+ * Uses abortSignal (not signal) to match the metadata pattern used in stream path.
+ */
+export interface CompletePromptOptions extends Pick<ApiHandlerCreateMessageMetadata, "abortSignal"> {
+	/** Optional timeout override (ms) — falls back to provider default if omitted */
+	timeoutMs?: number
+}
+
 export interface SingleCompletionHandler {
-	completePrompt(prompt: string): Promise<string>
+	completePrompt(prompt: string, metadata?: CompletePromptOptions): Promise<string>
 }
 
 export interface ApiHandlerCreateMessageMetadata {
diff --git a/src/api/providers/anthropic-vertex.ts b/src/api/providers/anthropic-vertex.ts
@@ -25,7 +25,7 @@ import {
 
 import { BaseProvider } from "./base-provider"
 import { parseVertexJsonCredentials } from "./utils/vertex-credentials"
-import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata } from "../index"
+import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata, CompletePromptOptions } from "../index"
 
 // https://docs.anthropic.com/en/api/claude-on-vertex-ai
 export class AnthropicVertexHandler extends BaseProvider implements SingleCompletionHandler {
@@ -270,7 +270,7 @@ export class AnthropicVertexHandler extends BaseProvider implements SingleComple
 		}
 	}
 
-	async completePrompt(prompt: string) {
+	async completePrompt(prompt: string, options?: CompletePromptOptions) {
 		try {
 			const {
 				id,
diff --git a/src/api/providers/anthropic.ts b/src/api/providers/anthropic.ts
@@ -22,7 +22,7 @@ import { getAnthropicProviderReasoning } from "../transform/reasoning"
 import { handleProviderError } from "./utils/error-handler"
 
 import { BaseProvider } from "./base-provider"
-import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata } from "../index"
+import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata, CompletePromptOptions } from "../index"
 import { calculateApiCostAnthropic } from "../../shared/cost"
 import {
 	convertOpenAIToolsToAnthropic,
@@ -429,7 +429,7 @@ export class AnthropicHandler extends BaseProvider implements SingleCompletionHa
 		}
 	}
 
-	async completePrompt(prompt: string) {
+	async completePrompt(prompt: string, options?: CompletePromptOptions) {
 		const { id: model, temperature } = this.getModel()
 
 		let message
diff --git a/src/api/providers/base-openai-compatible-provider.ts b/src/api/providers/base-openai-compatible-provider.ts
@@ -8,7 +8,7 @@ import { TagMatcher } from "../../utils/tag-matcher"
 import { ApiStream, ApiStreamUsageChunk } from "../transform/stream"
 import { convertToOpenAiMessages } from "../transform/openai-format"
 
-import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata } from "../index"
+import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata, CompletePromptOptions } from "../index"
 import { DEFAULT_HEADERS } from "./constants"
 import { BaseProvider } from "./base-provider"
 import { handleOpenAIError } from "./utils/error-handler"
@@ -212,7 +212,7 @@ export abstract class BaseOpenAiCompatibleProvider<ModelName extends string>
 		}
 	}
 
-	async completePrompt(prompt: string): Promise<string> {
+	async completePrompt(prompt: string, options?: CompletePromptOptions): Promise<string> {
 		const { id: modelId, info: modelInfo } = this.getModel()
 
 		const params: OpenAI.Chat.Completions.ChatCompletionCreateParams = {
diff --git a/src/api/providers/bedrock.ts b/src/api/providers/bedrock.ts
@@ -44,7 +44,7 @@ import { convertToBedrockConverseMessages as sharedConverter } from "../transfor
 import { getModelParams } from "../transform/model-params"
 import { shouldUseReasoningBudget } from "../../shared/api"
 import { normalizeToolSchema } from "../../utils/json-schema"
-import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata } from "../index"
+import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata, CompletePromptOptions } from "../index"
 
 /************************************************************************************
  *
@@ -798,7 +798,7 @@ export class AwsBedrockHandler extends BaseProvider implements SingleCompletionH
 		}
 	}
 
-	async completePrompt(prompt: string): Promise<string> {
+	async completePrompt(prompt: string, options?: CompletePromptOptions): Promise<string> {
 		try {
 			const modelConfig = this.getModel()
 
diff --git a/src/api/providers/fake-ai.ts b/src/api/providers/fake-ai.ts
@@ -2,7 +2,12 @@ import { Anthropic } from "@anthropic-ai/sdk"
 
 import type { ModelInfo } from "@roo-code/types"
 
-import type { ApiHandler, SingleCompletionHandler, ApiHandlerCreateMessageMetadata } from "../index"
+import type {
+	ApiHandler,
+	SingleCompletionHandler,
+	ApiHandlerCreateMessageMetadata,
+	CompletePromptOptions,
+} from "../index"
 import type { ApiHandlerOptions } from "../../shared/api"
 import { ApiStream } from "../transform/stream"
 
@@ -28,7 +33,7 @@ interface FakeAI {
 	): ApiStream
 	getModel(): { id: string; info: ModelInfo }
 	countTokens(content: Array<Anthropic.Messages.ContentBlockParam>): Promise<number>
-	completePrompt(prompt: string): Promise<string>
+	completePrompt(prompt: string, options?: CompletePromptOptions): Promise<string>
 }
 
 /**
@@ -75,7 +80,7 @@ export class FakeAIHandler implements ApiHandler, SingleCompletionHandler {
 		return this.ai.countTokens(content)
 	}
 
-	completePrompt(prompt: string): Promise<string> {
-		return this.ai.completePrompt(prompt)
+	completePrompt(prompt: string, options?: CompletePromptOptions): Promise<string> {
+		return this.ai.completePrompt(prompt, options)
 	}
 }
diff --git a/src/api/providers/gemini.ts b/src/api/providers/gemini.ts
@@ -23,7 +23,7 @@ import { t } from "i18next"
 import type { ApiStream, GroundingSource } from "../transform/stream"
 import { getModelParams } from "../transform/model-params"
 
-import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata } from "../index"
+import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata, CompletePromptOptions } from "../index"
 import { BaseProvider } from "./base-provider"
 import { parseVertexJsonCredentials } from "./utils/vertex-credentials"
 
@@ -576,7 +576,7 @@ export class GeminiHandler extends BaseProvider implements SingleCompletionHandl
 		return citationLinks.join(", ")
 	}
 
-	async completePrompt(prompt: string): Promise<string> {
+	async completePrompt(prompt: string, options?: CompletePromptOptions): Promise<string> {
 		const { id: model, info } = this.getModel()
 
 		try {
diff --git a/src/api/providers/kenari.ts b/src/api/providers/kenari.ts
@@ -13,7 +13,7 @@ import { ApiHandlerOptions } from "../../shared/api"
 import { ApiStream } from "../transform/stream"
 import { convertToOpenAiMessages } from "../transform/openai-format"
 
-import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata } from "../index"
+import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata, CompletePromptOptions } from "../index"
 import { RouterProvider } from "./router-provider"
 import { extractReasoningFromDelta } from "./utils/extract-reasoning"
 
@@ -123,7 +123,7 @@ export class KenariHandler extends RouterProvider implements SingleCompletionHan
 	 * @returns The model's reply text, or an empty string if no content is returned.
 	 * @throws Error with a Kenari-specific prefix if the request fails.
 	 */
-	async completePrompt(prompt: string): Promise<string> {
+	async completePrompt(prompt: string, options?: CompletePromptOptions): Promise<string> {
 		const { id: modelId, info } = await this.fetchModel()
 
 		try {
diff --git a/src/api/providers/lite-llm.ts b/src/api/providers/lite-llm.ts
@@ -12,7 +12,7 @@ import { convertToOpenAiMessages } from "../transform/openai-format"
 import { GEMINI_THOUGHT_SIGNATURE_BYPASS } from "../transform/gemini-format"
 import { sanitizeOpenAiCallId } from "../../utils/tool-id"
 
-import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata } from "../index"
+import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata, CompletePromptOptions } from "../index"
 import { RouterProvider } from "./router-provider"
 import { extractReasoningFromDelta } from "./utils/extract-reasoning"
 
@@ -311,7 +311,7 @@ export class LiteLLMHandler extends RouterProvider implements SingleCompletionHa
 		}
 	}
 
-	async completePrompt(prompt: string): Promise<string> {
+	async completePrompt(prompt: string, options?: CompletePromptOptions): Promise<string> {
 		const { id: modelId, info } = await this.fetchModel()
 
 		// Check if this is a GPT-5 model that requires max_completion_tokens instead of max_tokens
diff --git a/src/api/providers/lm-studio.ts b/src/api/providers/lm-studio.ts
@@ -13,7 +13,7 @@ import { convertToOpenAiMessages } from "../transform/openai-format"
 import { ApiStream } from "../transform/stream"
 
 import { BaseProvider } from "./base-provider"
-import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata } from "../index"
+import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata, CompletePromptOptions } from "../index"
 import { getModelsFromCache } from "./fetchers/modelCache"
 import { handleOpenAIError } from "./utils/error-handler"
 
@@ -187,7 +187,7 @@ export class LmStudioHandler extends BaseProvider implements SingleCompletionHan
 		}
 	}
 
-	async completePrompt(prompt: string): Promise<string> {
+	async completePrompt(prompt: string, options?: CompletePromptOptions): Promise<string> {
 		try {
 			// Create params object with optional draft model
 			const params: any = {
diff --git a/src/api/providers/minimax.ts b/src/api/providers/minimax.ts
@@ -12,7 +12,7 @@ import { getModelParams } from "../transform/model-params"
 import { mergeEnvironmentDetailsForMiniMax } from "../transform/minimax-format"
 
 import { BaseProvider } from "./base-provider"
-import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata } from "../index"
+import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata, CompletePromptOptions } from "../index"
 import { calculateApiCostAnthropic } from "../../shared/cost"
 import { convertOpenAIToolsToAnthropic } from "../../core/prompts/tools/native-tools/converters"
 
@@ -289,7 +289,7 @@ export class MiniMaxHandler extends BaseProvider implements SingleCompletionHand
 		}
 	}
 
-	async completePrompt(prompt: string) {
+	async completePrompt(prompt: string, options?: CompletePromptOptions) {
 		const { id: model, temperature } = this.getModel()
 
 		const message = await this.client.messages.create({
diff --git a/src/api/providers/mistral.ts b/src/api/providers/mistral.ts
@@ -18,7 +18,7 @@ import { ApiStream } from "../transform/stream"
 import { handleProviderError } from "./utils/error-handler"
 
 import { BaseProvider } from "./base-provider"
-import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata } from "../index"
+import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata, CompletePromptOptions } from "../index"
 
 // Type helper to handle thinking chunks from Mistral API
 // The SDK includes ThinkChunk but TypeScript has trouble with the discriminated union
@@ -192,8 +192,7 @@ export class MistralHandler extends BaseProvider implements SingleCompletionHand
 
 		return { id, info, maxTokens, temperature }
 	}
-
-	async completePrompt(prompt: string): Promise<string> {
+	async completePrompt(prompt: string, options?: CompletePromptOptions): Promise<string> {
 		const { id: model, temperature } = this.getModel()
 
 		try {
diff --git a/src/api/providers/native-ollama.ts b/src/api/providers/native-ollama.ts
@@ -7,7 +7,7 @@ import { BaseProvider } from "./base-provider"
 import type { ApiHandlerOptions } from "../../shared/api"
 import { getOllamaModels } from "./fetchers/ollama"
 import { TagMatcher } from "../../utils/tag-matcher"
-import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata } from "../index"
+import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata, CompletePromptOptions } from "../index"
 
 interface OllamaChatOptions {
 	temperature: number
@@ -550,7 +550,7 @@ export class NativeOllamaHandler extends BaseProvider implements SingleCompletio
 		}
 	}
 
-	async completePrompt(prompt: string): Promise<string> {
+	async completePrompt(prompt: string, options?: CompletePromptOptions): Promise<string> {
 		try {
 			const client = this.ensureClient()
 			const { id: modelId } = await this.fetchModel()
diff --git a/src/api/providers/openai-codex.ts b/src/api/providers/openai-codex.ts
@@ -21,7 +21,7 @@ import { ApiStream, ApiStreamUsageChunk } from "../transform/stream"
 import { getModelParams } from "../transform/model-params"
 
 import { BaseProvider } from "./base-provider"
-import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata } from "../index"
+import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata, CompletePromptOptions } from "../index"
 import { isMcpTool } from "../../utils/mcp-name"
 import { sanitizeOpenAiCallId } from "../../utils/tool-id"
 import { openAiCodexOAuthManager } from "../../integrations/openai-codex/oauth"
@@ -1154,7 +1154,7 @@ export class OpenAiCodexHandler extends BaseProvider implements SingleCompletion
 		return this.lastResponseId
 	}
 
-	async completePrompt(prompt: string): Promise<string> {
+	async completePrompt(prompt: string, options?: CompletePromptOptions): Promise<string> {
 		this.abortController = new AbortController()
 
 		try {
diff --git a/src/api/providers/openai-compatible.ts b/src/api/providers/openai-compatible.ts
@@ -17,7 +17,7 @@ import { ApiStream, ApiStreamUsageChunk } from "../transform/stream"
 
 import { DEFAULT_HEADERS } from "./constants"
 import { BaseProvider } from "./base-provider"
-import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata } from "../index"
+import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata, CompletePromptOptions } from "../index"
 
 /**
  * Configuration options for creating an OpenAI-compatible provider.
@@ -197,7 +197,7 @@ export abstract class OpenAICompatibleHandler extends BaseProvider implements Si
 	/**
 	 * Complete a prompt using the AI SDK generateText.
 	 */
-	async completePrompt(prompt: string): Promise<string> {
+	async completePrompt(prompt: string, options?: CompletePromptOptions): Promise<string> {
 		const languageModel = this.getLanguageModel()
 
 		const { text } = await generateText({
diff --git a/src/api/providers/openai-native.ts b/src/api/providers/openai-native.ts
@@ -26,7 +26,7 @@ import { ApiStream, ApiStreamUsageChunk } from "../transform/stream"
 import { getModelParams } from "../transform/model-params"
 
 import { BaseProvider } from "./base-provider"
-import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata } from "../index"
+import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata, CompletePromptOptions } from "../index"
 import { isMcpTool } from "../../utils/mcp-name"
 import { sanitizeOpenAiCallId } from "../../utils/tool-id"
 
@@ -1484,14 +1484,12 @@ export class OpenAiNativeHandler extends BaseProvider implements SingleCompletio
 	getResponseId(): string | undefined {
 		return this.lastResponseId
 	}
-
-	async completePrompt(prompt: string): Promise<string> {
-		// Create AbortController for cancellation
-		this.abortController = new AbortController()
-
+	async completePrompt(prompt: string, options?: CompletePromptOptions): Promise<string> {
 		try {
+			this.abortController = new AbortController()
+
 			const model = this.getModel()
-			const { verbosity, reasoning } = model
+			const { verbosity } = model
 
 			// Resolve reasoning effort for models that support it
 			const reasoningEffort = this.getReasoningEffort(model)
diff --git a/src/api/providers/openai.ts b/src/api/providers/openai.ts
@@ -296,7 +296,7 @@ export class OpenAiHandler extends BaseProvider implements SingleCompletionHandl
 		return { id, info, ...params }
 	}
 
-	async completePrompt(prompt: string): Promise<string> {
+	async completePrompt(prompt: string, options?: import("../../api").CompletePromptOptions): Promise<string> {
 		try {
 			const isAzureAiInference = this._isAzureAiInference(this.options.openAiBaseUrl)
 			const model = this.getModel()
diff --git a/src/api/providers/opencode-go.ts b/src/api/providers/opencode-go.ts
@@ -18,7 +18,7 @@ import { convertToR1Format } from "../transform/r1-format"
 import { filterNonAnthropicBlocks } from "../transform/anthropic-filter"
 import { getModelParams } from "../transform/model-params"
 
-import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata } from "../index"
+import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata, CompletePromptOptions } from "../index"
 import { RouterProvider } from "./router-provider"
 import { extractReasoningFromDelta } from "./utils/extract-reasoning"
 import { DEFAULT_HEADERS } from "./constants"
@@ -485,7 +485,7 @@ export class OpencodeGoHandler extends RouterProvider implements SingleCompletio
 	 * @returns The model's reply text, or an empty string if no content is returned.
 	 * @throws Error with an Opencode Go-specific prefix if the request fails.
 	 */
-	async completePrompt(prompt: string): Promise<string> {
+	async completePrompt(prompt: string, options?: CompletePromptOptions): Promise<string> {
 		const { id: modelId, format, temperature, reasoningEffort, maxTokens } = await this.resolveModel()
 
 		if (format === "anthropic") {
diff --git a/src/api/providers/openrouter.ts b/src/api/providers/openrouter.ts
@@ -35,7 +35,7 @@ import { getModelEndpoints } from "./fetchers/modelEndpointCache"
 
 import { DEFAULT_HEADERS } from "./constants"
 import { BaseProvider } from "./base-provider"
-import type { ApiHandlerCreateMessageMetadata, SingleCompletionHandler } from "../index"
+import type { ApiHandlerCreateMessageMetadata, CompletePromptOptions, SingleCompletionHandler } from "../index"
 import { handleOpenAIError } from "./utils/error-handler"
 import { generateImageWithProvider, ImageGenerationResult } from "./utils/image-generation"
 import { applyRouterToolPreferences } from "./utils/router-tool-preferences"
@@ -574,7 +574,7 @@ export class OpenRouterHandler extends BaseProvider implements SingleCompletionH
 		return { id, info, topP: isDeepSeekR1 ? 0.95 : undefined, ...params }
 	}
 
-	async completePrompt(prompt: string) {
+	async completePrompt(prompt: string, options?: CompletePromptOptions) {
 		const { id: modelId, maxTokens, temperature, reasoning } = await this.fetchModel()
 
 		const completionParams: OpenRouterChatCompletionParams = {
diff --git a/src/api/providers/poe.ts b/src/api/providers/poe.ts
@@ -18,7 +18,7 @@ import { convertToAiSdkMessages, convertToolsForAiSdk, processAiSdkStreamPart }
 import { ApiStream } from "../transform/stream"
 
 import { BaseProvider } from "./base-provider"
-import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata } from "../index"
+import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata, CompletePromptOptions } from "../index"
 import { getModelsFromCache } from "./fetchers/modelCache"
 
 const DEFAULT_THINKING_BUDGET = 8192
@@ -138,7 +138,7 @@ export class PoeHandler extends BaseProvider implements SingleCompletionHandler
 		}
 	}
 
-	async completePrompt(prompt: string): Promise<string> {
+	async completePrompt(prompt: string, options?: CompletePromptOptions): Promise<string> {
 		const { id } = this.getModel()
 		try {
 			const { text } = await generateText({
diff --git a/src/api/providers/qwen-code.ts b/src/api/providers/qwen-code.ts
@@ -15,7 +15,7 @@ import { ApiStream } from "../transform/stream"
 
 import { BaseProvider } from "./base-provider"
 import { extractReasoningFromDelta } from "./utils/extract-reasoning"
-import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata } from "../index"
+import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata, CompletePromptOptions } from "../index"
 
 const QWEN_OAUTH_BASE_URL = "https://chat.qwen.ai"
 const QWEN_OAUTH_TOKEN_ENDPOINT = `${QWEN_OAUTH_BASE_URL}/api/v1/oauth2/token`
@@ -327,7 +327,7 @@ export class QwenCodeHandler extends BaseProvider implements SingleCompletionHan
 		return { id, info }
 	}
 
-	async completePrompt(prompt: string): Promise<string> {
+	async completePrompt(prompt: string, options?: CompletePromptOptions): Promise<string> {
 		await this.ensureAuthenticated()
 		const client = this.ensureClient()
 		const model = this.getModel()
diff --git a/src/api/providers/requesty.ts b/src/api/providers/requesty.ts
diff --git a/src/api/providers/unbound.ts b/src/api/providers/unbound.ts
diff --git a/src/api/providers/vercel-ai-gateway.ts b/src/api/providers/vercel-ai-gateway.ts
diff --git a/src/api/providers/vscode-lm.ts b/src/api/providers/vscode-lm.ts
diff --git a/src/api/providers/xai.ts b/src/api/providers/xai.ts
diff --git a/src/api/providers/zoo-gateway.ts b/src/api/providers/zoo-gateway.ts
diff --git a/src/core/task/__tests__/Task.spec.ts b/src/core/task/__tests__/Task.spec.ts
diff --git a/src/utils/__tests__/enhance-prompt.spec.ts b/src/utils/__tests__/enhance-prompt.spec.ts
diff --git a/src/utils/single-completion-handler.ts b/src/utils/single-completion-handler.ts

Original file line number	Diff line number	Diff line change
`@@ -25,7 +25,7 @@ import {`
`25`	`25`
`26`	`26`	`import { BaseProvider } from "./base-provider"`
`27`	`27`	`import { parseVertexJsonCredentials } from "./utils/vertex-credentials"`
`28`		`-import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata } from "../index"`
	`28`	`+import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata, CompletePromptOptions } from "../index"`
`29`	`29`
`30`	`30`	`// https://docs.anthropic.com/en/api/claude-on-vertex-ai`
`31`	`31`	`export class AnthropicVertexHandler extends BaseProvider implements SingleCompletionHandler {`
`@@ -270,7 +270,7 @@ export class AnthropicVertexHandler extends BaseProvider implements SingleComple`
`270`	`270`	`}`
`271`	`271`	`}`
`272`	`272`
`273`		`- async completePrompt(prompt: string) {`
	`273`	`+ async completePrompt(prompt: string, options?: CompletePromptOptions) {`
`274`	`274`	`try {`
`275`	`275`	`const {`
`276`	`276`	`id,`
Original file line number	Diff line number	Diff line change
`@@ -22,7 +22,7 @@ import { getAnthropicProviderReasoning } from "../transform/reasoning"`
`22`	`22`	`import { handleProviderError } from "./utils/error-handler"`
`23`	`23`
`24`	`24`	`import { BaseProvider } from "./base-provider"`
`25`		`-import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata } from "../index"`
	`25`	`+import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata, CompletePromptOptions } from "../index"`
`26`	`26`	`import { calculateApiCostAnthropic } from "../../shared/cost"`
`27`	`27`	`import {`
`28`	`28`	`convertOpenAIToolsToAnthropic,`
`@@ -429,7 +429,7 @@ export class AnthropicHandler extends BaseProvider implements SingleCompletionHa`
`429`	`429`	`}`
`430`	`430`	`}`
`431`	`431`
`432`		`- async completePrompt(prompt: string) {`
	`432`	`+ async completePrompt(prompt: string, options?: CompletePromptOptions) {`
`433`	`433`	`const { id: model, temperature } = this.getModel()`
`434`	`434`
`435`	`435`	`let message`
Original file line number	Diff line number	Diff line change
`@@ -8,7 +8,7 @@ import { TagMatcher } from "../../utils/tag-matcher"`
`8`	`8`	`import { ApiStream, ApiStreamUsageChunk } from "../transform/stream"`
`9`	`9`	`import { convertToOpenAiMessages } from "../transform/openai-format"`
`10`	`10`
`11`		`-import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata } from "../index"`
	`11`	`+import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata, CompletePromptOptions } from "../index"`
`12`	`12`	`import { DEFAULT_HEADERS } from "./constants"`
`13`	`13`	`import { BaseProvider } from "./base-provider"`
`14`	`14`	`import { handleOpenAIError } from "./utils/error-handler"`
`@@ -212,7 +212,7 @@ export abstract class BaseOpenAiCompatibleProvider<ModelName extends string>`
`212`	`212`	`}`
`213`	`213`	`}`
`214`	`214`
`215`		`- async completePrompt(prompt: string): Promise<string> {`
	`215`	`+ async completePrompt(prompt: string, options?: CompletePromptOptions): Promise<string> {`
`216`	`216`	`const { id: modelId, info: modelInfo } = this.getModel()`
`217`	`217`
`218`	`218`	`const params: OpenAI.Chat.Completions.ChatCompletionCreateParams = {`
Original file line number	Diff line number	Diff line change
`@@ -44,7 +44,7 @@ import { convertToBedrockConverseMessages as sharedConverter } from "../transfor`
`44`	`44`	`import { getModelParams } from "../transform/model-params"`
`45`	`45`	`import { shouldUseReasoningBudget } from "../../shared/api"`
`46`	`46`	`import { normalizeToolSchema } from "../../utils/json-schema"`
`47`		`-import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata } from "../index"`
	`47`	`+import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata, CompletePromptOptions } from "../index"`
`48`	`48`
`49`	`49`	`/************************************************************************************`
`50`	`50`	`*`
`@@ -798,7 +798,7 @@ export class AwsBedrockHandler extends BaseProvider implements SingleCompletionH`
`798`	`798`	`}`
`799`	`799`	`}`
`800`	`800`
`801`		`- async completePrompt(prompt: string): Promise<string> {`
	`801`	`+ async completePrompt(prompt: string, options?: CompletePromptOptions): Promise<string> {`
`802`	`802`	`try {`
`803`	`803`	`const modelConfig = this.getModel()`
`804`	`804`
Original file line number	Diff line number	Diff line change
`@@ -12,7 +12,7 @@ import { convertToOpenAiMessages } from "../transform/openai-format"`
`12`	`12`	`import { GEMINI_THOUGHT_SIGNATURE_BYPASS } from "../transform/gemini-format"`
`13`	`13`	`import { sanitizeOpenAiCallId } from "../../utils/tool-id"`
`14`	`14`
`15`		`-import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata } from "../index"`
	`15`	`+import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata, CompletePromptOptions } from "../index"`
`16`	`16`	`import { RouterProvider } from "./router-provider"`
`17`	`17`	`import { extractReasoningFromDelta } from "./utils/extract-reasoning"`
`18`	`18`
`@@ -311,7 +311,7 @@ export class LiteLLMHandler extends RouterProvider implements SingleCompletionHa`
`311`	`311`	`}`
`312`	`312`	`}`
`313`	`313`
`314`		`- async completePrompt(prompt: string): Promise<string> {`
	`314`	`+ async completePrompt(prompt: string, options?: CompletePromptOptions): Promise<string> {`
`315`	`315`	`const { id: modelId, info } = await this.fetchModel()`
`316`	`316`
`317`	`317`	`// Check if this is a GPT-5 model that requires max_completion_tokens instead of max_tokens`
Original file line number	Diff line number	Diff line change
`@@ -13,7 +13,7 @@ import { convertToOpenAiMessages } from "../transform/openai-format"`
`13`	`13`	`import { ApiStream } from "../transform/stream"`
`14`	`14`
`15`	`15`	`import { BaseProvider } from "./base-provider"`
`16`		`-import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata } from "../index"`
	`16`	`+import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata, CompletePromptOptions } from "../index"`
`17`	`17`	`import { getModelsFromCache } from "./fetchers/modelCache"`
`18`	`18`	`import { handleOpenAIError } from "./utils/error-handler"`
`19`	`19`
`@@ -187,7 +187,7 @@ export class LmStudioHandler extends BaseProvider implements SingleCompletionHan`
`187`	`187`	`}`
`188`	`188`	`}`
`189`	`189`
`190`		`- async completePrompt(prompt: string): Promise<string> {`
	`190`	`+ async completePrompt(prompt: string, options?: CompletePromptOptions): Promise<string> {`
`191`	`191`	`try {`
`192`	`192`	`// Create params object with optional draft model`
`193`	`193`	`const params: any = {`