Update vertex_ai_prompt_optimizer_ui.ipynb

GoogleCloudPlatform · Nov 12, 2024 · b7053a9 · b7053a9
1 parent b5dd69e
commit b7053a9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/gemini/prompts/prompt_optimizer/vertex_ai_prompt_optimizer_ui.ipynb b/gemini/prompts/prompt_optimizer/vertex_ai_prompt_optimizer_ui.ipynb
@@ -265,7 +265,8 @@
         "\n",
         "# @markdown **Misc Configs**: <br/>\n",
         "PLACEHOLDER_TO_VALUE = \"{}\"  # @param\n",
-        "RESPONSE_MIME_TYPE = \"application/json\"  # @param [\"text/plain\", \"application/json\"]\n",
+        "RESPONSE_MIME_TYPE = \"application/json\"  # @param [\"text/plain\", \"application/json\", \"text/x.enum\"]\n",
+        "RESPONSE_SCHEMA = \"\"\n",
         "TARGET_LANGUAGE = \"English\"  # @param [\"English\", \"French\", \"German\", \"Hebrew\", \"Hindi\", \"Japanese\", \"Korean\", \"Portuguese\", \"Simplified Chinese\", \"Spanish\", \"Traditional Chinese\"]"
       ]
     },
@@ -335,6 +336,7 @@
         "    \"input_data_path\": input_data_path,\n",
         "    \"output_path\": output_path,\n",
         "    \"response_mime_type\": RESPONSE_MIME_TYPE,\n",
+        "    \"response_schema\": RESPONSE_SCHEMA,\n",
         "    \"language\": TARGET_LANGUAGE,\n",
         "    \"placeholder_to_content\": json.loads(PLACEHOLDER_TO_VALUE),\n",
         "}\n",