google-cloud-gemini_data_analytics-v1beta 0.3.1 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 55b039e1e4274fd79d3b0a220e590ec4c266b03fe8ce954432f93bb57bd4b88a
-  data.tar.gz: 5dc82a92dc257827cc5a5bbd6464c6a8d43581b26f95e5cafab5a756d5e1b879
+  metadata.gz: 1a9447d0889a80ad09f1986cb668dd16ca1c6a08a6a9ded2df2693de58a0174e
+  data.tar.gz: babc6a47fa8934dcdb6f7d76affc696f8096b2dd4db393dc51dd496bb874f956
 SHA512:
-  metadata.gz: 0dc015e761e373a10da1118be13fe1a6661abca3e8f22842ba03ee7683689ff6294db049bcb91a1c090c611036178b643d560c6e941bd9287f78cef5aceaef38
-  data.tar.gz: 47a48b4d5d0e21fbbe2db7a796e57d8121d48fe3216486fa0e419920e196bcf4ea4e8990839d920195fa8b978f808b04656fac8d207075110e259c59d6ec1136
+  metadata.gz: 4764d0385193c5b52f82dd5827ba5ea30059a8b0db054902e540c8721098cae325942eaec9e240e21e7c5cc82f5187c5f73231dd8cfd31e59691fc323e1972bd
+  data.tar.gz: 11aab4e5c8d49b5540b1a7bcbed510be1c959f8503160e328da290c36f272060465d5f0c4fae3848a25239298be60344d7edc01bcfdafb3ec0f4520afad34f7f

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

@@ -71,7 +71,7 @@ module Google
                                 end
                 default_config = Client::Configuration.new parent_config
 
-                default_config.timeout = 60.0
+                default_config.timeout = 600.0
                 default_config.retry_policy = {
                   initial_delay: 1.0, max_delay: 10.0, multiplier: 1.3, retry_codes: [14]
                 }

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

@@ -73,7 +73,7 @@ module Google
                                   end
                   default_config = Client::Configuration.new parent_config
 
-                  default_config.timeout = 60.0
+                  default_config.timeout = 600.0
                   default_config.retry_policy = {
                     initial_delay: 1.0, max_delay: 10.0, multiplier: 1.3, retry_codes: [14]
                   }

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

@@ -73,7 +73,7 @@ module Google
                                 end
                 default_config = Client::Configuration.new parent_config
 
-                default_config.timeout = 60.0
+                default_config.timeout = 600.0
                 default_config.retry_policy = {
                   initial_delay: 1.0, max_delay: 10.0, multiplier: 1.3, retry_codes: [14]
                 }
@@ -220,7 +220,7 @@ module Google
             #   @param options [::Gapic::CallOptions, ::Hash]
             #     Overrides the default settings for this call, e.g, timeout, retries, etc. Optional.
             #
-            # @overload chat(inline_context: nil, conversation_reference: nil, data_agent_context: nil, project: nil, parent: nil, messages: nil)
+            # @overload chat(inline_context: nil, conversation_reference: nil, data_agent_context: nil, client_managed_resource_context: nil, project: nil, parent: nil, messages: nil)
             #   Pass arguments to `chat` 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).
@@ -230,17 +230,24 @@ module Google
             #     statelessly (without managed conversation persistence and without an
             #     Agent) by passing all context inline.
             #
-            #     Note: The following parameters are mutually exclusive: `inline_context`, `conversation_reference`, `data_agent_context`. At most one of these parameters can be set. If more than one is set, only one will be used, and it is not defined which one.
+            #     Note: The following parameters are mutually exclusive: `inline_context`, `conversation_reference`, `data_agent_context`, `client_managed_resource_context`. At most one of these parameters can be set. If more than one is set, only one will be used, and it is not defined which one.
             #   @param conversation_reference [::Google::Cloud::GeminiDataAnalytics::V1beta::ConversationReference, ::Hash]
             #     Optional. Reference to a persisted conversation and agent context.
             #     Use this to chat with an Agent using managed conversation persistence.
             #
-            #     Note: The following parameters are mutually exclusive: `conversation_reference`, `inline_context`, `data_agent_context`. At most one of these parameters can be set. If more than one is set, only one will be used, and it is not defined which one.
+            #     Note: The following parameters are mutually exclusive: `conversation_reference`, `inline_context`, `data_agent_context`, `client_managed_resource_context`. At most one of these parameters can be set. If more than one is set, only one will be used, and it is not defined which one.
             #   @param data_agent_context [::Google::Cloud::GeminiDataAnalytics::V1beta::DataAgentContext, ::Hash]
             #     Optional. Context for the chat request. Use this to chat with an Agent
             #     statelessly, without managed conversation persistence.
             #
-            #     Note: The following parameters are mutually exclusive: `data_agent_context`, `inline_context`, `conversation_reference`. At most one of these parameters can be set. If more than one is set, only one will be used, and it is not defined which one.
+            #     Note: The following parameters are mutually exclusive: `data_agent_context`, `inline_context`, `conversation_reference`, `client_managed_resource_context`. At most one of these parameters can be set. If more than one is set, only one will be used, and it is not defined which one.
+            #   @param client_managed_resource_context [::Google::Cloud::GeminiDataAnalytics::V1beta::ClientManagedResourceContext, ::Hash]
+            #     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.
+            #
+            #     Note: The following parameters are mutually exclusive: `client_managed_resource_context`, `inline_context`, `conversation_reference`, `data_agent_context`. At most one of these parameters can be set. If more than one is set, only one will be used, and it is not defined which one.
             #   @param project [::String]
             #     Optional. The Google Cloud project to be used for quota and billing.
             #   @param parent [::String]
@@ -415,6 +422,93 @@ module Google
               raise ::Google::Cloud::Error.from_error(e)
             end
 
+            ##
+            # Deletes a conversation.
+            #
+            # @overload delete_conversation(request, options = nil)
+            #   Pass arguments to `delete_conversation` via a request object, either of type
+            #   {::Google::Cloud::GeminiDataAnalytics::V1beta::DeleteConversationRequest} or an equivalent Hash.
+            #
+            #   @param request [::Google::Cloud::GeminiDataAnalytics::V1beta::DeleteConversationRequest, ::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 delete_conversation(name: nil)
+            #   Pass arguments to `delete_conversation` 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 name [::String]
+            #     Required. Name of the resource.
+            #     Format:
+            #     `projects/{project}/locations/{location}/conversations/{conversation}`
+            #
+            # @yield [response, operation] Access the result along with the RPC operation
+            # @yieldparam response [::Google::Protobuf::Empty]
+            # @yieldparam operation [::GRPC::ActiveCall::Operation]
+            #
+            # @return [::Google::Protobuf::Empty]
+            #
+            # @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::DeleteConversationRequest.new
+            #
+            #   # Call the delete_conversation method.
+            #   result = client.delete_conversation request
+            #
+            #   # The returned object is of type Google::Protobuf::Empty.
+            #   p result
+            #
+            def delete_conversation request, options = nil
+              raise ::ArgumentError, "request must be provided" if request.nil?
+
+              request = ::Gapic::Protobuf.coerce request, to: ::Google::Cloud::GeminiDataAnalytics::V1beta::DeleteConversationRequest
+
+              # 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.delete_conversation.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.name
+                header_params["name"] = request.name
+              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.delete_conversation.timeout,
+                                     metadata:     metadata,
+                                     retry_policy: @config.rpcs.delete_conversation.retry_policy
+
+              options.apply_defaults timeout:      @config.timeout,
+                                     metadata:     @config.metadata,
+                                     retry_policy: @config.retry_policy
+
+              @data_chat_service_stub.call_rpc :delete_conversation, 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
+
             ##
             # Gets details of a single conversation by using conversation id and parent.
             #
@@ -532,7 +626,7 @@ module Google
             #   @param filter [::String]
             #     Optional. Returned conversations will match criteria specified within the
             #     filter. ListConversations allows filtering by:
-            #      * agent_id
+            #      * agents
             #      * labels
             #
             # @yield [response, operation] Access the result along with the RPC operation
@@ -711,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.
             #
@@ -905,6 +1093,11 @@ module Google
                 #
                 attr_reader :create_conversation
                 ##
+                # RPC-specific configuration for `delete_conversation`
+                # @return [::Gapic::Config::Method]
+                #
+                attr_reader :delete_conversation
+                ##
                 # RPC-specific configuration for `get_conversation`
                 # @return [::Gapic::Config::Method]
                 #
@@ -919,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
@@ -926,12 +1124,16 @@ module Google
                   @chat = ::Gapic::Config::Method.new chat_config
                   create_conversation_config = parent_rpcs.create_conversation if parent_rpcs.respond_to? :create_conversation
                   @create_conversation = ::Gapic::Config::Method.new create_conversation_config
+                  delete_conversation_config = parent_rpcs.delete_conversation if parent_rpcs.respond_to? :delete_conversation
+                  @delete_conversation = ::Gapic::Config::Method.new delete_conversation_config
                   get_conversation_config = parent_rpcs.get_conversation if parent_rpcs.respond_to? :get_conversation
                   @get_conversation = ::Gapic::Config::Method.new get_conversation_config
                   list_conversations_config = parent_rpcs.list_conversations if parent_rpcs.respond_to? :list_conversations
                   @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

@@ -75,7 +75,7 @@ module Google
                                   end
                   default_config = Client::Configuration.new parent_config
 
-                  default_config.timeout = 60.0
+                  default_config.timeout = 600.0
                   default_config.retry_policy = {
                     initial_delay: 1.0, max_delay: 10.0, multiplier: 1.3, retry_codes: [14]
                   }
@@ -214,7 +214,7 @@ module Google
               #   @param options [::Gapic::CallOptions, ::Hash]
               #     Overrides the default settings for this call, e.g, timeout, retries etc. Optional.
               #
-              # @overload chat(inline_context: nil, conversation_reference: nil, data_agent_context: nil, project: nil, parent: nil, messages: nil)
+              # @overload chat(inline_context: nil, conversation_reference: nil, data_agent_context: nil, client_managed_resource_context: nil, project: nil, parent: nil, messages: nil)
               #   Pass arguments to `chat` 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).
@@ -224,17 +224,24 @@ module Google
               #     statelessly (without managed conversation persistence and without an
               #     Agent) by passing all context inline.
               #
-              #     Note: The following parameters are mutually exclusive: `inline_context`, `conversation_reference`, `data_agent_context`. At most one of these parameters can be set. If more than one is set, only one will be used, and it is not defined which one.
+              #     Note: The following parameters are mutually exclusive: `inline_context`, `conversation_reference`, `data_agent_context`, `client_managed_resource_context`. At most one of these parameters can be set. If more than one is set, only one will be used, and it is not defined which one.
               #   @param conversation_reference [::Google::Cloud::GeminiDataAnalytics::V1beta::ConversationReference, ::Hash]
               #     Optional. Reference to a persisted conversation and agent context.
               #     Use this to chat with an Agent using managed conversation persistence.
               #
-              #     Note: The following parameters are mutually exclusive: `conversation_reference`, `inline_context`, `data_agent_context`. At most one of these parameters can be set. If more than one is set, only one will be used, and it is not defined which one.
+              #     Note: The following parameters are mutually exclusive: `conversation_reference`, `inline_context`, `data_agent_context`, `client_managed_resource_context`. At most one of these parameters can be set. If more than one is set, only one will be used, and it is not defined which one.
               #   @param data_agent_context [::Google::Cloud::GeminiDataAnalytics::V1beta::DataAgentContext, ::Hash]
               #     Optional. Context for the chat request. Use this to chat with an Agent
               #     statelessly, without managed conversation persistence.
               #
-              #     Note: The following parameters are mutually exclusive: `data_agent_context`, `inline_context`, `conversation_reference`. At most one of these parameters can be set. If more than one is set, only one will be used, and it is not defined which one.
+              #     Note: The following parameters are mutually exclusive: `data_agent_context`, `inline_context`, `conversation_reference`, `client_managed_resource_context`. At most one of these parameters can be set. If more than one is set, only one will be used, and it is not defined which one.
+              #   @param client_managed_resource_context [::Google::Cloud::GeminiDataAnalytics::V1beta::ClientManagedResourceContext, ::Hash]
+              #     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.
+              #
+              #     Note: The following parameters are mutually exclusive: `client_managed_resource_context`, `inline_context`, `conversation_reference`, `data_agent_context`. At most one of these parameters can be set. If more than one is set, only one will be used, and it is not defined which one.
               #   @param project [::String]
               #     Optional. The Google Cloud project to be used for quota and billing.
               #   @param parent [::String]
@@ -397,6 +404,86 @@ module Google
                 raise ::Google::Cloud::Error.from_error(e)
               end
 
+              ##
+              # Deletes a conversation.
+              #
+              # @overload delete_conversation(request, options = nil)
+              #   Pass arguments to `delete_conversation` via a request object, either of type
+              #   {::Google::Cloud::GeminiDataAnalytics::V1beta::DeleteConversationRequest} or an equivalent Hash.
+              #
+              #   @param request [::Google::Cloud::GeminiDataAnalytics::V1beta::DeleteConversationRequest, ::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 delete_conversation(name: nil)
+              #   Pass arguments to `delete_conversation` 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 name [::String]
+              #     Required. Name of the resource.
+              #     Format:
+              #     `projects/{project}/locations/{location}/conversations/{conversation}`
+              # @yield [result, operation] Access the result along with the TransportOperation object
+              # @yieldparam result [::Google::Protobuf::Empty]
+              # @yieldparam operation [::Gapic::Rest::TransportOperation]
+              #
+              # @return [::Google::Protobuf::Empty]
+              #
+              # @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::DeleteConversationRequest.new
+              #
+              #   # Call the delete_conversation method.
+              #   result = client.delete_conversation request
+              #
+              #   # The returned object is of type Google::Protobuf::Empty.
+              #   p result
+              #
+              def delete_conversation request, options = nil
+                raise ::ArgumentError, "request must be provided" if request.nil?
+
+                request = ::Gapic::Protobuf.coerce request, to: ::Google::Cloud::GeminiDataAnalytics::V1beta::DeleteConversationRequest
+
+                # 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.delete_conversation.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.delete_conversation.timeout,
+                                       metadata:     call_metadata,
+                                       retry_policy: @config.rpcs.delete_conversation.retry_policy
+
+                options.apply_defaults timeout:      @config.timeout,
+                                       metadata:     @config.metadata,
+                                       retry_policy: @config.retry_policy
+
+                @data_chat_service_stub.delete_conversation 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
+
               ##
               # Gets details of a single conversation by using conversation id and parent.
               #
@@ -507,7 +594,7 @@ module Google
               #   @param filter [::String]
               #     Optional. Returned conversations will match criteria specified within the
               #     filter. ListConversations allows filtering by:
-              #      * agent_id
+              #      * agents
               #      * labels
               # @yield [result, operation] Access the result along with the TransportOperation object
               # @yieldparam result [::Gapic::Rest::PagedEnumerable<::Google::Cloud::GeminiDataAnalytics::V1beta::Conversation>]
@@ -672,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.
               #
@@ -836,6 +1010,11 @@ module Google
                   #
                   attr_reader :create_conversation
                   ##
+                  # RPC-specific configuration for `delete_conversation`
+                  # @return [::Gapic::Config::Method]
+                  #
+                  attr_reader :delete_conversation
+                  ##
                   # RPC-specific configuration for `get_conversation`
                   # @return [::Gapic::Config::Method]
                   #
@@ -850,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
@@ -857,12 +1041,16 @@ module Google
                     @chat = ::Gapic::Config::Method.new chat_config
                     create_conversation_config = parent_rpcs.create_conversation if parent_rpcs.respond_to? :create_conversation
                     @create_conversation = ::Gapic::Config::Method.new create_conversation_config
+                    delete_conversation_config = parent_rpcs.delete_conversation if parent_rpcs.respond_to? :delete_conversation
+                    @delete_conversation = ::Gapic::Config::Method.new delete_conversation_config
                     get_conversation_config = parent_rpcs.get_conversation if parent_rpcs.respond_to? :get_conversation
                     @get_conversation = ::Gapic::Config::Method.new get_conversation_config
                     list_conversations_config = parent_rpcs.list_conversations if parent_rpcs.respond_to? :list_conversations
                     @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