trieve_ruby_client 0.8.0 → 0.10.7

data/lib/trieve_ruby_client/api/chunk_api.rb

@@ -3,7 +3,7 @@
 
 #Trieve OpenAPI Specification. This document describes all of the operations available through the Trieve API.
 
-The version of the OpenAPI document: 0.8.0
+The version of the OpenAPI document: 0.10.7
 Contact: developers@trieve.ai
 Generated by: https://openapi-generator.tech
 Generator version: 7.4.0
@@ -19,37 +19,37 @@ module TrieveRubyClient
     def initialize(api_client = ApiClient.default)
       @api_client = api_client
     end
-    # Create or Upsert Chunk or Chunks
-    # Create or Upsert Chunk or Chunks  Create a new chunk. If the chunk has the same tracking_id as an existing chunk, the request will fail. Once a chunk is created, it can be searched for using the search endpoint. If uploading in bulk, the maximum amount of chunks that can be uploaded at once is 120 chunks.
+    # Autocomplete
+    # Autocomplete  This route provides the primary autocomplete functionality for the API. This prioritize prefix matching with semantic or full-text search.
     # @param tr_dataset [String] The dataset id to use for the request
-    # @param create_chunk_data [CreateChunkData] JSON request payload to create a new chunk (chunk)
+    # @param autocomplete_req_payload [AutocompleteReqPayload] JSON request payload to semantically search for chunks (chunks)
     # @param [Hash] opts the optional parameters
-    # @return [ReturnQueuedChunk]
-    def create_chunk(tr_dataset, create_chunk_data, opts = {})
-      data, _status_code, _headers = create_chunk_with_http_info(tr_dataset, create_chunk_data, opts)
+    # @return [SearchChunkQueryResponseBody]
+    def autocomplete(tr_dataset, autocomplete_req_payload, opts = {})
+      data, _status_code, _headers = autocomplete_with_http_info(tr_dataset, autocomplete_req_payload, opts)
       data
     end
 
-    # Create or Upsert Chunk or Chunks
-    # Create or Upsert Chunk or Chunks  Create a new chunk. If the chunk has the same tracking_id as an existing chunk, the request will fail. Once a chunk is created, it can be searched for using the search endpoint. If uploading in bulk, the maximum amount of chunks that can be uploaded at once is 120 chunks.
+    # Autocomplete
+    # Autocomplete  This route provides the primary autocomplete functionality for the API. This prioritize prefix matching with semantic or full-text search.
     # @param tr_dataset [String] The dataset id to use for the request
-    # @param create_chunk_data [CreateChunkData] JSON request payload to create a new chunk (chunk)
+    # @param autocomplete_req_payload [AutocompleteReqPayload] JSON request payload to semantically search for chunks (chunks)
     # @param [Hash] opts the optional parameters
-    # @return [Array<(ReturnQueuedChunk, Integer, Hash)>] ReturnQueuedChunk data, response status code and response headers
-    def create_chunk_with_http_info(tr_dataset, create_chunk_data, opts = {})
+    # @return [Array<(SearchChunkQueryResponseBody, Integer, Hash)>] SearchChunkQueryResponseBody data, response status code and response headers
+    def autocomplete_with_http_info(tr_dataset, autocomplete_req_payload, opts = {})
       if @api_client.config.debugging
-        @api_client.config.logger.debug 'Calling API: ChunkApi.create_chunk ...'
+        @api_client.config.logger.debug 'Calling API: ChunkApi.autocomplete ...'
       end
       # verify the required parameter 'tr_dataset' is set
       if @api_client.config.client_side_validation && tr_dataset.nil?
-        fail ArgumentError, "Missing the required parameter 'tr_dataset' when calling ChunkApi.create_chunk"
+        fail ArgumentError, "Missing the required parameter 'tr_dataset' when calling ChunkApi.autocomplete"
       end
-      # verify the required parameter 'create_chunk_data' is set
-      if @api_client.config.client_side_validation && create_chunk_data.nil?
-        fail ArgumentError, "Missing the required parameter 'create_chunk_data' when calling ChunkApi.create_chunk"
+      # verify the required parameter 'autocomplete_req_payload' is set
+      if @api_client.config.client_side_validation && autocomplete_req_payload.nil?
+        fail ArgumentError, "Missing the required parameter 'autocomplete_req_payload' when calling ChunkApi.autocomplete"
       end
       # resource path
-      local_var_path = '/api/chunk'
+      local_var_path = '/api/chunk/autocomplete'
 
       # query parameters
       query_params = opts[:query_params] || {}
@@ -69,16 +69,16 @@ module TrieveRubyClient
       form_params = opts[:form_params] || {}
 
       # http body (model)
-      post_body = opts[:debug_body] || @api_client.object_to_http_body(create_chunk_data)
+      post_body = opts[:debug_body] || @api_client.object_to_http_body(autocomplete_req_payload)
 
       # return_type
-      return_type = opts[:debug_return_type] || 'ReturnQueuedChunk'
+      return_type = opts[:debug_return_type] || 'SearchChunkQueryResponseBody'
 
       # auth_names
       auth_names = opts[:debug_auth_names] || ['ApiKey']
 
       new_options = opts.merge(
-        :operation => :"ChunkApi.create_chunk",
+        :operation => :"ChunkApi.autocomplete",
         :header_params => header_params,
         :query_params => query_params,
         :form_params => form_params,
@@ -89,42 +89,42 @@ module TrieveRubyClient
 
       data, status_code, headers = @api_client.call_api(:POST, local_var_path, new_options)
       if @api_client.config.debugging
-        @api_client.config.logger.debug "API called: ChunkApi#create_chunk\nData: #{data.inspect}\nStatus code: #{status_code}\nHeaders: #{headers}"
+        @api_client.config.logger.debug "API called: ChunkApi#autocomplete\nData: #{data.inspect}\nStatus code: #{status_code}\nHeaders: #{headers}"
       end
       return data, status_code, headers
     end
 
-    # Generate suggested queries
-    # Generate suggested queries  This endpoint will generate 3 suggested queries based off the query provided in the request body and return them as a JSON object.
+    # Create or Upsert Chunk or Chunks
+    # Create or Upsert Chunk or Chunks  Create a new chunk. If the chunk has the same tracking_id as an existing chunk, the request will fail. Once a chunk is created, it can be searched for using the search endpoint. If uploading in bulk, the maximum amount of chunks that can be uploaded at once is 120 chunks. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
     # @param tr_dataset [String] The dataset id to use for the request
-    # @param suggested_queries_request [SuggestedQueriesRequest] JSON request payload to get alternative suggested queries
+    # @param create_chunk_req_payload_enum [CreateChunkReqPayloadEnum] JSON request payload to create a new chunk (chunk)
     # @param [Hash] opts the optional parameters
-    # @return [SuggestedQueriesResponse]
-    def create_suggested_queries_handler(tr_dataset, suggested_queries_request, opts = {})
-      data, _status_code, _headers = create_suggested_queries_handler_with_http_info(tr_dataset, suggested_queries_request, opts)
+    # @return [ReturnQueuedChunk]
+    def create_chunk(tr_dataset, create_chunk_req_payload_enum, opts = {})
+      data, _status_code, _headers = create_chunk_with_http_info(tr_dataset, create_chunk_req_payload_enum, opts)
       data
     end
 
-    # Generate suggested queries
-    # Generate suggested queries  This endpoint will generate 3 suggested queries based off the query provided in the request body and return them as a JSON object.
+    # Create or Upsert Chunk or Chunks
+    # Create or Upsert Chunk or Chunks  Create a new chunk. If the chunk has the same tracking_id as an existing chunk, the request will fail. Once a chunk is created, it can be searched for using the search endpoint. If uploading in bulk, the maximum amount of chunks that can be uploaded at once is 120 chunks. Auth&#39;ed user or api key must have an admin or owner role for the specified dataset&#39;s organization.
     # @param tr_dataset [String] The dataset id to use for the request
-    # @param suggested_queries_request [SuggestedQueriesRequest] JSON request payload to get alternative suggested queries
+    # @param create_chunk_req_payload_enum [CreateChunkReqPayloadEnum] JSON request payload to create a new chunk (chunk)
     # @param [Hash] opts the optional parameters
-    # @return [Array<(SuggestedQueriesResponse, Integer, Hash)>] SuggestedQueriesResponse data, response status code and response headers
-    def create_suggested_queries_handler_with_http_info(tr_dataset, suggested_queries_request, opts = {})
+    # @return [Array<(ReturnQueuedChunk, Integer, Hash)>] ReturnQueuedChunk data, response status code and response headers
+    def create_chunk_with_http_info(tr_dataset, create_chunk_req_payload_enum, opts = {})
       if @api_client.config.debugging
-        @api_client.config.logger.debug 'Calling API: ChunkApi.create_suggested_queries_handler ...'
+        @api_client.config.logger.debug 'Calling API: ChunkApi.create_chunk ...'
       end
       # verify the required parameter 'tr_dataset' is set
       if @api_client.config.client_side_validation && tr_dataset.nil?
-        fail ArgumentError, "Missing the required parameter 'tr_dataset' when calling ChunkApi.create_suggested_queries_handler"
+        fail ArgumentError, "Missing the required parameter 'tr_dataset' when calling ChunkApi.create_chunk"
       end
-      # verify the required parameter 'suggested_queries_request' is set
-      if @api_client.config.client_side_validation && suggested_queries_request.nil?
-        fail ArgumentError, "Missing the required parameter 'suggested_queries_request' when calling ChunkApi.create_suggested_queries_handler"
+      # verify the required parameter 'create_chunk_req_payload_enum' is set
+      if @api_client.config.client_side_validation && create_chunk_req_payload_enum.nil?
+        fail ArgumentError, "Missing the required parameter 'create_chunk_req_payload_enum' when calling ChunkApi.create_chunk"
       end
       # resource path
-      local_var_path = '/api/chunk/gen_suggestions'
+      local_var_path = '/api/chunk'
 
       # query parameters
       query_params = opts[:query_params] || {}
@@ -144,16 +144,16 @@ module TrieveRubyClient
       form_params = opts[:form_params] || {}
 
       # http body (model)
-      post_body = opts[:debug_body] || @api_client.object_to_http_body(suggested_queries_request)
+      post_body = opts[:debug_body] || @api_client.object_to_http_body(create_chunk_req_payload_enum)
 
       # return_type
-      return_type = opts[:debug_return_type] || 'SuggestedQueriesResponse'
+      return_type = opts[:debug_return_type] || 'ReturnQueuedChunk'
 
       # auth_names
       auth_names = opts[:debug_auth_names] || ['ApiKey']
 
       new_options = opts.merge(
-        :operation => :"ChunkApi.create_suggested_queries_handler",
+        :operation => :"ChunkApi.create_chunk",
         :header_params => header_params,
         :query_params => query_params,
         :form_params => form_params,
@@ -164,13 +164,13 @@ module TrieveRubyClient
 
       data, status_code, headers = @api_client.call_api(:POST, local_var_path, new_options)
       if @api_client.config.debugging
-        @api_client.config.logger.debug "API called: ChunkApi#create_suggested_queries_handler\nData: #{data.inspect}\nStatus code: #{status_code}\nHeaders: #{headers}"
+        @api_client.config.logger.debug "API called: ChunkApi#create_chunk\nData: #{data.inspect}\nStatus code: #{status_code}\nHeaders: #{headers}"
       end
       return data, status_code, headers
     end
 
     # Delete Chunk
-    # Delete Chunk  Delete a chunk by its id. If deleting a root chunk which has a collision, the most recently created collision will become a new root chunk.
+    # Delete Chunk  Delete a chunk by its id. If deleting a root chunk which has a collision, the most recently created collision will become a new root chunk. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
     # @param tr_dataset [String] The dataset id to use for the request
     # @param chunk_id [String] Id of the chunk you want to fetch.
     # @param [Hash] opts the optional parameters
@@ -181,7 +181,7 @@ module TrieveRubyClient
     end
 
     # Delete Chunk
-    # Delete Chunk  Delete a chunk by its id. If deleting a root chunk which has a collision, the most recently created collision will become a new root chunk.
+    # Delete Chunk  Delete a chunk by its id. If deleting a root chunk which has a collision, the most recently created collision will become a new root chunk. Auth&#39;ed user or api key must have an admin or owner role for the specified dataset&#39;s organization.
     # @param tr_dataset [String] The dataset id to use for the request
     # @param chunk_id [String] Id of the chunk you want to fetch.
     # @param [Hash] opts the optional parameters
@@ -240,7 +240,7 @@ module TrieveRubyClient
     end
 
     # Delete Chunk By Tracking Id
-    # Delete Chunk By Tracking Id  Delete a chunk by tracking_id. This is useful for when you are coordinating with an external system and want to use the tracking_id to identify the chunk. If deleting a root chunk which has a collision, the most recently created collision will become a new root chunk.
+    # Delete Chunk By Tracking Id  Delete a chunk by tracking_id. This is useful for when you are coordinating with an external system and want to use the tracking_id to identify the chunk. If deleting a root chunk which has a collision, the most recently created collision will become a new root chunk. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
     # @param tr_dataset [String] The dataset id to use for the request
     # @param tracking_id [String] tracking_id of the chunk you want to delete
     # @param [Hash] opts the optional parameters
@@ -251,7 +251,7 @@ module TrieveRubyClient
     end
 
     # Delete Chunk By Tracking Id
-    # Delete Chunk By Tracking Id  Delete a chunk by tracking_id. This is useful for when you are coordinating with an external system and want to use the tracking_id to identify the chunk. If deleting a root chunk which has a collision, the most recently created collision will become a new root chunk.
+    # Delete Chunk By Tracking Id  Delete a chunk by tracking_id. This is useful for when you are coordinating with an external system and want to use the tracking_id to identify the chunk. If deleting a root chunk which has a collision, the most recently created collision will become a new root chunk. Auth&#39;ed user or api key must have an admin or owner role for the specified dataset&#39;s organization.
     # @param tr_dataset [String] The dataset id to use for the request
     # @param tracking_id [String] tracking_id of the chunk you want to delete
     # @param [Hash] opts the optional parameters
@@ -389,7 +389,7 @@ module TrieveRubyClient
     # @param tr_dataset [String] The dataset id to use for the request
     # @param chunk_id [String] Id of the chunk you want to fetch.
     # @param [Hash] opts the optional parameters
-    # @return [ChunkMetadata]
+    # @return [ChunkMetadataStringTagSet]
     def get_chunk_by_id(tr_dataset, chunk_id, opts = {})
       data, _status_code, _headers = get_chunk_by_id_with_http_info(tr_dataset, chunk_id, opts)
       data
@@ -400,7 +400,7 @@ module TrieveRubyClient
     # @param tr_dataset [String] The dataset id to use for the request
     # @param chunk_id [String] Id of the chunk you want to fetch.
     # @param [Hash] opts the optional parameters
-    # @return [Array<(ChunkMetadata, Integer, Hash)>] ChunkMetadata data, response status code and response headers
+    # @return [Array<(ChunkMetadataStringTagSet, Integer, Hash)>] ChunkMetadataStringTagSet data, response status code and response headers
     def get_chunk_by_id_with_http_info(tr_dataset, chunk_id, opts = {})
       if @api_client.config.debugging
         @api_client.config.logger.debug 'Calling API: ChunkApi.get_chunk_by_id ...'
@@ -432,7 +432,7 @@ module TrieveRubyClient
       post_body = opts[:debug_body]
 
       # return_type
-      return_type = opts[:debug_return_type] || 'ChunkMetadata'
+      return_type = opts[:debug_return_type] || 'ChunkMetadataStringTagSet'
 
       # auth_names
       auth_names = opts[:debug_auth_names] || ['ApiKey']
@@ -459,7 +459,7 @@ module TrieveRubyClient
     # @param tr_dataset [String] The dataset id to use for the request
     # @param tracking_id [String] tracking_id of the chunk you want to fetch
     # @param [Hash] opts the optional parameters
-    # @return [ChunkMetadata]
+    # @return [ChunkMetadataStringTagSet]
     def get_chunk_by_tracking_id(tr_dataset, tracking_id, opts = {})
       data, _status_code, _headers = get_chunk_by_tracking_id_with_http_info(tr_dataset, tracking_id, opts)
       data
@@ -470,7 +470,7 @@ module TrieveRubyClient
     # @param tr_dataset [String] The dataset id to use for the request
     # @param tracking_id [String] tracking_id of the chunk you want to fetch
     # @param [Hash] opts the optional parameters
-    # @return [Array<(ChunkMetadata, Integer, Hash)>] ChunkMetadata data, response status code and response headers
+    # @return [Array<(ChunkMetadataStringTagSet, Integer, Hash)>] ChunkMetadataStringTagSet data, response status code and response headers
     def get_chunk_by_tracking_id_with_http_info(tr_dataset, tracking_id, opts = {})
       if @api_client.config.debugging
         @api_client.config.logger.debug 'Calling API: ChunkApi.get_chunk_by_tracking_id ...'
@@ -502,7 +502,7 @@ module TrieveRubyClient
       post_body = opts[:debug_body]
 
       # return_type
-      return_type = opts[:debug_return_type] || 'ChunkMetadata'
+      return_type = opts[:debug_return_type] || 'ChunkMetadataStringTagSet'
 
       # auth_names
       auth_names = opts[:debug_auth_names] || ['ApiKey']
@@ -529,7 +529,7 @@ module TrieveRubyClient
     # @param tr_dataset [String] The dataset id to use for the request
     # @param get_chunks_data [GetChunksData] JSON request payload to get the chunks in the request
     # @param [Hash] opts the optional parameters
-    # @return [Array<ChunkMetadata>]
+    # @return [Array<ChunkMetadataStringTagSet>]
     def get_chunks_by_ids(tr_dataset, get_chunks_data, opts = {})
       data, _status_code, _headers = get_chunks_by_ids_with_http_info(tr_dataset, get_chunks_data, opts)
       data
@@ -540,7 +540,7 @@ module TrieveRubyClient
     # @param tr_dataset [String] The dataset id to use for the request
     # @param get_chunks_data [GetChunksData] JSON request payload to get the chunks in the request
     # @param [Hash] opts the optional parameters
-    # @return [Array<(Array<ChunkMetadata>, Integer, Hash)>] Array<ChunkMetadata> data, response status code and response headers
+    # @return [Array<(Array<ChunkMetadataStringTagSet>, Integer, Hash)>] Array<ChunkMetadataStringTagSet> data, response status code and response headers
     def get_chunks_by_ids_with_http_info(tr_dataset, get_chunks_data, opts = {})
       if @api_client.config.debugging
         @api_client.config.logger.debug 'Calling API: ChunkApi.get_chunks_by_ids ...'
@@ -577,7 +577,7 @@ module TrieveRubyClient
       post_body = opts[:debug_body] || @api_client.object_to_http_body(get_chunks_data)
 
       # return_type
-      return_type = opts[:debug_return_type] || 'Array<ChunkMetadata>'
+      return_type = opts[:debug_return_type] || 'Array<ChunkMetadataStringTagSet>'
 
       # auth_names
       auth_names = opts[:debug_auth_names] || ['ApiKey']
@@ -604,7 +604,7 @@ module TrieveRubyClient
     # @param tr_dataset [String] The dataset id to use for the request
     # @param get_tracking_chunks_data [GetTrackingChunksData] JSON request payload to get the chunks in the request
     # @param [Hash] opts the optional parameters
-    # @return [ChunkMetadata]
+    # @return [Array<ChunkMetadataStringTagSet>]
     def get_chunks_by_tracking_ids(tr_dataset, get_tracking_chunks_data, opts = {})
       data, _status_code, _headers = get_chunks_by_tracking_ids_with_http_info(tr_dataset, get_tracking_chunks_data, opts)
       data
@@ -615,7 +615,7 @@ module TrieveRubyClient
     # @param tr_dataset [String] The dataset id to use for the request
     # @param get_tracking_chunks_data [GetTrackingChunksData] JSON request payload to get the chunks in the request
     # @param [Hash] opts the optional parameters
-    # @return [Array<(ChunkMetadata, Integer, Hash)>] ChunkMetadata data, response status code and response headers
+    # @return [Array<(Array<ChunkMetadataStringTagSet>, Integer, Hash)>] Array<ChunkMetadataStringTagSet> data, response status code and response headers
     def get_chunks_by_tracking_ids_with_http_info(tr_dataset, get_tracking_chunks_data, opts = {})
       if @api_client.config.debugging
         @api_client.config.logger.debug 'Calling API: ChunkApi.get_chunks_by_tracking_ids ...'
@@ -652,7 +652,7 @@ module TrieveRubyClient
       post_body = opts[:debug_body] || @api_client.object_to_http_body(get_tracking_chunks_data)
 
       # return_type
-      return_type = opts[:debug_return_type] || 'ChunkMetadata'
+      return_type = opts[:debug_return_type] || 'Array<ChunkMetadataStringTagSet>'
 
       # auth_names
       auth_names = opts[:debug_auth_names] || ['ApiKey']
@@ -675,7 +675,7 @@ module TrieveRubyClient
     end
 
     # Get Recommended Chunks
-    # Get Recommended Chunks  Get recommendations of chunks similar to the chunks in the request. Think about this as a feature similar to the \"add to playlist\" recommendation feature on Spotify. This request pairs especially well with our groups endpoint.
+    # Get Recommended Chunks  Get recommendations of chunks similar to the positive samples in the request and dissimilar to the negative. You must provide at least one of either positive_chunk_ids or positive_tracking_ids.
     # @param tr_dataset [String] The dataset id to use for the request
     # @param recommend_chunks_request [RecommendChunksRequest] JSON request payload to get recommendations of chunks similar to the chunks in the request
     # @param [Hash] opts the optional parameters
@@ -686,7 +686,7 @@ module TrieveRubyClient
     end
 
     # Get Recommended Chunks
-    # Get Recommended Chunks  Get recommendations of chunks similar to the chunks in the request. Think about this as a feature similar to the \&quot;add to playlist\&quot; recommendation feature on Spotify. This request pairs especially well with our groups endpoint.
+    # Get Recommended Chunks  Get recommendations of chunks similar to the positive samples in the request and dissimilar to the negative. You must provide at least one of either positive_chunk_ids or positive_tracking_ids.
     # @param tr_dataset [String] The dataset id to use for the request
     # @param recommend_chunks_request [RecommendChunksRequest] JSON request payload to get recommendations of chunks similar to the chunks in the request
     # @param [Hash] opts the optional parameters
@@ -749,24 +749,99 @@ module TrieveRubyClient
       return data, status_code, headers
     end
 
+    # Generate suggested queries
+    # Generate suggested queries  This endpoint will generate 3 suggested queries based off a hybrid search using RAG with the query provided in the request body and return them as a JSON object.
+    # @param tr_dataset [String] The dataset id to use for the request
+    # @param suggested_queries_req_payload [SuggestedQueriesReqPayload] JSON request payload to get alternative suggested queries
+    # @param [Hash] opts the optional parameters
+    # @return [SuggestedQueriesResponse]
+    def get_suggested_queries(tr_dataset, suggested_queries_req_payload, opts = {})
+      data, _status_code, _headers = get_suggested_queries_with_http_info(tr_dataset, suggested_queries_req_payload, opts)
+      data
+    end
+
+    # Generate suggested queries
+    # Generate suggested queries  This endpoint will generate 3 suggested queries based off a hybrid search using RAG with the query provided in the request body and return them as a JSON object.
+    # @param tr_dataset [String] The dataset id to use for the request
+    # @param suggested_queries_req_payload [SuggestedQueriesReqPayload] JSON request payload to get alternative suggested queries
+    # @param [Hash] opts the optional parameters
+    # @return [Array<(SuggestedQueriesResponse, Integer, Hash)>] SuggestedQueriesResponse data, response status code and response headers
+    def get_suggested_queries_with_http_info(tr_dataset, suggested_queries_req_payload, opts = {})
+      if @api_client.config.debugging
+        @api_client.config.logger.debug 'Calling API: ChunkApi.get_suggested_queries ...'
+      end
+      # verify the required parameter 'tr_dataset' is set
+      if @api_client.config.client_side_validation && tr_dataset.nil?
+        fail ArgumentError, "Missing the required parameter 'tr_dataset' when calling ChunkApi.get_suggested_queries"
+      end
+      # verify the required parameter 'suggested_queries_req_payload' is set
+      if @api_client.config.client_side_validation && suggested_queries_req_payload.nil?
+        fail ArgumentError, "Missing the required parameter 'suggested_queries_req_payload' when calling ChunkApi.get_suggested_queries"
+      end
+      # resource path
+      local_var_path = '/api/chunk/gen_suggestions'
+
+      # query parameters
+      query_params = opts[:query_params] || {}
+
+      # header parameters
+      header_params = opts[:header_params] || {}
+      # HTTP header 'Accept' (if needed)
+      header_params['Accept'] = @api_client.select_header_accept(['application/json'])
+      # HTTP header 'Content-Type'
+      content_type = @api_client.select_header_content_type(['application/json'])
+      if !content_type.nil?
+          header_params['Content-Type'] = content_type
+      end
+      header_params[:'TR-Dataset'] = tr_dataset
+
+      # form parameters
+      form_params = opts[:form_params] || {}
+
+      # http body (model)
+      post_body = opts[:debug_body] || @api_client.object_to_http_body(suggested_queries_req_payload)
+
+      # return_type
+      return_type = opts[:debug_return_type] || 'SuggestedQueriesResponse'
+
+      # auth_names
+      auth_names = opts[:debug_auth_names] || ['ApiKey']
+
+      new_options = opts.merge(
+        :operation => :"ChunkApi.get_suggested_queries",
+        :header_params => header_params,
+        :query_params => query_params,
+        :form_params => form_params,
+        :body => post_body,
+        :auth_names => auth_names,
+        :return_type => return_type
+      )
+
+      data, status_code, headers = @api_client.call_api(:POST, local_var_path, new_options)
+      if @api_client.config.debugging
+        @api_client.config.logger.debug "API called: ChunkApi#get_suggested_queries\nData: #{data.inspect}\nStatus code: #{status_code}\nHeaders: #{headers}"
+      end
+      return data, status_code, headers
+    end
+
     # Search
     # Search  This route provides the primary search functionality for the API. It can be used to search for chunks by semantic similarity, full-text similarity, or a combination of both. Results' `chunk_html` values will be modified with `<b><mark>` tags for sub-sentence highlighting.
     # @param tr_dataset [String] The dataset id to use for the request
-    # @param search_chunk_data [SearchChunkData] JSON request payload to semantically search for chunks (chunks)
+    # @param search_chunks_req_payload [SearchChunksReqPayload] JSON request payload to semantically search for chunks (chunks)
     # @param [Hash] opts the optional parameters
     # @return [SearchChunkQueryResponseBody]
-    def search_chunks(tr_dataset, search_chunk_data, opts = {})
-      data, _status_code, _headers = search_chunks_with_http_info(tr_dataset, search_chunk_data, opts)
+    def search_chunks(tr_dataset, search_chunks_req_payload, opts = {})
+      data, _status_code, _headers = search_chunks_with_http_info(tr_dataset, search_chunks_req_payload, opts)
       data
     end
 
     # Search
     # Search  This route provides the primary search functionality for the API. It can be used to search for chunks by semantic similarity, full-text similarity, or a combination of both. Results&#39; &#x60;chunk_html&#x60; values will be modified with &#x60;&lt;b&gt;&lt;mark&gt;&#x60; tags for sub-sentence highlighting.
     # @param tr_dataset [String] The dataset id to use for the request
-    # @param search_chunk_data [SearchChunkData] JSON request payload to semantically search for chunks (chunks)
+    # @param search_chunks_req_payload [SearchChunksReqPayload] JSON request payload to semantically search for chunks (chunks)
     # @param [Hash] opts the optional parameters
     # @return [Array<(SearchChunkQueryResponseBody, Integer, Hash)>] SearchChunkQueryResponseBody data, response status code and response headers
-    def search_chunks_with_http_info(tr_dataset, search_chunk_data, opts = {})
+    def search_chunks_with_http_info(tr_dataset, search_chunks_req_payload, opts = {})
       if @api_client.config.debugging
         @api_client.config.logger.debug 'Calling API: ChunkApi.search_chunks ...'
       end
@@ -774,9 +849,9 @@ module TrieveRubyClient
       if @api_client.config.client_side_validation && tr_dataset.nil?
         fail ArgumentError, "Missing the required parameter 'tr_dataset' when calling ChunkApi.search_chunks"
       end
-      # verify the required parameter 'search_chunk_data' is set
-      if @api_client.config.client_side_validation && search_chunk_data.nil?
-        fail ArgumentError, "Missing the required parameter 'search_chunk_data' when calling ChunkApi.search_chunks"
+      # verify the required parameter 'search_chunks_req_payload' is set
+      if @api_client.config.client_side_validation && search_chunks_req_payload.nil?
+        fail ArgumentError, "Missing the required parameter 'search_chunks_req_payload' when calling ChunkApi.search_chunks"
       end
       # resource path
       local_var_path = '/api/chunk/search'
@@ -799,7 +874,7 @@ module TrieveRubyClient
       form_params = opts[:form_params] || {}
 
       # http body (model)
-      post_body = opts[:debug_body] || @api_client.object_to_http_body(search_chunk_data)
+      post_body = opts[:debug_body] || @api_client.object_to_http_body(search_chunks_req_payload)
 
       # return_type
       return_type = opts[:debug_return_type] || 'SearchChunkQueryResponseBody'
@@ -825,23 +900,23 @@ module TrieveRubyClient
     end
 
     # Update Chunk
-    # Update Chunk  Update a chunk. If you try to change the tracking_id of the chunk to have the same tracking_id as an existing chunk, the request will fail.
+    # Update Chunk  Update a chunk. If you try to change the tracking_id of the chunk to have the same tracking_id as an existing chunk, the request will fail. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
     # @param tr_dataset [String] The dataset id to use for the request
-    # @param update_chunk_data [UpdateChunkData] JSON request payload to update a chunk (chunk)
+    # @param update_chunk_req_payload [UpdateChunkReqPayload] JSON request payload to update a chunk (chunk)
     # @param [Hash] opts the optional parameters
     # @return [nil]
-    def update_chunk(tr_dataset, update_chunk_data, opts = {})
-      update_chunk_with_http_info(tr_dataset, update_chunk_data, opts)
+    def update_chunk(tr_dataset, update_chunk_req_payload, opts = {})
+      update_chunk_with_http_info(tr_dataset, update_chunk_req_payload, opts)
       nil
     end
 
     # Update Chunk
-    # Update Chunk  Update a chunk. If you try to change the tracking_id of the chunk to have the same tracking_id as an existing chunk, the request will fail.
+    # Update Chunk  Update a chunk. If you try to change the tracking_id of the chunk to have the same tracking_id as an existing chunk, the request will fail. Auth&#39;ed user or api key must have an admin or owner role for the specified dataset&#39;s organization.
     # @param tr_dataset [String] The dataset id to use for the request
-    # @param update_chunk_data [UpdateChunkData] JSON request payload to update a chunk (chunk)
+    # @param update_chunk_req_payload [UpdateChunkReqPayload] JSON request payload to update a chunk (chunk)
     # @param [Hash] opts the optional parameters
     # @return [Array<(nil, Integer, Hash)>] nil, response status code and response headers
-    def update_chunk_with_http_info(tr_dataset, update_chunk_data, opts = {})
+    def update_chunk_with_http_info(tr_dataset, update_chunk_req_payload, opts = {})
       if @api_client.config.debugging
         @api_client.config.logger.debug 'Calling API: ChunkApi.update_chunk ...'
       end
@@ -849,9 +924,9 @@ module TrieveRubyClient
       if @api_client.config.client_side_validation && tr_dataset.nil?
         fail ArgumentError, "Missing the required parameter 'tr_dataset' when calling ChunkApi.update_chunk"
       end
-      # verify the required parameter 'update_chunk_data' is set
-      if @api_client.config.client_side_validation && update_chunk_data.nil?
-        fail ArgumentError, "Missing the required parameter 'update_chunk_data' when calling ChunkApi.update_chunk"
+      # verify the required parameter 'update_chunk_req_payload' is set
+      if @api_client.config.client_side_validation && update_chunk_req_payload.nil?
+        fail ArgumentError, "Missing the required parameter 'update_chunk_req_payload' when calling ChunkApi.update_chunk"
       end
       # resource path
       local_var_path = '/api/chunk'
@@ -874,7 +949,7 @@ module TrieveRubyClient
       form_params = opts[:form_params] || {}
 
       # http body (model)
-      post_body = opts[:debug_body] || @api_client.object_to_http_body(update_chunk_data)
+      post_body = opts[:debug_body] || @api_client.object_to_http_body(update_chunk_req_payload)
 
       # return_type
       return_type = opts[:debug_return_type]
@@ -900,7 +975,7 @@ module TrieveRubyClient
     end
 
     # Update Chunk By Tracking Id
-    # Update Chunk By Tracking Id  Update a chunk by tracking_id. This is useful for when you are coordinating with an external system and want to use the tracking_id to identify the chunk.
+    # Update Chunk By Tracking Id  Update a chunk by tracking_id. This is useful for when you are coordinating with an external system and want to use the tracking_id to identify the chunk. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
     # @param tr_dataset [String] The dataset id to use for the request
     # @param update_chunk_by_tracking_id_data [UpdateChunkByTrackingIdData] JSON request payload to update a chunk by tracking_id (chunks)
     # @param [Hash] opts the optional parameters
@@ -911,7 +986,7 @@ module TrieveRubyClient
     end
 
     # Update Chunk By Tracking Id
-    # Update Chunk By Tracking Id  Update a chunk by tracking_id. This is useful for when you are coordinating with an external system and want to use the tracking_id to identify the chunk.
+    # Update Chunk By Tracking Id  Update a chunk by tracking_id. This is useful for when you are coordinating with an external system and want to use the tracking_id to identify the chunk. Auth&#39;ed user or api key must have an admin or owner role for the specified dataset&#39;s organization.
     # @param tr_dataset [String] The dataset id to use for the request
     # @param update_chunk_by_tracking_id_data [UpdateChunkByTrackingIdData] JSON request payload to update a chunk by tracking_id (chunks)
     # @param [Hash] opts the optional parameters