Brettapps commited on
Commit
0ad783b
·
verified ·
1 Parent(s): b78bb32

Upload folder using huggingface_hub

Browse files
.gemini/extensions/vertex/commands/vertex/analyze_data_driven_optimize_results.toml ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name = "analyze_data_driven_optimize_results"
2
+ description = "Analyzes the output of a Data-Driven Optimize job and saves the results."
3
+ prompt = """
4
+ You are tasked with generating parameters for a `prompt_optimizer.analyzer.analyze_results` tool call.
5
+
6
+ **Parameters:**
7
+
8
+ 1. **`output_path` (string, required):**
9
+ * **Source:** The GCS, Google Drive, or local path containing the job results. This is usually provided by the user.
10
+
11
+ 2. **`analysis_data_path` (string, optional):**
12
+ * **Default:** "analysis_data.json"
13
+ * **Source:** The local path where the analysis metadata will be saved.
14
+
15
+ 3. **`top_n_prompts` (integer, optional):**
16
+ * **Default:** 10
17
+ * **Source:** The number of top-performing prompts to include in the analysis.
18
+
19
+ **Workflow Integration:**
20
+ Refer to the "Analysis and Suggestions" workflow in `GEMINI.md`. This tool is the first step in a chain that includes formulating suggestions and generating an HTML report.
21
+ """
.gemini/extensions/vertex/commands/vertex/create_prompt.toml ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name = "create_prompt"
2
+ description = "Writes generated prompt content, system instructions, and specified model information to Vertex AI Prompt Management."
3
+ prompt = """
4
+ You are tasked with generating parameters for a `create_prompt` SDK API call to Vertex AI Prompt Management.
5
+
6
+ **Parameters for the `create_prompt` function:**
7
+
8
+ 1. **`project_id` (string, optional):**
9
+ * **Source:** If the user explicitly provides a 'project_id' in the conversation, use that. Otherwise, **do not ask the user for it**. The underlying Python tool will handle defaulting to the `GOOGLE_CLOUD_PROJECT` environment variable.
10
+
11
+ 2. **`location_id` (string, optional):**
12
+ * **Source:** If the user explicitly provides a 'location_id' in the conversation, use that. Otherwise, **do not ask the user for it**. The underlying Python tool defaults to "us-central1" or uses `GOOGLE_CLOUD_LOCATION`.
13
+
14
+ 3. **`content` (string, required):**
15
+ * **Source:** Refer to the "Detailed Instructions for `create_prompt` Parameters" section in `GEMINI.md` for how to source the `content` argument.
16
+
17
+ 4. **`system_instruction` (string, required):**
18
+ * **Source:** Refer to the "Detailed Instructions for `create_prompt` Parameters" section in `GEMINI.md` for how to source the `system_instruction` argument.
19
+
20
+ 5. **`model` (string, required):**
21
+ * **Source:** Refer to the "Detailed Instructions for `create_prompt` Parameters" section in `GEMINI.md` for how to source the `model` argument.
22
+
23
+ 6. **`display_name` (string, optional):**
24
+ * **Source:** Refer to the "Detailed Instructions for `create_prompt` Parameters" section in `GEMINI.md` for how to source the `display_name` argument.
25
+
26
+ **Clarification Policy:**
27
+ If any required information (`content`, `system_instruction`, `model`, or `display_name` when not already set) is missing or ambiguous based on the current context, you *must* ask the user for clarification before proceeding with the `create_prompt` call.
28
+ """
.gemini/extensions/vertex/commands/vertex/delete_prompt.toml ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name = "delete_prompt"
2
+ description = "Deletes a prompt in Vertex AI Prompt Management."
3
+ prompt = """
4
+ You are tasked with generating parameters for a `delete_prompt` SDK API call to Vertex AI Prompt Management.
5
+
6
+ **Parameters for the `delete_prompt` function:**
7
+
8
+ 1. **`prompt_id` (string, required):**
9
+ * **Source:** Must be explicitly requested from the user. If not provided, you *must* ask the user for the prompt ID.
10
+
11
+ 2. **`project_id` (string, optional):**
12
+ * **Source:** If the user explicitly provides a 'project_id' in the conversation, use that. Otherwise, **do not ask the user for it**. The underlying Python tool will handle defaulting to the `GOOGLE_CLOUD_PROJECT` environment variable.
13
+
14
+ 3. **`location_id` (string, optional):**
15
+ * **Source:** If the user explicitly provides a 'location_id' in the conversation, use that. Otherwise, **do not ask the user for it**. The underlying Python tool defaults to "us-central1" or uses `GOOGLE_CLOUD_LOCATION`.
16
+
17
+ **Clarification Policy:**
18
+ If the `prompt_id` is missing, you *must* ask the user for clarification before proceeding with the `delete_prompt` call.
19
+ """
.gemini/extensions/vertex/commands/vertex/generate_html_report.toml ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name = "generate_html_report"
2
+ description = "Generates a comprehensive HTML report from Data-Driven Optimize analysis results."
3
+ prompt = """
4
+ You are tasked with generating parameters for a `prompt_optimizer.analyzer.generate_report` tool call.
5
+
6
+ **Parameters:**
7
+
8
+ 1. **`analysis_data_path` (string, required):**
9
+ * **Source:** The path to the JSON file generated by `analyze_results`.
10
+
11
+ 2. **`report_path` (string, optional):**
12
+ * **Default:** "data_driven_optimize_analysis_report.html"
13
+ * **Source:** The path where the HTML report will be saved.
14
+
15
+ 3. **`suggested_config_data` (object, optional):**
16
+ * **Source:** A dictionary containing `"suggested_config"` (modifications) and `"rationale"`. You must formulate this based on the `analysis_data` from the previous step using the guidelines in `src/vertex/prompt_optimizer/docs/data_driven_optimize_tuning_guide.md`.
17
+
18
+ 4. **`top_n_prompts` (integer, optional):**
19
+ * **Default:** 10
20
+
21
+ **Workflow Integration:**
22
+ Refer to the "Analysis and Suggestions" workflow in `GEMINI.md`. This tool is the final step in the analysis chain.
23
+ """
.gemini/extensions/vertex/commands/vertex/list_prompts.toml ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name = "list_prompts"
2
+ description = "List Vertex AI prompts, optionally filtered by display name."
3
+ prompt = """
4
+ You are tasked with generating parameters for an `list_prompts` SDK API call to Vertex AI Prompt Management.
5
+
6
+ **Parameters for the `list_prompts` function:**
7
+
8
+ 1. **`display_name` (string, optional):**
9
+ * **Source:** User's input for the display name to filter by.
10
+
11
+ 2. **`page_size` (int, optional):**
12
+ * **Source:** The user's input for the maximum number of prompts to return. Defaults to 10.
13
+
14
+ 3. **`project_id` (string, optional):**
15
+ * **Source:** If the user explicitly provides a 'project_id' in the conversation, use that. Otherwise, **do not ask the user for it**. The underlying Python tool will handle defaulting to the `GOOGLE_CLOUD_PROJECT` environment variable.
16
+
17
+ 4. **`location_id` (string, optional):**
18
+ * **Source:** If the user explicitly provides a 'location_id' in the conversation, use that. Otherwise, **do not ask the user for it**. The underlying Python tool defaults to "us-central1" or uses `GOOGLE_CLOUD_LOCATION`.
19
+ """
.gemini/extensions/vertex/commands/vertex/read_prompt.toml ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name = "read_prompt"
2
+ description = "Read the prompt content with given prompt id from Vertex AI Prompt Management."
3
+ prompt = """
4
+ You are tasked with generating parameters for a `read_prompt` SDK API call to Vertex AI Prompt Management.
5
+
6
+ **Parameters for the `read_prompt` function:**
7
+
8
+ 1. **`prompt_id` (string, required):**
9
+ * **Source:** Must be explicitly requested from the user. If not provided, you *must* ask the user for the prompt ID.
10
+
11
+ 2. **`project_id` (string, optional):**
12
+ * **Source:** If the user explicitly provides a 'project_id' in the conversation, use that. Otherwise, **do not ask the user for it**. The underlying Python tool will handle defaulting to the `GOOGLE_CLOUD_PROJECT` environment variable.
13
+
14
+ 3. **`location_id` (string, optional):**
15
+ * **Source:** If the user explicitly provides a 'location_id' in the conversation, use that. Otherwise, **do not ask the user for it**. The underlying Python tool defaults to "us-central1" or uses `GOOGLE_CLOUD_LOCATION`.
16
+
17
+ **Clarification Policy:**
18
+ If the `prompt_id` is missing, you *must* ask the user for clarification before proceeding with the `read_prompt` call.
19
+ """
.gemini/extensions/vertex/commands/vertex/run_data_driven_optimize.toml ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name = "run_data_driven_optimize"
2
+ description = "Starts a data-driven prompt optimization job on Vertex AI."
3
+ prompt = """
4
+ You are tasked with generating parameters for a `run_data_driven_optimize` tool call.
5
+
6
+ **Parameters:**
7
+
8
+ 1. **`config_gcs_path` (string, required):**
9
+ * **Source:** The GCS URI to the configuration JSON file.
10
+
11
+ 2. **`service_account` (string, required):**
12
+ * **Source:** The service account email to run the job (typically `{project_number}-compute@developer.gserviceaccount.com`). Ask the user if not known.
13
+
14
+ 3. **`prompt_optimizer_method` (string, required):**
15
+ * **Source:** The method for prompt optimization. Either "VAPO" or "OPTIMIZATION_TARGET_GEMINI_NANO".
16
+
17
+ 4. **`wait_for_completion` (boolean, optional):**
18
+ * **Default:** false
19
+ * **Source:** Whether to block until the job completes.
20
+
21
+ **Clarification Policy:**
22
+ If `config_gcs_path`, `service_account` or `prompt_optimizer_method` is missing, ask the user for it.
23
+ """
.gemini/extensions/vertex/commands/vertex/run_few_shot_prompt_optimization.toml ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name = "run_few_shot_prompt_optimization"
2
+ description = "Optimize a prompt based on few examples using few-shot optimization from Vertex AI SDK."
3
+ prompt = """
4
+ You are tasked to call `run_few_shot_prompt_optimization` tool call.
5
+
6
+ **Parameters:**
7
+
8
+ 1. **`prompt_to_optimize` (string, required):**
9
+ * **Source:** Prompt (or system instructions) that will be optimized.
10
+ 2. **`example_path` (string, required):**
11
+ * **Source:** The GCS URI to the CSV file containing examples
12
+ 3. **`method:` (string, required):**
13
+ * **Source:** The method for prompt optimization. Either "TARGET_RESPONSE" or "RUBRICS"
14
+
15
+
16
+ **Clarification Policy:**
17
+ If `prompt_to_optimize`, `example_path` or `method` is missing, ask the user for it.
18
+ """
.gemini/extensions/vertex/commands/vertex/update_prompt.toml ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name = "update_prompt"
2
+ description = "Updates a prompt in Vertex AI Prompt Management with new content, system instructions, model, or display name."
3
+ prompt = """
4
+ You are tasked with generating parameters for an `update_prompt` SDK API call to Vertex AI Prompt Management.
5
+
6
+ **Parameters for the `update_prompt` function:**
7
+
8
+ 1. **`prompt_id` (string, required):**
9
+ * **Source:** Must be explicitly requested from the user. If not provided, you *must* ask the user for the prompt ID.
10
+
11
+ 2. **`project_id` (string, optional):**
12
+ * **Source:** If the user explicitly provides a 'project_id' in the conversation, use that. Otherwise, **do not ask the user for it**. The underlying Python tool will handle defaulting to the `GOOGLE_CLOUD_PROJECT` environment variable.
13
+
14
+ 3. **`location_id` (string, optional):**
15
+ * **Source:** If the user explicitly provides a 'location_id' in the conversation, use that. Otherwise, **do not ask the user for it**. The underlying Python tool defaults to "us-central1" or uses `GOOGLE_CLOUD_LOCATION`.
16
+
17
+ 4. **`content` (string, optional):**
18
+ * **Source:** The user's input for the new prompt content.
19
+
20
+ 5. **`system_instruction` (string, optional):**
21
+ * **Source:** The user's input for the new system instruction.
22
+
23
+ 6. **`model` (string, optional):**
24
+ * **Source:** The user's input for the new model.
25
+
26
+ 7. **`display_name` (string, optional):**
27
+ * **Source:** The user's input for the new display name.
28
+
29
+ **Clarification Policy:**
30
+ # If the `prompt_id` is missing, you *must* ask the user for clarification before proceeding with the `update_prompt` call.
31
+
32
+ # - If the user provides a display name in the initial request without an ID, the agent should first call 'list_prompts'.
33
+ """
.gemini/extensions/vertex/commands/vertex/write_data_driven_optimize_config.toml ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name = "write_data_driven_optimize_config"
2
+ description = "Constructs and uploads a Data-Driven Optimize configuration file to GCS."
3
+ prompt = """
4
+ You are tasked with generating parameters for a `prompt_optimizer.config.write_config` tool call.
5
+
6
+ **Parameters:**
7
+
8
+ 1. **`gcs_config_uri` (string, required):**
9
+ * **Source:** The destination GCS URI for the config file.
10
+
11
+ 2. **`prompt_optimizer_method` (string, required):**
12
+ * **Source:** The method for prompt optimization. Either "VAPO" or "OPTIMIZATION_TARGET_GEMINI_NANO". If "VAPO", `batch_size` will be automatically removed from the config.
13
+
14
+ 3. **`target_model_endpoint_url` (string, optional):**
15
+ * **Source:** The custom endpoint URL for the target model. This is required ONLY for the Gemini Nano optimization target.
16
+
17
+ 4. **`base_config` (object, optional):**
18
+ * **Source:** A dictionary representing the base configuration.
19
+
20
+ 5. **`modifications` (object, optional):**
21
+ * **Source:** A dictionary of modifications to apply to the JSON configuration file. Only allowed tuning knobs (from `src/vertex/prompt_optimizer/docs/data_driven_optimize_tuning_guide.md`) and path-related fields should be included. Do NOT include tool-level parameters like `prompt_optimizer_method` here.
22
+
23
+ 6. **`base_config_path` (string, optional):**
24
+ * **Source:** A GCS path to an existing config file to use as a base.
25
+
26
+ **Clarification Policy:**
27
+ If `gcs_config_uri` or `prompt_optimizer_method` is missing, ask the user for it.
28
+ """