From 028e89b080d535fd9c48f699daf69476fa174888 Mon Sep 17 00:00:00 2001 From: Kenneth Reitz Date: Sat, 2 Nov 2024 16:48:54 -0400 Subject: [PATCH] Refactor generate_stream_text.py to use the "gemini" provider for streaming text generation --- 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 e432e0f..fe15b87 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="xai", stream=True) +r = sm.generate_text("Write a poem about the moon", llm_provider="gemini", stream=True) for chunk in r: print(chunk, end="", flush=True)