Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
41 changes: 41 additions & 0 deletions datasets/bird/test_agent_runtime_run_config.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,41 @@
############################################################
### Dataset / Eval Items (Vertex AI Multi-Agent Test)
############################################################
dataset_config: datasets/bird/prompts.json

databases:
- california_schools
num_trials: 1

database_configs:
- datasets/bat/db_configs/bigquery.yaml
- datasets/bird/db_configs/sqlite.yaml
dialects:
- bigquery
dialect: bigquery
query_types:
- dql
dataset_format: bird-standard-format

############################################################
### Prompt and Generation Modules
############################################################
model_config: datasets/model_configs/agent_runtime.yaml
prompt_generator: 'NOOPGenerator'

############################################################
### Scorer Related Configs
############################################################
scorers:
python_scorer:
script_path: 'evalbench/scorers/judges/hybrid_xa_judge.py'
scorer_name: 'hybrid_cross_db'

############################################################
### Reporting Related Configs
############################################################
reporting:
bigquery:
dataset_location: "US"
gcp_project_id: !ENV ${EVAL_GCP_PROJECT_ID}
csv: {}
14 changes: 14 additions & 0 deletions datasets/model_configs/agent_runtime.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
# Agent Runtime (Gemini Enterprise Agent Platform) Generator Manifest
# Enables native NL2SQL evaluation benchmarking against deployed Agent Runtime endpoints.
generator: agent_runtime

# Explicit resource URI string (e.g. "projects/12345/locations/us-central1/reasoningEngines/67890")
# Leave empty string "" to dynamically read AGENT_ENGINE_RESOURCE from env.
resource_name: !ENV ${AGENT_ENGINE_RESOURCE:""}

gcp_project_id: !ENV ${EVAL_GCP_PROJECT_ID}
gcp_region: !ENV ${EVAL_GCP_PROJECT_REGION:us-central1}

# Rate limiting and retry mechanics
execs_per_minute: 10
max_attempts: 3
28 changes: 26 additions & 2 deletions docs/configs/model-config.md
Original file line number Diff line number Diff line change
Expand Up @@ -28,6 +28,16 @@ These settings are **required only** for generators that utilize Google Cloud Ve

> Required*, you can globally set your GCP project_id and gcp_region using the environment variables `EVAL_GCP_PROJECT_ID` and `EVAL_GCP_PROJECT_REGION`.

## Agent Runtime (Gemini Enterprise Agent Platform) Configuration

These settings are **required only** for the `agent_runtime` generator, which connects to a live deployed Agent Runtime (Gemini Enterprise Agent Platform) instance.

| **Key** | **Required** | **Default Value** | **Description** |
| ----------------- | ------------ | ----------------- | ---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- |
| `resource_name` | Yes | N/A | The fully-qualified resource ID of the deployed Agent Engine resource, e.g. `projects/<GCP_PROJECT>/locations/<REGION>/reasoningEngines/<RESOURCE_ID>`. |
| `gcp_project_id` | Optional | `""` | The Google Cloud Project ID that hosts your Vertex AI resources. Can also be set via `EVAL_GCP_PROJECT_ID` environment variable. |
| `gcp_region` | Optional | `""` | The Google Cloud region where the Vertex AI service is deployed. Can also be set via `EVAL_GCP_PROJECT_REGION` environment variable. |

## Important Notes

- **Customization:** This configuration is fully customizable to the needs of the selected generator. You can add or remove keys as necessary.
Expand All @@ -37,10 +47,11 @@ These settings are **required only** for generators that utilize Google Cloud Ve
- **Rate Limiting & Retries:** The `execs_per_minute` and `max_attempts` keys help control the query generation process, ensuring that you can stay below project quota limits.


## Example Configuration
## Example Configurations

Below is an example of the updated YAML configuration file:
Below are examples of the YAML configuration files:

### Gemini Model Example
```yaml
# General Generator Configuration
generator: gcp_vertex_gemini
Expand All @@ -54,3 +65,16 @@ gcp_project_id: my_cool_gcp_project
gcp_region: us-east5
vertex_model: gemini-2.0-pro-exp-02-05
```

### Agent Runtime Example
```yaml
# General Generator Configuration
generator: agent_runtime
execs_per_minute: 10
max_attempts: 3

# Agent Runtime Configuration (Required for agent_runtime)
resource_name: "projects/my-gcp-project/locations/us-central1/reasoningEngines/1234567890"
gcp_project_id: "my-gcp-project"
gcp_region: "us-central1"
```
Loading
Loading