|
19 | 19 |
|
20 | 20 | import proto # type: ignore
|
21 | 21 |
|
| 22 | +from google.cloud.aiplatform_v1beta1.types import openapi |
22 | 23 | from google.cloud.aiplatform_v1beta1.types import tool
|
23 | 24 | from google.protobuf import duration_pb2 # type: ignore
|
24 | 25 | from google.type import date_pb2 # type: ignore
|
@@ -309,6 +310,17 @@ class GenerationConfig(proto.Message):
|
309 | 310 | The model needs to be prompted to output the appropriate
|
310 | 311 | response type, otherwise the behavior is undefined. This
|
311 | 312 | is a preview feature.
|
| 313 | + response_schema (google.cloud.aiplatform_v1beta1.types.Schema): |
| 314 | + Optional. The ``Schema`` object allows the definition of |
| 315 | + input and output data types. These types can be objects, but |
| 316 | + also primitives and arrays. Represents a select subset of an |
| 317 | + `OpenAPI 3.0 schema |
| 318 | + object <https://spec.openapis.org/oas/v3.0.3#schema>`__. If |
| 319 | + set, a compatible response_mime_type must also be set. |
| 320 | + Compatible mimetypes: ``application/json``: Schema for JSON |
| 321 | + response. |
| 322 | +
|
| 323 | + This field is a member of `oneof`_ ``_response_schema``. |
312 | 324 | """
|
313 | 325 |
|
314 | 326 | temperature: float = proto.Field(
|
@@ -354,6 +366,12 @@ class GenerationConfig(proto.Message):
|
354 | 366 | proto.STRING,
|
355 | 367 | number=13,
|
356 | 368 | )
|
| 369 | + response_schema: openapi.Schema = proto.Field( |
| 370 | + proto.MESSAGE, |
| 371 | + number=16, |
| 372 | + optional=True, |
| 373 | + message=openapi.Schema, |
| 374 | + ) |
357 | 375 |
|
358 | 376 |
|
359 | 377 | class SafetySetting(proto.Message):
|
|
0 commit comments