Refactor generate_stream_text.py to use the "gemini" provider for streaming text generation

This commit is contained in:
2024-11-02 16:48:54 -04:00
parent e13d03f40b
commit 028e89b080
+1 -1
View File
@@ -1,7 +1,7 @@
from _context import sm from _context import sm
# Defaults to the default provider (openai) # 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: for chunk in r:
print(chunk, end="", flush=True) print(chunk, end="", flush=True)