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
2 changes: 2 additions & 0 deletions src/gaos/models/interactions/index.ts
Original file line number Diff line number Diff line change
Expand Up @@ -111,7 +111,9 @@ export * from "./response-format.js";
export * from "./response-modality.js";
export * from "./retrieval-call-arguments.js";
export * from "./retrieval-call-delta.js";
export * from "./retrieval-call-step.js";
export * from "./retrieval-result-delta.js";
export * from "./retrieval-result-step.js";
export * from "./retrieval.js";
export * from "./review-snippet.js";
export * from "./safety-setting.js";
Expand Down
50 changes: 50 additions & 0 deletions src/gaos/models/interactions/retrieval-call-step.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,50 @@
/**
* @license
* Copyright 2026 Google LLC
* SPDX-License-Identifier: Apache-2.0
*
* g3-prettier-ignore-file
*/

/*
* Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
*/

import { RetrievalCallArguments } from "./retrieval-call-arguments.js";

/**
* The type of retrieval tools.
*/
export type RetrievalCallStepRetrievalType =
| "vertex_ai_search"
| "rag_store"
| "exa_ai_search"
| "parallel_ai_search"
| (string & {});

/**
* Retrieval call step.
*
* @remarks
* Used by Vertex Retrieval tools such as Parallel AI, Exa AI, Vertex AI Search,
* etc. RetrievalType decides which tool is used.
*/
export type RetrievalCallStep = {
/**
* The arguments to pass to Retrieval tools.
*/
arguments: RetrievalCallArguments;
/**
* Required. A unique ID for this specific tool call.
*/
id: string;
/**
* The type of retrieval tools.
*/
retrieval_type?: RetrievalCallStepRetrievalType | undefined;
/**
* A signature hash for backend validation.
*/
signature?: string | undefined;
type: "retrieval_call";
};
34 changes: 34 additions & 0 deletions src/gaos/models/interactions/retrieval-result-step.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,34 @@
/**
* @license
* Copyright 2026 Google LLC
* SPDX-License-Identifier: Apache-2.0
*
* g3-prettier-ignore-file
*/

/*
* Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
*/

/**
* Vertex Retrieval result step.
*
* @remarks
* Used by Vertex Retrieval tools such as Parallel AI, Exa AI, Vertex AI Search,
* etc.
*/
export type RetrievalResultStep = {
/**
* Required. ID to match the ID from the function call block.
*/
call_id: string;
/**
* Whether the retrieval resulted in an error.
*/
is_error?: boolean | undefined;
/**
* A signature hash for backend validation.
*/
signature?: string | undefined;
type: "retrieval_result";
};
4 changes: 4 additions & 0 deletions src/gaos/models/interactions/step.ts
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,8 @@ import { MCPServerToolResultStep } from "./mcp-server-tool-result-step.js";
import { ModelOutputStep } from "./model-output-step.js";
import { ProcessingCallStep } from "./processing-call-step.js";
import { ProcessingResultStep } from "./processing-result-step.js";
import { RetrievalCallStep } from "./retrieval-call-step.js";
import { RetrievalResultStep } from "./retrieval-result-step.js";
import { ThoughtStep } from "./thought-step.js";
import { URLContextCallStep } from "./url-context-call-step.js";
import { URLContextResultStep } from "./url-context-result-step.js";
Expand All @@ -49,6 +51,8 @@ export type Step =
| ModelOutputStep
| ProcessingCallStep
| ProcessingResultStep
| RetrievalCallStep
| RetrievalResultStep
| ThoughtStep
| URLContextCallStep
| URLContextResultStep
Expand Down
16 changes: 14 additions & 2 deletions src/gaos/resources.ts
Original file line number Diff line number Diff line change
Expand Up @@ -100,7 +100,9 @@ import type { ProcessingResultDelta as ProcessingResultDelta$Import } from "./mo
import type { ProcessingResultStep as ProcessingResultStep$Import } from "./models/interactions/processing-result-step.js";
import type { RetrievalCallArguments as RetrievalCallArguments$Import } from "./models/interactions/retrieval-call-arguments.js";
import type { RetrievalCallDelta as RetrievalCallDelta$Import } from "./models/interactions/retrieval-call-delta.js";
import type { RetrievalCallStep as RetrievalCallStep$Import } from "./models/interactions/retrieval-call-step.js";
import type { RetrievalResultDelta as RetrievalResultDelta$Import } from "./models/interactions/retrieval-result-delta.js";
import type { RetrievalResultStep as RetrievalResultStep$Import } from "./models/interactions/retrieval-result-step.js";
import type { Retrieval as Retrieval$Import } from "./models/interactions/retrieval.js";
import type { ReviewSnippet as ReviewSnippet$Import } from "./models/interactions/review-snippet.js";
import type { SafetySetting as SafetySetting$Import } from "./models/interactions/safety-setting.js";
Expand Down Expand Up @@ -265,7 +267,9 @@ type ProcessingCallStep$ = ProcessingCallStep$Import;
type ProcessingResultStep$ = ProcessingResultStep$Import;
type RetrievalCallArguments$ = RetrievalCallArguments$Import;
type RetrievalCallDelta$ = RetrievalCallDelta$Import;
type RetrievalCallStep$ = RetrievalCallStep$Import;
type RetrievalResultDelta$ = RetrievalResultDelta$Import;
type RetrievalResultStep$ = RetrievalResultStep$Import;
type SafetySetting$ = SafetySetting$Import;
type SpeechConfig$ = SpeechConfig$Import;
type Step$ = Step$Import;
Expand Down Expand Up @@ -312,6 +316,7 @@ type Arguments$3 = GoogleSearchCallArguments$Import;
type Result$2 = GoogleSearchResult$Import;
type ReviewSnippet$3 = ReviewSnippet$Import;
type Arguments$4 = RetrievalCallArguments$Import;
type Arguments$5 = RetrievalCallArguments$Import;
type ArgumentsDelta$ = ArgumentsDelta$Import;
type Audio$ = AudioDelta$Import;
type CodeExecutionCall$ = CodeExecutionCallDelta$Import;
Expand Down Expand Up @@ -348,7 +353,7 @@ type URLContext$ = URLContext$Import;
type VertexAISearchConfig$ = VertexAISearchConfig$Import;
type Smart$ = SmartTranscriptionMode$Import;
type Verbatim$ = VerbatimTranscriptionMode$Import;
type Arguments$5 = URLContextCallArguments$Import;
type Arguments$6 = URLContextCallArguments$Import;
type Result$3 = URLContextResult$Import;
type CachedTokensByModality$ = ModalityTokens$Import;
type GroundingToolCount$ = GroundingToolCount$Import;
Expand Down Expand Up @@ -503,7 +508,9 @@ export type Result = Result$;
export type Retrieval = Retrieval$;
export type RetrievalCallArguments = RetrievalCallArguments$;
export type RetrievalCallDelta = RetrievalCallDelta$;
export type RetrievalCallStep = RetrievalCallStep$;
export type RetrievalResultDelta = RetrievalResultDelta$;
export type RetrievalResultStep = RetrievalResultStep$;
export type ReviewSnippet = ReviewSnippet$;
export type RotateSigningSecretRequest = RotateSigningSecretRequest$;
export type SafetySetting = SafetySetting$;
Expand Down Expand Up @@ -668,7 +675,9 @@ export declare namespace Interactions {
export type ProcessingResultStep = ProcessingResultStep$;
export type RetrievalCallArguments = RetrievalCallArguments$;
export interface RetrievalCallDelta extends RetrievalCallDelta$ {}
export interface RetrievalCallStep extends RetrievalCallStep$ {}
export type RetrievalResultDelta = RetrievalResultDelta$;
export type RetrievalResultStep = RetrievalResultStep$;
export type SafetySetting = SafetySetting$;
export type SpeechConfig = SpeechConfig$;
export type Step = Step$;
Expand Down Expand Up @@ -752,6 +761,9 @@ export declare namespace Interactions {
export namespace RetrievalCallDelta {
export type Arguments = Arguments$4;
}
export namespace RetrievalCallStep {
export type Arguments = Arguments$5;
}
export namespace StepDelta {
export type ArgumentsDelta = ArgumentsDelta$;
export type Audio = Audio$;
Expand Down Expand Up @@ -797,7 +809,7 @@ export declare namespace Interactions {
export type Verbatim = Verbatim$;
}
export namespace URLContextCallStep {
export type Arguments = Arguments$5;
export type Arguments = Arguments$6;
}
export namespace URLContextResultStep {
export type Result = Result$3;
Expand Down
Loading