File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -25,14 +25,14 @@ class UnitTests(absltest.TestCase):
25
25
def test_text_gen_text_only_prompt (self ):
26
26
# [START text_gen_text_only_prompt]
27
27
model = genai .GenerativeModel ("gemini-1.5-flash" )
28
- response = model .generate_content ("Give me python code to sort a list " )
28
+ response = model .generate_content ("Write a story about a magic backpack. " )
29
29
print (response .text )
30
30
# [END text_gen_text_only_prompt]
31
31
32
32
def test_text_gen_text_only_prompt_streaming (self ):
33
33
# [START text_gen_text_only_prompt_streaming]
34
34
model = genai .GenerativeModel ("gemini-1.5-flash" )
35
- response = model .generate_content ("Give me python code to sort a list " , stream = True )
35
+ response = model .generate_content ("Write a story about a magic backpack. " , stream = True )
36
36
for chunk in response :
37
37
print (chunk .text )
38
38
print ("_" * 80 )
You can’t perform that action at this time.
0 commit comments