Skip to content
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.

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

feat: enable streaming in GoogleAIGemini #1016

Merged
merged 8 commits into from
Aug 27, 2024
Merged

Conversation

Amnah199
Copy link
Contributor

Related Issues

Proposed Changes:

Add streaming_callback to init and run methods of GoogleAIGeminiGenerator and GoogleAIGeminiChatGenerator.

How did you test it?

  • Updated and ran existing unit tests
  • Added a test for streaming
  • Tested locally on my system

Notes for the reviewer

Similar to the PR for VertexAI: 1014

Checklist

@github-actions github-actions bot added integration:google-ai type:documentation Improvements or additions to documentation labels Aug 22, 2024
@Amnah199 Amnah199 marked this pull request as ready for review August 26, 2024 16:44
@Amnah199 Amnah199 requested a review from a team as a code owner August 26, 2024 16:44
@Amnah199 Amnah199 requested review from silvanocerza and anakin87 and removed request for a team and silvanocerza August 26, 2024 16:44
Copy link
Member

@anakin87 anakin87 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Good work!

However, I discovered a strange issue (see the comments).

@anakin87 anakin87 self-requested a review August 27, 2024 13:15
@Amnah199 Amnah199 merged commit 611b05d into main Aug 27, 2024
10 checks passed
@Amnah199 Amnah199 deleted the feat-add-streaming-googleai branch August 27, 2024 13:16
@Amnah199 Amnah199 changed the title Enable streaming in GoogleAIGemini feat: enable streaming in GoogleAIGemini Sep 30, 2024
Amnah199 added a commit that referenced this pull request Oct 2, 2024
* Add streaming to GoogleAIGeminiGenerator
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
integration:google-ai type:documentation Improvements or additions to documentation
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Support Streaming for Gemini Generators
2 participants