Skip to content

Commit a78b583

Browse files
Google APIscopybara-github
authored andcommitted
feat: Add asyncQueryReasoningEngine to aiplatform v1beta1 API
PiperOrigin-RevId: 902815104
1 parent 4ad1b67 commit a78b583

1 file changed

Lines changed: 54 additions & 1 deletion

File tree

‎google/cloud/aiplatform/v1beta1/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/v1beta1/operation.proto";
25+
import "google/longrunning/operations.proto";
2426
import "google/protobuf/struct.proto";
2527

2628
option csharp_namespace = "Google.Cloud.AIPlatform.V1Beta1";
@@ -62,6 +64,23 @@ service ReasoningEngineExecutionService {
6264
}
6365
};
6466
}
67+
68+
// Async query using a reasoning engine.
69+
rpc AsyncQueryReasoningEngine(AsyncQueryReasoningEngineRequest)
70+
returns (google.longrunning.Operation) {
71+
option (google.api.http) = {
72+
post: "/v1beta1/{name=projects/*/locations/*/reasoningEngines/*}:asyncQuery"
73+
body: "*"
74+
additional_bindings {
75+
post: "/v1beta1/{name=reasoningEngines/*}:asyncQuery"
76+
body: "*"
77+
}
78+
};
79+
option (google.longrunning.operation_info) = {
80+
response_type: "AsyncQueryReasoningEngineResponse"
81+
metadata_type: "AsyncQueryReasoningEngineOperationMetadata"
82+
};
83+
}
6584
}
6685

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

0 commit comments

Comments
 (0)