Grounding with Vertex AI Search

This page explains how you can ground responses by using your data from Vertex AI Search.

Grounding Gemini to your data

If you want to do retrieval-augmented generation (RAG), connect your model to your website data or your sets of documents, then use Grounding with Vertex AI Search .

Grounding to your data supports a maximum of 10 Vertex AI Search data sources and can be combined with Grounding with Google Search .

Supported models

This section lists the models that support grounding with your data.

Prerequisites

Before you can ground model output to your data, do the following:

  1. In the Google Cloud console, go to the IAMpage, and search for the discoveryengine.servingConfigs.search permission, which is required for the grounding service to work.

    Go to IAM

  2. Enable AI Applications and activate the API.

  3. Create a AI Applications data source and application.

See the Introduction to Vertex AI Search for more.

Enable AI Applications

  1. In the Google Cloud console, go to the AI Applicationspage.

    AI Applications

  2. Read and agree to the terms of service, then click Continue and activate the API.

AI Applications is available in the global location, or the eu and us multi-region. To learn more, see AI Applications locations

Create a data store in AI Applications

To create a data store in AI Applications, you can choose to ground with website data or documents.

Website

  1. Open the Create Data Store page from the Google Cloud console.

  2. In Website Contentbox, click Select .
    Specify the websites for your data storepane displays.

  3. If Advanced website indexingisn't checked, then select the Advanced website indexingcheckbox to turn it on.
    Configure your data storepane displays.

  4. In the Specify URL patterns to indexsection, do the following:

    • Add URLs for Sites to include.
    • Optional: Add URLs for Sites to exclude.
  5. Click Continue.

  6. In the Configure your data storepane,

    1. Select a value from the Location of your data storelist.
    2. Enter a name in the Your data store namefield. The ID is generated. Use this ID when you generate your grounded responses with your data store. For more information, see Generate grounded responses with your data store .
    3. Click Create.

Documents

  1. Open the Create Data Store page from the Google Cloud console.

  2. In Cloud Storagebox, click Select .
    Import data from Cloud Storagepane displays.

  3. In the Unstructured documents (PDF, HTML, TXT and more)section, select Unstructured documents (PDF, HTML, TXT and more).

  4. Select a Synchronization frequencyoption.

  5. Select a Select a folder or a file you want to importoption, and enter the path in the field.

  6. Click Continue.
    Configure your data storepane displays.

  7. In the Configure your data storepane,

    1. Select a value from the Location of your data storelist.
    2. Enter a name in the Your data store namefield. The ID is generated.
    3. To select parsing and chunking options for your documents, expand the Document Processing Optionssection. For more information about different parsers, see Parse documents .
    4. Click Create.
  8. Click Create.

Generate grounded responses with your data store

Use the following instructions to ground a model with your data. A maximum of 10 data stores is supported.

If you don't know your data store ID, follow these steps:

  1. In the Google Cloud console, go to the AI Applicationspage and in the navigation menu, click Data stores.

    Go to the Data stores page

  2. Click the name of your data store.

  3. On the Data page for your data store, get the data store ID.

Console

To ground your model output to AI Applications by using Vertex AI Studio in the Google Cloud console, follow these steps:

  1. In the Google Cloud console, go to the Vertex AI Studio Freeform page.

    Go to Freeform

  2. To turn on grounding, click the Grounding: your datatoggle.
  3. Click Customize.
    1. Select Vertex AI Searchas your source.
    2. Using this path format, replace your data store's Project ID and the ID of the data store:

      projects/ project_id /locations/global/collections/default_collection/dataStores/ data_store_id .

  4. Click Save.
  5. Enter your prompt in the text box, and click Submit.

Your prompt responses are grounded to AI Applications.

Python

Before trying this sample, follow the Python setup instructions in the Vertex AI quickstart using client libraries . For more information, see the Vertex AI Python API reference documentation .

To authenticate to Vertex AI, set up Application Default Credentials. For more information, see Set up authentication for a local development environment .

  from 
  
 google 
  
 import 
 genai 
 from 
  
 google.genai.types 
  
 import 
 ( 
 GenerateContentConfig 
 , 
 HttpOptions 
 , 
 Retrieval 
 , 
 Tool 
 , 
 VertexAISearch 
 , 
 ) 
 client 
 = 
 genai 
 . 
 Client 
 ( 
 http_options 
 = 
 HttpOptions 
 ( 
 api_version 
 = 
 "v1" 
 )) 
 # Load Data Store ID from Vertex AI Search 
 # datastore = "projects/111111111111/locations/global/collections/default_collection/dataStores/data-store-id" 
 response 
 = 
 client 
 . 
 models 
 . 
 generate_content 
 ( 
 model 
 = 
 "gemini-2.5-flash" 
 , 
 contents 
 = 
 "How do I make an appointment to renew my driver's license?" 
 , 
 config 
 = 
 GenerateContentConfig 
 ( 
 tools 
 = 
 [ 
 # Use Vertex AI Search Tool 
 Tool 
 ( 
 retrieval 
 = 
 Retrieval 
 ( 
 vertex_ai_search 
 = 
 VertexAISearch 
 ( 
 datastore 
 = 
 datastore 
 , 
 ) 
 ) 
 ) 
 ], 
 ), 
 ) 
 print 
 ( 
 response 
 . 
 text 
 ) 
 # Example response: 
 # 'The process for making an appointment to renew your driver's license varies depending on your location. To provide you with the most accurate instructions...' 
 

REST

To test a text prompt by using the Vertex AI API, send a POST request to the publisher model endpoint.

Before using any of the request data, make the following replacements:

  • LOCATION : The region to process the request.
  • PROJECT_ID : Your project ID .
  • MODEL_ID : The model ID of the multimodal model.
  • TEXT : The text instructions to include in the prompt.

HTTP method and URL:

POST https:// LOCATION 
-aiplatform.googleapis.com/v1beta1/projects/ PROJECT_ID 
/locations/ LOCATION 
/publishers/google/models/ MODEL_ID 
:generateContent

Request JSON body:

{
  "contents": [{
    "role": "user",
    "parts": [{
      "text": " TEXT 
"
    }]
  }],
  "tools": [{
    "retrieval": {
      "vertexAiSearch": {
        "datastore": projects/ PROJECT_ID 
/locations/global/collections/default_collection/dataStores/ DATA_STORE_ID 
}
    }
  }],
  "model": "projects/ PROJECT_ID 
/locations/ LOCATION 
/publishers/google/models/ MODEL_ID 
"
}

To send your request, expand one of these options:

You should receive a JSON response similar to the following:

{
  "candidates": [
    {
      "content": {
        "role": "model",
        "parts": [
          {
            "text": "You can make an appointment on the website https://dmv.gov/"
          }
        ]
      },
      "finishReason": "STOP",
      "safetyRatings": [
        "..."
      ],
      "groundingMetadata": {
        "retrievalQueries": [
          "How to make appointment to renew driving license?"
        ],
        "groundingChunks": [
          {
            "retrievedContext": {
              "uri": "https://vertexaisearch.cloud.google.com/grounding-api-redirect/AXiHM.....QTN92V5ePQ==",
              "title": "dmv"
            }
          }
        ],
        "groundingSupport": [
          {
            "segment": {
              "startIndex": 25,
              "endIndex": 147
            },
            "segment_text": "ipsum lorem ...",
            "supportChunkIndices": [1, 2],
            "confidenceScore": [0.9541752, 0.97726375]
          },
          {
            "segment": {
              "startIndex": 294,
              "endIndex": 439
            },
            "segment_text": "ipsum lorem ...",
            "supportChunkIndices": [1],
            "confidenceScore": [0.9541752, 0.9325467]
          }
        ]
      }
    }
  ],
  "usageMetadata": {
    "..."
  }
}

Understand your response

The response from both APIs include the LLM-generated text, which is called a candidate . If your model prompt successfully grounds to your data source, then the responses include grounding metadata, which identifies the parts of the response that were derived from your data. However, there are several reasons this metadata may not be provided, and the prompt response won't be grounded. These reasons include low-source relevance or incomplete information within the model's response.

The following is a breakdown of the output data:

  • Role: Indicates the sender of the grounded answer. Because the response always contains grounded text, the role is always model .
  • Text: The grounded answer generated by the LLM.
  • Grounding metadata: Information about the grounding source, which contains the following elements:
    • Grounding chunks: A list of results from your index that support the answer.
    • Grounding supports: Information about a specific claim within the answer that can be used to show citations:
    • Segment: The part of the model's answer that is substantiated by a grounding chunk.
    • Grounding chunk index: The index of the grounding chunks in the grounding chunks list that corresponds to this claim.
    • Confidence scores: A number from 0 to 1 that indicates how grounded the claim is in the provided set of grounding chunks. Not available for Gemini 2.5 and later.

What's next

Create a Mobile Website
View Site in Mobile | Classic
Share by: