google-cloud-gemini_data_analytics-v1beta 0.4.0 → 0.5.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0f4bb4c6c74223477e145231e4ce970757e30ab994371107a08a5e3a7375c04b
-  data.tar.gz: 454a60a2b55e7263f57c16c7a39c0c86aac4d5ac0f3cfab5706f100c7a502eb5
+  metadata.gz: a8ed5529cd7f778c606e6939e0c4c1bf2acd654d403136ac66bf9d44639cb58a
+  data.tar.gz: c5edb60e80d42bef069610c5d4263f987b725bba0bbc1af451f69847e711f52f
 SHA512:
-  metadata.gz: cc03e6b1efaf2d0986c1299d259c2d2e778785cafc51848542482e3dabe9c54e2aadc02709e8d058182d5a208f6065385a58579e174f97c3f3e6123cf0e6dfb0
-  data.tar.gz: 891c283185713485e61a155186482b66b9fe7db729c09a8841474826c531806509609cfdc141055464b42109238e67506e32c1a60b9596d2f37b0112f7e205a0
+  metadata.gz: '09b2cfa220c77557b0b6bd80689a74a5ce84a065a5c92309195506ba3b685f116c8198229b2846db17d283fdcd467d1d7536a3d2f2b8556b02fddc2d6917d317'
+  data.tar.gz: 6c3fe238e1a026e9bf6f836d8e4d120f06c25d943bef78cef9e0874bca8dc788ecef9f5a4a0acff7c40df1f464e504ae58fcbb5ce7e89dd702c2b5498269d9ba

data/lib/google/cloud/gemini_data_analytics/v1beta/data_chat_service/client.rb

@@ -805,6 +805,100 @@ module Google
               raise ::Google::Cloud::Error.from_error(e)
             end
 
+            ##
+            # Queries data from a natural language user query.
+            #
+            # @overload query_data(request, options = nil)
+            #   Pass arguments to `query_data` via a request object, either of type
+            #   {::Google::Cloud::GeminiDataAnalytics::V1beta::QueryDataRequest} or an equivalent Hash.
+            #
+            #   @param request [::Google::Cloud::GeminiDataAnalytics::V1beta::QueryDataRequest, ::Hash]
+            #     A request object representing the call parameters. Required. To specify no
+            #     parameters, or to keep all the default parameter values, pass an empty Hash.
+            #   @param options [::Gapic::CallOptions, ::Hash]
+            #     Overrides the default settings for this call, e.g, timeout, retries, etc. Optional.
+            #
+            # @overload query_data(parent: nil, prompt: nil, context: nil, generation_options: nil)
+            #   Pass arguments to `query_data` via keyword arguments. Note that at
+            #   least one keyword argument is required. To specify no parameters, or to keep all
+            #   the default parameter values, pass an empty Hash as a request object (see above).
+            #
+            #   @param parent [::String]
+            #     Required. The parent resource to generate the query for.
+            #     Format: projects/\\{project}/locations/\\{location}
+            #   @param prompt [::String]
+            #     Required. The natural language query for which to generate query.
+            #     Example: "What are the top 5 best selling products this month?"
+            #   @param context [::Google::Cloud::GeminiDataAnalytics::V1beta::QueryDataContext, ::Hash]
+            #     Required. The context for the data query, including the data sources to
+            #     use.
+            #   @param generation_options [::Google::Cloud::GeminiDataAnalytics::V1beta::GenerationOptions, ::Hash]
+            #     Optional. Options to control query generation and execution behavior.
+            #
+            # @yield [response, operation] Access the result along with the RPC operation
+            # @yieldparam response [::Google::Cloud::GeminiDataAnalytics::V1beta::QueryDataResponse]
+            # @yieldparam operation [::GRPC::ActiveCall::Operation]
+            #
+            # @return [::Google::Cloud::GeminiDataAnalytics::V1beta::QueryDataResponse]
+            #
+            # @raise [::Google::Cloud::Error] if the RPC is aborted.
+            #
+            # @example Basic example
+            #   require "google/cloud/gemini_data_analytics/v1beta"
+            #
+            #   # Create a client object. The client can be reused for multiple calls.
+            #   client = Google::Cloud::GeminiDataAnalytics::V1beta::DataChatService::Client.new
+            #
+            #   # Create a request. To set request fields, pass in keyword arguments.
+            #   request = Google::Cloud::GeminiDataAnalytics::V1beta::QueryDataRequest.new
+            #
+            #   # Call the query_data method.
+            #   result = client.query_data request
+            #
+            #   # The returned object is of type Google::Cloud::GeminiDataAnalytics::V1beta::QueryDataResponse.
+            #   p result
+            #
+            def query_data request, options = nil
+              raise ::ArgumentError, "request must be provided" if request.nil?
+
+              request = ::Gapic::Protobuf.coerce request, to: ::Google::Cloud::GeminiDataAnalytics::V1beta::QueryDataRequest
+
+              # Converts hash and nil to an options object
+              options = ::Gapic::CallOptions.new(**options.to_h) if options.respond_to? :to_h
+
+              # Customize the options with defaults
+              metadata = @config.rpcs.query_data.metadata.to_h
+
+              # Set x-goog-api-client, x-goog-user-project and x-goog-api-version headers
+              metadata[:"x-goog-api-client"] ||= ::Gapic::Headers.x_goog_api_client \
+                lib_name: @config.lib_name, lib_version: @config.lib_version,
+                gapic_version: ::Google::Cloud::GeminiDataAnalytics::V1beta::VERSION
+              metadata[:"x-goog-api-version"] = API_VERSION unless API_VERSION.empty?
+              metadata[:"x-goog-user-project"] = @quota_project_id if @quota_project_id
+
+              header_params = {}
+              if request.parent
+                header_params["parent"] = request.parent
+              end
+
+              request_params_header = header_params.map { |k, v| "#{k}=#{v}" }.join("&")
+              metadata[:"x-goog-request-params"] ||= request_params_header
+
+              options.apply_defaults timeout:      @config.rpcs.query_data.timeout,
+                                     metadata:     metadata,
+                                     retry_policy: @config.rpcs.query_data.retry_policy
+
+              options.apply_defaults timeout:      @config.timeout,
+                                     metadata:     @config.metadata,
+                                     retry_policy: @config.retry_policy
+
+              @data_chat_service_stub.call_rpc :query_data, request, options: options do |response, operation|
+                yield response, operation if block_given?
+              end
+            rescue ::GRPC::BadStatus => e
+              raise ::Google::Cloud::Error.from_error(e)
+            end
+
             ##
             # Configuration class for the DataChatService API.
             #
@@ -1018,6 +1112,11 @@ module Google
                 # @return [::Gapic::Config::Method]
                 #
                 attr_reader :list_messages
+                ##
+                # RPC-specific configuration for `query_data`
+                # @return [::Gapic::Config::Method]
+                #
+                attr_reader :query_data
 
                 # @private
                 def initialize parent_rpcs = nil
@@ -1033,6 +1132,8 @@ module Google
                   @list_conversations = ::Gapic::Config::Method.new list_conversations_config
                   list_messages_config = parent_rpcs.list_messages if parent_rpcs.respond_to? :list_messages
                   @list_messages = ::Gapic::Config::Method.new list_messages_config
+                  query_data_config = parent_rpcs.query_data if parent_rpcs.respond_to? :query_data
+                  @query_data = ::Gapic::Config::Method.new query_data_config
 
                   yield self if block_given?
                 end

data/lib/google/cloud/gemini_data_analytics/v1beta/data_chat_service/rest/client.rb

@@ -759,6 +759,93 @@ module Google
                 raise ::Google::Cloud::Error.from_error(e)
               end
 
+              ##
+              # Queries data from a natural language user query.
+              #
+              # @overload query_data(request, options = nil)
+              #   Pass arguments to `query_data` via a request object, either of type
+              #   {::Google::Cloud::GeminiDataAnalytics::V1beta::QueryDataRequest} or an equivalent Hash.
+              #
+              #   @param request [::Google::Cloud::GeminiDataAnalytics::V1beta::QueryDataRequest, ::Hash]
+              #     A request object representing the call parameters. Required. To specify no
+              #     parameters, or to keep all the default parameter values, pass an empty Hash.
+              #   @param options [::Gapic::CallOptions, ::Hash]
+              #     Overrides the default settings for this call, e.g, timeout, retries etc. Optional.
+              #
+              # @overload query_data(parent: nil, prompt: nil, context: nil, generation_options: nil)
+              #   Pass arguments to `query_data` via keyword arguments. Note that at
+              #   least one keyword argument is required. To specify no parameters, or to keep all
+              #   the default parameter values, pass an empty Hash as a request object (see above).
+              #
+              #   @param parent [::String]
+              #     Required. The parent resource to generate the query for.
+              #     Format: projects/\\{project}/locations/\\{location}
+              #   @param prompt [::String]
+              #     Required. The natural language query for which to generate query.
+              #     Example: "What are the top 5 best selling products this month?"
+              #   @param context [::Google::Cloud::GeminiDataAnalytics::V1beta::QueryDataContext, ::Hash]
+              #     Required. The context for the data query, including the data sources to
+              #     use.
+              #   @param generation_options [::Google::Cloud::GeminiDataAnalytics::V1beta::GenerationOptions, ::Hash]
+              #     Optional. Options to control query generation and execution behavior.
+              # @yield [result, operation] Access the result along with the TransportOperation object
+              # @yieldparam result [::Google::Cloud::GeminiDataAnalytics::V1beta::QueryDataResponse]
+              # @yieldparam operation [::Gapic::Rest::TransportOperation]
+              #
+              # @return [::Google::Cloud::GeminiDataAnalytics::V1beta::QueryDataResponse]
+              #
+              # @raise [::Google::Cloud::Error] if the REST call is aborted.
+              #
+              # @example Basic example
+              #   require "google/cloud/gemini_data_analytics/v1beta"
+              #
+              #   # Create a client object. The client can be reused for multiple calls.
+              #   client = Google::Cloud::GeminiDataAnalytics::V1beta::DataChatService::Rest::Client.new
+              #
+              #   # Create a request. To set request fields, pass in keyword arguments.
+              #   request = Google::Cloud::GeminiDataAnalytics::V1beta::QueryDataRequest.new
+              #
+              #   # Call the query_data method.
+              #   result = client.query_data request
+              #
+              #   # The returned object is of type Google::Cloud::GeminiDataAnalytics::V1beta::QueryDataResponse.
+              #   p result
+              #
+              def query_data request, options = nil
+                raise ::ArgumentError, "request must be provided" if request.nil?
+
+                request = ::Gapic::Protobuf.coerce request, to: ::Google::Cloud::GeminiDataAnalytics::V1beta::QueryDataRequest
+
+                # Converts hash and nil to an options object
+                options = ::Gapic::CallOptions.new(**options.to_h) if options.respond_to? :to_h
+
+                # Customize the options with defaults
+                call_metadata = @config.rpcs.query_data.metadata.to_h
+
+                # Set x-goog-api-client, x-goog-user-project and x-goog-api-version headers
+                call_metadata[:"x-goog-api-client"] ||= ::Gapic::Headers.x_goog_api_client \
+                  lib_name: @config.lib_name, lib_version: @config.lib_version,
+                  gapic_version: ::Google::Cloud::GeminiDataAnalytics::V1beta::VERSION,
+                  transports_version_send: [:rest]
+
+                call_metadata[:"x-goog-api-version"] = API_VERSION unless API_VERSION.empty?
+                call_metadata[:"x-goog-user-project"] = @quota_project_id if @quota_project_id
+
+                options.apply_defaults timeout:      @config.rpcs.query_data.timeout,
+                                       metadata:     call_metadata,
+                                       retry_policy: @config.rpcs.query_data.retry_policy
+
+                options.apply_defaults timeout:      @config.timeout,
+                                       metadata:     @config.metadata,
+                                       retry_policy: @config.retry_policy
+
+                @data_chat_service_stub.query_data request, options do |result, operation|
+                  yield result, operation if block_given?
+                end
+              rescue ::Gapic::Rest::Error => e
+                raise ::Google::Cloud::Error.from_error(e)
+              end
+
               ##
               # Configuration class for the DataChatService REST API.
               #
@@ -942,6 +1029,11 @@ module Google
                   # @return [::Gapic::Config::Method]
                   #
                   attr_reader :list_messages
+                  ##
+                  # RPC-specific configuration for `query_data`
+                  # @return [::Gapic::Config::Method]
+                  #
+                  attr_reader :query_data
 
                   # @private
                   def initialize parent_rpcs = nil
@@ -957,6 +1049,8 @@ module Google
                     @list_conversations = ::Gapic::Config::Method.new list_conversations_config
                     list_messages_config = parent_rpcs.list_messages if parent_rpcs.respond_to? :list_messages
                     @list_messages = ::Gapic::Config::Method.new list_messages_config
+                    query_data_config = parent_rpcs.query_data if parent_rpcs.respond_to? :query_data
+                    @query_data = ::Gapic::Config::Method.new query_data_config
 
                     yield self if block_given?
                   end

data/lib/google/cloud/gemini_data_analytics/v1beta/data_chat_service/rest/service_stub.rb

@@ -307,6 +307,46 @@ module Google
                 end
               end
 
+              ##
+              # Baseline implementation for the query_data REST call
+              #
+              # @param request_pb [::Google::Cloud::GeminiDataAnalytics::V1beta::QueryDataRequest]
+              #   A request object representing the call parameters. Required.
+              # @param options [::Gapic::CallOptions]
+              #   Overrides the default settings for this call, e.g, timeout, retries etc. Optional.
+              #
+              # @yield [result, operation] Access the result along with the TransportOperation object
+              # @yieldparam result [::Google::Cloud::GeminiDataAnalytics::V1beta::QueryDataResponse]
+              # @yieldparam operation [::Gapic::Rest::TransportOperation]
+              #
+              # @return [::Google::Cloud::GeminiDataAnalytics::V1beta::QueryDataResponse]
+              #   A result object deserialized from the server's reply
+              def query_data request_pb, options = nil
+                raise ::ArgumentError, "request must be provided" if request_pb.nil?
+
+                verb, uri, query_string_params, body = ServiceStub.transcode_query_data_request request_pb
+                query_string_params = if query_string_params.any?
+                                        query_string_params.to_h { |p| p.split "=", 2 }
+                                      else
+                                        {}
+                                      end
+
+                response = @client_stub.make_http_request(
+                  verb,
+                  uri: uri,
+                  body: body || "",
+                  params: query_string_params,
+                  method_name: "query_data",
+                  options: options
+                )
+                operation = ::Gapic::Rest::TransportOperation.new response
+                result = ::Google::Cloud::GeminiDataAnalytics::V1beta::QueryDataResponse.decode_json response.body, ignore_unknown_fields: true
+                catch :response do
+                  yield result, operation if block_given?
+                  result
+                end
+              end
+
               ##
               # @private
               #
@@ -434,6 +474,28 @@ module Google
                                                         )
                 transcoder.transcode request_pb
               end
+
+              ##
+              # @private
+              #
+              # GRPC transcoding helper method for the query_data REST call
+              #
+              # @param request_pb [::Google::Cloud::GeminiDataAnalytics::V1beta::QueryDataRequest]
+              #   A request object representing the call parameters. Required.
+              # @return [Array(String, [String, nil], Hash{String => String})]
+              #   Uri, Body, Query string parameters
+              def self.transcode_query_data_request request_pb
+                transcoder = Gapic::Rest::GrpcTranscoder.new
+                                                        .with_bindings(
+                                                          uri_method: :post,
+                                                          uri_template: "/v1beta/{parent}:queryData",
+                                                          body: "*",
+                                                          matches: [
+                                                            ["parent", %r{^projects/[^/]+/locations/[^/]+/?$}, false]
+                                                          ]
+                                                        )
+                transcoder.transcode request_pb
+              end
             end
           end
         end

data/lib/google/cloud/gemini_data_analytics/v1beta/version.rb

@@ -21,7 +21,7 @@ module Google
   module Cloud
     module GeminiDataAnalytics
       module V1beta
-        VERSION = "0.4.0"
+        VERSION = "0.5.1"
       end
     end
   end

data/lib/google/cloud/geminidataanalytics/v1beta/agent_context_pb.rb

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: google/cloud/geminidataanalytics/v1beta/agent_context.proto
+
+require 'google/protobuf'
+
+require 'google/api/field_behavior_pb'
+
+
+descriptor_data = "\n;google/cloud/geminidataanalytics/v1beta/agent_context.proto\x12\'google.cloud.geminidataanalytics.v1beta\x1a\x1fgoogle/api/field_behavior.proto\"4\n\x15\x41gentContextReference\x12\x1b\n\x0e\x63ontext_set_id\x18\x01 \x01(\tB\x03\xe0\x41\x02\x42\xa2\x02\n+com.google.cloud.geminidataanalytics.v1betaB\x11\x41gentContextProtoP\x01Z]cloud.google.com/go/geminidataanalytics/apiv1beta/geminidataanalyticspb;geminidataanalyticspb\xaa\x02\'Google.Cloud.GeminiDataAnalytics.V1Beta\xca\x02\'Google\\Cloud\\GeminiDataAnalytics\\V1beta\xea\x02*Google::Cloud::GeminiDataAnalytics::V1betab\x06proto3"
+
+pool = Google::Protobuf::DescriptorPool.generated_pool
+
+begin
+  pool.add_serialized_file(descriptor_data)
+rescue TypeError
+  # Compatibility code: will be removed in the next major version.
+  require 'google/protobuf/descriptor_pb'
+  parsed = Google::Protobuf::FileDescriptorProto.decode(descriptor_data)
+  parsed.clear_dependency
+  serialized = parsed.class.encode(parsed)
+  file = pool.add_serialized_file(serialized)
+  warn "Warning: Protobuf detected an import path issue while loading generated file #{__FILE__}"
+  imports = [
+  ]
+  imports.each do |type_name, expected_filename|
+    import_file = pool.lookup(type_name).file_descriptor
+    if import_file.name != expected_filename
+      warn "- #{file.name} imports #{expected_filename}, but that import was loaded as #{import_file.name}"
+    end
+  end
+  warn "Each proto file must use a consistent fully-qualified name."
+  warn "This will become an error in the next major version."
+end
+
+module Google
+  module Cloud
+    module GeminiDataAnalytics
+      module V1beta
+        AgentContextReference = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.geminidataanalytics.v1beta.AgentContextReference").msgclass
+      end
+    end
+  end
+end

data/lib/google/cloud/geminidataanalytics/v1beta/data_chat_service_pb.rb

@@ -17,7 +17,7 @@ require 'google/protobuf/struct_pb'
 require 'google/protobuf/timestamp_pb'
 
 
-descriptor_data = "\n?google/cloud/geminidataanalytics/v1beta/data_chat_service.proto\x12\'google.cloud.geminidataanalytics.v1beta\x1a\x1cgoogle/api/annotations.proto\x1a\x17google/api/client.proto\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a\x35google/cloud/geminidataanalytics/v1beta/context.proto\x1a:google/cloud/geminidataanalytics/v1beta/conversation.proto\x1a\x39google/cloud/geminidataanalytics/v1beta/credentials.proto\x1a\x38google/cloud/geminidataanalytics/v1beta/datasource.proto\x1a\x1bgoogle/protobuf/empty.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\xa4\x01\n\x13ListMessagesRequest\x12G\n\x06parent\x18\x01 \x01(\tB7\xe0\x41\x02\xfa\x41\x31\n/geminidataanalytics.googleapis.com/Conversation\x12\x16\n\tpage_size\x18\x03 \x01(\x05\x42\x03\xe0\x41\x01\x12\x17\n\npage_token\x18\x04 \x01(\tB\x03\xe0\x41\x01\x12\x13\n\x06\x66ilter\x18\x05 \x01(\tB\x03\xe0\x41\x01\"z\n\x14ListMessagesResponse\x12I\n\x08messages\x18\x01 \x03(\x0b\x32\x37.google.cloud.geminidataanalytics.v1beta.StorageMessage\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"g\n\x0eStorageMessage\x12\x12\n\nmessage_id\x18\x01 \x01(\t\x12\x41\n\x07message\x18\x02 \x01(\x0b\x32\x30.google.cloud.geminidataanalytics.v1beta.Message\"\xd4\x04\n\x0b\x43hatRequest\x12O\n\x0einline_context\x18\x65 \x01(\x0b\x32\x30.google.cloud.geminidataanalytics.v1beta.ContextB\x03\xe0\x41\x01H\x00\x12\x65\n\x16\x63onversation_reference\x18g \x01(\x0b\x32>.google.cloud.geminidataanalytics.v1beta.ConversationReferenceB\x03\xe0\x41\x01H\x00\x12\\\n\x12\x64\x61ta_agent_context\x18h \x01(\x0b\x32\x39.google.cloud.geminidataanalytics.v1beta.DataAgentContextB\x03\xe0\x41\x01H\x00\x12u\n\x1f\x63lient_managed_resource_context\x18i \x01(\x0b\x32\x45.google.cloud.geminidataanalytics.v1beta.ClientManagedResourceContextB\x03\xe0\x41\x01H\x00\x12\x46\n\x07project\x18\x01 \x01(\tB5\x18\x01\xe0\x41\x01\xfa\x41-\n+cloudresourcemanager.googleapis.com/Project\x12\x13\n\x06parent\x18\x03 \x01(\tB\x03\xe0\x41\x02\x12G\n\x08messages\x18\x02 \x03(\x0b\x32\x30.google.cloud.geminidataanalytics.v1beta.MessageB\x03\xe0\x41\x02\x42\x12\n\x10\x63ontext_provider\"\xe3\x02\n\x10\x44\x61taAgentContext\x12H\n\ndata_agent\x18\x01 \x01(\tB4\xe0\x41\x02\xfa\x41.\n,geminidataanalytics.googleapis.com/DataAgent\x12N\n\x0b\x63redentials\x18\x02 \x01(\x0b\x32\x34.google.cloud.geminidataanalytics.v1beta.CredentialsB\x03\xe0\x41\x01\x12\x66\n\x0f\x63ontext_version\x18\x03 \x01(\x0e\x32H.google.cloud.geminidataanalytics.v1beta.DataAgentContext.ContextVersionB\x03\xe0\x41\x01\"M\n\x0e\x43ontextVersion\x12\x1f\n\x1b\x43ONTEXT_VERSION_UNSPECIFIED\x10\x00\x12\x0b\n\x07STAGING\x10\x01\x12\r\n\tPUBLISHED\x10\x02\"\xc2\x01\n\x15\x43onversationReference\x12M\n\x0c\x63onversation\x18\x01 \x01(\tB7\xe0\x41\x02\xfa\x41\x31\n/geminidataanalytics.googleapis.com/Conversation\x12Z\n\x12\x64\x61ta_agent_context\x18\x03 \x01(\x0b\x32\x39.google.cloud.geminidataanalytics.v1beta.DataAgentContextB\x03\xe0\x41\x02\"\xa2\x01\n\x1c\x43lientManagedResourceContext\x12M\n\x0einline_context\x18\x01 \x01(\x0b\x32\x30.google.cloud.geminidataanalytics.v1beta.ContextB\x03\xe0\x41\x02\x12\x1c\n\x0f\x63onversation_id\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12\x15\n\x08\x61gent_id\x18\x03 \x01(\tB\x03\xe0\x41\x01\"\xfe\x01\n\x07Message\x12L\n\x0cuser_message\x18\x02 \x01(\x0b\x32\x34.google.cloud.geminidataanalytics.v1beta.UserMessageH\x00\x12P\n\x0esystem_message\x18\x03 \x01(\x0b\x32\x36.google.cloud.geminidataanalytics.v1beta.SystemMessageH\x00\x12\x32\n\ttimestamp\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x17\n\nmessage_id\x18\x04 \x01(\tB\x03\xe0\x41\x01\x42\x06\n\x04kind\"%\n\x0bUserMessage\x12\x0e\n\x04text\x18\x01 \x01(\tH\x00\x42\x06\n\x04kind\"\xc8\x04\n\rSystemMessage\x12\x44\n\x04text\x18\x01 \x01(\x0b\x32\x34.google.cloud.geminidataanalytics.v1beta.TextMessageH\x00\x12H\n\x06schema\x18\x02 \x01(\x0b\x32\x36.google.cloud.geminidataanalytics.v1beta.SchemaMessageH\x00\x12\x44\n\x04\x64\x61ta\x18\x03 \x01(\x0b\x32\x34.google.cloud.geminidataanalytics.v1beta.DataMessageH\x00\x12L\n\x08\x61nalysis\x18\x04 \x01(\x0b\x32\x38.google.cloud.geminidataanalytics.v1beta.AnalysisMessageH\x00\x12\x46\n\x05\x63hart\x18\x05 \x01(\x0b\x32\x35.google.cloud.geminidataanalytics.v1beta.ChartMessageH\x00\x12\x46\n\x05\x65rror\x18\x06 \x01(\x0b\x32\x35.google.cloud.geminidataanalytics.v1beta.ErrorMessageH\x00\x12W\n\x0f\x65xample_queries\x18\r \x01(\x0b\x32\x37.google.cloud.geminidataanalytics.v1beta.ExampleQueriesB\x03\xe0\x41\x01H\x00\x12\x15\n\x08group_id\x18\x0c \x01(\x05H\x01\x88\x01\x01\x42\x06\n\x04kindB\x0b\n\t_group_id\"\xce\x01\n\x0bTextMessage\x12\x12\n\x05parts\x18\x01 \x03(\tB\x03\xe0\x41\x01\x12U\n\ttext_type\x18\x02 \x01(\x0e\x32=.google.cloud.geminidataanalytics.v1beta.TextMessage.TextTypeB\x03\xe0\x41\x01\"T\n\x08TextType\x12\x19\n\x15TEXT_TYPE_UNSPECIFIED\x10\x00\x12\x12\n\x0e\x46INAL_RESPONSE\x10\x01\x12\x0b\n\x07THOUGHT\x10\x02\x12\x0c\n\x08PROGRESS\x10\x03\"\xa7\x01\n\rSchemaMessage\x12\x45\n\x05query\x18\x01 \x01(\x0b\x32\x34.google.cloud.geminidataanalytics.v1beta.SchemaQueryH\x00\x12G\n\x06result\x18\x02 \x01(\x0b\x32\x35.google.cloud.geminidataanalytics.v1beta.SchemaResultH\x00\x42\x06\n\x04kind\"$\n\x0bSchemaQuery\x12\x15\n\x08question\x18\x01 \x01(\tB\x03\xe0\x41\x01\"]\n\x0cSchemaResult\x12M\n\x0b\x64\x61tasources\x18\x01 \x03(\x0b\x32\x33.google.cloud.geminidataanalytics.v1beta.DatasourceB\x03\xe0\x41\x01\"\xe5\x02\n\x0b\x44\x61taMessage\x12\x43\n\x05query\x18\x01 \x01(\x0b\x32\x32.google.cloud.geminidataanalytics.v1beta.DataQueryH\x00\x12\x17\n\rgenerated_sql\x18\x02 \x01(\tH\x00\x12\x45\n\x06result\x18\x03 \x01(\x0b\x32\x33.google.cloud.geminidataanalytics.v1beta.DataResultH\x00\x12Z\n\x16generated_looker_query\x18\x04 \x01(\x0b\x32\x34.google.cloud.geminidataanalytics.v1beta.LookerQueryB\x02\x18\x01H\x00\x12M\n\rbig_query_job\x18\x05 \x01(\x0b\x32\x34.google.cloud.geminidataanalytics.v1beta.BigQueryJobH\x00\x42\x06\n\x04kind\"\x88\x02\n\x0bLookerQuery\x12\x12\n\x05model\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x14\n\x07\x65xplore\x18\x02 \x01(\tB\x03\xe0\x41\x02\x12\x13\n\x06\x66ields\x18\x03 \x03(\tB\x03\xe0\x41\x01\x12Q\n\x07\x66ilters\x18\x04 \x03(\x0b\x32;.google.cloud.geminidataanalytics.v1beta.LookerQuery.FilterB\x03\xe0\x41\x01\x12\x12\n\x05sorts\x18\x05 \x03(\tB\x03\xe0\x41\x01\x12\x17\n\x05limit\x18\x06 \x01(\tB\x03\xe0\x41\x01H\x00\x88\x01\x01\x1a\x30\n\x06\x46ilter\x12\x12\n\x05\x66ield\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x12\n\x05value\x18\x02 \x01(\tB\x03\xe0\x41\x02\x42\x08\n\x06_limit\"\xdf\x01\n\tDataQuery\x12K\n\x06looker\x18\x04 \x01(\x0b\x32\x34.google.cloud.geminidataanalytics.v1beta.LookerQueryB\x03\xe0\x41\x01H\x00\x12\x15\n\x08question\x18\x01 \x01(\tB\x03\xe0\x41\x01\x12\x11\n\x04name\x18\x03 \x01(\tB\x03\xe0\x41\x01\x12M\n\x0b\x64\x61tasources\x18\x02 \x03(\x0b\x32\x33.google.cloud.geminidataanalytics.v1beta.DatasourceB\x03\xe0\x41\x01\x42\x0c\n\nquery_type\"\x91\x01\n\nDataResult\x12\x11\n\x04name\x18\x03 \x01(\tB\x03\xe0\x41\x01\x12\x44\n\x06schema\x18\x05 \x01(\x0b\x32/.google.cloud.geminidataanalytics.v1beta.SchemaB\x03\xe0\x41\x01\x12*\n\x04\x64\x61ta\x18\x02 \x03(\x0b\x32\x17.google.protobuf.StructB\x03\xe0\x41\x01\"\xf9\x01\n\x0b\x42igQueryJob\x12\x17\n\nproject_id\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x13\n\x06job_id\x18\x02 \x01(\tB\x03\xe0\x41\x02\x12\x15\n\x08location\x18\x05 \x01(\tB\x03\xe0\x41\x01\x12_\n\x11\x64\x65stination_table\x18\x03 \x01(\x0b\x32?.google.cloud.geminidataanalytics.v1beta.BigQueryTableReferenceB\x03\xe0\x41\x01\x12\x44\n\x06schema\x18\x07 \x01(\x0b\x32/.google.cloud.geminidataanalytics.v1beta.SchemaB\x03\xe0\x41\x01\"\xb4\x01\n\x0f\x41nalysisMessage\x12G\n\x05query\x18\x01 \x01(\x0b\x32\x36.google.cloud.geminidataanalytics.v1beta.AnalysisQueryH\x00\x12P\n\x0eprogress_event\x18\x02 \x01(\x0b\x32\x36.google.cloud.geminidataanalytics.v1beta.AnalysisEventH\x00\x42\x06\n\x04kind\"F\n\rAnalysisQuery\x12\x15\n\x08question\x18\x01 \x01(\tB\x03\xe0\x41\x01\x12\x1e\n\x11\x64\x61ta_result_names\x18\x02 \x03(\tB\x03\xe0\x41\x01\"\xaa\x02\n\rAnalysisEvent\x12\x1b\n\x11planner_reasoning\x18\x02 \x01(\tH\x00\x12\x1b\n\x11\x63oder_instruction\x18\x03 \x01(\tH\x00\x12\x0e\n\x04\x63ode\x18\x04 \x01(\tH\x00\x12\x1a\n\x10\x65xecution_output\x18\x05 \x01(\tH\x00\x12\x19\n\x0f\x65xecution_error\x18\x06 \x01(\tH\x00\x12 \n\x16result_vega_chart_json\x18\x07 \x01(\tH\x00\x12!\n\x17result_natural_language\x18\x08 \x01(\tH\x00\x12\x19\n\x0fresult_csv_data\x18\t \x01(\tH\x00\x12\x1f\n\x15result_reference_data\x18\n \x01(\tH\x00\x12\x0f\n\x05\x65rror\x18\x0b \x01(\tH\x00\x42\x06\n\x04kind\"\xa4\x01\n\x0c\x43hartMessage\x12\x44\n\x05query\x18\x01 \x01(\x0b\x32\x33.google.cloud.geminidataanalytics.v1beta.ChartQueryH\x00\x12\x46\n\x06result\x18\x02 \x01(\x0b\x32\x34.google.cloud.geminidataanalytics.v1beta.ChartResultH\x00\x42\x06\n\x04kind\"F\n\nChartQuery\x12\x19\n\x0cinstructions\x18\x01 \x01(\tB\x03\xe0\x41\x01\x12\x1d\n\x10\x64\x61ta_result_name\x18\x02 \x01(\tB\x03\xe0\x41\x01\"\x83\x01\n\x0b\x43hartResult\x12\x31\n\x0bvega_config\x18\x02 \x01(\x0b\x32\x17.google.protobuf.StructB\x03\xe0\x41\x01\x12\x41\n\x05image\x18\x03 \x01(\x0b\x32-.google.cloud.geminidataanalytics.v1beta.BlobB\x03\xe0\x41\x01\"!\n\x0c\x45rrorMessage\x12\x11\n\x04text\x18\x01 \x01(\tB\x03\xe0\x41\x03\"e\n\x0e\x45xampleQueries\x12S\n\x0f\x65xample_queries\x18\x01 \x03(\x0b\x32\x35.google.cloud.geminidataanalytics.v1beta.ExampleQueryB\x03\xe0\x41\x01\"1\n\x04\x42lob\x12\x16\n\tmime_type\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x11\n\x04\x64\x61ta\x18\x02 \x01(\x0c\x42\x03\xe0\x41\x02\x32\xeb\n\n\x0f\x44\x61taChatService\x12\xa9\x01\n\x04\x43hat\x12\x34.google.cloud.geminidataanalytics.v1beta.ChatRequest\x1a\x30.google.cloud.geminidataanalytics.v1beta.Message\"7\x82\xd3\xe4\x93\x02\x31\",/v1beta/{parent=projects/*/locations/*}:chat:\x01*0\x01\x12\x82\x02\n\x12\x43reateConversation\x12\x42.google.cloud.geminidataanalytics.v1beta.CreateConversationRequest\x1a\x35.google.cloud.geminidataanalytics.v1beta.Conversation\"q\xda\x41#parent,conversation,conversation_id\x82\xd3\xe4\x93\x02\x45\"5/v1beta/{parent=projects/*/locations/*}/conversations:\x0c\x63onversation\x12\xb6\x01\n\x12\x44\x65leteConversation\x12\x42.google.cloud.geminidataanalytics.v1beta.DeleteConversationRequest\x1a\x16.google.protobuf.Empty\"D\xda\x41\x04name\x82\xd3\xe4\x93\x02\x37*5/v1beta/{name=projects/*/locations/*/conversations/*}\x12\xcf\x01\n\x0fGetConversation\x12?.google.cloud.geminidataanalytics.v1beta.GetConversationRequest\x1a\x35.google.cloud.geminidataanalytics.v1beta.Conversation\"D\xda\x41\x04name\x82\xd3\xe4\x93\x02\x37\x12\x35/v1beta/{name=projects/*/locations/*/conversations/*}\x12\xe2\x01\n\x11ListConversations\x12\x41.google.cloud.geminidataanalytics.v1beta.ListConversationsRequest\x1a\x42.google.cloud.geminidataanalytics.v1beta.ListConversationsResponse\"F\xda\x41\x06parent\x82\xd3\xe4\x93\x02\x37\x12\x35/v1beta/{parent=projects/*/locations/*}/conversations\x12\xde\x01\n\x0cListMessages\x12<.google.cloud.geminidataanalytics.v1beta.ListMessagesRequest\x1a=.google.cloud.geminidataanalytics.v1beta.ListMessagesResponse\"Q\xda\x41\x06parent\x82\xd3\xe4\x93\x02\x42\x12@/v1beta/{parent=projects/*/locations/*/conversations/*}/messages\x1aV\xca\x41\"geminidataanalytics.googleapis.com\xd2\x41.https://www.googleapis.com/auth/cloud-platformB\xa5\x02\n+com.google.cloud.geminidataanalytics.v1betaB\x14\x44\x61taChatServiceProtoP\x01Z]cloud.google.com/go/geminidataanalytics/apiv1beta/geminidataanalyticspb;geminidataanalyticspb\xaa\x02\'Google.Cloud.GeminiDataAnalytics.V1Beta\xca\x02\'Google\\Cloud\\GeminiDataAnalytics\\V1beta\xea\x02*Google::Cloud::GeminiDataAnalytics::V1betab\x06proto3"
+descriptor_data = "\n?google/cloud/geminidataanalytics/v1beta/data_chat_service.proto\x12\'google.cloud.geminidataanalytics.v1beta\x1a\x1cgoogle/api/annotations.proto\x1a\x17google/api/client.proto\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a\x35google/cloud/geminidataanalytics/v1beta/context.proto\x1a:google/cloud/geminidataanalytics/v1beta/conversation.proto\x1a\x39google/cloud/geminidataanalytics/v1beta/credentials.proto\x1a\x38google/cloud/geminidataanalytics/v1beta/datasource.proto\x1a\x1bgoogle/protobuf/empty.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\x90\x02\n\x10QueryDataRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12\x13\n\x06prompt\x18\x02 \x01(\tB\x03\xe0\x41\x02\x12O\n\x07\x63ontext\x18\x03 \x01(\x0b\x32\x39.google.cloud.geminidataanalytics.v1beta.QueryDataContextB\x03\xe0\x41\x02\x12[\n\x12generation_options\x18\x04 \x01(\x0b\x32:.google.cloud.geminidataanalytics.v1beta.GenerationOptionsB\x03\xe0\x41\x01\"\xb8\x01\n\x11GenerationOptions\x12\"\n\x15generate_query_result\x18\x01 \x01(\x08\x42\x03\xe0\x41\x01\x12-\n generate_natural_language_answer\x18\x02 \x01(\x08\x42\x03\xe0\x41\x01\x12!\n\x14generate_explanation\x18\x03 \x01(\x08\x42\x03\xe0\x41\x01\x12-\n generate_disambiguation_question\x18\x04 \x01(\x08\x42\x03\xe0\x41\x01\"u\n\x10QueryDataContext\x12\x61\n\x15\x64\x61tasource_references\x18\x01 \x01(\x0b\x32=.google.cloud.geminidataanalytics.v1beta.DatasourceReferencesB\x03\xe0\x41\x02\"\xde\x01\n\x11QueryDataResponse\x12\x17\n\x0fgenerated_query\x18\x01 \x01(\t\x12\x1a\n\x12intent_explanation\x18\x02 \x01(\t\x12R\n\x0cquery_result\x18\x03 \x01(\x0b\x32<.google.cloud.geminidataanalytics.v1beta.ExecutedQueryResult\x12\x1f\n\x17natural_language_answer\x18\x04 \x01(\t\x12\x1f\n\x17\x64isambiguation_question\x18\x05 \x03(\t\"\xa4\x03\n\x13\x45xecutedQueryResult\x12T\n\x07\x63olumns\x18\x01 \x03(\x0b\x32\x43.google.cloud.geminidataanalytics.v1beta.ExecutedQueryResult.Column\x12N\n\x04rows\x18\x02 \x03(\x0b\x32@.google.cloud.geminidataanalytics.v1beta.ExecutedQueryResult.Row\x12\x17\n\x0ftotal_row_count\x18\x03 \x01(\x03\x12\x16\n\x0epartial_result\x18\x04 \x01(\x08\x12\x1d\n\x15query_execution_error\x18\x05 \x01(\t\x1a$\n\x06\x43olumn\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0c\n\x04type\x18\x02 \x01(\t\x1a\x16\n\x05Value\x12\r\n\x05value\x18\x01 \x01(\t\x1aY\n\x03Row\x12R\n\x06values\x18\x01 \x03(\x0b\x32\x42.google.cloud.geminidataanalytics.v1beta.ExecutedQueryResult.Value\"\xa4\x01\n\x13ListMessagesRequest\x12G\n\x06parent\x18\x01 \x01(\tB7\xe0\x41\x02\xfa\x41\x31\n/geminidataanalytics.googleapis.com/Conversation\x12\x16\n\tpage_size\x18\x03 \x01(\x05\x42\x03\xe0\x41\x01\x12\x17\n\npage_token\x18\x04 \x01(\tB\x03\xe0\x41\x01\x12\x13\n\x06\x66ilter\x18\x05 \x01(\tB\x03\xe0\x41\x01\"z\n\x14ListMessagesResponse\x12I\n\x08messages\x18\x01 \x03(\x0b\x32\x37.google.cloud.geminidataanalytics.v1beta.StorageMessage\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"g\n\x0eStorageMessage\x12\x12\n\nmessage_id\x18\x01 \x01(\t\x12\x41\n\x07message\x18\x02 \x01(\x0b\x32\x30.google.cloud.geminidataanalytics.v1beta.Message\"\xd4\x04\n\x0b\x43hatRequest\x12O\n\x0einline_context\x18\x65 \x01(\x0b\x32\x30.google.cloud.geminidataanalytics.v1beta.ContextB\x03\xe0\x41\x01H\x00\x12\x65\n\x16\x63onversation_reference\x18g \x01(\x0b\x32>.google.cloud.geminidataanalytics.v1beta.ConversationReferenceB\x03\xe0\x41\x01H\x00\x12\\\n\x12\x64\x61ta_agent_context\x18h \x01(\x0b\x32\x39.google.cloud.geminidataanalytics.v1beta.DataAgentContextB\x03\xe0\x41\x01H\x00\x12u\n\x1f\x63lient_managed_resource_context\x18i \x01(\x0b\x32\x45.google.cloud.geminidataanalytics.v1beta.ClientManagedResourceContextB\x03\xe0\x41\x01H\x00\x12\x46\n\x07project\x18\x01 \x01(\tB5\x18\x01\xe0\x41\x01\xfa\x41-\n+cloudresourcemanager.googleapis.com/Project\x12\x13\n\x06parent\x18\x03 \x01(\tB\x03\xe0\x41\x02\x12G\n\x08messages\x18\x02 \x03(\x0b\x32\x30.google.cloud.geminidataanalytics.v1beta.MessageB\x03\xe0\x41\x02\x42\x12\n\x10\x63ontext_provider\"\xe3\x02\n\x10\x44\x61taAgentContext\x12H\n\ndata_agent\x18\x01 \x01(\tB4\xe0\x41\x02\xfa\x41.\n,geminidataanalytics.googleapis.com/DataAgent\x12N\n\x0b\x63redentials\x18\x02 \x01(\x0b\x32\x34.google.cloud.geminidataanalytics.v1beta.CredentialsB\x03\xe0\x41\x01\x12\x66\n\x0f\x63ontext_version\x18\x03 \x01(\x0e\x32H.google.cloud.geminidataanalytics.v1beta.DataAgentContext.ContextVersionB\x03\xe0\x41\x01\"M\n\x0e\x43ontextVersion\x12\x1f\n\x1b\x43ONTEXT_VERSION_UNSPECIFIED\x10\x00\x12\x0b\n\x07STAGING\x10\x01\x12\r\n\tPUBLISHED\x10\x02\"\xc2\x01\n\x15\x43onversationReference\x12M\n\x0c\x63onversation\x18\x01 \x01(\tB7\xe0\x41\x02\xfa\x41\x31\n/geminidataanalytics.googleapis.com/Conversation\x12Z\n\x12\x64\x61ta_agent_context\x18\x03 \x01(\x0b\x32\x39.google.cloud.geminidataanalytics.v1beta.DataAgentContextB\x03\xe0\x41\x02\"\xa2\x01\n\x1c\x43lientManagedResourceContext\x12M\n\x0einline_context\x18\x01 \x01(\x0b\x32\x30.google.cloud.geminidataanalytics.v1beta.ContextB\x03\xe0\x41\x02\x12\x1c\n\x0f\x63onversation_id\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12\x15\n\x08\x61gent_id\x18\x03 \x01(\tB\x03\xe0\x41\x01\"\xfe\x01\n\x07Message\x12L\n\x0cuser_message\x18\x02 \x01(\x0b\x32\x34.google.cloud.geminidataanalytics.v1beta.UserMessageH\x00\x12P\n\x0esystem_message\x18\x03 \x01(\x0b\x32\x36.google.cloud.geminidataanalytics.v1beta.SystemMessageH\x00\x12\x32\n\ttimestamp\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x17\n\nmessage_id\x18\x04 \x01(\tB\x03\xe0\x41\x01\x42\x06\n\x04kind\"%\n\x0bUserMessage\x12\x0e\n\x04text\x18\x01 \x01(\tH\x00\x42\x06\n\x04kind\"\xc8\x04\n\rSystemMessage\x12\x44\n\x04text\x18\x01 \x01(\x0b\x32\x34.google.cloud.geminidataanalytics.v1beta.TextMessageH\x00\x12H\n\x06schema\x18\x02 \x01(\x0b\x32\x36.google.cloud.geminidataanalytics.v1beta.SchemaMessageH\x00\x12\x44\n\x04\x64\x61ta\x18\x03 \x01(\x0b\x32\x34.google.cloud.geminidataanalytics.v1beta.DataMessageH\x00\x12L\n\x08\x61nalysis\x18\x04 \x01(\x0b\x32\x38.google.cloud.geminidataanalytics.v1beta.AnalysisMessageH\x00\x12\x46\n\x05\x63hart\x18\x05 \x01(\x0b\x32\x35.google.cloud.geminidataanalytics.v1beta.ChartMessageH\x00\x12\x46\n\x05\x65rror\x18\x06 \x01(\x0b\x32\x35.google.cloud.geminidataanalytics.v1beta.ErrorMessageH\x00\x12W\n\x0f\x65xample_queries\x18\r \x01(\x0b\x32\x37.google.cloud.geminidataanalytics.v1beta.ExampleQueriesB\x03\xe0\x41\x01H\x00\x12\x15\n\x08group_id\x18\x0c \x01(\x05H\x01\x88\x01\x01\x42\x06\n\x04kindB\x0b\n\t_group_id\"\xce\x01\n\x0bTextMessage\x12\x12\n\x05parts\x18\x01 \x03(\tB\x03\xe0\x41\x01\x12U\n\ttext_type\x18\x02 \x01(\x0e\x32=.google.cloud.geminidataanalytics.v1beta.TextMessage.TextTypeB\x03\xe0\x41\x01\"T\n\x08TextType\x12\x19\n\x15TEXT_TYPE_UNSPECIFIED\x10\x00\x12\x12\n\x0e\x46INAL_RESPONSE\x10\x01\x12\x0b\n\x07THOUGHT\x10\x02\x12\x0c\n\x08PROGRESS\x10\x03\"\xa7\x01\n\rSchemaMessage\x12\x45\n\x05query\x18\x01 \x01(\x0b\x32\x34.google.cloud.geminidataanalytics.v1beta.SchemaQueryH\x00\x12G\n\x06result\x18\x02 \x01(\x0b\x32\x35.google.cloud.geminidataanalytics.v1beta.SchemaResultH\x00\x42\x06\n\x04kind\"$\n\x0bSchemaQuery\x12\x15\n\x08question\x18\x01 \x01(\tB\x03\xe0\x41\x01\"]\n\x0cSchemaResult\x12M\n\x0b\x64\x61tasources\x18\x01 \x03(\x0b\x32\x33.google.cloud.geminidataanalytics.v1beta.DatasourceB\x03\xe0\x41\x01\"\xe5\x02\n\x0b\x44\x61taMessage\x12\x43\n\x05query\x18\x01 \x01(\x0b\x32\x32.google.cloud.geminidataanalytics.v1beta.DataQueryH\x00\x12\x17\n\rgenerated_sql\x18\x02 \x01(\tH\x00\x12\x45\n\x06result\x18\x03 \x01(\x0b\x32\x33.google.cloud.geminidataanalytics.v1beta.DataResultH\x00\x12Z\n\x16generated_looker_query\x18\x04 \x01(\x0b\x32\x34.google.cloud.geminidataanalytics.v1beta.LookerQueryB\x02\x18\x01H\x00\x12M\n\rbig_query_job\x18\x05 \x01(\x0b\x32\x34.google.cloud.geminidataanalytics.v1beta.BigQueryJobH\x00\x42\x06\n\x04kind\"\x88\x02\n\x0bLookerQuery\x12\x12\n\x05model\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x14\n\x07\x65xplore\x18\x02 \x01(\tB\x03\xe0\x41\x02\x12\x13\n\x06\x66ields\x18\x03 \x03(\tB\x03\xe0\x41\x01\x12Q\n\x07\x66ilters\x18\x04 \x03(\x0b\x32;.google.cloud.geminidataanalytics.v1beta.LookerQuery.FilterB\x03\xe0\x41\x01\x12\x12\n\x05sorts\x18\x05 \x03(\tB\x03\xe0\x41\x01\x12\x17\n\x05limit\x18\x06 \x01(\tB\x03\xe0\x41\x01H\x00\x88\x01\x01\x1a\x30\n\x06\x46ilter\x12\x12\n\x05\x66ield\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x12\n\x05value\x18\x02 \x01(\tB\x03\xe0\x41\x02\x42\x08\n\x06_limit\"\xdf\x01\n\tDataQuery\x12K\n\x06looker\x18\x04 \x01(\x0b\x32\x34.google.cloud.geminidataanalytics.v1beta.LookerQueryB\x03\xe0\x41\x01H\x00\x12\x15\n\x08question\x18\x01 \x01(\tB\x03\xe0\x41\x01\x12\x11\n\x04name\x18\x03 \x01(\tB\x03\xe0\x41\x01\x12M\n\x0b\x64\x61tasources\x18\x02 \x03(\x0b\x32\x33.google.cloud.geminidataanalytics.v1beta.DatasourceB\x03\xe0\x41\x01\x42\x0c\n\nquery_type\"\x91\x01\n\nDataResult\x12\x11\n\x04name\x18\x03 \x01(\tB\x03\xe0\x41\x01\x12\x44\n\x06schema\x18\x05 \x01(\x0b\x32/.google.cloud.geminidataanalytics.v1beta.SchemaB\x03\xe0\x41\x01\x12*\n\x04\x64\x61ta\x18\x02 \x03(\x0b\x32\x17.google.protobuf.StructB\x03\xe0\x41\x01\"\xf9\x01\n\x0b\x42igQueryJob\x12\x17\n\nproject_id\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x13\n\x06job_id\x18\x02 \x01(\tB\x03\xe0\x41\x02\x12\x15\n\x08location\x18\x05 \x01(\tB\x03\xe0\x41\x01\x12_\n\x11\x64\x65stination_table\x18\x03 \x01(\x0b\x32?.google.cloud.geminidataanalytics.v1beta.BigQueryTableReferenceB\x03\xe0\x41\x01\x12\x44\n\x06schema\x18\x07 \x01(\x0b\x32/.google.cloud.geminidataanalytics.v1beta.SchemaB\x03\xe0\x41\x01\"\xb4\x01\n\x0f\x41nalysisMessage\x12G\n\x05query\x18\x01 \x01(\x0b\x32\x36.google.cloud.geminidataanalytics.v1beta.AnalysisQueryH\x00\x12P\n\x0eprogress_event\x18\x02 \x01(\x0b\x32\x36.google.cloud.geminidataanalytics.v1beta.AnalysisEventH\x00\x42\x06\n\x04kind\"F\n\rAnalysisQuery\x12\x15\n\x08question\x18\x01 \x01(\tB\x03\xe0\x41\x01\x12\x1e\n\x11\x64\x61ta_result_names\x18\x02 \x03(\tB\x03\xe0\x41\x01\"\xaa\x02\n\rAnalysisEvent\x12\x1b\n\x11planner_reasoning\x18\x02 \x01(\tH\x00\x12\x1b\n\x11\x63oder_instruction\x18\x03 \x01(\tH\x00\x12\x0e\n\x04\x63ode\x18\x04 \x01(\tH\x00\x12\x1a\n\x10\x65xecution_output\x18\x05 \x01(\tH\x00\x12\x19\n\x0f\x65xecution_error\x18\x06 \x01(\tH\x00\x12 \n\x16result_vega_chart_json\x18\x07 \x01(\tH\x00\x12!\n\x17result_natural_language\x18\x08 \x01(\tH\x00\x12\x19\n\x0fresult_csv_data\x18\t \x01(\tH\x00\x12\x1f\n\x15result_reference_data\x18\n \x01(\tH\x00\x12\x0f\n\x05\x65rror\x18\x0b \x01(\tH\x00\x42\x06\n\x04kind\"\xa4\x01\n\x0c\x43hartMessage\x12\x44\n\x05query\x18\x01 \x01(\x0b\x32\x33.google.cloud.geminidataanalytics.v1beta.ChartQueryH\x00\x12\x46\n\x06result\x18\x02 \x01(\x0b\x32\x34.google.cloud.geminidataanalytics.v1beta.ChartResultH\x00\x42\x06\n\x04kind\"F\n\nChartQuery\x12\x19\n\x0cinstructions\x18\x01 \x01(\tB\x03\xe0\x41\x01\x12\x1d\n\x10\x64\x61ta_result_name\x18\x02 \x01(\tB\x03\xe0\x41\x01\"\x83\x01\n\x0b\x43hartResult\x12\x31\n\x0bvega_config\x18\x02 \x01(\x0b\x32\x17.google.protobuf.StructB\x03\xe0\x41\x01\x12\x41\n\x05image\x18\x03 \x01(\x0b\x32-.google.cloud.geminidataanalytics.v1beta.BlobB\x03\xe0\x41\x01\"!\n\x0c\x45rrorMessage\x12\x11\n\x04text\x18\x01 \x01(\tB\x03\xe0\x41\x03\"e\n\x0e\x45xampleQueries\x12S\n\x0f\x65xample_queries\x18\x01 \x03(\x0b\x32\x35.google.cloud.geminidataanalytics.v1beta.ExampleQueryB\x03\xe0\x41\x01\"1\n\x04\x42lob\x12\x16\n\tmime_type\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x11\n\x04\x64\x61ta\x18\x02 \x01(\x0c\x42\x03\xe0\x41\x02\x32\xae\x0c\n\x0f\x44\x61taChatService\x12\xa9\x01\n\x04\x43hat\x12\x34.google.cloud.geminidataanalytics.v1beta.ChatRequest\x1a\x30.google.cloud.geminidataanalytics.v1beta.Message\"7\x82\xd3\xe4\x93\x02\x31\",/v1beta/{parent=projects/*/locations/*}:chat:\x01*0\x01\x12\x82\x02\n\x12\x43reateConversation\x12\x42.google.cloud.geminidataanalytics.v1beta.CreateConversationRequest\x1a\x35.google.cloud.geminidataanalytics.v1beta.Conversation\"q\xda\x41#parent,conversation,conversation_id\x82\xd3\xe4\x93\x02\x45\"5/v1beta/{parent=projects/*/locations/*}/conversations:\x0c\x63onversation\x12\xb6\x01\n\x12\x44\x65leteConversation\x12\x42.google.cloud.geminidataanalytics.v1beta.DeleteConversationRequest\x1a\x16.google.protobuf.Empty\"D\xda\x41\x04name\x82\xd3\xe4\x93\x02\x37*5/v1beta/{name=projects/*/locations/*/conversations/*}\x12\xcf\x01\n\x0fGetConversation\x12?.google.cloud.geminidataanalytics.v1beta.GetConversationRequest\x1a\x35.google.cloud.geminidataanalytics.v1beta.Conversation\"D\xda\x41\x04name\x82\xd3\xe4\x93\x02\x37\x12\x35/v1beta/{name=projects/*/locations/*/conversations/*}\x12\xe2\x01\n\x11ListConversations\x12\x41.google.cloud.geminidataanalytics.v1beta.ListConversationsRequest\x1a\x42.google.cloud.geminidataanalytics.v1beta.ListConversationsResponse\"F\xda\x41\x06parent\x82\xd3\xe4\x93\x02\x37\x12\x35/v1beta/{parent=projects/*/locations/*}/conversations\x12\xde\x01\n\x0cListMessages\x12<.google.cloud.geminidataanalytics.v1beta.ListMessagesRequest\x1a=.google.cloud.geminidataanalytics.v1beta.ListMessagesResponse\"Q\xda\x41\x06parent\x82\xd3\xe4\x93\x02\x42\x12@/v1beta/{parent=projects/*/locations/*/conversations/*}/messages\x12\xc0\x01\n\tQueryData\x12\x39.google.cloud.geminidataanalytics.v1beta.QueryDataRequest\x1a:.google.cloud.geminidataanalytics.v1beta.QueryDataResponse\"<\x82\xd3\xe4\x93\x02\x36\"1/v1beta/{parent=projects/*/locations/*}:queryData:\x01*\x1aV\xca\x41\"geminidataanalytics.googleapis.com\xd2\x41.https://www.googleapis.com/auth/cloud-platformB\xa5\x02\n+com.google.cloud.geminidataanalytics.v1betaB\x14\x44\x61taChatServiceProtoP\x01Z]cloud.google.com/go/geminidataanalytics/apiv1beta/geminidataanalyticspb;geminidataanalyticspb\xaa\x02\'Google.Cloud.GeminiDataAnalytics.V1Beta\xca\x02\'Google\\Cloud\\GeminiDataAnalytics\\V1beta\xea\x02*Google::Cloud::GeminiDataAnalytics::V1betab\x06proto3"
 
 pool = Google::Protobuf::DescriptorPool.generated_pool
 
@@ -32,10 +32,10 @@ rescue TypeError
   file = pool.add_serialized_file(serialized)
   warn "Warning: Protobuf detected an import path issue while loading generated file #{__FILE__}"
   imports = [
+    ["google.cloud.geminidataanalytics.v1beta.DatasourceReferences", "google/cloud/geminidataanalytics/v1beta/datasource.proto"],
     ["google.cloud.geminidataanalytics.v1beta.Context", "google/cloud/geminidataanalytics/v1beta/context.proto"],
     ["google.cloud.geminidataanalytics.v1beta.Credentials", "google/cloud/geminidataanalytics/v1beta/credentials.proto"],
     ["google.protobuf.Timestamp", "google/protobuf/timestamp.proto"],
-    ["google.cloud.geminidataanalytics.v1beta.Datasource", "google/cloud/geminidataanalytics/v1beta/datasource.proto"],
     ["google.protobuf.Struct", "google/protobuf/struct.proto"],
   ]
   imports.each do |type_name, expected_filename|
@@ -52,6 +52,14 @@ module Google
   module Cloud
     module GeminiDataAnalytics
       module V1beta
+        QueryDataRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.geminidataanalytics.v1beta.QueryDataRequest").msgclass
+        GenerationOptions = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.geminidataanalytics.v1beta.GenerationOptions").msgclass
+        QueryDataContext = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.geminidataanalytics.v1beta.QueryDataContext").msgclass
+        QueryDataResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.geminidataanalytics.v1beta.QueryDataResponse").msgclass
+        ExecutedQueryResult = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.geminidataanalytics.v1beta.ExecutedQueryResult").msgclass
+        ExecutedQueryResult::Column = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.geminidataanalytics.v1beta.ExecutedQueryResult.Column").msgclass
+        ExecutedQueryResult::Value = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.geminidataanalytics.v1beta.ExecutedQueryResult.Value").msgclass
+        ExecutedQueryResult::Row = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.geminidataanalytics.v1beta.ExecutedQueryResult.Row").msgclass
         ListMessagesRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.geminidataanalytics.v1beta.ListMessagesRequest").msgclass
         ListMessagesResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.geminidataanalytics.v1beta.ListMessagesResponse").msgclass
         StorageMessage = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.geminidataanalytics.v1beta.StorageMessage").msgclass

data/lib/google/cloud/geminidataanalytics/v1beta/data_chat_service_services_pb.rb

@@ -49,6 +49,8 @@ module Google
             rpc :ListConversations, ::Google::Cloud::GeminiDataAnalytics::V1beta::ListConversationsRequest, ::Google::Cloud::GeminiDataAnalytics::V1beta::ListConversationsResponse
             # Lists all messages for a given conversation.
             rpc :ListMessages, ::Google::Cloud::GeminiDataAnalytics::V1beta::ListMessagesRequest, ::Google::Cloud::GeminiDataAnalytics::V1beta::ListMessagesResponse
+            # Queries data from a natural language user query.
+            rpc :QueryData, ::Google::Cloud::GeminiDataAnalytics::V1beta::QueryDataRequest, ::Google::Cloud::GeminiDataAnalytics::V1beta::QueryDataResponse
           end
 
           Stub = Service.rpc_stub_class

data/lib/google/cloud/geminidataanalytics/v1beta/datasource_pb.rb

@@ -5,11 +5,12 @@
 require 'google/protobuf'
 
 require 'google/api/field_behavior_pb'
+require 'google/cloud/geminidataanalytics/v1beta/agent_context_pb'
 require 'google/cloud/geminidataanalytics/v1beta/credentials_pb'
 require 'google/protobuf/struct_pb'
 
 
-descriptor_data = "\n8google/cloud/geminidataanalytics/v1beta/datasource.proto\x12\'google.cloud.geminidataanalytics.v1beta\x1a\x1fgoogle/api/field_behavior.proto\x1a\x39google/cloud/geminidataanalytics/v1beta/credentials.proto\x1a\x1cgoogle/protobuf/struct.proto\"\x9f\x02\n\x14\x44\x61tasourceReferences\x12N\n\x02\x62q\x18\x01 \x01(\x0b\x32@.google.cloud.geminidataanalytics.v1beta.BigQueryTableReferencesH\x00\x12U\n\x06studio\x18\x02 \x01(\x0b\x32\x43.google.cloud.geminidataanalytics.v1beta.StudioDatasourceReferencesH\x00\x12R\n\x06looker\x18\x03 \x01(\x0b\x32@.google.cloud.geminidataanalytics.v1beta.LookerExploreReferencesH\x00\x42\x0c\n\nreferences\"y\n\x17\x42igQueryTableReferences\x12^\n\x10table_references\x18\x01 \x03(\x0b\x32?.google.cloud.geminidataanalytics.v1beta.BigQueryTableReferenceB\x03\xe0\x41\x02\"\xa7\x01\n\x16\x42igQueryTableReference\x12\x17\n\nproject_id\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x17\n\ndataset_id\x18\x03 \x01(\tB\x03\xe0\x41\x02\x12\x15\n\x08table_id\x18\x04 \x01(\tB\x03\xe0\x41\x02\x12\x44\n\x06schema\x18\x06 \x01(\x0b\x32/.google.cloud.geminidataanalytics.v1beta.SchemaB\x03\xe0\x41\x01\"{\n\x1aStudioDatasourceReferences\x12]\n\x11studio_references\x18\x02 \x03(\x0b\x32\x42.google.cloud.geminidataanalytics.v1beta.StudioDatasourceReference\"7\n\x19StudioDatasourceReference\x12\x1a\n\rdatasource_id\x18\x01 \x01(\tB\x03\xe0\x41\x02\"\xcb\x01\n\x17LookerExploreReferences\x12`\n\x12\x65xplore_references\x18\x01 \x03(\x0b\x32?.google.cloud.geminidataanalytics.v1beta.LookerExploreReferenceB\x03\xe0\x41\x02\x12N\n\x0b\x63redentials\x18\x02 \x01(\x0b\x32\x34.google.cloud.geminidataanalytics.v1beta.CredentialsB\x03\xe0\x41\x01\"\xa6\x02\n\x16LookerExploreReference\x12\x1d\n\x13looker_instance_uri\x18\t \x01(\tH\x00\x12j\n\x1cprivate_looker_instance_info\x18\n \x01(\x0b\x32\x42.google.cloud.geminidataanalytics.v1beta.PrivateLookerInstanceInfoH\x00\x12\x19\n\x0clookml_model\x18\x04 \x01(\tB\x03\xe0\x41\x02\x12\x14\n\x07\x65xplore\x18\x05 \x01(\tB\x03\xe0\x41\x02\x12\x44\n\x06schema\x18\x08 \x01(\x0b\x32/.google.cloud.geminidataanalytics.v1beta.SchemaB\x03\xe0\x41\x01\x42\n\n\x08instance\"W\n\x19PrivateLookerInstanceInfo\x12\x1a\n\x12looker_instance_id\x18\x01 \x01(\t\x12\x1e\n\x16service_directory_name\x18\x02 \x01(\t\"\xfe\x02\n\nDatasource\x12\x63\n\x18\x62igquery_table_reference\x18\x01 \x01(\x0b\x32?.google.cloud.geminidataanalytics.v1beta.BigQueryTableReferenceH\x00\x12\x1e\n\x14studio_datasource_id\x18\x02 \x01(\tH\x00\x12\x63\n\x18looker_explore_reference\x18\x04 \x01(\x0b\x32?.google.cloud.geminidataanalytics.v1beta.LookerExploreReferenceH\x00\x12\x44\n\x06schema\x18\x07 \x01(\x0b\x32/.google.cloud.geminidataanalytics.v1beta.SchemaB\x03\xe0\x41\x01\x12\x33\n\rstruct_schema\x18\n \x01(\x0b\x32\x17.google.protobuf.StructB\x03\xe0\x41\x01\x42\x0b\n\treference\"\xf7\x01\n\x06Schema\x12\x43\n\x06\x66ields\x18\x01 \x03(\x0b\x32..google.cloud.geminidataanalytics.v1beta.FieldB\x03\xe0\x41\x01\x12\x18\n\x0b\x64\x65scription\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12\x15\n\x08synonyms\x18\x03 \x03(\tB\x03\xe0\x41\x01\x12\x11\n\x04tags\x18\x04 \x03(\tB\x03\xe0\x41\x01\x12\x19\n\x0c\x64isplay_name\x18\x05 \x01(\tB\x03\xe0\x41\x01\x12I\n\x07\x66ilters\x18\x06 \x03(\x0b\x32\x33.google.cloud.geminidataanalytics.v1beta.DataFilterB\x03\xe0\x41\x01\"\x99\x02\n\x05\x46ield\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x01\x12\x11\n\x04type\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12\x18\n\x0b\x64\x65scription\x18\x03 \x01(\tB\x03\xe0\x41\x01\x12\x11\n\x04mode\x18\x04 \x01(\tB\x03\xe0\x41\x01\x12\x15\n\x08synonyms\x18\x06 \x03(\tB\x03\xe0\x41\x01\x12\x11\n\x04tags\x18\x07 \x03(\tB\x03\xe0\x41\x01\x12\x19\n\x0c\x64isplay_name\x18\x08 \x01(\tB\x03\xe0\x41\x01\x12\x46\n\tsubfields\x18\t \x03(\x0b\x32..google.cloud.geminidataanalytics.v1beta.FieldB\x03\xe0\x41\x01\x12\x15\n\x08\x63\x61tegory\x18\n \x01(\tB\x03\xe0\x41\x01\x12\x19\n\x0cvalue_format\x18\x0b \x01(\tB\x03\xe0\x41\x01\"\x80\x01\n\nDataFilter\x12\x12\n\x05\x66ield\x18\x01 \x01(\tB\x03\xe0\x41\x01\x12\x12\n\x05value\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12J\n\x04type\x18\x03 \x01(\x0e\x32\x37.google.cloud.geminidataanalytics.v1beta.DataFilterTypeB\x03\xe0\x41\x01*E\n\x0e\x44\x61taFilterType\x12 \n\x1c\x44\x41TA_FILTER_TYPE_UNSPECIFIED\x10\x00\x12\x11\n\rALWAYS_FILTER\x10\x01\x42\xa0\x02\n+com.google.cloud.geminidataanalytics.v1betaB\x0f\x44\x61tasourceProtoP\x01Z]cloud.google.com/go/geminidataanalytics/apiv1beta/geminidataanalyticspb;geminidataanalyticspb\xaa\x02\'Google.Cloud.GeminiDataAnalytics.V1Beta\xca\x02\'Google\\Cloud\\GeminiDataAnalytics\\V1beta\xea\x02*Google::Cloud::GeminiDataAnalytics::V1betab\x06proto3"
+descriptor_data = "\n8google/cloud/geminidataanalytics/v1beta/datasource.proto\x12\'google.cloud.geminidataanalytics.v1beta\x1a\x1fgoogle/api/field_behavior.proto\x1a;google/cloud/geminidataanalytics/v1beta/agent_context.proto\x1a\x39google/cloud/geminidataanalytics/v1beta/credentials.proto\x1a\x1cgoogle/protobuf/struct.proto\"\xa0\x04\n\x14\x44\x61tasourceReferences\x12N\n\x02\x62q\x18\x01 \x01(\x0b\x32@.google.cloud.geminidataanalytics.v1beta.BigQueryTableReferencesH\x00\x12U\n\x06studio\x18\x02 \x01(\x0b\x32\x43.google.cloud.geminidataanalytics.v1beta.StudioDatasourceReferencesH\x00\x12R\n\x06looker\x18\x03 \x01(\x0b\x32@.google.cloud.geminidataanalytics.v1beta.LookerExploreReferencesH\x00\x12L\n\x07\x61lloydb\x18\x08 \x01(\x0b\x32\x39.google.cloud.geminidataanalytics.v1beta.AlloyDbReferenceH\x00\x12V\n\x11spanner_reference\x18\t \x01(\x0b\x32\x39.google.cloud.geminidataanalytics.v1beta.SpannerReferenceH\x00\x12Y\n\x13\x63loud_sql_reference\x18\n \x01(\x0b\x32:.google.cloud.geminidataanalytics.v1beta.CloudSqlReferenceH\x00\x42\x0c\n\nreferences\"y\n\x17\x42igQueryTableReferences\x12^\n\x10table_references\x18\x01 \x03(\x0b\x32?.google.cloud.geminidataanalytics.v1beta.BigQueryTableReferenceB\x03\xe0\x41\x01\"\xa7\x01\n\x16\x42igQueryTableReference\x12\x17\n\nproject_id\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x17\n\ndataset_id\x18\x03 \x01(\tB\x03\xe0\x41\x02\x12\x15\n\x08table_id\x18\x04 \x01(\tB\x03\xe0\x41\x02\x12\x44\n\x06schema\x18\x06 \x01(\x0b\x32/.google.cloud.geminidataanalytics.v1beta.SchemaB\x03\xe0\x41\x01\"{\n\x1aStudioDatasourceReferences\x12]\n\x11studio_references\x18\x02 \x03(\x0b\x32\x42.google.cloud.geminidataanalytics.v1beta.StudioDatasourceReference\"7\n\x19StudioDatasourceReference\x12\x1a\n\rdatasource_id\x18\x01 \x01(\tB\x03\xe0\x41\x02\"\xdc\x01\n\x10\x41lloyDbReference\x12\x62\n\x12\x64\x61tabase_reference\x18\x01 \x01(\x0b\x32\x41.google.cloud.geminidataanalytics.v1beta.AlloyDbDatabaseReferenceB\x03\xe0\x41\x02\x12\x64\n\x17\x61gent_context_reference\x18\x03 \x01(\x0b\x32>.google.cloud.geminidataanalytics.v1beta.AgentContextReferenceB\x03\xe0\x41\x01\"\xad\x01\n\x18\x41lloyDbDatabaseReference\x12\x17\n\nproject_id\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x13\n\x06region\x18\x02 \x01(\tB\x03\xe0\x41\x02\x12\x17\n\ncluster_id\x18\x03 \x01(\tB\x03\xe0\x41\x02\x12\x18\n\x0binstance_id\x18\x04 \x01(\tB\x03\xe0\x41\x02\x12\x18\n\x0b\x64\x61tabase_id\x18\x05 \x01(\tB\x03\xe0\x41\x02\x12\x16\n\ttable_ids\x18\x06 \x03(\tB\x03\xe0\x41\x01\"\xdc\x01\n\x10SpannerReference\x12\x62\n\x12\x64\x61tabase_reference\x18\x01 \x01(\x0b\x32\x41.google.cloud.geminidataanalytics.v1beta.SpannerDatabaseReferenceB\x03\xe0\x41\x02\x12\x64\n\x17\x61gent_context_reference\x18\x02 \x01(\x0b\x32>.google.cloud.geminidataanalytics.v1beta.AgentContextReferenceB\x03\xe0\x41\x01\"\xb5\x02\n\x18SpannerDatabaseReference\x12]\n\x06\x65ngine\x18\x06 \x01(\x0e\x32H.google.cloud.geminidataanalytics.v1beta.SpannerDatabaseReference.EngineB\x03\xe0\x41\x02\x12\x17\n\nproject_id\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x13\n\x06region\x18\x02 \x01(\tB\x03\xe0\x41\x02\x12\x18\n\x0binstance_id\x18\x03 \x01(\tB\x03\xe0\x41\x02\x12\x18\n\x0b\x64\x61tabase_id\x18\x04 \x01(\tB\x03\xe0\x41\x02\x12\x16\n\ttable_ids\x18\x05 \x03(\tB\x03\xe0\x41\x01\"@\n\x06\x45ngine\x12\x16\n\x12\x45NGINE_UNSPECIFIED\x10\x00\x12\x0e\n\nGOOGLE_SQL\x10\x01\x12\x0e\n\nPOSTGRESQL\x10\x02\"\xde\x01\n\x11\x43loudSqlReference\x12\x63\n\x12\x64\x61tabase_reference\x18\x01 \x01(\x0b\x32\x42.google.cloud.geminidataanalytics.v1beta.CloudSqlDatabaseReferenceB\x03\xe0\x41\x02\x12\x64\n\x17\x61gent_context_reference\x18\x02 \x01(\x0b\x32>.google.cloud.geminidataanalytics.v1beta.AgentContextReferenceB\x03\xe0\x41\x01\"\xb2\x02\n\x19\x43loudSqlDatabaseReference\x12^\n\x06\x65ngine\x18\x01 \x01(\x0e\x32I.google.cloud.geminidataanalytics.v1beta.CloudSqlDatabaseReference.EngineB\x03\xe0\x41\x02\x12\x17\n\nproject_id\x18\x02 \x01(\tB\x03\xe0\x41\x02\x12\x13\n\x06region\x18\x03 \x01(\tB\x03\xe0\x41\x02\x12\x18\n\x0binstance_id\x18\x04 \x01(\tB\x03\xe0\x41\x02\x12\x18\n\x0b\x64\x61tabase_id\x18\x05 \x01(\tB\x03\xe0\x41\x02\x12\x16\n\ttable_ids\x18\x07 \x03(\tB\x03\xe0\x41\x01\";\n\x06\x45ngine\x12\x16\n\x12\x45NGINE_UNSPECIFIED\x10\x00\x12\x0e\n\nPOSTGRESQL\x10\x01\x12\t\n\x05MYSQL\x10\x02\"\xcb\x01\n\x17LookerExploreReferences\x12`\n\x12\x65xplore_references\x18\x01 \x03(\x0b\x32?.google.cloud.geminidataanalytics.v1beta.LookerExploreReferenceB\x03\xe0\x41\x02\x12N\n\x0b\x63redentials\x18\x02 \x01(\x0b\x32\x34.google.cloud.geminidataanalytics.v1beta.CredentialsB\x03\xe0\x41\x01\"\xa6\x02\n\x16LookerExploreReference\x12\x1d\n\x13looker_instance_uri\x18\t \x01(\tH\x00\x12j\n\x1cprivate_looker_instance_info\x18\n \x01(\x0b\x32\x42.google.cloud.geminidataanalytics.v1beta.PrivateLookerInstanceInfoH\x00\x12\x19\n\x0clookml_model\x18\x04 \x01(\tB\x03\xe0\x41\x02\x12\x14\n\x07\x65xplore\x18\x05 \x01(\tB\x03\xe0\x41\x02\x12\x44\n\x06schema\x18\x08 \x01(\x0b\x32/.google.cloud.geminidataanalytics.v1beta.SchemaB\x03\xe0\x41\x01\x42\n\n\x08instance\"W\n\x19PrivateLookerInstanceInfo\x12\x1a\n\x12looker_instance_id\x18\x01 \x01(\t\x12\x1e\n\x16service_directory_name\x18\x02 \x01(\t\"\x8a\x05\n\nDatasource\x12\x63\n\x18\x62igquery_table_reference\x18\x01 \x01(\x0b\x32?.google.cloud.geminidataanalytics.v1beta.BigQueryTableReferenceH\x00\x12\x1e\n\x14studio_datasource_id\x18\x02 \x01(\tH\x00\x12\x63\n\x18looker_explore_reference\x18\x04 \x01(\x0b\x32?.google.cloud.geminidataanalytics.v1beta.LookerExploreReferenceH\x00\x12W\n\x12\x61lloy_db_reference\x18\x0c \x01(\x0b\x32\x39.google.cloud.geminidataanalytics.v1beta.AlloyDbReferenceH\x00\x12V\n\x11spanner_reference\x18\r \x01(\x0b\x32\x39.google.cloud.geminidataanalytics.v1beta.SpannerReferenceH\x00\x12Y\n\x13\x63loud_sql_reference\x18\x0e \x01(\x0b\x32:.google.cloud.geminidataanalytics.v1beta.CloudSqlReferenceH\x00\x12\x44\n\x06schema\x18\x07 \x01(\x0b\x32/.google.cloud.geminidataanalytics.v1beta.SchemaB\x03\xe0\x41\x01\x12\x33\n\rstruct_schema\x18\n \x01(\x0b\x32\x17.google.protobuf.StructB\x03\xe0\x41\x01\x42\x0b\n\treference\"\xf7\x01\n\x06Schema\x12\x43\n\x06\x66ields\x18\x01 \x03(\x0b\x32..google.cloud.geminidataanalytics.v1beta.FieldB\x03\xe0\x41\x01\x12\x18\n\x0b\x64\x65scription\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12\x15\n\x08synonyms\x18\x03 \x03(\tB\x03\xe0\x41\x01\x12\x11\n\x04tags\x18\x04 \x03(\tB\x03\xe0\x41\x01\x12\x19\n\x0c\x64isplay_name\x18\x05 \x01(\tB\x03\xe0\x41\x01\x12I\n\x07\x66ilters\x18\x06 \x03(\x0b\x32\x33.google.cloud.geminidataanalytics.v1beta.DataFilterB\x03\xe0\x41\x01\"\x99\x02\n\x05\x46ield\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x01\x12\x11\n\x04type\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12\x18\n\x0b\x64\x65scription\x18\x03 \x01(\tB\x03\xe0\x41\x01\x12\x11\n\x04mode\x18\x04 \x01(\tB\x03\xe0\x41\x01\x12\x15\n\x08synonyms\x18\x06 \x03(\tB\x03\xe0\x41\x01\x12\x11\n\x04tags\x18\x07 \x03(\tB\x03\xe0\x41\x01\x12\x19\n\x0c\x64isplay_name\x18\x08 \x01(\tB\x03\xe0\x41\x01\x12\x46\n\tsubfields\x18\t \x03(\x0b\x32..google.cloud.geminidataanalytics.v1beta.FieldB\x03\xe0\x41\x01\x12\x15\n\x08\x63\x61tegory\x18\n \x01(\tB\x03\xe0\x41\x01\x12\x19\n\x0cvalue_format\x18\x0b \x01(\tB\x03\xe0\x41\x01\"\x80\x01\n\nDataFilter\x12\x12\n\x05\x66ield\x18\x01 \x01(\tB\x03\xe0\x41\x01\x12\x12\n\x05value\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12J\n\x04type\x18\x03 \x01(\x0e\x32\x37.google.cloud.geminidataanalytics.v1beta.DataFilterTypeB\x03\xe0\x41\x01*E\n\x0e\x44\x61taFilterType\x12 \n\x1c\x44\x41TA_FILTER_TYPE_UNSPECIFIED\x10\x00\x12\x11\n\rALWAYS_FILTER\x10\x01\x42\xa0\x02\n+com.google.cloud.geminidataanalytics.v1betaB\x0f\x44\x61tasourceProtoP\x01Z]cloud.google.com/go/geminidataanalytics/apiv1beta/geminidataanalyticspb;geminidataanalyticspb\xaa\x02\'Google.Cloud.GeminiDataAnalytics.V1Beta\xca\x02\'Google\\Cloud\\GeminiDataAnalytics\\V1beta\xea\x02*Google::Cloud::GeminiDataAnalytics::V1betab\x06proto3"
 
 pool = Google::Protobuf::DescriptorPool.generated_pool
 
@@ -24,6 +25,7 @@ rescue TypeError
   file = pool.add_serialized_file(serialized)
   warn "Warning: Protobuf detected an import path issue while loading generated file #{__FILE__}"
   imports = [
+    ["google.cloud.geminidataanalytics.v1beta.AgentContextReference", "google/cloud/geminidataanalytics/v1beta/agent_context.proto"],
     ["google.cloud.geminidataanalytics.v1beta.Credentials", "google/cloud/geminidataanalytics/v1beta/credentials.proto"],
     ["google.protobuf.Struct", "google/protobuf/struct.proto"],
   ]
@@ -46,6 +48,14 @@ module Google
         BigQueryTableReference = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.geminidataanalytics.v1beta.BigQueryTableReference").msgclass
         StudioDatasourceReferences = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.geminidataanalytics.v1beta.StudioDatasourceReferences").msgclass
         StudioDatasourceReference = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.geminidataanalytics.v1beta.StudioDatasourceReference").msgclass
+        AlloyDbReference = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.geminidataanalytics.v1beta.AlloyDbReference").msgclass
+        AlloyDbDatabaseReference = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.geminidataanalytics.v1beta.AlloyDbDatabaseReference").msgclass
+        SpannerReference = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.geminidataanalytics.v1beta.SpannerReference").msgclass
+        SpannerDatabaseReference = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.geminidataanalytics.v1beta.SpannerDatabaseReference").msgclass
+        SpannerDatabaseReference::Engine = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.geminidataanalytics.v1beta.SpannerDatabaseReference.Engine").enummodule
+        CloudSqlReference = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.geminidataanalytics.v1beta.CloudSqlReference").msgclass
+        CloudSqlDatabaseReference = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.geminidataanalytics.v1beta.CloudSqlDatabaseReference").msgclass
+        CloudSqlDatabaseReference::Engine = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.geminidataanalytics.v1beta.CloudSqlDatabaseReference.Engine").enummodule
         LookerExploreReferences = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.geminidataanalytics.v1beta.LookerExploreReferences").msgclass
         LookerExploreReference = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.geminidataanalytics.v1beta.LookerExploreReference").msgclass
         PrivateLookerInstanceInfo = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.geminidataanalytics.v1beta.PrivateLookerInstanceInfo").msgclass

data/proto_docs/google/cloud/geminidataanalytics/v1beta/agent_context.rb

@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# Auto-generated by gapic-generator-ruby. DO NOT EDIT!
+
+
+module Google
+  module Cloud
+    module GeminiDataAnalytics
+      module V1beta
+        # Message representing a reference to Agent Context.
+        # @!attribute [rw] context_set_id
+        #   @return [::String]
+        #     Required. Context set ID to retrieve.
+        class AgentContextReference
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+      end
+    end
+  end
+end

data/proto_docs/google/cloud/geminidataanalytics/v1beta/data_chat_service.rb

@@ -21,6 +21,148 @@ module Google
   module Cloud
     module GeminiDataAnalytics
       module V1beta
+        # Request to query data from a natural language query.
+        # @!attribute [rw] parent
+        #   @return [::String]
+        #     Required. The parent resource to generate the query for.
+        #     Format: projects/\\{project}/locations/\\{location}
+        # @!attribute [rw] prompt
+        #   @return [::String]
+        #     Required. The natural language query for which to generate query.
+        #     Example: "What are the top 5 best selling products this month?"
+        # @!attribute [rw] context
+        #   @return [::Google::Cloud::GeminiDataAnalytics::V1beta::QueryDataContext]
+        #     Required. The context for the data query, including the data sources to
+        #     use.
+        # @!attribute [rw] generation_options
+        #   @return [::Google::Cloud::GeminiDataAnalytics::V1beta::GenerationOptions]
+        #     Optional. Options to control query generation and execution behavior.
+        class QueryDataRequest
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+
+        # Options to control query generation, execution, and response format.
+        # @!attribute [rw] generate_query_result
+        #   @return [::Boolean]
+        #     Optional. If true, the generated query will be executed, and the result
+        #     data will be returned in the response.
+        # @!attribute [rw] generate_natural_language_answer
+        #   @return [::Boolean]
+        #     Optional. If true, a natural language answer based on the query execution
+        #     result will be generated and returned in the response.
+        # @!attribute [rw] generate_explanation
+        #   @return [::Boolean]
+        #     Optional. If true, an explanation of the generated query will be returned
+        #     in the response.
+        # @!attribute [rw] generate_disambiguation_question
+        #   @return [::Boolean]
+        #     Optional. If true (default to false), the service may return a
+        #     clarifying_question if the input query is ambiguous.
+        class GenerationOptions
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+
+        # References to data sources and context to use for the query.
+        # @!attribute [rw] datasource_references
+        #   @return [::Google::Cloud::GeminiDataAnalytics::V1beta::DatasourceReferences]
+        #     Required. The datasource references to use for the query.
+        class QueryDataContext
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+
+        # Response containing the generated query and related information.
+        # @!attribute [rw] generated_query
+        #   @return [::String]
+        #     Generated query for the given user prompt.
+        # @!attribute [rw] intent_explanation
+        #   @return [::String]
+        #     A natural language explanation of the generated query.
+        #     Populated if options.generate_explanation was true in the request.
+        # @!attribute [rw] query_result
+        #   @return [::Google::Cloud::GeminiDataAnalytics::V1beta::ExecutedQueryResult]
+        #     The result of executing the query.
+        #     Populated if options.generate_query_result or
+        #     options.generate_natural_language_answer was true in the request, and
+        #     execution was successful or attempted.
+        # @!attribute [rw] natural_language_answer
+        #   @return [::String]
+        #     A natural language answer to the query, based on the query_result.
+        #     Populated if options.generate_natural_language_answer was true in the
+        #     request and query execution was successful based in the response from
+        #     executeSql API.
+        # @!attribute [rw] disambiguation_question
+        #   @return [::Array<::String>]
+        #     If ambiguity was detected in the natural language query and
+        #     options.generate_disambiguation_question was true, this field contains a
+        #     question to the user for clarification. The returned represents the
+        #     service's best effort based on the ambiguous input.
+        class QueryDataResponse
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+
+        # The result of a query execution. The design is generic for all dialects.
+        # @!attribute [rw] columns
+        #   @return [::Array<::Google::Cloud::GeminiDataAnalytics::V1beta::ExecutedQueryResult::Column>]
+        #     The columns in the result set, in order.
+        # @!attribute [rw] rows
+        #   @return [::Array<::Google::Cloud::GeminiDataAnalytics::V1beta::ExecutedQueryResult::Row>]
+        #     The rows returned by the query.
+        # @!attribute [rw] total_row_count
+        #   @return [::Integer]
+        #     The total number of rows in the full result set, if known.
+        #     This may be an estimate or an exact count.
+        # @!attribute [rw] partial_result
+        #   @return [::Boolean]
+        #     Set to true if the returned rows in `query_result` are a subset of the
+        #     full result. This can happen, for example, if the query execution hits a
+        #     row limit. When true, the `query_result` does not contain all
+        #     rows. To retrieve the complete result, consider using the
+        #     `generated_query` in `QueryDataResponse` and executing it in your own
+        #     environment.
+        # @!attribute [rw] query_execution_error
+        #   @return [::String]
+        #     The error message if the query execution failed.
+        class ExecutedQueryResult
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+
+          # Describes a single column in the result set.
+          # @!attribute [rw] name
+          #   @return [::String]
+          #     The name of the column.
+          # @!attribute [rw] type
+          #   @return [::String]
+          #     The type of the column (e.g., "VARCHAR", "INT64", "TIMESTAMP").
+          class Column
+            include ::Google::Protobuf::MessageExts
+            extend ::Google::Protobuf::MessageExts::ClassMethods
+          end
+
+          # Represents a single value within a row.
+          # @!attribute [rw] value
+          #   @return [::String]
+          #     The cell value, represented in a string format.
+          #     Timestamps could be formatted, for example, using RFC3339Nano.
+          #     This field is used if the value is not null.
+          class Value
+            include ::Google::Protobuf::MessageExts
+            extend ::Google::Protobuf::MessageExts::ClassMethods
+          end
+
+          # Represents a single row in the result set.
+          # @!attribute [rw] values
+          #   @return [::Array<::Google::Cloud::GeminiDataAnalytics::V1beta::ExecutedQueryResult::Value>]
+          #     The values in the row, corresponding positionally to the columns.
+          class Row
+            include ::Google::Protobuf::MessageExts
+            extend ::Google::Protobuf::MessageExts::ClassMethods
+          end
+        end
+
         # Request for listing chat messages based on parent and conversation_id.
         # @!attribute [rw] parent
         #   @return [::String]

data/proto_docs/google/cloud/geminidataanalytics/v1beta/datasource.rb

@@ -26,26 +26,46 @@ module Google
         #   @return [::Google::Cloud::GeminiDataAnalytics::V1beta::BigQueryTableReferences]
         #     References to BigQuery tables.
         #
-        #     Note: The following fields are mutually exclusive: `bq`, `studio`, `looker`. If a field in that set is populated, all other fields in the set will automatically be cleared.
+        #     Note: The following fields are mutually exclusive: `bq`, `studio`, `looker`, `alloydb`, `spanner_reference`, `cloud_sql_reference`. If a field in that set is populated, all other fields in the set will automatically be cleared.
         # @!attribute [rw] studio
         #   @return [::Google::Cloud::GeminiDataAnalytics::V1beta::StudioDatasourceReferences]
         #     References to Looker Studio datasources.
         #
-        #     Note: The following fields are mutually exclusive: `studio`, `bq`, `looker`. If a field in that set is populated, all other fields in the set will automatically be cleared.
+        #     Note: The following fields are mutually exclusive: `studio`, `bq`, `looker`, `alloydb`, `spanner_reference`, `cloud_sql_reference`. If a field in that set is populated, all other fields in the set will automatically be cleared.
         # @!attribute [rw] looker
         #   @return [::Google::Cloud::GeminiDataAnalytics::V1beta::LookerExploreReferences]
         #     References to Looker Explores.
         #
-        #     Note: The following fields are mutually exclusive: `looker`, `bq`, `studio`. If a field in that set is populated, all other fields in the set will automatically be cleared.
+        #     Note: The following fields are mutually exclusive: `looker`, `bq`, `studio`, `alloydb`, `spanner_reference`, `cloud_sql_reference`. If a field in that set is populated, all other fields in the set will automatically be cleared.
+        # @!attribute [rw] alloydb
+        #   @return [::Google::Cloud::GeminiDataAnalytics::V1beta::AlloyDbReference]
+        #     Reference to an AlloyDB database. Only supported for the `QueryData`
+        #     method.
+        #
+        #     Note: The following fields are mutually exclusive: `alloydb`, `bq`, `studio`, `looker`, `spanner_reference`, `cloud_sql_reference`. If a field in that set is populated, all other fields in the set will automatically be cleared.
+        # @!attribute [rw] spanner_reference
+        #   @return [::Google::Cloud::GeminiDataAnalytics::V1beta::SpannerReference]
+        #     Reference to a Spanner database. Only supported for the `QueryData`
+        #     method.
+        #
+        #     Note: The following fields are mutually exclusive: `spanner_reference`, `bq`, `studio`, `looker`, `alloydb`, `cloud_sql_reference`. If a field in that set is populated, all other fields in the set will automatically be cleared.
+        # @!attribute [rw] cloud_sql_reference
+        #   @return [::Google::Cloud::GeminiDataAnalytics::V1beta::CloudSqlReference]
+        #     Reference to a CloudSql database. Only supported for the `QueryData`
+        #     method.
+        #
+        #     Note: The following fields are mutually exclusive: `cloud_sql_reference`, `bq`, `studio`, `looker`, `alloydb`, `spanner_reference`. If a field in that set is populated, all other fields in the set will automatically be cleared.
         class DatasourceReferences
           include ::Google::Protobuf::MessageExts
           extend ::Google::Protobuf::MessageExts::ClassMethods
         end
 
-        # Message representing references to BigQuery tables.
+        # Message representing references to BigQuery tables and property graphs.
+        # At least one of `table_references` or `property_graph_references` must be
+        # populated.
         # @!attribute [rw] table_references
         #   @return [::Array<::Google::Cloud::GeminiDataAnalytics::V1beta::BigQueryTableReference>]
-        #     Required. References to BigQuery tables.
+        #     Optional. References to BigQuery tables.
         class BigQueryTableReferences
           include ::Google::Protobuf::MessageExts
           extend ::Google::Protobuf::MessageExts::ClassMethods
@@ -87,6 +107,144 @@ module Google
           extend ::Google::Protobuf::MessageExts::ClassMethods
         end
 
+        # Message representing reference to an AlloyDB database and agent context.
+        # Only supported for the `QueryData` method.
+        # @!attribute [rw] database_reference
+        #   @return [::Google::Cloud::GeminiDataAnalytics::V1beta::AlloyDbDatabaseReference]
+        #     Required. Singular proto that supports specifying which database and tables
+        #     to include.
+        # @!attribute [rw] agent_context_reference
+        #   @return [::Google::Cloud::GeminiDataAnalytics::V1beta::AgentContextReference]
+        #     Optional. Parameters for retrieving data from Agent Context.
+        class AlloyDbReference
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+
+        # Message representing a reference to a single AlloyDB database.
+        # @!attribute [rw] project_id
+        #   @return [::String]
+        #     Required. The project the instance belongs to.
+        # @!attribute [rw] region
+        #   @return [::String]
+        #     Required. The region of the instance.
+        # @!attribute [rw] cluster_id
+        #   @return [::String]
+        #     Required. The cluster id.
+        # @!attribute [rw] instance_id
+        #   @return [::String]
+        #     Required. The instance id.
+        # @!attribute [rw] database_id
+        #   @return [::String]
+        #     Required. The database id.
+        # @!attribute [rw] table_ids
+        #   @return [::Array<::String>]
+        #     Optional. The table ids. Denotes all tables if unset.
+        class AlloyDbDatabaseReference
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+
+        # Message representing reference to a Spanner database and agent context.
+        # Only supported for the `QueryData` method.
+        # @!attribute [rw] database_reference
+        #   @return [::Google::Cloud::GeminiDataAnalytics::V1beta::SpannerDatabaseReference]
+        #     Required. Singular proto that supports specifying which database and tables
+        #     to include.
+        # @!attribute [rw] agent_context_reference
+        #   @return [::Google::Cloud::GeminiDataAnalytics::V1beta::AgentContextReference]
+        #     Optional. Parameters for retrieving data from Agent Context.
+        class SpannerReference
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+
+        # Message representing a reference to a single Spanner database.
+        # @!attribute [rw] engine
+        #   @return [::Google::Cloud::GeminiDataAnalytics::V1beta::SpannerDatabaseReference::Engine]
+        #     Required. The engine of the Spanner instance.
+        # @!attribute [rw] project_id
+        #   @return [::String]
+        #     Required. The project the instance belongs to.
+        # @!attribute [rw] region
+        #   @return [::String]
+        #     Required. The region of the instance.
+        # @!attribute [rw] instance_id
+        #   @return [::String]
+        #     Required. The instance id.
+        # @!attribute [rw] database_id
+        #   @return [::String]
+        #     Required. The database id.
+        # @!attribute [rw] table_ids
+        #   @return [::Array<::String>]
+        #     Optional. The table ids. Denotes all tables if unset.
+        class SpannerDatabaseReference
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+
+          # The database engine.
+          module Engine
+            # Engine is not specified.
+            ENGINE_UNSPECIFIED = 0
+
+            # Google SQL
+            GOOGLE_SQL = 1
+
+            # PostgreSQL
+            POSTGRESQL = 2
+          end
+        end
+
+        # Message representing reference to a CloudSQL database and agent context.
+        # Only supported for the `QueryData` method.
+        # @!attribute [rw] database_reference
+        #   @return [::Google::Cloud::GeminiDataAnalytics::V1beta::CloudSqlDatabaseReference]
+        #     Required. Singular proto that supports specifying which database and tables
+        #     to include.
+        # @!attribute [rw] agent_context_reference
+        #   @return [::Google::Cloud::GeminiDataAnalytics::V1beta::AgentContextReference]
+        #     Optional. Parameters for retrieving data from Agent Context.
+        class CloudSqlReference
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+
+        # Message representing a reference to a single CloudSQL database.
+        # @!attribute [rw] engine
+        #   @return [::Google::Cloud::GeminiDataAnalytics::V1beta::CloudSqlDatabaseReference::Engine]
+        #     Required. The engine of the Cloud SQL instance.
+        # @!attribute [rw] project_id
+        #   @return [::String]
+        #     Required. The project the instance belongs to.
+        # @!attribute [rw] region
+        #   @return [::String]
+        #     Required. The region of the instance.
+        # @!attribute [rw] instance_id
+        #   @return [::String]
+        #     Required. The instance id.
+        # @!attribute [rw] database_id
+        #   @return [::String]
+        #     Required. The database id.
+        # @!attribute [rw] table_ids
+        #   @return [::Array<::String>]
+        #     Optional. The table ids. Denotes all tables if unset.
+        class CloudSqlDatabaseReference
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+
+          # The database engine.
+          module Engine
+            # Engine is not specified.
+            ENGINE_UNSPECIFIED = 0
+
+            # PostgreSQL
+            POSTGRESQL = 1
+
+            # MySQL
+            MYSQL = 2
+          end
+        end
+
         # Message representing references to Looker explores.
         # @!attribute [rw] explore_references
         #   @return [::Array<::Google::Cloud::GeminiDataAnalytics::V1beta::LookerExploreReference>]
@@ -153,17 +311,32 @@ module Google
         #   @return [::Google::Cloud::GeminiDataAnalytics::V1beta::BigQueryTableReference]
         #     A reference to a BigQuery table.
         #
-        #     Note: The following fields are mutually exclusive: `bigquery_table_reference`, `studio_datasource_id`, `looker_explore_reference`. If a field in that set is populated, all other fields in the set will automatically be cleared.
+        #     Note: The following fields are mutually exclusive: `bigquery_table_reference`, `studio_datasource_id`, `looker_explore_reference`, `alloy_db_reference`, `spanner_reference`, `cloud_sql_reference`. If a field in that set is populated, all other fields in the set will automatically be cleared.
         # @!attribute [rw] studio_datasource_id
         #   @return [::String]
         #     A reference to a Looker Studio datasource.
         #
-        #     Note: The following fields are mutually exclusive: `studio_datasource_id`, `bigquery_table_reference`, `looker_explore_reference`. If a field in that set is populated, all other fields in the set will automatically be cleared.
+        #     Note: The following fields are mutually exclusive: `studio_datasource_id`, `bigquery_table_reference`, `looker_explore_reference`, `alloy_db_reference`, `spanner_reference`, `cloud_sql_reference`. If a field in that set is populated, all other fields in the set will automatically be cleared.
         # @!attribute [rw] looker_explore_reference
         #   @return [::Google::Cloud::GeminiDataAnalytics::V1beta::LookerExploreReference]
         #     A reference to a Looker explore.
         #
-        #     Note: The following fields are mutually exclusive: `looker_explore_reference`, `bigquery_table_reference`, `studio_datasource_id`. If a field in that set is populated, all other fields in the set will automatically be cleared.
+        #     Note: The following fields are mutually exclusive: `looker_explore_reference`, `bigquery_table_reference`, `studio_datasource_id`, `alloy_db_reference`, `spanner_reference`, `cloud_sql_reference`. If a field in that set is populated, all other fields in the set will automatically be cleared.
+        # @!attribute [rw] alloy_db_reference
+        #   @return [::Google::Cloud::GeminiDataAnalytics::V1beta::AlloyDbReference]
+        #     A reference to an AlloyDB database.
+        #
+        #     Note: The following fields are mutually exclusive: `alloy_db_reference`, `bigquery_table_reference`, `studio_datasource_id`, `looker_explore_reference`, `spanner_reference`, `cloud_sql_reference`. If a field in that set is populated, all other fields in the set will automatically be cleared.
+        # @!attribute [rw] spanner_reference
+        #   @return [::Google::Cloud::GeminiDataAnalytics::V1beta::SpannerReference]
+        #     A reference to a Spanner database.
+        #
+        #     Note: The following fields are mutually exclusive: `spanner_reference`, `bigquery_table_reference`, `studio_datasource_id`, `looker_explore_reference`, `alloy_db_reference`, `cloud_sql_reference`. If a field in that set is populated, all other fields in the set will automatically be cleared.
+        # @!attribute [rw] cloud_sql_reference
+        #   @return [::Google::Cloud::GeminiDataAnalytics::V1beta::CloudSqlReference]
+        #     A reference to a CloudSQL database.
+        #
+        #     Note: The following fields are mutually exclusive: `cloud_sql_reference`, `bigquery_table_reference`, `studio_datasource_id`, `looker_explore_reference`, `alloy_db_reference`, `spanner_reference`. If a field in that set is populated, all other fields in the set will automatically be cleared.
         # @!attribute [rw] schema
         #   @return [::Google::Cloud::GeminiDataAnalytics::V1beta::Schema]
         #     Optional. The schema of the datasource.

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: google-cloud-gemini_data_analytics-v1beta
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.5.1
 platform: ruby
 authors:
 - Google LLC
@@ -100,6 +100,7 @@ files:
 - lib/google/cloud/gemini_data_analytics/v1beta/data_chat_service/rest/service_stub.rb
 - lib/google/cloud/gemini_data_analytics/v1beta/rest.rb
 - lib/google/cloud/gemini_data_analytics/v1beta/version.rb
+- lib/google/cloud/geminidataanalytics/v1beta/agent_context_pb.rb
 - lib/google/cloud/geminidataanalytics/v1beta/context_pb.rb
 - lib/google/cloud/geminidataanalytics/v1beta/conversation_pb.rb
 - lib/google/cloud/geminidataanalytics/v1beta/credentials_pb.rb
@@ -116,6 +117,7 @@ files:
 - proto_docs/google/api/field_info.rb
 - proto_docs/google/api/launch_stage.rb
 - proto_docs/google/api/resource.rb
+- proto_docs/google/cloud/geminidataanalytics/v1beta/agent_context.rb
 - proto_docs/google/cloud/geminidataanalytics/v1beta/context.rb
 - proto_docs/google/cloud/geminidataanalytics/v1beta/conversation.rb
 - proto_docs/google/cloud/geminidataanalytics/v1beta/credentials.rb