MCP Server for Vertex AI Search

by ubie-oss

35 stars
411 downloads
Not rated
GitHub Website

About

MCP Server for Vertex AI Search is a Model Context Protocol server that searches documents using Vertex AI grounding with Gemini. It integrates one or multiple Vertex AI data stores to ground responses in private data.

Details

Author
ubie-oss
GitHub stars
35
Downloads
411
Categories
Search

- Uses Gemini with Vertex AI grounding for accurate search
- Supports one or multiple Vertex AI data stores
- Offers both SSE and stdio transport modes
- Provides a dedicated search test command
- Can run locally or via Docker
- Configurable YAML configuration file

Setting up with Highlight

This MCP is not yet compatible with Highlight’s one-click setup. However, you can still use it with Highlight by following these steps:

  1. Download and install Highlight from highlightai.com/download
  2. Navigate to the plugins tab and select "Add Custom Plugin"
  3. Configure the plugin with the settings below
    Plugin Name MCP Server for Vertex AI Search
    Command (node, npx, python, etc.)

    Please refer to the README for specific instructions on how to obtain API keys or other required environment variables.

  4. Enable "Start Automatically" if you want the plugin to start when Highlight launches

From the repository

Clone the repository, create a virtual environment with uv, and install dependencies. Then run the server with uv run mcp-vertexai-search serve --config config.yml --transport <stdio|sse>. Alternatively, install the Python package from GitHub and use the same command. A YAML config file (template provided) is required.

Claude Desktop / Cursor

Paste into your MCP client config file to install this server.

{
    "mcpServers": {
        "mcp server for vertex ai search": {
            "mcp-vertexai-search": {
                "command": "uv",
                "args": [
                    "venv"
                ]
            }
        }
    }
}

McpServers

{
    "mcp-vertexai-search": {
        "command": "uv",
        "args": [
            "venv"
        ]
    }
}

MCP Server for Vertex AI Search

This is a MCP server to search documents using Vertex AI.

Architecture

This solution uses Gemini with Vertex AI grounding to search documents using your private data.
Grounding improves the quality of search results by grounding Gemini's responses in your data stored in Vertex AI Datastore.
We can integrate one or multiple Vertex AI data stores to the MCP server.
For more details on grounding, refer to Vertex AI Grounding Documentation.

Architecture

How to use

There are two ways to use this MCP server.
If you want to run this on Docker, the first approach would be good as Dockerfile is provided in the project.

1. Clone the repository

```shell

No reviews yet — be the first

Sign in to leave a review

Use Google, GitHub, or an email account so ratings stay tied to real people.

Email sign in

No reviews posted yet.