Add response_schema parameter

google/generativeai/types/generation_types.py

@@ -17,11 +17,11 @@
 import collections
 import contextlib
 import sys
-from collections.abc import Iterable, AsyncIterable
+from collections.abc import Iterable, AsyncIterable, Mapping
 import dataclasses
 import itertools
 import textwrap
-from typing import Union
+from typing import Union, Any
 from typing_extensions import TypedDict
 
 import google.protobuf.json_format
@@ -80,6 +80,7 @@ class GenerationConfigDict(TypedDict, total=False):
     max_output_tokens: int
     temperature: float
     response_mime_type: str
+    response_schema: glm.Schema | Mapping[str, str]  # fmt: off
 
 
 @dataclasses.dataclass
@@ -146,6 +147,10 @@ class GenerationConfig:
             Supported mimetype:
                 `text/plain`: (default) Text output.
                 `application/json`: JSON response in the candidates.
+
+        response_schema:
+            Optional. Specifies the format of the JSON requested if response_mime_type is
+            `application/json`.
     """
 
     candidate_count: int | None = None
@@ -155,6 +160,7 @@ class GenerationConfig:
     top_p: float | None = None
     top_k: int | None = None
     response_mime_type: str | None = None
+    response_schema: glm.Schema | Mapping[str, Any] | None = None
 
 
 GenerationConfigType = Union[glm.GenerationConfig, GenerationConfigDict, GenerationConfig]