trieve_ruby_client 0.5.7 → 0.6.2

data/docs/GroupScoreChunk.md

@@ -0,0 +1,22 @@
+# TrieveRubyClient::GroupScoreChunk
+
+## Properties
+
+| Name | Type | Description | Notes |
+| ---- | ---- | ----------- | ----- |
+| **group_id** | **String** |  |  |
+| **group_tracking_id** | **String** |  | [optional] |
+| **metadata** | [**Array<ScoreChunkDTO>**](ScoreChunkDTO.md) |  |  |
+
+## Example
+
+```ruby
+require 'trieve_ruby_client'
+
+instance = TrieveRubyClient::GroupScoreChunk.new(
+  group_id: null,
+  group_tracking_id: null,
+  metadata: null
+)
+```
+

data/docs/GroupScoreSlimChunks.md

@@ -0,0 +1,20 @@
+# TrieveRubyClient::GroupScoreSlimChunks
+
+## Properties
+
+| Name | Type | Description | Notes |
+| ---- | ---- | ----------- | ----- |
+| **group_id** | **String** |  |  |
+| **metadata** | [**Array<ScoreSlimChunks>**](ScoreSlimChunks.md) |  |  |
+
+## Example
+
+```ruby
+require 'trieve_ruby_client'
+
+instance = TrieveRubyClient::GroupScoreSlimChunks.new(
+  group_id: null,
+  metadata: null
+)
+```
+

data/docs/RecommendChunksRequest.md

@@ -11,6 +11,7 @@
 | **positive_chunk_ids** | **Array<String>** | The ids of the chunks to be used as positive examples for the recommendation. The chunks in this array will be used to find similar chunks. | [optional] |
 | **positive_tracking_ids** | **Array<String>** | The tracking_ids of the chunks to be used as positive examples for the recommendation. The chunks in this array will be used to find similar chunks. | [optional] |
 | **slim_chunks** | **Boolean** | Set slim_chunks to true to avoid returning the content and chunk_html of the chunks. This is useful for when you want to reduce amount of data over the wire for latency improvement. Default is false. | [optional] |
+| **strategy** | **String** | Strategy to use for recommendations, either \"average_vector\" or \"best_score\". The default is \"average_vector\". The \"average_vector\" strategy will construct a single average vector from the positive and negative samples then use it to perform a pseudo-search. The \"best_score\" strategy is more advanced and navigates the HNSW with a heuristic of picking edges where the point is closer to the positive samples than it is the negatives. | [optional] |
 
 ## Example
 
@@ -24,7 +25,8 @@ instance = TrieveRubyClient::RecommendChunksRequest.new(
   negative_tracking_ids: null,
   positive_chunk_ids: null,
   positive_tracking_ids: null,
-  slim_chunks: null
+  slim_chunks: null,
+  strategy: null
 )
 ```
 

data/docs/RecommendGroupChunksRequest.md

@@ -0,0 +1,34 @@
+# TrieveRubyClient::RecommendGroupChunksRequest
+
+## Properties
+
+| Name | Type | Description | Notes |
+| ---- | ---- | ----------- | ----- |
+| **filters** | [**ChunkFilter**](ChunkFilter.md) |  | [optional] |
+| **group_size** | **Integer** | The number of chunks to fetch for each group. This is the number of chunks which will be returned in the response for each group. The default is 3. If this is set to a large number, we recommend setting slim_chunks to true to avoid returning the content and chunk_html of the chunks so as to reduce latency due to content download and serialization. | [optional] |
+| **limit** | **Integer** | The number of groups to return. This is the number of groups which will be returned in the response. The default is 10. | [optional] |
+| **negative_group_ids** | **Array<String>** | The ids of the groups to be used as negative examples for the recommendation. The groups in this array will be used to filter out similar groups. | [optional] |
+| **negative_group_tracking_ids** | **Array<String>** | The ids of the groups to be used as negative examples for the recommendation. The groups in this array will be used to filter out similar groups. | [optional] |
+| **positive_group_ids** | **Array<String>** | The ids of the groups to be used as positive examples for the recommendation. The groups in this array will be used to find similar groups. | [optional] |
+| **positive_group_tracking_ids** | **Array<String>** | The ids of the groups to be used as positive examples for the recommendation. The groups in this array will be used to find similar groups. | [optional] |
+| **slim_chunks** | **Boolean** | Set slim_chunks to true to avoid returning the content and chunk_html of the chunks. This is useful for when you want to reduce amount of data over the wire for latency improvement. Default is false. | [optional] |
+| **strategy** | **String** | Strategy to use for recommendations, either \"average_vector\" or \"best_score\". The default is \"average_vector\". The \"average_vector\" strategy will construct a single average vector from the positive and negative samples then use it to perform a pseudo-search. The \"best_score\" strategy is more advanced and navigates the HNSW with a heuristic of picking edges where the point is closer to the positive samples than it is the negatives. | [optional] |
+
+## Example
+
+```ruby
+require 'trieve_ruby_client'
+
+instance = TrieveRubyClient::RecommendGroupChunksRequest.new(
+  filters: null,
+  group_size: null,
+  limit: null,
+  negative_group_ids: null,
+  negative_group_tracking_ids: null,
+  positive_group_ids: null,
+  positive_group_tracking_ids: null,
+  slim_chunks: null,
+  strategy: null
+)
+```
+

data/docs/ScoreChunkDTO.md

@@ -4,7 +4,7 @@
 
 | Name | Type | Description | Notes |
 | ---- | ---- | ----------- | ----- |
-| **metadata** | [**Array<ChunkMetadataWithFileData>**](ChunkMetadataWithFileData.md) |  |  |
+| **metadata** | [**Array<ChunkMetadata>**](ChunkMetadata.md) |  |  |
 | **score** | **Float** |  |  |
 
 ## Example

data/docs/SearchChunkData.md

@@ -9,10 +9,9 @@
 | **get_collisions** | **Boolean** | Set get_collisions to true to get the collisions for each chunk. This will only apply if environment variable COLLISIONS_ENABLED is set to true. | [optional] |
 | **highlight_delimiters** | **Array<String>** | Set highlight_delimiters to a list of strings to use as delimiters for highlighting. If not specified, this defaults to [\"?\", \",\", \".\", \"!\"]. | [optional] |
 | **highlight_results** | **Boolean** | Set highlight_results to true to highlight the results. If not specified, this defaults to true. | [optional] |
-| **page** | **Integer** | Page of chunks to fetch. Each page is 10 chunks. Support for custom page size is coming soon. | [optional] |
+| **page** | **Integer** | Page of chunks to fetch. Page is 1-indexed. | [optional] |
 | **page_size** | **Integer** | Page size is the number of chunks to fetch. This can be used to fetch more than 10 chunks at a time. | [optional] |
 | **query** | **String** | Query is the search query. This can be any string. The query will be used to create an embedding vector and/or SPLADE vector which will be used to find the result set. |  |
-| **quote_negated_words** | **Boolean** | Turn on quote words and negated words to search for exact phrases and exclude words from the search results. Default is false. | [optional] |
 | **score_threshold** | **Float** | Set score_threshold to a float to filter out chunks with a score below the threshold. | [optional] |
 | **search_type** | **String** | Can be either \"semantic\", \"fulltext\", or \"hybrid\". \"hybrid\" will pull in one page (10 chunks) of both semantic and full-text results then re-rank them using BAAI/bge-reranker-large. \"semantic\" will pull in one page (10 chunks) of the nearest cosine distant vectors. \"fulltext\" will pull in one page (10 chunks) of full-text results based on SPLADE. |  |
 | **slim_chunks** | **Boolean** | Set slim_chunks to true to avoid returning the content and chunk_html of the chunks. This is useful for when you want to reduce amount of data over the wire for latency improvement. Default is false. | [optional] |
@@ -32,7 +31,6 @@ instance = TrieveRubyClient::SearchChunkData.new(
   page: null,
   page_size: null,
   query: null,
-  quote_negated_words: null,
   score_threshold: null,
   search_type: null,
   slim_chunks: null,

data/docs/SearchOverGroupsData.md

@@ -6,13 +6,12 @@
 | ---- | ---- | ----------- | ----- |
 | **filters** | [**ChunkFilter**](ChunkFilter.md) |  | [optional] |
 | **get_collisions** | **Boolean** | Set get_collisions to true to get the collisions for each chunk. This will only apply if environment variable COLLISIONS_ENABLED is set to true. | [optional] |
-| **group_size** | **Integer** |  | [optional] |
+| **group_size** | **Integer** | Group_size is the number of chunks to fetch for each group. The default is 3. If a group has less than group_size chunks, all chunks will be returned. If this is set to a large number, we recommend setting slim_chunks to true to avoid returning the content and chunk_html of the chunks so as to lower the amount of time required for content download and serialization. | [optional] |
 | **highlight_delimiters** | **Array<String>** | Set highlight_delimiters to a list of strings to use as delimiters for highlighting. If not specified, this defaults to [\"?\", \",\", \".\", \"!\"]. | [optional] |
 | **highlight_results** | **Boolean** | Set highlight_results to true to highlight the results. If not specified, this defaults to true. | [optional] |
-| **page** | **Integer** | Page of chunks to fetch. Each page is 10 chunks. Support for custom page size is coming soon. | [optional] |
-| **page_size** | **Integer** | Page size is the number of chunks to fetch. This can be used to fetch more than 10 chunks at a time. | [optional] |
+| **page** | **Integer** | Page of group results to fetch. Page is 1-indexed. | [optional] |
+| **page_size** | **Integer** | Page size is the number of group results to fetch. The default is 10. | [optional] |
 | **query** | **String** | Query is the search query. This can be any string. The query will be used to create an embedding vector and/or SPLADE vector which will be used to find the result set. |  |
-| **quote_negated_words** | **Boolean** | Turn on quote words and negated words to search for exact phrases and exclude words from the search results. Default is false. | [optional] |
 | **score_threshold** | **Float** | Set score_threshold to a float to filter out chunks with a score below the threshold. | [optional] |
 | **search_type** | **String** | Can be either \"semantic\", \"fulltext\", or \"hybrid\". \"hybrid\" will pull in one page (10 chunks) of both semantic and full-text results then re-rank them using BAAI/bge-reranker-large. \"semantic\" will pull in one page (10 chunks) of the nearest cosine distant vectors. \"fulltext\" will pull in one page (10 chunks) of full-text results based on SPLADE. |  |
 | **slim_chunks** | **Boolean** | Set slim_chunks to true to avoid returning the content and chunk_html of the chunks. This is useful for when you want to reduce amount of data over the wire for latency improvement. Default is false. | [optional] |
@@ -31,7 +30,6 @@ instance = TrieveRubyClient::SearchOverGroupsData.new(
   page: null,
   page_size: null,
   query: null,
-  quote_negated_words: null,
   score_threshold: null,
   search_type: null,
   slim_chunks: null

data/docs/SearchOverGroupsResults.md

@@ -0,0 +1,20 @@
+# TrieveRubyClient::SearchOverGroupsResults
+
+## Properties
+
+| Name | Type | Description | Notes |
+| ---- | ---- | ----------- | ----- |
+| **group_chunks** | [**Array<GroupScoreChunk>**](GroupScoreChunk.md) |  |  |
+| **total_chunk_pages** | **Integer** |  |  |
+
+## Example
+
+```ruby
+require 'trieve_ruby_client'
+
+instance = TrieveRubyClient::SearchOverGroupsResults.new(
+  group_chunks: null,
+  total_chunk_pages: null
+)
+```
+

data/docs/SearchOverGroupsSlimResults.md

@@ -0,0 +1,20 @@
+# TrieveRubyClient::SearchOverGroupsSlimResults
+
+## Properties
+
+| Name | Type | Description | Notes |
+| ---- | ---- | ----------- | ----- |
+| **group_chunks** | [**Array<GroupScoreSlimChunks>**](GroupScoreSlimChunks.md) |  |  |
+| **total_chunk_pages** | **Integer** |  |  |
+
+## Example
+
+```ruby
+require 'trieve_ruby_client'
+
+instance = TrieveRubyClient::SearchOverGroupsSlimResults.new(
+  group_chunks: null,
+  total_chunk_pages: null
+)
+```
+

data/docs/SearchWithinGroupData.md

@@ -10,10 +10,9 @@
 | **group_tracking_id** | **String** | Group_tracking_id specifies the group to search within by tracking id. Results will only consist of chunks which are bookmarks within the specified group. If both group_id and group_tracking_id are provided, group_id will be used. | [optional] |
 | **highlight_delimiters** | **Array<String>** | Set highlight_delimiters to a list of strings to use as delimiters for highlighting. If not specified, this defaults to [\"?\", \",\", \".\", \"!\"]. | [optional] |
 | **highlight_results** | **Boolean** | Set highlight_results to true to highlight the results. If not specified, this defaults to true. | [optional] |
-| **page** | **Integer** | The page of chunks to fetch. Each page is 10 chunks. Support for custom page size is coming soon. | [optional] |
+| **page** | **Integer** | The page of chunks to fetch. Page is 1-indexed. | [optional] |
 | **page_size** | **Integer** | The page size is the number of chunks to fetch. This can be used to fetch more than 10 chunks at a time. | [optional] |
 | **query** | **String** | The query is the search query. This can be any string. The query will be used to create an embedding vector and/or SPLADE vector which will be used to find the result set. |  |
-| **quote_negated_words** | **Boolean** | Turn on quote words and negated words to search for exact phrases and exclude words from the search results. Default is false. | [optional] |
 | **score_threshold** | **Float** | Set score_threshold to a float to filter out chunks with a score below the threshold. | [optional] |
 | **search_type** | **String** | Search_type can be either \"semantic\", \"fulltext\", or \"hybrid\". \"hybrid\" will pull in one page (10 chunks) of both semantic and full-text results then re-rank them using BAAI/bge-reranker-large. \"semantic\" will pull in one page (10 chunks) of the nearest cosine distant vectors. \"fulltext\" will pull in one page (10 chunks) of full-text results based on SPLADE. |  |
 | **slim_chunks** | **Boolean** | Set slim_chunks to true to avoid returning the content and chunk_html of the chunks. This is useful for when you want to reduce amount of data over the wire for latency improvement. Default is false. | [optional] |
@@ -34,7 +33,6 @@ instance = TrieveRubyClient::SearchWithinGroupData.new(
   page: null,
   page_size: null,
   query: null,
-  quote_negated_words: null,
   score_threshold: null,
   search_type: null,
   slim_chunks: null,

data/docs/SearchWithinGroupResults.md

@@ -0,0 +1,22 @@
+# TrieveRubyClient::SearchWithinGroupResults
+
+## Properties
+
+| Name | Type | Description | Notes |
+| ---- | ---- | ----------- | ----- |
+| **bookmarks** | [**Array<ScoreChunkDTO>**](ScoreChunkDTO.md) |  |  |
+| **group** | [**ChunkGroup**](ChunkGroup.md) |  |  |
+| **total_pages** | **Integer** |  |  |
+
+## Example
+
+```ruby
+require 'trieve_ruby_client'
+
+instance = TrieveRubyClient::SearchWithinGroupResults.new(
+  bookmarks: null,
+  group: null,
+  total_pages: null
+)
+```
+

data/docs/SearchWithinGroupSlimResults.md

@@ -0,0 +1,22 @@
+# TrieveRubyClient::SearchWithinGroupSlimResults
+
+## Properties
+
+| Name | Type | Description | Notes |
+| ---- | ---- | ----------- | ----- |
+| **bookmarks** | [**Array<ScoreSlimChunks>**](ScoreSlimChunks.md) |  |  |
+| **group** | [**ChunkGroup**](ChunkGroup.md) |  |  |
+| **total_pages** | **Integer** |  |  |
+
+## Example
+
+```ruby
+require 'trieve_ruby_client'
+
+instance = TrieveRubyClient::SearchWithinGroupSlimResults.new(
+  bookmarks: null,
+  group: null,
+  total_pages: null
+)
+```
+

data/docs/SlimChunkMetadataWithScore.md

@@ -0,0 +1,38 @@
+# TrieveRubyClient::SlimChunkMetadataWithScore
+
+## Properties
+
+| Name | Type | Description | Notes |
+| ---- | ---- | ----------- | ----- |
+| **created_at** | **Time** |  |  |
+| **id** | **String** |  |  |
+| **link** | **String** |  | [optional] |
+| **metadata** | **Object** |  | [optional] |
+| **qdrant_point_id** | **String** |  | [optional] |
+| **score** | **Float** |  |  |
+| **tag_set** | **String** |  | [optional] |
+| **time_stamp** | **Time** |  | [optional] |
+| **tracking_id** | **String** |  | [optional] |
+| **updated_at** | **Time** |  |  |
+| **weight** | **Float** |  |  |
+
+## Example
+
+```ruby
+require 'trieve_ruby_client'
+
+instance = TrieveRubyClient::SlimChunkMetadataWithScore.new(
+  created_at: null,
+  id: null,
+  link: null,
+  metadata: null,
+  qdrant_point_id: null,
+  score: null,
+  tag_set: null,
+  time_stamp: null,
+  tracking_id: null,
+  updated_at: null,
+  weight: null
+)
+```
+

data/docs/UpdateChunkGroupData.md

@@ -6,7 +6,9 @@
 | ---- | ---- | ----------- | ----- |
 | **description** | **String** | Description to assign to the chunk_group. Convenience field for you to avoid having to remember what the group is for. If not provided, the description will not be updated. | [optional] |
 | **group_id** | **String** | Id of the chunk_group to update. | [optional] |
+| **metadata** | **Object** | Optional metadata to assign to the chunk_group. This is a JSON object that can store any additional information you want to associate with the chunks inside of the chunk_group. | [optional] |
 | **name** | **String** | Name to assign to the chunk_group. Does not need to be unique. If not provided, the name will not be updated. | [optional] |
+| **tag_set** | **Array<String>** | Optional tags to assign to the chunk_group. This is a list of strings that can be used to categorize the chunks inside the chunk_group. | [optional] |
 | **tracking_id** | **String** | Tracking Id of the chunk_group to update. | [optional] |
 
 ## Example
@@ -17,7 +19,9 @@ require 'trieve_ruby_client'
 instance = TrieveRubyClient::UpdateChunkGroupData.new(
   description: null,
   group_id: null,
+  metadata: null,
   name: null,
+  tag_set: null,
   tracking_id: null
 )
 ```

data/docs/UpdateGroupByTrackingIDData.md

@@ -5,7 +5,9 @@
 | Name | Type | Description | Notes |
 | ---- | ---- | ----------- | ----- |
 | **description** | **String** | Description to assign to the chunk_group. Convenience field for you to avoid having to remember what the group is for. If not provided, the description will not be updated. | [optional] |
+| **metadata** | **Object** | Optional metadata to assign to the chunk_group. This is a JSON object that can store any additional information you want to associate with the chunks inside of the chunk_group. | [optional] |
 | **name** | **String** | Name to assign to the chunk_group. Does not need to be unique. If not provided, the name will not be updated. | [optional] |
+| **tag_set** | **Array<String>** | Optional tags to assign to the chunk_group. This is a list of strings that can be used to categorize the chunks inside the chunk_group. | [optional] |
 | **tracking_id** | **String** | Tracking Id of the chunk_group to update. |  |
 
 ## Example
@@ -15,7 +17,9 @@ require 'trieve_ruby_client'
 
 instance = TrieveRubyClient::UpdateGroupByTrackingIDData.new(
   description: null,
+  metadata: null,
   name: null,
+  tag_set: null,
   tracking_id: null
 )
 ```

data/docs/UserApi.md

@@ -4,8 +4,8 @@ All URIs are relative to *https://api.trieve.ai*
 
 | Method | HTTP request | Description |
 | ------ | ------------ | ----------- |
-| [**delete_user_api_key**](UserApi.md#delete_user_api_key) | **DELETE** /api/user/delete_api_key/{api_key_id} | Delete User Api Key |
-| [**set_user_api_key**](UserApi.md#set_user_api_key) | **POST** /api/user/set_api_key | Set User Api Key |
+| [**delete_user_api_key**](UserApi.md#delete_user_api_key) | **DELETE** /api/user/api_key/{api_key_id} | Delete User Api Key |
+| [**set_user_api_key**](UserApi.md#set_user_api_key) | **POST** /api/user/api_key | Set User Api Key |
 | [**update_user**](UserApi.md#update_user) | **PUT** /api/user | Update User |
 
 

data/lib/trieve_ruby_client/api/auth_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.5.7
+The version of the OpenAPI document: 0.6.2
 Contact: developers@trieve.ai
 Generated by: https://openapi-generator.tech
 Generator version: 7.4.0
@@ -135,33 +135,35 @@ module TrieveRubyClient
 
     # Login
     # Login  This will redirect you to the OAuth provider for authentication with email/pass, SSO, Google, Github, etc.
-    # @param content [AuthQuery] Query parameters for login to be included as kv pairs after ? on the request URL.
     # @param [Hash] opts the optional parameters
+    # @option opts [String] :organization_id ID of organization to authenticate into
+    # @option opts [String] :redirect_uri URL to redirect to after successful login
+    # @option opts [String] :inv_code Code sent via email as a result of successful call to send_invitation
     # @return [nil]
-    def login(content, opts = {})
-      login_with_http_info(content, opts)
+    def login(opts = {})
+      login_with_http_info(opts)
       nil
     end
 
     # Login
     # Login  This will redirect you to the OAuth provider for authentication with email/pass, SSO, Google, Github, etc.
-    # @param content [AuthQuery] Query parameters for login to be included as kv pairs after ? on the request URL.
     # @param [Hash] opts the optional parameters
+    # @option opts [String] :organization_id ID of organization to authenticate into
+    # @option opts [String] :redirect_uri URL to redirect to after successful login
+    # @option opts [String] :inv_code Code sent via email as a result of successful call to send_invitation
     # @return [Array<(nil, Integer, Hash)>] nil, response status code and response headers
-    def login_with_http_info(content, opts = {})
+    def login_with_http_info(opts = {})
       if @api_client.config.debugging
         @api_client.config.logger.debug 'Calling API: AuthApi.login ...'
       end
-      # verify the required parameter 'content' is set
-      if @api_client.config.client_side_validation && content.nil?
-        fail ArgumentError, "Missing the required parameter 'content' when calling AuthApi.login"
-      end
       # resource path
       local_var_path = '/api/auth'
 
       # query parameters
       query_params = opts[:query_params] || {}
-      query_params[:'content'] = content
+      query_params[:'organization_id'] = opts[:'organization_id'] if !opts[:'organization_id'].nil?
+      query_params[:'redirect_uri'] = opts[:'redirect_uri'] if !opts[:'redirect_uri'].nil?
+      query_params[:'inv_code'] = opts[:'inv_code'] if !opts[:'inv_code'].nil?
 
       # header parameters
       header_params = opts[:header_params] || {}

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.5.7
+The version of the OpenAPI document: 0.6.2
 Contact: developers@trieve.ai
 Generated by: https://openapi-generator.tech
 Generator version: 7.4.0
@@ -529,7 +529,7 @@ module TrieveRubyClient
     # @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 [RecommendChunksResponseTypes]
+    # @return [Array<ChunkMetadataWithScore>]
     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
@@ -540,7 +540,7 @@ module TrieveRubyClient
     # @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<(RecommendChunksResponseTypes, Integer, Hash)>] RecommendChunksResponseTypes data, response status code and response headers
+    # @return [Array<(Array<ChunkMetadataWithScore>, Integer, Hash)>] Array<ChunkMetadataWithScore> 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 ...'
@@ -577,7 +577,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] || 'RecommendChunksResponseTypes'
+      return_type = opts[:debug_return_type] || 'Array<ChunkMetadataWithScore>'
 
       # 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 search_chunk_data [SearchChunkData] JSON request payload to semantically search for chunks (chunks)
     # @param [Hash] opts the optional parameters
-    # @return [SearchChunkResponseTypes]
+    # @return [SearchChunkQueryResponseBody]
     def search_chunk(tr_dataset, search_chunk_data, opts = {})
       data, _status_code, _headers = search_chunk_with_http_info(tr_dataset, search_chunk_data, opts)
       data
@@ -615,7 +615,7 @@ module TrieveRubyClient
     # @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 [Hash] opts the optional parameters
-    # @return [Array<(SearchChunkResponseTypes, Integer, Hash)>] SearchChunkResponseTypes data, response status code and response headers
+    # @return [Array<(SearchChunkQueryResponseBody, Integer, Hash)>] SearchChunkQueryResponseBody data, response status code and response headers
     def search_chunk_with_http_info(tr_dataset, search_chunk_data, opts = {})
       if @api_client.config.debugging
         @api_client.config.logger.debug 'Calling API: ChunkApi.search_chunk ...'
@@ -652,7 +652,7 @@ module TrieveRubyClient
       post_body = opts[:debug_body] || @api_client.object_to_http_body(search_chunk_data)
 
       # return_type
-      return_type = opts[:debug_return_type] || 'SearchChunkResponseTypes'
+      return_type = opts[:debug_return_type] || 'SearchChunkQueryResponseBody'
 
       # auth_names
       auth_names = opts[:debug_auth_names] || ['ApiKey']