From de6c759c28169d786213aa50186aa574b45c2795 Mon Sep 17 00:00:00 2001
From: Wanlin Du <wanlindu@google.com>
Date: Fri, 15 Aug 2025 13:59:01 -0700
Subject: [PATCH] fix: allow user to use responseSchema as well. (#6336)

---
 packages/core/src/code_assist/converter.ts | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/packages/core/src/code_assist/converter.ts b/packages/core/src/code_assist/converter.ts
index 9c8a2434..7ca17386 100644
--- a/packages/core/src/code_assist/converter.ts
+++ b/packages/core/src/code_assist/converter.ts
@@ -61,6 +61,7 @@ interface VertexGenerationConfig {
   seed?: number;
   responseMimeType?: string;
   responseJsonSchema?: unknown;
+  responseSchema?: unknown;
   routingConfig?: GenerationConfigRoutingConfig;
   modelSelectionConfig?: ModelSelectionConfig;
   responseModalities?: string[];
@@ -229,6 +230,7 @@ function toVertexGenerationConfig(
     frequencyPenalty: config.frequencyPenalty,
     seed: config.seed,
     responseMimeType: config.responseMimeType,
+    responseSchema: config.responseSchema,
     responseJsonSchema: config.responseJsonSchema,
     routingConfig: config.routingConfig,
     modelSelectionConfig: config.modelSelectionConfig,