trieve_ruby_client 0.10.7 → 0.11.0

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.10.7
+The version of the OpenAPI document: 0.11.0
 Contact: developers@trieve.ai
 Generated by: https://openapi-generator.tech
 Generator version: 7.4.0
@@ -20,22 +20,24 @@ module TrieveRubyClient
       @api_client = api_client
     end
     # Autocomplete
-    # Autocomplete  This route provides the primary autocomplete functionality for the API. This prioritize prefix matching with semantic or full-text search.
+    # 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 autocomplete_req_payload [AutocompleteReqPayload] JSON request payload to semantically search for chunks (chunks)
     # @param [Hash] opts the optional parameters
-    # @return [SearchChunkQueryResponseBody]
+    # @option opts [APIVersion] :x_api_version The API version to use for this request. Defaults to V2 for orgs created after July 12, 2024 and V1 otherwise.
+    # @return [SearchResponseTypes]
     def autocomplete(tr_dataset, autocomplete_req_payload, opts = {})
       data, _status_code, _headers = autocomplete_with_http_info(tr_dataset, autocomplete_req_payload, opts)
       data
     end
 
     # Autocomplete
-    # Autocomplete  This route provides the primary autocomplete functionality for the API. This prioritize prefix matching with semantic or full-text search.
+    # 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 autocomplete_req_payload [AutocompleteReqPayload] 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
+    # @option opts [APIVersion] :x_api_version The API version to use for this request. Defaults to V2 for orgs created after July 12, 2024 and V1 otherwise.
+    # @return [Array<(SearchResponseTypes, Integer, Hash)>] SearchResponseTypes 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.autocomplete ...'
@@ -64,6 +66,7 @@ module TrieveRubyClient
           header_params['Content-Type'] = content_type
       end
       header_params[:'TR-Dataset'] = tr_dataset
+      header_params[:'X-API-Version'] = opts[:'x_api_version'] if !opts[:'x_api_version'].nil?
 
       # form parameters
       form_params = opts[:form_params] || {}
@@ -72,7 +75,7 @@ module TrieveRubyClient
       post_body = opts[:debug_body] || @api_client.object_to_http_body(autocomplete_req_payload)
 
       # return_type
-      return_type = opts[:debug_return_type] || 'SearchChunkQueryResponseBody'
+      return_type = opts[:debug_return_type] || 'SearchResponseTypes'
 
       # auth_names
       auth_names = opts[:debug_auth_names] || ['ApiKey']
@@ -94,8 +97,83 @@ module TrieveRubyClient
       return data, status_code, headers
     end
 
+    # Count chunks above threshold
+    # This route can be used to determine the number of chunks that match a given search criteria including filters and score threshold. It may be high latency for large datasets. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization. There is a dataset configuration imposed restriction on the maximum limit value (default 10,000) which is used to prevent DDOS attacks.
+    # @param tr_dataset [String] The dataset id to use for the request
+    # @param count_chunks_req_payload [CountChunksReqPayload] 
+    # @param [Hash] opts the optional parameters
+    # @return [CountChunkQueryResponseBody]
+    def count_chunks(tr_dataset, count_chunks_req_payload, opts = {})
+      data, _status_code, _headers = count_chunks_with_http_info(tr_dataset, count_chunks_req_payload, opts)
+      data
+    end
+
+    # Count chunks above threshold
+    # This route can be used to determine the number of chunks that match a given search criteria including filters and score threshold. It may be high latency for large datasets. Auth&#39;ed user or api key must have an admin or owner role for the specified dataset&#39;s organization. There is a dataset configuration imposed restriction on the maximum limit value (default 10,000) which is used to prevent DDOS attacks.
+    # @param tr_dataset [String] The dataset id to use for the request
+    # @param count_chunks_req_payload [CountChunksReqPayload] 
+    # @param [Hash] opts the optional parameters
+    # @return [Array<(CountChunkQueryResponseBody, Integer, Hash)>] CountChunkQueryResponseBody data, response status code and response headers
+    def count_chunks_with_http_info(tr_dataset, count_chunks_req_payload, opts = {})
+      if @api_client.config.debugging
+        @api_client.config.logger.debug 'Calling API: ChunkApi.count_chunks ...'
+      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.count_chunks"
+      end
+      # verify the required parameter 'count_chunks_req_payload' is set
+      if @api_client.config.client_side_validation && count_chunks_req_payload.nil?
+        fail ArgumentError, "Missing the required parameter 'count_chunks_req_payload' when calling ChunkApi.count_chunks"
+      end
+      # resource path
+      local_var_path = '/api/chunk/count'
+
+      # 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(count_chunks_req_payload)
+
+      # return_type
+      return_type = opts[:debug_return_type] || 'CountChunkQueryResponseBody'
+
+      # auth_names
+      auth_names = opts[:debug_auth_names] || ['ApiKey']
+
+      new_options = opts.merge(
+        :operation => :"ChunkApi.count_chunks",
+        :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#count_chunks\nData: #{data.inspect}\nStatus code: #{status_code}\nHeaders: #{headers}"
+      end
+      return data, status_code, headers
+    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. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
+    # 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 create_chunk_req_payload_enum [CreateChunkReqPayloadEnum] JSON request payload to create a new chunk (chunk)
     # @param [Hash] opts the optional parameters
@@ -106,7 +184,7 @@ module TrieveRubyClient
     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. Auth&#39;ed user or api key must have an admin or owner role for the specified dataset&#39;s organization.
+    # 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 create_chunk_req_payload_enum [CreateChunkReqPayloadEnum] JSON request payload to create a new chunk (chunk)
     # @param [Hash] opts the optional parameters
@@ -170,7 +248,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. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
+    # 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 +259,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. Auth&#39;ed user or api key must have an admin or owner role for the specified dataset&#39;s organization.
+    # 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 +318,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. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
+    # 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 +329,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. Auth&#39;ed user or api key must have an admin or owner role for the specified dataset&#39;s organization.
+    # 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
@@ -310,7 +388,7 @@ module TrieveRubyClient
     end
 
     # RAG on Specified Chunks
-    # RAG on Specified Chunks  This endpoint exists as an alternative to the topic+message concept where our API handles chat memory. With this endpoint, the user is responsible for providing the context window and the prompt. See more in the \"search before generate\" page at docs.trieve.ai.
+    # This endpoint exists as an alternative to the topic+message concept where our API handles chat memory. With this endpoint, the user is responsible for providing the context window and the prompt. See more in the \"search before generate\" page at docs.trieve.ai.
     # @param tr_dataset [String] The dataset id to use for the request
     # @param generate_chunks_request [GenerateChunksRequest] JSON request payload to perform RAG on some chunks (chunks)
     # @param [Hash] opts the optional parameters
@@ -321,7 +399,7 @@ module TrieveRubyClient
     end
 
     # RAG on Specified Chunks
-    # RAG on Specified Chunks  This endpoint exists as an alternative to the topic+message concept where our API handles chat memory. With this endpoint, the user is responsible for providing the context window and the prompt. See more in the \&quot;search before generate\&quot; page at docs.trieve.ai.
+    # This endpoint exists as an alternative to the topic+message concept where our API handles chat memory. With this endpoint, the user is responsible for providing the context window and the prompt. See more in the \&quot;search before generate\&quot; page at docs.trieve.ai.
     # @param tr_dataset [String] The dataset id to use for the request
     # @param generate_chunks_request [GenerateChunksRequest] JSON request payload to perform RAG on some chunks (chunks)
     # @param [Hash] opts the optional parameters
@@ -385,22 +463,24 @@ module TrieveRubyClient
     end
 
     # Get Chunk By Id
-    # Get Chunk By Id  Get a singular chunk by id.
+    # Get a singular chunk by id.
     # @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 [ChunkMetadataStringTagSet]
+    # @option opts [APIVersion] :x_api_version The API version to use for this request. Defaults to V2 for orgs created after July 12, 2024 and V1 otherwise.
+    # @return [ChunkReturnTypes]
     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
     end
 
     # Get Chunk By Id
-    # Get Chunk By Id  Get a singular chunk by id.
+    # Get a singular chunk by id.
     # @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<(ChunkMetadataStringTagSet, Integer, Hash)>] ChunkMetadataStringTagSet data, response status code and response headers
+    # @option opts [APIVersion] :x_api_version The API version to use for this request. Defaults to V2 for orgs created after July 12, 2024 and V1 otherwise.
+    # @return [Array<(ChunkReturnTypes, Integer, Hash)>] ChunkReturnTypes 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 ...'
@@ -424,6 +504,7 @@ module TrieveRubyClient
       # HTTP header 'Accept' (if needed)
       header_params['Accept'] = @api_client.select_header_accept(['application/json'])
       header_params[:'TR-Dataset'] = tr_dataset
+      header_params[:'X-API-Version'] = opts[:'x_api_version'] if !opts[:'x_api_version'].nil?
 
       # form parameters
       form_params = opts[:form_params] || {}
@@ -432,7 +513,7 @@ module TrieveRubyClient
       post_body = opts[:debug_body]
 
       # return_type
-      return_type = opts[:debug_return_type] || 'ChunkMetadataStringTagSet'
+      return_type = opts[:debug_return_type] || 'ChunkReturnTypes'
 
       # auth_names
       auth_names = opts[:debug_auth_names] || ['ApiKey']
@@ -455,22 +536,24 @@ module TrieveRubyClient
     end
 
     # Get Chunk By Tracking Id
-    # Get Chunk By Tracking Id  Get a singular chunk by tracking_id. This is useful for when you are coordinating with an external system and want to use your own id as the primary reference for a chunk.
+    # Get a singular chunk by tracking_id. This is useful for when you are coordinating with an external system and want to use your own id as the primary reference for a chunk.
     # @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 [ChunkMetadataStringTagSet]
+    # @option opts [APIVersion] :x_api_version The API version to use for this request. Defaults to V2 for orgs created after July 12, 2024 and V1 otherwise.
+    # @return [ChunkReturnTypes]
     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
     end
 
     # Get Chunk By Tracking Id
-    # Get Chunk By Tracking Id  Get a singular chunk by tracking_id. This is useful for when you are coordinating with an external system and want to use your own id as the primary reference for a chunk.
+    # Get a singular chunk by tracking_id. This is useful for when you are coordinating with an external system and want to use your own id as the primary reference for a chunk.
     # @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<(ChunkMetadataStringTagSet, Integer, Hash)>] ChunkMetadataStringTagSet data, response status code and response headers
+    # @option opts [APIVersion] :x_api_version The API version to use for this request. Defaults to V2 for orgs created after July 12, 2024 and V1 otherwise.
+    # @return [Array<(ChunkReturnTypes, Integer, Hash)>] ChunkReturnTypes 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 ...'
@@ -494,6 +577,7 @@ module TrieveRubyClient
       # HTTP header 'Accept' (if needed)
       header_params['Accept'] = @api_client.select_header_accept(['application/json'])
       header_params[:'TR-Dataset'] = tr_dataset
+      header_params[:'X-API-Version'] = opts[:'x_api_version'] if !opts[:'x_api_version'].nil?
 
       # form parameters
       form_params = opts[:form_params] || {}
@@ -502,7 +586,7 @@ module TrieveRubyClient
       post_body = opts[:debug_body]
 
       # return_type
-      return_type = opts[:debug_return_type] || 'ChunkMetadataStringTagSet'
+      return_type = opts[:debug_return_type] || 'ChunkReturnTypes'
 
       # auth_names
       auth_names = opts[:debug_auth_names] || ['ApiKey']
@@ -525,22 +609,24 @@ module TrieveRubyClient
     end
 
     # Get Chunks By Ids
-    # Get Chunks By Ids  Get multiple chunks by multiple ids.
+    # Get multiple chunks by multiple ids.
     # @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<ChunkMetadataStringTagSet>]
+    # @option opts [APIVersion] :x_api_version The API version to use for this request. Defaults to V2 for orgs created after July 12, 2024 and V1 otherwise.
+    # @return [Array<ChunkReturnTypes>]
     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
     end
 
     # Get Chunks By Ids
-    # Get Chunks By Ids  Get multiple chunks by multiple ids.
+    # Get multiple chunks by multiple ids.
     # @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<ChunkMetadataStringTagSet>, Integer, Hash)>] Array<ChunkMetadataStringTagSet> data, response status code and response headers
+    # @option opts [APIVersion] :x_api_version The API version to use for this request. Defaults to V2 for orgs created after July 12, 2024 and V1 otherwise.
+    # @return [Array<(Array<ChunkReturnTypes>, Integer, Hash)>] Array<ChunkReturnTypes> 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 ...'
@@ -569,6 +655,7 @@ module TrieveRubyClient
           header_params['Content-Type'] = content_type
       end
       header_params[:'TR-Dataset'] = tr_dataset
+      header_params[:'X-API-Version'] = opts[:'x_api_version'] if !opts[:'x_api_version'].nil?
 
       # form parameters
       form_params = opts[:form_params] || {}
@@ -577,7 +664,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<ChunkMetadataStringTagSet>'
+      return_type = opts[:debug_return_type] || 'Array<ChunkReturnTypes>'
 
       # auth_names
       auth_names = opts[:debug_auth_names] || ['ApiKey']
@@ -599,23 +686,25 @@ module TrieveRubyClient
       return data, status_code, headers
     end
 
-    # Get Chunks By TrackingIds
-    # Get Chunks By TrackingIds  Get multiple chunks by ids.
+    # Get Chunks By Tracking Ids
+    # Get multiple chunks by ids.
     # @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<ChunkMetadataStringTagSet>]
+    # @option opts [APIVersion] :x_api_version The API version to use for this request. Defaults to V2 for orgs created after July 12, 2024 and V1 otherwise.
+    # @return [Array<ChunkReturnTypes>]
     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
     end
 
-    # Get Chunks By TrackingIds
-    # Get Chunks By TrackingIds  Get multiple chunks by ids.
+    # Get Chunks By Tracking Ids
+    # Get multiple chunks by ids.
     # @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<(Array<ChunkMetadataStringTagSet>, Integer, Hash)>] Array<ChunkMetadataStringTagSet> data, response status code and response headers
+    # @option opts [APIVersion] :x_api_version The API version to use for this request. Defaults to V2 for orgs created after July 12, 2024 and V1 otherwise.
+    # @return [Array<(Array<ChunkReturnTypes>, Integer, Hash)>] Array<ChunkReturnTypes> 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 ...'
@@ -644,6 +733,7 @@ module TrieveRubyClient
           header_params['Content-Type'] = content_type
       end
       header_params[:'TR-Dataset'] = tr_dataset
+      header_params[:'X-API-Version'] = opts[:'x_api_version'] if !opts[:'x_api_version'].nil?
 
       # form parameters
       form_params = opts[:form_params] || {}
@@ -652,7 +742,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] || 'Array<ChunkMetadataStringTagSet>'
+      return_type = opts[:debug_return_type] || 'Array<ChunkReturnTypes>'
 
       # auth_names
       auth_names = opts[:debug_auth_names] || ['ApiKey']
@@ -675,22 +765,24 @@ module TrieveRubyClient
     end
 
     # Get Recommended Chunks
-    # 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.
+    # 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
-    # @return [Array<ChunkMetadataWithScore>]
+    # @option opts [APIVersion] :x_api_version The API version to use for this request. Defaults to V2 for orgs created after July 12, 2024 and V1 otherwise.
+    # @return [RecommendResponseTypes]
     def get_recommended_chunks(tr_dataset, recommend_chunks_request, opts = {})
       data, _status_code, _headers = get_recommended_chunks_with_http_info(tr_dataset, recommend_chunks_request, opts)
       data
     end
 
     # Get Recommended Chunks
-    # 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.
+    # 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
-    # @return [Array<(Array<ChunkMetadataWithScore>, Integer, Hash)>] Array<ChunkMetadataWithScore> data, response status code and response headers
+    # @option opts [APIVersion] :x_api_version The API version to use for this request. Defaults to V2 for orgs created after July 12, 2024 and V1 otherwise.
+    # @return [Array<(RecommendResponseTypes, Integer, Hash)>] RecommendResponseTypes data, response status code and response headers
     def get_recommended_chunks_with_http_info(tr_dataset, recommend_chunks_request, opts = {})
       if @api_client.config.debugging
         @api_client.config.logger.debug 'Calling API: ChunkApi.get_recommended_chunks ...'
@@ -719,6 +811,7 @@ module TrieveRubyClient
           header_params['Content-Type'] = content_type
       end
       header_params[:'TR-Dataset'] = tr_dataset
+      header_params[:'X-API-Version'] = opts[:'x_api_version'] if !opts[:'x_api_version'].nil?
 
       # form parameters
       form_params = opts[:form_params] || {}
@@ -727,7 +820,7 @@ module TrieveRubyClient
       post_body = opts[:debug_body] || @api_client.object_to_http_body(recommend_chunks_request)
 
       # return_type
-      return_type = opts[:debug_return_type] || 'Array<ChunkMetadataWithScore>'
+      return_type = opts[:debug_return_type] || 'RecommendResponseTypes'
 
       # auth_names
       auth_names = opts[:debug_auth_names] || ['ApiKey']
@@ -750,7 +843,7 @@ module TrieveRubyClient
     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.
+    # 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
@@ -761,7 +854,7 @@ module TrieveRubyClient
     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.
+    # 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
@@ -825,22 +918,24 @@ module TrieveRubyClient
     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.
+    # 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_chunks_req_payload [SearchChunksReqPayload] JSON request payload to semantically search for chunks (chunks)
     # @param [Hash] opts the optional parameters
-    # @return [SearchChunkQueryResponseBody]
+    # @option opts [APIVersion] :x_api_version The API version to use for this request. Defaults to V2 for orgs created after July 12, 2024 and V1 otherwise.
+    # @return [SearchResponseTypes]
     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.
+    # 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_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
+    # @option opts [APIVersion] :x_api_version The API version to use for this request. Defaults to V2 for orgs created after July 12, 2024 and V1 otherwise.
+    # @return [Array<(SearchResponseTypes, Integer, Hash)>] SearchResponseTypes data, response status code and response headers
     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 ...'
@@ -869,6 +964,7 @@ module TrieveRubyClient
           header_params['Content-Type'] = content_type
       end
       header_params[:'TR-Dataset'] = tr_dataset
+      header_params[:'X-API-Version'] = opts[:'x_api_version'] if !opts[:'x_api_version'].nil?
 
       # form parameters
       form_params = opts[:form_params] || {}
@@ -877,7 +973,7 @@ module TrieveRubyClient
       post_body = opts[:debug_body] || @api_client.object_to_http_body(search_chunks_req_payload)
 
       # return_type
-      return_type = opts[:debug_return_type] || 'SearchChunkQueryResponseBody'
+      return_type = opts[:debug_return_type] || 'SearchResponseTypes'
 
       # auth_names
       auth_names = opts[:debug_auth_names] || ['ApiKey']
@@ -900,7 +996,7 @@ 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. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
+    # 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_req_payload [UpdateChunkReqPayload] JSON request payload to update a chunk (chunk)
     # @param [Hash] opts the optional parameters
@@ -911,7 +1007,7 @@ 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. Auth&#39;ed user or api key must have an admin or owner role for the specified dataset&#39;s organization.
+    # 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_req_payload [UpdateChunkReqPayload] JSON request payload to update a chunk (chunk)
     # @param [Hash] opts the optional parameters
@@ -975,7 +1071,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. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
+    # 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
@@ -986,7 +1082,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. Auth&#39;ed user or api key must have an admin or owner role for the specified dataset&#39;s organization.
+    # 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