-
Notifications
You must be signed in to change notification settings - Fork 1.1k
Description
Is your feature request related to a problem? Please describe.
Currently, when using the Java Vertex AI SDK’s generateContent() (e.g. via ContentMaker), we only get back a raw JSON string. Client code must separately construct a JSON Schema, attach it in GenerationConfig, then manually parse the returned JSON—either via Jackson’s ObjectMapper or via Protobuf’s JsonFormat. This boilerplate is brittle, error-prone, and hides what fields the model will actually return.
Describe the solution you’d like
-
Expose the active response schema programmatically. For example:
Schema schema = model.getResponseSchema();
or as metadata on
GenerateContentResponse. -
Provide a one-step parser that binds the response into a POJO, Map, or Protobuf message, e.g.
MyResult result = response.getTypedContent(MyResult.class);
under the hood handling all JSON Schema validation and mapping.
Describe alternatives you’ve considered
- Continuing to manually wire up JSON Schema definitions +
GenerationConfig. - Writing custom helper utilities in each codebase to wrap
ObjectMapperorJsonFormat. - Using generic
Map<String,Object>parsing and then casting fields at runtime.
Additional context
-
Sample desired usage:
Schema schema = model.getResponseSchema(); MyResponse resp = model.generateContent(prompt, config) .getTypedContent(MyResponse.class);
-
This would eliminate repetitive boilerplate, reduce parsing errors, and let developers discover response fields at compile time.