trieve_ruby_client 0.10.7 → 0.11.0

data/lib/trieve_ruby_client/api/chunk_group_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,7 +20,7 @@ module TrieveRubyClient
       @api_client = api_client
     end
     # Add Chunk to Group
-    # Add Chunk to Group  Route to add a chunk to a group. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
+    # Route to add a chunk to a group. 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 group_id [String] Id of the group to add the chunk to as a bookmark
     # @param add_chunk_to_group_data [AddChunkToGroupData] JSON request payload to add a chunk to a group (bookmark it)
@@ -32,7 +32,7 @@ module TrieveRubyClient
     end
 
     # Add Chunk to Group
-    # Add Chunk to Group  Route to add a chunk to a group. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
+    # Route to add a chunk to a group. 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 group_id [String] Id of the group to add the chunk to as a bookmark
     # @param add_chunk_to_group_data [AddChunkToGroupData] JSON request payload to add a chunk to a group (bookmark it)
@@ -101,7 +101,7 @@ module TrieveRubyClient
     end
 
     # Add Chunk to Group by Tracking ID
-    # Add Chunk to Group by Tracking ID  Route to add a chunk to a group by tracking id. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
+    # Route to add a chunk to a group by tracking id. 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] Id of the group to add the chunk to as a bookmark
     # @param add_chunk_to_group_data [AddChunkToGroupData] JSON request payload to add a chunk to a group (bookmark it)
@@ -113,7 +113,7 @@ module TrieveRubyClient
     end
 
     # Add Chunk to Group by Tracking ID
-    # Add Chunk to Group by Tracking ID  Route to add a chunk to a group by tracking id. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
+    # Route to add a chunk to a group by tracking id. 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] Id of the group to add the chunk to as a bookmark
     # @param add_chunk_to_group_data [AddChunkToGroupData] JSON request payload to add a chunk to a group (bookmark it)
@@ -182,7 +182,7 @@ module TrieveRubyClient
     end
 
     # Create Chunk Group
-    # Create Chunk Group  Create a new chunk_group. This is a way to group chunks together. If you try to create a chunk_group with the same tracking_id as an existing chunk_group, this operation will fail. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
+    # Create a new chunk_group. This is a way to group chunks together. If you try to create a chunk_group with the same tracking_id as an existing chunk_group, this operation 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 create_chunk_group_req_payload [CreateChunkGroupReqPayload] JSON request payload to cretea a chunkGroup
     # @param [Hash] opts the optional parameters
@@ -193,7 +193,7 @@ module TrieveRubyClient
     end
 
     # Create Chunk Group
-    # Create Chunk Group  Create a new chunk_group. This is a way to group chunks together. If you try to create a chunk_group with the same tracking_id as an existing chunk_group, this operation will fail. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
+    # Create a new chunk_group. This is a way to group chunks together. If you try to create a chunk_group with the same tracking_id as an existing chunk_group, this operation 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 create_chunk_group_req_payload [CreateChunkGroupReqPayload] JSON request payload to cretea a chunkGroup
     # @param [Hash] opts the optional parameters
@@ -257,7 +257,7 @@ module TrieveRubyClient
     end
 
     # Delete Group
-    # Delete Group  This will delete a chunk_group. If you set delete_chunks to true, it will also delete the chunks within the group. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
+    # This will delete a chunk_group. If you set delete_chunks to true, it will also delete the chunks within the group. 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 group_id [String] Id of the group you want to fetch.
     # @param delete_chunks [Boolean] Delete the chunks within the group
@@ -269,7 +269,7 @@ module TrieveRubyClient
     end
 
     # Delete Group
-    # Delete Group  This will delete a chunk_group. If you set delete_chunks to true, it will also delete the chunks within the group. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
+    # This will delete a chunk_group. If you set delete_chunks to true, it will also delete the chunks within the group. 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 group_id [String] Id of the group you want to fetch.
     # @param delete_chunks [Boolean] Delete the chunks within the group
@@ -334,7 +334,7 @@ module TrieveRubyClient
     end
 
     # Delete Group by Tracking ID
-    # Delete Group by Tracking ID  Delete a chunk_group with the given tracking id. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
+    # Delete a chunk_group with the given tracking id. 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_group to delete
     # @param delete_chunks [Boolean] Delete the chunks within the group
@@ -346,7 +346,7 @@ module TrieveRubyClient
     end
 
     # Delete Group by Tracking ID
-    # Delete Group by Tracking ID  Delete a chunk_group with the given tracking id. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
+    # Delete a chunk_group with the given tracking id. 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_group to delete
     # @param delete_chunks [Boolean] Delete the chunks within the group
@@ -411,22 +411,22 @@ module TrieveRubyClient
     end
 
     # Get Group
-    # Get Group  Fetch the group with the given id. get_group
+    # Fetch the group with the given id. get_group
     # @param tr_dataset [String] The dataset id to use for the request
     # @param group_id [String] Id of the group you want to fetch.
     # @param [Hash] opts the optional parameters
-    # @return [ChunkGroup]
+    # @return [ChunkGroupAndFileId]
     def get_chunk_group(tr_dataset, group_id, opts = {})
       data, _status_code, _headers = get_chunk_group_with_http_info(tr_dataset, group_id, opts)
       data
     end
 
     # Get Group
-    # Get Group  Fetch the group with the given id. get_group
+    # Fetch the group with the given id. get_group
     # @param tr_dataset [String] The dataset id to use for the request
     # @param group_id [String] Id of the group you want to fetch.
     # @param [Hash] opts the optional parameters
-    # @return [Array<(ChunkGroup, Integer, Hash)>] ChunkGroup data, response status code and response headers
+    # @return [Array<(ChunkGroupAndFileId, Integer, Hash)>] ChunkGroupAndFileId data, response status code and response headers
     def get_chunk_group_with_http_info(tr_dataset, group_id, opts = {})
       if @api_client.config.debugging
         @api_client.config.logger.debug 'Calling API: ChunkGroupApi.get_chunk_group ...'
@@ -458,7 +458,7 @@ module TrieveRubyClient
       post_body = opts[:debug_body]
 
       # return_type
-      return_type = opts[:debug_return_type] || 'ChunkGroup'
+      return_type = opts[:debug_return_type] || 'ChunkGroupAndFileId'
 
       # auth_names
       auth_names = opts[:debug_auth_names] || ['ApiKey']
@@ -481,24 +481,26 @@ module TrieveRubyClient
     end
 
     # Get Chunks in Group
-    # Get Chunks in Group  Route to get all chunks for a group. The response is paginated, with each page containing 10 chunks. Page is 1-indexed.
+    # Route to get all chunks for a group. The response is paginated, with each page containing 10 chunks. Page is 1-indexed.
     # @param tr_dataset [String] The dataset id to use for the request
     # @param group_id [String] Id of the group you want to fetch.
     # @param page [Integer] The page of chunks to get from the group
     # @param [Hash] opts the optional parameters
-    # @return [BookmarkData]
+    # @option opts [APIVersion] :x_api_version The version of the API to use for the request
+    # @return [BookmarkGroupResponse]
     def get_chunks_in_group(tr_dataset, group_id, page, opts = {})
       data, _status_code, _headers = get_chunks_in_group_with_http_info(tr_dataset, group_id, page, opts)
       data
     end
 
     # Get Chunks in Group
-    # Get Chunks in Group  Route to get all chunks for a group. The response is paginated, with each page containing 10 chunks. Page is 1-indexed.
+    # Route to get all chunks for a group. The response is paginated, with each page containing 10 chunks. Page is 1-indexed.
     # @param tr_dataset [String] The dataset id to use for the request
     # @param group_id [String] Id of the group you want to fetch.
     # @param page [Integer] The page of chunks to get from the group
     # @param [Hash] opts the optional parameters
-    # @return [Array<(BookmarkData, Integer, Hash)>] BookmarkData data, response status code and response headers
+    # @option opts [APIVersion] :x_api_version The version of the API to use for the request
+    # @return [Array<(BookmarkGroupResponse, Integer, Hash)>] BookmarkGroupResponse data, response status code and response headers
     def get_chunks_in_group_with_http_info(tr_dataset, group_id, page, opts = {})
       if @api_client.config.debugging
         @api_client.config.logger.debug 'Calling API: ChunkGroupApi.get_chunks_in_group ...'
@@ -526,6 +528,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] || {}
@@ -534,7 +537,7 @@ module TrieveRubyClient
       post_body = opts[:debug_body]
 
       # return_type
-      return_type = opts[:debug_return_type] || 'BookmarkData'
+      return_type = opts[:debug_return_type] || 'BookmarkGroupResponse'
 
       # auth_names
       auth_names = opts[:debug_auth_names] || ['ApiKey']
@@ -557,24 +560,26 @@ module TrieveRubyClient
     end
 
     # Get Chunks in Group by Tracking ID
-    # Get Chunks in Group by Tracking ID  Route to get all chunks for a group. The response is paginated, with each page containing 10 chunks. Support for custom page size is coming soon. Page is 1-indexed.
+    # Route to get all chunks for a group. The response is paginated, with each page containing 10 chunks. Support for custom page size is coming soon. Page is 1-indexed.
     # @param tr_dataset [String] The dataset id to use for the request
     # @param group_tracking_id [String] The id of the group to get the chunks from
     # @param page [Integer] The page of chunks to get from the group
     # @param [Hash] opts the optional parameters
-    # @return [BookmarkData]
+    # @option opts [APIVersion] :x_api_version The version of the API to use for the request
+    # @return [BookmarkGroupResponse]
     def get_chunks_in_group_by_tracking_id(tr_dataset, group_tracking_id, page, opts = {})
       data, _status_code, _headers = get_chunks_in_group_by_tracking_id_with_http_info(tr_dataset, group_tracking_id, page, opts)
       data
     end
 
     # Get Chunks in Group by Tracking ID
-    # Get Chunks in Group by Tracking ID  Route to get all chunks for a group. The response is paginated, with each page containing 10 chunks. Support for custom page size is coming soon. Page is 1-indexed.
+    # Route to get all chunks for a group. The response is paginated, with each page containing 10 chunks. Support for custom page size is coming soon. Page is 1-indexed.
     # @param tr_dataset [String] The dataset id to use for the request
     # @param group_tracking_id [String] The id of the group to get the chunks from
     # @param page [Integer] The page of chunks to get from the group
     # @param [Hash] opts the optional parameters
-    # @return [Array<(BookmarkData, Integer, Hash)>] BookmarkData data, response status code and response headers
+    # @option opts [APIVersion] :x_api_version The version of the API to use for the request
+    # @return [Array<(BookmarkGroupResponse, Integer, Hash)>] BookmarkGroupResponse data, response status code and response headers
     def get_chunks_in_group_by_tracking_id_with_http_info(tr_dataset, group_tracking_id, page, opts = {})
       if @api_client.config.debugging
         @api_client.config.logger.debug 'Calling API: ChunkGroupApi.get_chunks_in_group_by_tracking_id ...'
@@ -606,6 +611,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] || {}
@@ -614,7 +620,7 @@ module TrieveRubyClient
       post_body = opts[:debug_body]
 
       # return_type
-      return_type = opts[:debug_return_type] || 'BookmarkData'
+      return_type = opts[:debug_return_type] || 'BookmarkGroupResponse'
 
       # auth_names
       auth_names = opts[:debug_auth_names] || ['ApiKey']
@@ -637,22 +643,22 @@ module TrieveRubyClient
     end
 
     # Get Group by Tracking ID
-    # Get Group by Tracking ID  Fetch the group with the given tracking id. get_group_by_tracking_id
+    # Fetch the group with the given tracking id. get_group_by_tracking_id
     # @param tr_dataset [String] The dataset id to use for the request
     # @param tracking_id [String] The tracking id of the group to fetch.
     # @param [Hash] opts the optional parameters
-    # @return [ChunkGroup]
+    # @return [ChunkGroupAndFileId]
     def get_group_by_tracking_id(tr_dataset, tracking_id, opts = {})
       data, _status_code, _headers = get_group_by_tracking_id_with_http_info(tr_dataset, tracking_id, opts)
       data
     end
 
     # Get Group by Tracking ID
-    # Get Group by Tracking ID  Fetch the group with the given tracking id. get_group_by_tracking_id
+    # Fetch the group with the given tracking id. get_group_by_tracking_id
     # @param tr_dataset [String] The dataset id to use for the request
     # @param tracking_id [String] The tracking id of the group to fetch.
     # @param [Hash] opts the optional parameters
-    # @return [Array<(ChunkGroup, Integer, Hash)>] ChunkGroup data, response status code and response headers
+    # @return [Array<(ChunkGroupAndFileId, Integer, Hash)>] ChunkGroupAndFileId data, response status code and response headers
     def get_group_by_tracking_id_with_http_info(tr_dataset, tracking_id, opts = {})
       if @api_client.config.debugging
         @api_client.config.logger.debug 'Calling API: ChunkGroupApi.get_group_by_tracking_id ...'
@@ -684,7 +690,7 @@ module TrieveRubyClient
       post_body = opts[:debug_body]
 
       # return_type
-      return_type = opts[:debug_return_type] || 'ChunkGroup'
+      return_type = opts[:debug_return_type] || 'ChunkGroupAndFileId'
 
       # auth_names
       auth_names = opts[:debug_auth_names] || ['ApiKey']
@@ -707,22 +713,22 @@ module TrieveRubyClient
     end
 
     # Get Groups for Chunks
-    # Get Groups for Chunks  Route to get the groups that a chunk is in.
+    # Route to get the groups that a chunk is in.
     # @param tr_dataset [String] The dataset id to use for the request
     # @param get_groups_for_chunks_data [GetGroupsForChunksData] JSON request payload to get the groups that a chunk is in
     # @param [Hash] opts the optional parameters
-    # @return [Array<BookmarkGroupResult>]
+    # @return [Array<BookmarkGroupResponse>]
     def get_groups_chunk_is_in(tr_dataset, get_groups_for_chunks_data, opts = {})
       data, _status_code, _headers = get_groups_chunk_is_in_with_http_info(tr_dataset, get_groups_for_chunks_data, opts)
       data
     end
 
     # Get Groups for Chunks
-    # Get Groups for Chunks  Route to get the groups that a chunk is in.
+    # Route to get the groups that a chunk is in.
     # @param tr_dataset [String] The dataset id to use for the request
     # @param get_groups_for_chunks_data [GetGroupsForChunksData] JSON request payload to get the groups that a chunk is in
     # @param [Hash] opts the optional parameters
-    # @return [Array<(Array<BookmarkGroupResult>, Integer, Hash)>] Array<BookmarkGroupResult> data, response status code and response headers
+    # @return [Array<(Array<BookmarkGroupResponse>, Integer, Hash)>] Array<BookmarkGroupResponse> data, response status code and response headers
     def get_groups_chunk_is_in_with_http_info(tr_dataset, get_groups_for_chunks_data, opts = {})
       if @api_client.config.debugging
         @api_client.config.logger.debug 'Calling API: ChunkGroupApi.get_groups_chunk_is_in ...'
@@ -759,7 +765,7 @@ module TrieveRubyClient
       post_body = opts[:debug_body] || @api_client.object_to_http_body(get_groups_for_chunks_data)
 
       # return_type
-      return_type = opts[:debug_return_type] || 'Array<BookmarkGroupResult>'
+      return_type = opts[:debug_return_type] || 'Array<BookmarkGroupResponse>'
 
       # auth_names
       auth_names = opts[:debug_auth_names] || ['ApiKey']
@@ -781,37 +787,43 @@ module TrieveRubyClient
       return data, status_code, headers
     end
 
-    # Get Recommended Groups
-    # Get Recommended Groups  Route to get recommended groups. This route will return groups which are similar to the groups in the request body. You must provide at least one positive group id or group tracking id.
+    # Get Groups for Dataset
+    # Fetch the groups which belong to a dataset specified by its id.
     # @param tr_dataset [String] The dataset id to use for the request
-    # @param recommend_group_chunks_request [RecommendGroupChunksRequest] JSON request payload to get recommendations of chunks similar to the chunks in the request
+    # @param dataset_id [String] The id of the dataset to fetch groups for.
+    # @param page [Integer] The page of groups to fetch. Page is 1-indexed.
     # @param [Hash] opts the optional parameters
-    # @return [Array<GroupScoreChunk>]
-    def get_recommended_groups(tr_dataset, recommend_group_chunks_request, opts = {})
-      data, _status_code, _headers = get_recommended_groups_with_http_info(tr_dataset, recommend_group_chunks_request, opts)
+    # @return [GroupData]
+    def get_groups_for_dataset(tr_dataset, dataset_id, page, opts = {})
+      data, _status_code, _headers = get_groups_for_dataset_with_http_info(tr_dataset, dataset_id, page, opts)
       data
     end
 
-    # Get Recommended Groups
-    # Get Recommended Groups  Route to get recommended groups. This route will return groups which are similar to the groups in the request body. You must provide at least one positive group id or group tracking id.
+    # Get Groups for Dataset
+    # Fetch the groups which belong to a dataset specified by its id.
     # @param tr_dataset [String] The dataset id to use for the request
-    # @param recommend_group_chunks_request [RecommendGroupChunksRequest] JSON request payload to get recommendations of chunks similar to the chunks in the request
+    # @param dataset_id [String] The id of the dataset to fetch groups for.
+    # @param page [Integer] The page of groups to fetch. Page is 1-indexed.
     # @param [Hash] opts the optional parameters
-    # @return [Array<(Array<GroupScoreChunk>, Integer, Hash)>] Array<GroupScoreChunk> data, response status code and response headers
-    def get_recommended_groups_with_http_info(tr_dataset, recommend_group_chunks_request, opts = {})
+    # @return [Array<(GroupData, Integer, Hash)>] GroupData data, response status code and response headers
+    def get_groups_for_dataset_with_http_info(tr_dataset, dataset_id, page, opts = {})
       if @api_client.config.debugging
-        @api_client.config.logger.debug 'Calling API: ChunkGroupApi.get_recommended_groups ...'
+        @api_client.config.logger.debug 'Calling API: ChunkGroupApi.get_groups_for_dataset ...'
       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 ChunkGroupApi.get_recommended_groups"
+        fail ArgumentError, "Missing the required parameter 'tr_dataset' when calling ChunkGroupApi.get_groups_for_dataset"
       end
-      # verify the required parameter 'recommend_group_chunks_request' is set
-      if @api_client.config.client_side_validation && recommend_group_chunks_request.nil?
-        fail ArgumentError, "Missing the required parameter 'recommend_group_chunks_request' when calling ChunkGroupApi.get_recommended_groups"
+      # verify the required parameter 'dataset_id' is set
+      if @api_client.config.client_side_validation && dataset_id.nil?
+        fail ArgumentError, "Missing the required parameter 'dataset_id' when calling ChunkGroupApi.get_groups_for_dataset"
+      end
+      # verify the required parameter 'page' is set
+      if @api_client.config.client_side_validation && page.nil?
+        fail ArgumentError, "Missing the required parameter 'page' when calling ChunkGroupApi.get_groups_for_dataset"
       end
       # resource path
-      local_var_path = '/api/chunk_group/recommend'
+      local_var_path = '/api/dataset/groups/{dataset_id}/{page}'.sub('{' + 'dataset_id' + '}', CGI.escape(dataset_id.to_s)).sub('{' + 'page' + '}', CGI.escape(page.to_s))
 
       # query parameters
       query_params = opts[:query_params] || {}
@@ -820,27 +832,22 @@ module TrieveRubyClient
       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(recommend_group_chunks_request)
+      post_body = opts[:debug_body]
 
       # return_type
-      return_type = opts[:debug_return_type] || 'Array<GroupScoreChunk>'
+      return_type = opts[:debug_return_type] || 'GroupData'
 
       # auth_names
       auth_names = opts[:debug_auth_names] || ['ApiKey']
 
       new_options = opts.merge(
-        :operation => :"ChunkGroupApi.get_recommended_groups",
+        :operation => :"ChunkGroupApi.get_groups_for_dataset",
         :header_params => header_params,
         :query_params => query_params,
         :form_params => form_params,
@@ -849,50 +856,46 @@ module TrieveRubyClient
         :return_type => return_type
       )
 
-      data, status_code, headers = @api_client.call_api(:POST, local_var_path, new_options)
+      data, status_code, headers = @api_client.call_api(:GET, local_var_path, new_options)
       if @api_client.config.debugging
-        @api_client.config.logger.debug "API called: ChunkGroupApi#get_recommended_groups\nData: #{data.inspect}\nStatus code: #{status_code}\nHeaders: #{headers}"
+        @api_client.config.logger.debug "API called: ChunkGroupApi#get_groups_for_dataset\nData: #{data.inspect}\nStatus code: #{status_code}\nHeaders: #{headers}"
       end
       return data, status_code, headers
     end
 
-    # Get Groups for Dataset
-    # Get Groups for Dataset  Fetch the groups which belong to a dataset specified by its id.
+    # Get Recommended Groups
+    # Route to get recommended groups. This route will return groups which are similar to the groups in the request body. You must provide at least one positive group id or group tracking id.
     # @param tr_dataset [String] The dataset id to use for the request
-    # @param dataset_id [String] The id of the dataset to fetch groups for.
-    # @param page [Integer] The page of groups to fetch. Page is 1-indexed.
+    # @param recommend_group_chunks_request [RecommendGroupChunksRequest] JSON request payload to get recommendations of chunks similar to the chunks in the request
     # @param [Hash] opts the optional parameters
-    # @return [GroupData]
-    def get_specific_dataset_chunk_groups(tr_dataset, dataset_id, page, opts = {})
-      data, _status_code, _headers = get_specific_dataset_chunk_groups_with_http_info(tr_dataset, dataset_id, page, opts)
+    # @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 [RecommendGroupsResponse]
+    def get_recommended_groups(tr_dataset, recommend_group_chunks_request, opts = {})
+      data, _status_code, _headers = get_recommended_groups_with_http_info(tr_dataset, recommend_group_chunks_request, opts)
       data
     end
 
-    # Get Groups for Dataset
-    # Get Groups for Dataset  Fetch the groups which belong to a dataset specified by its id.
+    # Get Recommended Groups
+    # Route to get recommended groups. This route will return groups which are similar to the groups in the request body. You must provide at least one positive group id or group tracking id.
     # @param tr_dataset [String] The dataset id to use for the request
-    # @param dataset_id [String] The id of the dataset to fetch groups for.
-    # @param page [Integer] The page of groups to fetch. Page is 1-indexed.
+    # @param recommend_group_chunks_request [RecommendGroupChunksRequest] JSON request payload to get recommendations of chunks similar to the chunks in the request
     # @param [Hash] opts the optional parameters
-    # @return [Array<(GroupData, Integer, Hash)>] GroupData data, response status code and response headers
-    def get_specific_dataset_chunk_groups_with_http_info(tr_dataset, dataset_id, page, opts = {})
+    # @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<(RecommendGroupsResponse, Integer, Hash)>] RecommendGroupsResponse data, response status code and response headers
+    def get_recommended_groups_with_http_info(tr_dataset, recommend_group_chunks_request, opts = {})
       if @api_client.config.debugging
-        @api_client.config.logger.debug 'Calling API: ChunkGroupApi.get_specific_dataset_chunk_groups ...'
+        @api_client.config.logger.debug 'Calling API: ChunkGroupApi.get_recommended_groups ...'
       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 ChunkGroupApi.get_specific_dataset_chunk_groups"
-      end
-      # verify the required parameter 'dataset_id' is set
-      if @api_client.config.client_side_validation && dataset_id.nil?
-        fail ArgumentError, "Missing the required parameter 'dataset_id' when calling ChunkGroupApi.get_specific_dataset_chunk_groups"
+        fail ArgumentError, "Missing the required parameter 'tr_dataset' when calling ChunkGroupApi.get_recommended_groups"
       end
-      # verify the required parameter 'page' is set
-      if @api_client.config.client_side_validation && page.nil?
-        fail ArgumentError, "Missing the required parameter 'page' when calling ChunkGroupApi.get_specific_dataset_chunk_groups"
+      # verify the required parameter 'recommend_group_chunks_request' is set
+      if @api_client.config.client_side_validation && recommend_group_chunks_request.nil?
+        fail ArgumentError, "Missing the required parameter 'recommend_group_chunks_request' when calling ChunkGroupApi.get_recommended_groups"
       end
       # resource path
-      local_var_path = '/api/dataset/groups/{dataset_id}/{page}'.sub('{' + 'dataset_id' + '}', CGI.escape(dataset_id.to_s)).sub('{' + 'page' + '}', CGI.escape(page.to_s))
+      local_var_path = '/api/chunk_group/recommend'
 
       # query parameters
       query_params = opts[:query_params] || {}
@@ -901,22 +904,28 @@ module TrieveRubyClient
       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
+      header_params[:'X-API-Version'] = opts[:'x_api_version'] if !opts[:'x_api_version'].nil?
 
       # form parameters
       form_params = opts[:form_params] || {}
 
       # http body (model)
-      post_body = opts[:debug_body]
+      post_body = opts[:debug_body] || @api_client.object_to_http_body(recommend_group_chunks_request)
 
       # return_type
-      return_type = opts[:debug_return_type] || 'GroupData'
+      return_type = opts[:debug_return_type] || 'RecommendGroupsResponse'
 
       # auth_names
       auth_names = opts[:debug_auth_names] || ['ApiKey']
 
       new_options = opts.merge(
-        :operation => :"ChunkGroupApi.get_specific_dataset_chunk_groups",
+        :operation => :"ChunkGroupApi.get_recommended_groups",
         :header_params => header_params,
         :query_params => query_params,
         :form_params => form_params,
@@ -925,15 +934,15 @@ module TrieveRubyClient
         :return_type => return_type
       )
 
-      data, status_code, headers = @api_client.call_api(:GET, local_var_path, new_options)
+      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: ChunkGroupApi#get_specific_dataset_chunk_groups\nData: #{data.inspect}\nStatus code: #{status_code}\nHeaders: #{headers}"
+        @api_client.config.logger.debug "API called: ChunkGroupApi#get_recommended_groups\nData: #{data.inspect}\nStatus code: #{status_code}\nHeaders: #{headers}"
       end
       return data, status_code, headers
     end
 
     # Remove Chunk from Group
-    # Remove Chunk from Group  Route to remove a chunk from a group. Auth'ed user or api key must be an admin or owner of the dataset's organization to remove a chunk from a group.
+    # Route to remove a chunk from a group. Auth'ed user or api key must be an admin or owner of the dataset's organization to remove a chunk from a group.
     # @param tr_dataset [String] The dataset id to use for the request
     # @param group_id [String] Id of the group you want to remove the chunk from.
     # @param remove_chunk_from_group_req_payload [RemoveChunkFromGroupReqPayload] JSON request payload to remove a chunk from a group
@@ -945,7 +954,7 @@ module TrieveRubyClient
     end
 
     # Remove Chunk from Group
-    # Remove Chunk from Group  Route to remove a chunk from a group. Auth&#39;ed user or api key must be an admin or owner of the dataset&#39;s organization to remove a chunk from a group.
+    # Route to remove a chunk from a group. Auth&#39;ed user or api key must be an admin or owner of the dataset&#39;s organization to remove a chunk from a group.
     # @param tr_dataset [String] The dataset id to use for the request
     # @param group_id [String] Id of the group you want to remove the chunk from.
     # @param remove_chunk_from_group_req_payload [RemoveChunkFromGroupReqPayload] JSON request payload to remove a chunk from a group
@@ -1014,22 +1023,24 @@ module TrieveRubyClient
     end
 
     # Search Over Groups
-    # Search Over Groups  This route allows you to get groups as results instead of chunks. Each group returned will have the matching chunks sorted by similarity within the group. This is useful for when you want to get groups of chunks which are similar to the search query. If choosing hybrid search, the results will be re-ranked using BAAI/bge-reranker-large. Compatible with semantic, fulltext, or hybrid search modes.
+    # This route allows you to get groups as results instead of chunks. Each group returned will have the matching chunks sorted by similarity within the group. This is useful for when you want to get groups of chunks which are similar to the search query. If choosing hybrid search, the results will be re-ranked using scores from a cross encoder model. Compatible with semantic, fulltext, or hybrid search modes.
     # @param tr_dataset [String] The dataset id to use for the request
     # @param search_over_groups_data [SearchOverGroupsData] JSON request payload to semantically search over groups
     # @param [Hash] opts the optional parameters
-    # @return [SearchOverGroupsResults]
+    # @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 [SearchOverGroupsResponseTypes]
     def search_over_groups(tr_dataset, search_over_groups_data, opts = {})
       data, _status_code, _headers = search_over_groups_with_http_info(tr_dataset, search_over_groups_data, opts)
       data
     end
 
     # Search Over Groups
-    # Search Over Groups  This route allows you to get groups as results instead of chunks. Each group returned will have the matching chunks sorted by similarity within the group. This is useful for when you want to get groups of chunks which are similar to the search query. If choosing hybrid search, the results will be re-ranked using BAAI/bge-reranker-large. Compatible with semantic, fulltext, or hybrid search modes.
+    # This route allows you to get groups as results instead of chunks. Each group returned will have the matching chunks sorted by similarity within the group. This is useful for when you want to get groups of chunks which are similar to the search query. If choosing hybrid search, the results will be re-ranked using scores from a cross encoder model. Compatible with semantic, fulltext, or hybrid search modes.
     # @param tr_dataset [String] The dataset id to use for the request
     # @param search_over_groups_data [SearchOverGroupsData] JSON request payload to semantically search over groups
     # @param [Hash] opts the optional parameters
-    # @return [Array<(SearchOverGroupsResults, Integer, Hash)>] SearchOverGroupsResults 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<(SearchOverGroupsResponseTypes, Integer, Hash)>] SearchOverGroupsResponseTypes data, response status code and response headers
     def search_over_groups_with_http_info(tr_dataset, search_over_groups_data, opts = {})
       if @api_client.config.debugging
         @api_client.config.logger.debug 'Calling API: ChunkGroupApi.search_over_groups ...'
@@ -1058,6 +1069,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] || {}
@@ -1066,7 +1078,7 @@ module TrieveRubyClient
       post_body = opts[:debug_body] || @api_client.object_to_http_body(search_over_groups_data)
 
       # return_type
-      return_type = opts[:debug_return_type] || 'SearchOverGroupsResults'
+      return_type = opts[:debug_return_type] || 'SearchOverGroupsResponseTypes'
 
       # auth_names
       auth_names = opts[:debug_auth_names] || ['ApiKey']
@@ -1089,22 +1101,24 @@ module TrieveRubyClient
     end
 
     # Search Within Group
-    # Search Within Group  This route allows you to search only within a group. This is useful for when you only want search results to contain chunks which are members of a specific group. If choosing hybrid search, the results will be re-ranked using BAAI/bge-reranker-large.
+    # This route allows you to search only within a group. This is useful for when you only want search results to contain chunks which are members of a specific group. If choosing hybrid search, the results will be re-ranked using scores from a cross encoder model.
     # @param tr_dataset [String] The dataset id to use for the request
     # @param search_within_group_data [SearchWithinGroupData] JSON request payload to semantically search a group
     # @param [Hash] opts the optional parameters
-    # @return [SearchWithinGroupResults]
+    # @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 [SearchGroupResponseTypes]
     def search_within_group(tr_dataset, search_within_group_data, opts = {})
       data, _status_code, _headers = search_within_group_with_http_info(tr_dataset, search_within_group_data, opts)
       data
     end
 
     # Search Within Group
-    # Search Within Group  This route allows you to search only within a group. This is useful for when you only want search results to contain chunks which are members of a specific group. If choosing hybrid search, the results will be re-ranked using BAAI/bge-reranker-large.
+    # This route allows you to search only within a group. This is useful for when you only want search results to contain chunks which are members of a specific group. If choosing hybrid search, the results will be re-ranked using scores from a cross encoder model.
     # @param tr_dataset [String] The dataset id to use for the request
     # @param search_within_group_data [SearchWithinGroupData] JSON request payload to semantically search a group
     # @param [Hash] opts the optional parameters
-    # @return [Array<(SearchWithinGroupResults, Integer, Hash)>] SearchWithinGroupResults 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<(SearchGroupResponseTypes, Integer, Hash)>] SearchGroupResponseTypes data, response status code and response headers
     def search_within_group_with_http_info(tr_dataset, search_within_group_data, opts = {})
       if @api_client.config.debugging
         @api_client.config.logger.debug 'Calling API: ChunkGroupApi.search_within_group ...'
@@ -1133,6 +1147,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] || {}
@@ -1141,7 +1156,7 @@ module TrieveRubyClient
       post_body = opts[:debug_body] || @api_client.object_to_http_body(search_within_group_data)
 
       # return_type
-      return_type = opts[:debug_return_type] || 'SearchWithinGroupResults'
+      return_type = opts[:debug_return_type] || 'SearchGroupResponseTypes'
 
       # auth_names
       auth_names = opts[:debug_auth_names] || ['ApiKey']
@@ -1164,7 +1179,7 @@ module TrieveRubyClient
     end
 
     # Update Group
-    # Update Group  Update a chunk_group. If you try to change the tracking_id to one that already exists, this operation will fail. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
+    # Update a chunk_group. If you try to change the tracking_id to one that already exists, this operation 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_group_data [UpdateChunkGroupData] JSON request payload to update a chunkGroup
     # @param [Hash] opts the optional parameters
@@ -1175,7 +1190,7 @@ module TrieveRubyClient
     end
 
     # Update Group
-    # Update Group  Update a chunk_group. If you try to change the tracking_id to one that already exists, this operation 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_group. If you try to change the tracking_id to one that already exists, this operation 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_group_data [UpdateChunkGroupData] JSON request payload to update a chunkGroup
     # @param [Hash] opts the optional parameters
@@ -1239,7 +1254,7 @@ module TrieveRubyClient
     end
 
     # Update Group by Tracking ID
-    # Update Group by Tracking ID  Update a chunk_group with the given tracking id. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
+    # Update a chunk_group with the given tracking id. 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_group to update
     # @param update_group_by_tracking_id_req_payload [UpdateGroupByTrackingIDReqPayload] JSON request payload to update a chunkGroup
@@ -1251,7 +1266,7 @@ module TrieveRubyClient
     end
 
     # Update Group by Tracking ID
-    # Update Group by Tracking ID  Update a chunk_group with the given tracking id. Auth&#39;ed user or api key must have an admin or owner role for the specified dataset&#39;s organization.
+    # Update a chunk_group with the given tracking id. 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_group to update
     # @param update_group_by_tracking_id_req_payload [UpdateGroupByTrackingIDReqPayload] JSON request payload to update a chunkGroup