mirror of
https://github.com/RooVetGit/Roo-Code.git
synced 2026-09-05 08:10:14 +00:00
fix deprecated max_tokens
This commit is contained in:
parent
51eb938bea
commit
6abba41abb
2 changed files with 2 additions and 5 deletions
|
|
@ -20,9 +20,6 @@ import { RouterProvider } from "./router-provider"
|
|||
// Extend OpenAI's CompletionUsage to include Vercel AI Gateway specific fields
|
||||
interface VercelAiGatewayUsage extends OpenAI.CompletionUsage {
|
||||
cache_creation_input_tokens?: number
|
||||
prompt_tokens_details?: {
|
||||
cached_tokens?: number
|
||||
}
|
||||
cost?: number
|
||||
}
|
||||
|
||||
|
|
@ -61,7 +58,7 @@ export class VercelAiGatewayHandler extends RouterProvider implements SingleComp
|
|||
temperature: this.supportsTemperature(modelId)
|
||||
? (this.options.modelTemperature ?? VERCEL_AI_GATEWAY_DEFAULT_TEMPERATURE)
|
||||
: undefined,
|
||||
max_tokens: info.maxTokens,
|
||||
max_completion_tokens: info.maxTokens,
|
||||
stream: true,
|
||||
}
|
||||
|
||||
|
|
@ -104,7 +101,7 @@ export class VercelAiGatewayHandler extends RouterProvider implements SingleComp
|
|||
requestOptions.temperature = this.options.modelTemperature ?? VERCEL_AI_GATEWAY_DEFAULT_TEMPERATURE
|
||||
}
|
||||
|
||||
requestOptions.max_tokens = info.maxTokens
|
||||
requestOptions.max_completion_tokens = info.maxTokens
|
||||
|
||||
const response = await this.client.chat.completions.create(requestOptions)
|
||||
return response.choices[0]?.message.content || ""
|
||||
|
|
|
|||
Binary file not shown.
|
Before Width: | Height: | Size: 875 B |
Loading…
Add table
Reference in a new issue