Skip to content

Add response_schema parameter #302

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 15 commits into from
May 3, 2024
Merged

Add response_schema parameter #302

Changes from 1 commit
Commits
Show all changes
15 commits
Select commit Hold shift + click to select a range
e124a24
Add response_schema parameter
shilpakancharla Apr 25, 2024
164cfd5
Update types for response_schema
shilpakancharla Apr 29, 2024
44ba21e
fix type
MarkDaoust Apr 29, 2024
42e4922
Update type to Mapping[str, Any]
shilpakancharla Apr 30, 2024
7dbaede
TYPE_UNSPECIFIED
shilpakancharla Apr 30, 2024
51008df
Update Any import
shilpakancharla Apr 30, 2024
63582a1
Add black . format check
shilpakancharla Apr 30, 2024
0ff4408
check black . precheck
shilpakancharla Apr 30, 2024
2c04a73
Remove seed parameter for now
shilpakancharla Apr 30, 2024
9467222
Update google/generativeai/types/generation_types.py
MarkDaoust Apr 30, 2024
f9941eb
Added test cases for response_schema, function for normalizing schema…
shilpakancharla May 3, 2024
1c175af
Merge branch 'json_schema_2' of https://github.com/shilpakancharla/ge…
shilpakancharla May 3, 2024
ebc1dfd
Merge branch 'main' into json_schema_2
shilpakancharla May 3, 2024
937a621
merge conflict resolved
shilpakancharla May 3, 2024
74bb791
merge conflict resolved
shilpakancharla May 3, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 3 additions & 1 deletion google/generativeai/types/generation_types.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@
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
Expand Down Expand Up @@ -80,6 +80,7 @@ class GenerationConfigDict(TypedDict, total=False):
max_output_tokens: int
temperature: float
response_mime_type: str
response_schema: Mapping[str, str]


@dataclasses.dataclass
Expand Down Expand Up @@ -155,6 +156,7 @@ class GenerationConfig:
top_p: float | None = None
top_k: int | None = None
response_mime_type: str | None = None
response_schema: Mapping[str, str | None] | None = None


GenerationConfigType = Union[glm.GenerationConfig, GenerationConfigDict, GenerationConfig]
Expand Down