From c21f68aad6fc0f3ddc7dd5a9df0e03b389e4a1e2 Mon Sep 17 00:00:00 2001 From: Kenneth Reitz Date: Sat, 2 Nov 2024 16:54:49 -0400 Subject: [PATCH] Add streaming support to Gemini provider and update example script --- examples/generate_stream_text.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/generate_stream_text.py b/examples/generate_stream_text.py index fe15b87..3bc1443 100644 --- a/examples/generate_stream_text.py +++ b/examples/generate_stream_text.py @@ -1,7 +1,7 @@ from _context import sm # Defaults to the default provider (openai) -r = sm.generate_text("Write a poem about the moon", llm_provider="gemini", stream=True) +r = sm.generate_text("Write a poem about the moon", stream=True) for chunk in r: print(chunk, end="", flush=True)