Use flash more often.

Change-Id: If20e5d5e8462d160681d9dc2bfec965fd94fb633

Author: MarkDaoust

README.md

@@ -33,7 +33,7 @@ genai.configure(api_key=os.environ["GEMINI_API_KEY"])
 3. Create a model and run a prompt.
 
 ```python
-model = genai.GenerativeModel('gemini-1.0-pro-latest')
+model = genai.GenerativeModel('gemini-1.5-flash')
 response = model.generate_content("The opposite of hot is")
 print(response.text)
 ```

samples/code_execution.py

@@ -29,7 +29,7 @@ def test_code_execution_basic(self):
         )
 
         # Each `part` either contains `text`, `executable_code` or an `execution_result`
-        for part in result.candidates[0].content.parts:
+        for part in response.candidates[0].content.parts:
             print(part, "\n")
 
         print("-" * 80)
@@ -92,7 +92,7 @@ def test_code_execution_basic(self):
 
     def test_code_execution_request_override(self):
         # [START code_execution_request_override]
-        model = genai.GenerativeModel(model_name="gemini-1.5-pro")
+        model = genai.GenerativeModel(model_name="gemini-1.5-flash")
         response = model.generate_content(
             (
                 "What is the sum of the first 50 prime numbers? "
@@ -140,7 +140,7 @@ def test_code_execution_request_override(self):
 
     def test_code_execution_chat(self):
         # [START code_execution_chat]
-        model = genai.GenerativeModel(model_name="gemini-1.5-pro", tools="code_execution")
+        model = genai.GenerativeModel(model_name="gemini-1.5-flash", tools="code_execution")
         chat = model.start_chat()
         response = chat.send_message('Can you print "Hello world!"?')
         response = chat.send_message(

samples/count_tokens.py

@@ -23,7 +23,7 @@
 class UnitTests(absltest.TestCase):
     def test_tokens_context_window(self):
         # [START tokens_context_window]
-        model_info = genai.get_model("models/gemini-1.0-pro-001")
+        model_info = genai.get_model("models/gemini-1.5-flash")
 
         # Returns the "context window" for the model,
         # which is the combined input and output token limits.
@@ -91,7 +91,7 @@ def test_tokens_multimodal_image_inline(self):
         model = genai.GenerativeModel("models/gemini-1.5-flash")
 
         prompt = "Tell me about this image"
-        your_image_file = PIL.Image.open("image.jpg")
+        your_image_file = PIL.Image.open(media/"organ.jpg")
 
         # Call `count_tokens` to get the input token count
         # of the combined text and file (`total_tokens`).
@@ -115,7 +115,7 @@ def test_tokens_multimodal_image_file_api(self):
         model = genai.GenerativeModel("models/gemini-1.5-flash")
 
         prompt = "Tell me about this image"
-        your_image_file = genai.upload_file(path="image.jpg")
+        your_image_file = genai.upload_file(path=media/"organ.jpg")
 
         # Call `count_tokens` to get the input token count
         # of the combined text and file (`total_tokens`).

samples/rest/code_execution.sh

@@ -2,7 +2,7 @@ set -eu
 
 echo "[START code_execution_basic]"
 # [START code_execution_basic]
-curl "https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-pro-latest:generateContent?key=$GOOGLE_API_KEY" \
+curl "https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash:generateContent?key=$GOOGLE_API_KEY" \
 -H 'Content-Type: application/json' \
 -d ' {"tools": [{'code_execution': {}}],
     "contents": {
@@ -16,7 +16,7 @@ curl "https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-pro-lat
 
 echo "[START code_execution_chat]"
 # [START code_execution_chat]
-curl "https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-pro-latest:generateContent?key=$GOOGLE_API_KEY" \
+curl "https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash:generateContent?key=$GOOGLE_API_KEY" \
 -H 'Content-Type: application/json' \
 -d '{"tools": [{'code_execution': {}}],
     "contents": [

samples/rest/controlled_generation.sh

@@ -2,7 +2,7 @@ set -eu
 
 echo "json_controlled_generation"
 # [START json_controlled_generation]
-curl "https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-pro-latest:generateContent?key=$GOOGLE_API_KEY" \
+curl "https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash:generateContent?key=$GOOGLE_API_KEY" \
 -H 'Content-Type: application/json' \
 -d '{
     "contents": [{
@@ -27,7 +27,7 @@ curl "https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-pro-lat
 
 echo "json_no_schema"
 # [START json_no_schema]
-curl "https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-pro-latest:generateContent?key=$GOOGLE_API_KEY" \
+curl "https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash:generateContent?key=$GOOGLE_API_KEY" \
 -H 'Content-Type: application/json' \
 -d '{
     "contents": [{

samples/rest/models.sh

@@ -7,5 +7,5 @@ curl https://generativelanguage.googleapis.com/v1beta/models?key=$GOOGLE_API_KEY
 
 echo "[START models_get]"
 # [START models_get]
-curl https://generativelanguage.googleapis.com/v1beta/models/gemini-pro?key=$GOOGLE_API_KEY
+curl https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash?key=$GOOGLE_API_KEY
 # [END models_get]

samples/rest/safety_settings.sh

@@ -2,37 +2,33 @@ set -eu
 
 echo "[START safety_settings]"
 # [START safety_settings]
-echo '{
+    echo '{
     "safetySettings": [
-        {'category': HARM_CATEGORY_HARASSMENT, 'threshold': BLOCK_ONLY_HIGH}
+        {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_ONLY_HIGH"}
     ],
     "contents": [{
         "parts":[{
             "text": "'I support Martians Soccer Club and I think Jupiterians Football Club sucks! Write a ironic phrase about them.'"}]}]}' > request.json
 
-    curl "https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent?key=$GOOGLE_API_KEY" \
+    curl "https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash:generateContent?key=$GOOGLE_API_KEY" \
         -H 'Content-Type: application/json' \
         -X POST \
-        -d @request.json  2> /dev/null > tee response.json
-
-    jq .promptFeedback > response.json
+        -d @request.json 2> /dev/null
 # [END safety_settings]
 
 echo "[START safety_settings_multi]"
 # [START safety_settings_multi]
-echo '{
-    "safetySettings": [
-        {'category': HARM_CATEGORY_HARASSMENT, 'threshold': BLOCK_ONLY_HIGH},
-        {'category': HARM_CATEGORY_HATE_SPEECH, 'threshold': BLOCK_MEDIUM_AND_ABOVE}
-    ],
-    "contents": [{
-        "parts":[{
-            "text": "'I support Martians Soccer Club and I think Jupiterians Football Club sucks! Write a ironic phrase about them.'"}]}]}' > request.json
+    echo '{
+        "safetySettings": [
+            {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_ONLY_HIGH"},
+            {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_MEDIUM_AND_ABOVE"}
+        ],
+        "contents": [{
+            "parts":[{
+                "text": "'I support Martians Soccer Club and I think Jupiterians Football Club sucks! Write a ironic phrase about them.'"}]}]}' > request.json
 
-    curl "https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent?key=$GOOGLE_API_KEY" \
+    curl "https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash:generateContent?key=$GOOGLE_API_KEY" \
         -H 'Content-Type: application/json' \
         -X POST \
-        -d @request.json  2> /dev/null > response.json
-
-    jq .promptFeedback > response.json
+        -d @request.json 2> /dev/null
 # [END safety_settings_multi]

samples/rest/system_instruction.sh

@@ -2,7 +2,7 @@ set -eu
 
 echo "[START system_instruction]"
 # [START system_instruction]
-curl "https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-pro-latest:generateContent?key=$GOOGLE_API_KEY" \
+curl "https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash:generateContent?key=$GOOGLE_API_KEY" \
 -H 'Content-Type: application/json' \
 -d '{ "system_instruction": {
     "parts":

samples/tuned_models.py

@@ -27,7 +27,7 @@ def test_tuned_models_create(self):
         # [START tuned_models_create]
         import time
 
-        base_model = "models/gemini-1.0-pro-001"
+        base_model = "models/gemini-1.5-flash-001-tuning"
         training_data = [
             {"text_input": "1", "output": "2"},
             # ... more examples ...
@@ -94,7 +94,7 @@ def test_tuned_models_list(self):
     def test_tuned_models_delete(self):
         import time
 
-        base_model = "models/gemini-1.0-pro-001"
+        base_model = "models/gemini-1.5-flash-001-tuning"
         training_data = samples / "increment_tuning_data.json"
         try:
             operation = genai.create_tuned_model(