Skip to content

Commit ebc7180

Browse files
google-genai-botcopybara-github
authored andcommitted
fix: Fix the return type of generate_content_stream.
PiperOrigin-RevId: 805343984
1 parent aa2f97b commit ebc7180

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

google/genai/models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8342,7 +8342,7 @@ async def generate_content_stream(
83428342
model: str,
83438343
contents: Union[types.ContentListUnion, types.ContentListUnionDict],
83448344
config: Optional[types.GenerateContentConfigOrDict] = None,
8345-
) -> AsyncIterator[types.GenerateContentResponse]:
8345+
) -> Awaitable[AsyncIterator[types.GenerateContentResponse]]:
83468346
"""Makes an API request to generate content using a model and yields the model's response in chunks.
83478347
83488348
For the `model` parameter, supported formats for Vertex AI API include:

0 commit comments

Comments
 (0)