File tree Expand file tree Collapse file tree 1 file changed +5
-0
lines changed Expand file tree Collapse file tree 1 file changed +5
-0
lines changed Original file line number Diff line number Diff line change @@ -9,6 +9,7 @@ import { ApiStream, ApiStreamUsageChunk } from "../transform/stream"
99import { convertToOpenAiMessages } from "../transform/openai-format"
1010import { addCacheBreakpoints as addAnthropicCacheBreakpoints } from "../transform/caching/anthropic"
1111import { addCacheBreakpoints as addGeminiCacheBreakpoints } from "../transform/caching/gemini"
12+ import { addCacheBreakpoints as addVertexCacheBreakpoints } from "../transform/caching/vertex"
1213
1314import type { SingleCompletionHandler , ApiHandlerCreateMessageMetadata } from "../index"
1415import { RouterProvider } from "./router-provider"
@@ -70,6 +71,10 @@ export class UnboundHandler extends RouterProvider implements SingleCompletionHa
7071 addAnthropicCacheBreakpoints ( systemPrompt , openAiMessages )
7172 }
7273 }
74+ // Custom models from Vertex AI (no configuration) need to be handled differently.
75+ if ( modelId . startsWith ( "vertex-ai/google." ) || modelId . startsWith ( "vertex-ai/anthropic." ) ) {
76+ addVertexCacheBreakpoints ( messages )
77+ }
7378
7479 // Required by Anthropic; other providers default to max tokens allowed.
7580 let maxTokens : number | undefined
You can’t perform that action at this time.
0 commit comments