diff --git a/src/models/generative_models.ts b/src/models/generative_models.ts index 1e2820e9..0ec3c8f7 100644 --- a/src/models/generative_models.ts +++ b/src/models/generative_models.ts @@ -367,7 +367,7 @@ export class GenerativeModelPreview { request, this.systemInstruction ), - cachedContent: this.cachedContent?.name, + cachedContent: request.cachedContent?.name, }; return generateContent( this.location, diff --git a/src/vertex_ai.ts b/src/vertex_ai.ts index 21a67447..122f443c 100644 --- a/src/vertex_ai.ts +++ b/src/vertex_ai.ts @@ -215,6 +215,7 @@ class VertexAIPreview { toolConfig: modelParams.toolConfig, requestOptions: requestOptions, systemInstruction: modelParams.systemInstruction, + cachedContent: modelParams.cachedContent, }; return new GenerativeModelPreview(getGenerativeModelParams); }