Skip to content

Commit 372858b

Browse files
matthew29tangcopybara-github
authored andcommitted
feat: Make AgentEngine module methods public for prototyping
PiperOrigin-RevId: 889897915
1 parent cf692e3 commit 372858b

3 files changed

Lines changed: 13 additions & 12 deletions

File tree

google-cloud-vertexai/src/main/java/com/google/cloud/vertexai/genai/AgentEngines.java

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -499,7 +499,7 @@ AgentEngineOperation processResponseForPrivateCreate(
499499
return JsonSerializable.fromJsonNode(responseNode, AgentEngineOperation.class);
500500
}
501501

502-
AgentEngineOperation privateCreate(CreateAgentEngineConfig config) {
502+
public AgentEngineOperation privateCreate(CreateAgentEngineConfig config) {
503503
BuiltRequest builtRequest = buildRequestForPrivateCreate(config);
504504

505505
try (ApiResponse response =
@@ -574,7 +574,7 @@ DeleteAgentEngineOperation processResponseForPrivateDelete(
574574
return JsonSerializable.fromJsonNode(responseNode, DeleteAgentEngineOperation.class);
575575
}
576576

577-
DeleteAgentEngineOperation privateDelete(
577+
public DeleteAgentEngineOperation privateDelete(
578578
String name, boolean force, DeleteAgentEngineConfig config) {
579579
BuiltRequest builtRequest = buildRequestForPrivateDelete(name, force, config);
580580

@@ -645,7 +645,7 @@ ReasoningEngine processResponseForPrivateGet(ApiResponse response, GetAgentEngin
645645
return JsonSerializable.fromJsonNode(responseNode, ReasoningEngine.class);
646646
}
647647

648-
ReasoningEngine privateGet(String name, GetAgentEngineConfig config) {
648+
public ReasoningEngine privateGet(String name, GetAgentEngineConfig config) {
649649
BuiltRequest builtRequest = buildRequestForPrivateGet(name, config);
650650

651651
try (ApiResponse response =
@@ -713,7 +713,7 @@ ListReasoningEnginesResponse processResponseForPrivateList(
713713
return JsonSerializable.fromJsonNode(responseNode, ListReasoningEnginesResponse.class);
714714
}
715715

716-
ListReasoningEnginesResponse privateList(ListAgentEngineConfig config) {
716+
public ListReasoningEnginesResponse privateList(ListAgentEngineConfig config) {
717717
BuiltRequest builtRequest = buildRequestForPrivateList(config);
718718

719719
try (ApiResponse response =
@@ -785,7 +785,7 @@ AgentEngineOperation processResponseForPrivateGetAgentOperation(
785785
return JsonSerializable.fromJsonNode(responseNode, AgentEngineOperation.class);
786786
}
787787

788-
AgentEngineOperation privateGetAgentOperation(
788+
public AgentEngineOperation privateGetAgentOperation(
789789
String operationName, GetAgentEngineOperationConfig config) {
790790
BuiltRequest builtRequest = buildRequestForPrivateGetAgentOperation(operationName, config);
791791

@@ -857,7 +857,7 @@ QueryReasoningEngineResponse processResponseForPrivateQuery(
857857
return JsonSerializable.fromJsonNode(responseNode, QueryReasoningEngineResponse.class);
858858
}
859859

860-
QueryReasoningEngineResponse privateQuery(String name, QueryAgentEngineConfig config) {
860+
public QueryReasoningEngineResponse privateQuery(String name, QueryAgentEngineConfig config) {
861861
BuiltRequest builtRequest = buildRequestForPrivateQuery(name, config);
862862

863863
try (ApiResponse response =
@@ -928,7 +928,7 @@ AgentEngineOperation processResponseForPrivateUpdate(
928928
return JsonSerializable.fromJsonNode(responseNode, AgentEngineOperation.class);
929929
}
930930

931-
AgentEngineOperation privateUpdate(String name, UpdateAgentEngineConfig config) {
931+
public AgentEngineOperation privateUpdate(String name, UpdateAgentEngineConfig config) {
932932
BuiltRequest builtRequest = buildRequestForPrivateUpdate(name, config);
933933

934934
try (ApiResponse response =

google-cloud-vertexai/src/main/java/com/google/cloud/vertexai/genai/SessionEvents.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -342,7 +342,7 @@ ListAgentEngineSessionEventsResponse processResponseForPrivateList(
342342
return JsonSerializable.fromJsonNode(responseNode, ListAgentEngineSessionEventsResponse.class);
343343
}
344344

345-
ListAgentEngineSessionEventsResponse privateList(
345+
public ListAgentEngineSessionEventsResponse privateList(
346346
String name, ListAgentEngineSessionEventsConfig config) {
347347
BuiltRequest builtRequest = buildRequestForPrivateList(name, config);
348348

google-cloud-vertexai/src/main/java/com/google/cloud/vertexai/genai/Sessions.java

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -363,7 +363,7 @@ AgentEngineSessionOperation processResponseForPrivateCreate(
363363
return JsonSerializable.fromJsonNode(responseNode, AgentEngineSessionOperation.class);
364364
}
365365

366-
AgentEngineSessionOperation privateCreate(
366+
public AgentEngineSessionOperation privateCreate(
367367
String name, String userId, CreateAgentEngineSessionConfig config) {
368368
BuiltRequest builtRequest = buildRequestForPrivateCreate(name, userId, config);
369369

@@ -577,7 +577,7 @@ ListReasoningEnginesSessionsResponse processResponseForPrivateList(
577577
return JsonSerializable.fromJsonNode(responseNode, ListReasoningEnginesSessionsResponse.class);
578578
}
579579

580-
ListReasoningEnginesSessionsResponse privateList(
580+
public ListReasoningEnginesSessionsResponse privateList(
581581
String name, ListAgentEngineSessionsConfig config) {
582582
BuiltRequest builtRequest = buildRequestForPrivateList(name, config);
583583

@@ -650,7 +650,7 @@ AgentEngineSessionOperation processResponseForPrivateGetSessionOperation(
650650
return JsonSerializable.fromJsonNode(responseNode, AgentEngineSessionOperation.class);
651651
}
652652

653-
AgentEngineSessionOperation privateGetSessionOperation(
653+
public AgentEngineSessionOperation privateGetSessionOperation(
654654
String operationName, GetAgentEngineOperationConfig config) {
655655
BuiltRequest builtRequest = buildRequestForPrivateGetSessionOperation(operationName, config);
656656

@@ -722,7 +722,8 @@ AgentEngineSessionOperation processResponseForPrivateUpdate(
722722
return JsonSerializable.fromJsonNode(responseNode, AgentEngineSessionOperation.class);
723723
}
724724

725-
AgentEngineSessionOperation privateUpdate(String name, UpdateAgentEngineSessionConfig config) {
725+
public AgentEngineSessionOperation privateUpdate(
726+
String name, UpdateAgentEngineSessionConfig config) {
726727
BuiltRequest builtRequest = buildRequestForPrivateUpdate(name, config);
727728

728729
try (ApiResponse response =

0 commit comments

Comments
 (0)