Skip to content

Commit 1a00cec

Browse files
Google APIscopybara-github
authored andcommitted
feat: Add asyncQueryReasoningEngine to aiplatform v1 API
PiperOrigin-RevId: 900907134
1 parent 15cb2f9 commit 1a00cec

1 file changed

Lines changed: 54 additions & 1 deletion

File tree

‎google/cloud/aiplatform/v1/reasoning_engine_execution_service.proto‎

Lines changed: 54 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
// Copyright 2025 Google LLC
1+
// Copyright 2026 Google LLC
22
//
33
// Licensed under the Apache License, Version 2.0 (the "License");
44
// you may not use this file except in compliance with the License.
@@ -21,6 +21,8 @@ import "google/api/client.proto";
2121
import "google/api/field_behavior.proto";
2222
import "google/api/httpbody.proto";
2323
import "google/api/resource.proto";
24+
import "google/cloud/aiplatform/v1/operation.proto";
25+
import "google/longrunning/operations.proto";
2426
import "google/protobuf/struct.proto";
2527

2628
option csharp_namespace = "Google.Cloud.AIPlatform.V1";
@@ -54,6 +56,23 @@ service ReasoningEngineExecutionService {
5456
body: "*"
5557
};
5658
}
59+
60+
// Async query using a reasoning engine.
61+
rpc AsyncQueryReasoningEngine(AsyncQueryReasoningEngineRequest)
62+
returns (google.longrunning.Operation) {
63+
option (google.api.http) = {
64+
post: "/v1/{name=projects/*/locations/*/reasoningEngines/*}:asyncQuery"
65+
body: "*"
66+
additional_bindings {
67+
post: "/v1/{name=reasoningEngines/*}:asyncQuery"
68+
body: "*"
69+
}
70+
};
71+
option (google.longrunning.operation_info) = {
72+
response_type: "AsyncQueryReasoningEngineResponse"
73+
metadata_type: "AsyncQueryReasoningEngineOperationMetadata"
74+
};
75+
}
5776
}
5877

5978
// Request message for [ReasoningEngineExecutionService.Query][].
@@ -103,3 +122,37 @@ message StreamQueryReasoningEngineRequest {
103122
// It is optional and defaults to "stream_query" if unspecified.
104123
string class_method = 3 [(google.api.field_behavior) = OPTIONAL];
105124
}
125+
126+
// Request message for
127+
// [ReasoningEngineExecutionService.AsyncQueryReasoningEngine][google.cloud.aiplatform.v1.ReasoningEngineExecutionService.AsyncQueryReasoningEngine].
128+
message AsyncQueryReasoningEngineRequest {
129+
// Required. The name of the ReasoningEngine resource to use.
130+
// Format:
131+
// `projects/{project}/locations/{location}/reasoningEngines/{reasoning_engine}`
132+
string name = 1 [
133+
(google.api.field_behavior) = REQUIRED,
134+
(google.api.resource_reference) = {
135+
type: "aiplatform.googleapis.com/ReasoningEngine"
136+
}
137+
];
138+
139+
// Optional. Input Cloud Storage URI for the Async query.
140+
string input_gcs_uri = 2 [(google.api.field_behavior) = OPTIONAL];
141+
142+
// Optional. Output Cloud Storage URI for the Async query.
143+
string output_gcs_uri = 3 [(google.api.field_behavior) = OPTIONAL];
144+
}
145+
146+
// Operation metadata message for
147+
// [ReasoningEngineExecutionService.AsyncQueryReasoningEngine][google.cloud.aiplatform.v1.ReasoningEngineExecutionService.AsyncQueryReasoningEngine].
148+
message AsyncQueryReasoningEngineOperationMetadata {
149+
// The common part of the operation metadata.
150+
GenericOperationMetadata generic_metadata = 1;
151+
}
152+
153+
// Response message for
154+
// [ReasoningEngineExecutionService.AsyncQueryReasoningEngine][google.cloud.aiplatform.v1.ReasoningEngineExecutionService.AsyncQueryReasoningEngine].
155+
message AsyncQueryReasoningEngineResponse {
156+
// Output Cloud Storage URI for the Async query.
157+
string output_gcs_uri = 1;
158+
}

0 commit comments

Comments
 (0)