mirror of
https://github.com/kennethreitz/simplemind.git
synced 2026-06-05 06:46:18 +00:00
Refactor generate_stream_text.py to use the "gemini" provider for streaming text generation
This commit is contained in:
@@ -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)
|
||||||
|
|||||||
Reference in New Issue
Block a user