feat: Adding a new TextType PROGRESS to provide informational messages about an agent's progress for supporting more granular Agent RAG tools

Google APIs · copybara-github · commit 9fcfbea0aa5b · 2025-11-06T08:24:37.000-08:00
feat: Adding struct_schema to Datasource to support flexible schemas, particularly for Looker datasources
feat: Adding client_managed_resource_context to allow clients to manage their own conversation and agent resources

PiperOrigin-RevId: 828965493
diff --git a/google/cloud/geminidataanalytics/v1alpha/BUILD.bazel b/google/cloud/geminidataanalytics/v1alpha/BUILD.bazel
@@ -46,6 +46,7 @@ proto_library(
         "@com_google_protobuf//:field_mask_proto",
         "@com_google_protobuf//:struct_proto",
         "@com_google_protobuf//:timestamp_proto",
+        "@com_google_protobuf//:wrappers_proto",
     ],
 )
 
diff --git a/google/cloud/geminidataanalytics/v1alpha/context.proto b/google/cloud/geminidataanalytics/v1alpha/context.proto
@@ -18,6 +18,7 @@ package google.cloud.geminidataanalytics.v1alpha;
 
 import "google/api/field_behavior.proto";
 import "google/cloud/geminidataanalytics/v1alpha/datasource.proto";
+import "google/protobuf/wrappers.proto";
 
 option csharp_namespace = "Google.Cloud.GeminiDataAnalytics.V1Alpha";
 option go_package = "cloud.google.com/go/geminidataanalytics/apiv1alpha/geminidataanalyticspb;geminidataanalyticspb";
@@ -151,6 +152,19 @@ message ConversationOptions {
 
   // Optional. Options for analysis.
   AnalysisOptions analysis = 2 [(google.api.field_behavior) = OPTIONAL];
+
+  // Optional. Options for datasources.
+  DatasourceOptions datasource = 3 [(google.api.field_behavior) = OPTIONAL];
+}
+
+// Options for datasources configurations.
+message DatasourceOptions {
+  // Optional. This option applies to datasources that require BigQuery queries
+  // only. Limits the bytes billed for each BQ query job. Queries that will have
+  // bytes billed beyond this limit will fail (without incurring a charge).
+  // If unspecified, no limit will be applied.
+  google.protobuf.Int64Value big_query_max_billed_bytes = 1
+      [(google.api.field_behavior) = OPTIONAL];
 }
 
 // Options for chart generation.
diff --git a/google/cloud/geminidataanalytics/v1alpha/data_chat_service.proto b/google/cloud/geminidataanalytics/v1alpha/data_chat_service.proto
@@ -169,6 +169,13 @@ message ChatRequest {
     // statelessly, without managed conversation persistence.
     DataAgentContext data_agent_context = 104
         [(google.api.field_behavior) = OPTIONAL];
+
+    // Optional. Context with client managed resources.
+    // Some clients may not use GDA managed resources including
+    // conversations and agents, instead they create and manage their own
+    // conversations and agents resources.
+    ClientManagedResourceContext client_managed_resource_context = 105
+        [(google.api.field_behavior) = OPTIONAL];
   }
 
   // Optional. The Google Cloud project to be used for quota and billing.
@@ -243,6 +250,22 @@ message ConversationReference {
       [(google.api.field_behavior) = REQUIRED];
 }
 
+// Context with client managed resources.
+// Some clients may not use GDA managed resources including
+// conversations and agents, instead they create and manage their own
+// conversations and agents resources.
+message ClientManagedResourceContext {
+  // Required. Context for the chat request. Use this to chat without GDA API
+  // managed conversation and agent persistence by passing all context inline.
+  Context inline_context = 1 [(google.api.field_behavior) = REQUIRED];
+
+  // Optional. The client managed conversation id.
+  string conversation_id = 2 [(google.api.field_behavior) = OPTIONAL];
+
+  // Optional. The client managed agent id.
+  string agent_id = 3 [(google.api.field_behavior) = OPTIONAL];
+}
+
 // A message from an interaction between the user and the system.
 message Message {
   // The kind of message.
@@ -320,6 +343,13 @@ message TextMessage {
 
     // The text is a thinking plan generated by the thinking tool.
     THOUGHT = 2;
+
+    // The text is an informational message about the agent's progress, such as
+    // a tool being invoked. This is distinct from the agent's internal thought
+    // process (`THOUGHT`) and the final answer to the user
+    // (`FINAL_RESPONSE`). These messages provide insight into the agent's
+    // actions.
+    PROGRESS = 3;
   }
 
   // Optional. The parts of the message.
@@ -579,7 +609,20 @@ message ChartResult {
   Blob image = 3 [(google.api.field_behavior) = OPTIONAL];
 }
 
-// An error message.
+// An error message from a tool call.
+// This message is used to represent an error that occurred while an agent was
+// trying to use a tool. It's important to note that not all errors are
+// terminal. Many are recoverable, and the agent may use the information from
+// this error message to self-correct and retry the tool call or try a
+// different approach.
+//
+// For example, if a data query fails, the agent might receive an
+// `ErrorMessage`, analyze it, and then generate a corrected query.
+//
+// Clients should be cautious about interpreting this message as a definitive
+// failure. It can be part of the agent's normal, iterative process of
+// completing a task. Surfacing these errors directly to end-users without
+// context (e.g., as a "hard failure") may be misleading.
 message ErrorMessage {
   // Output only. The text of the error.
   string text = 1 [(google.api.field_behavior) = OUTPUT_ONLY];
diff --git a/google/cloud/geminidataanalytics/v1alpha/datasource.proto b/google/cloud/geminidataanalytics/v1alpha/datasource.proto
@@ -18,6 +18,7 @@ package google.cloud.geminidataanalytics.v1alpha;
 
 import "google/api/field_behavior.proto";
 import "google/cloud/geminidataanalytics/v1alpha/credentials.proto";
+import "google/protobuf/struct.proto";
 
 option csharp_namespace = "Google.Cloud.GeminiDataAnalytics.V1Alpha";
 option go_package = "cloud.google.com/go/geminidataanalytics/apiv1alpha/geminidataanalyticspb;geminidataanalyticspb";
@@ -155,6 +156,15 @@ message Datasource {
 
   // Optional. The schema of the datasource.
   Schema schema = 7 [(google.api.field_behavior) = OPTIONAL];
+
+  // Optional. A struct representation of the schema.
+  // This is populated for datasources with schemas that cannot be
+  // fully represented by the strongly-typed `schema` field.
+  //
+  // For Looker datasources, this maps to the LookmlModelExplore type:
+  // https://cloud.google.com/looker/docs/reference/looker-api/latest/types/LookmlModelExplore
+  google.protobuf.Struct struct_schema = 10
+      [(google.api.field_behavior) = OPTIONAL];
 }
 
 // The schema of a Datasource or QueryResult instance.

Original file line number	Diff line number	Diff line change
`@@ -46,6 +46,7 @@ proto_library(`
`46`	`46`	`"@com_google_protobuf//:field_mask_proto",`
`47`	`47`	`"@com_google_protobuf//:struct_proto",`
`48`	`48`	`"@com_google_protobuf//:timestamp_proto",`
	`49`	`+ "@com_google_protobuf//:wrappers_proto",`
`49`	`50`	`],`
`50`	`51`	`)`
`51`	`52`