Class VertexAiSearchTool

java.lang.Object

com.google.adk.tools.BaseTool

com.google.adk.tools.VertexAiSearchTool

public abstract class VertexAiSearchTool
extends BaseTool

A built-in tool using Vertex AI Search.

This tool can be configured with either a dataStoreId (the Vertex AI search data store resource ID) or a searchEngineId (the Vertex AI search engine resource ID).

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	VertexAiSearchTool.Builder	Builder for VertexAiSearchTool.

Nested classes/interfaces inherited from class BaseTool

BaseTool.ToolArgsConfig, BaseTool.ToolConfig

Method Summary

Modifier and Type	Method	Description
static VertexAiSearchTool.Builder	builder()
abstract Optional<String>	dataStore()
abstract Optional<String>	dataStoreId()
abstract com.google.common.collect.ImmutableList<com.google.genai.types.VertexAISearchDataStoreSpec>	dataStoreSpecs()
abstract Optional<String>	filter()
abstract Optional<String>	location()
abstract Optional<Integer>	maxResults()
io.reactivex.rxjava3.core.Completable	processLlmRequest(LlmRequest.Builder llmRequestBuilder, ToolContext toolContext)	Processes the outgoing LlmRequest.Builder.
abstract Optional<String>	project()
abstract Optional<String>	searchEngineId()

Methods inherited from class BaseTool

customMetadata, declaration, description, fromConfig, longRunning, name, runAsync, setCustomMetadata

Methods inherited from class Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Details

dataStoreId

public abstract Optional<String> dataStoreId()

dataStoreSpecs

public abstract com.google.common.collect.ImmutableList<com.google.genai.types.VertexAISearchDataStoreSpec> dataStoreSpecs()

searchEngineId

public abstract Optional<String> searchEngineId()

filter

public abstract Optional<String> filter()

maxResults

public abstract Optional<Integer> maxResults()

project

public abstract Optional<String> project()

location

public abstract Optional<String> location()

dataStore

public abstract Optional<String> dataStore()

builder

public static VertexAiSearchTool.Builder builder()

processLlmRequest

public io.reactivex.rxjava3.core.Completable processLlmRequest(LlmRequest.Builder llmRequestBuilder,
 ToolContext toolContext)

Description copied from class: BaseTool

Processes the outgoing LlmRequest.Builder.

This implementation adds the current tool's BaseTool.declaration() to the GenerateContentConfig within the builder. If a tool with function declarations already exists, the current tool's declaration is merged into it. Otherwise, a new tool definition with the current tool's declaration is created. The current tool itself is also added to the builder's internal list of tools. Override this method for processing the outgoing request.

Overrides:

processLlmRequest in class BaseTool