trieve_ruby_client 0.10.7 → 0.11.0

data/lib/trieve_ruby_client/api/stripe_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
     # Cancel Subscription
-    # Cancel Subscription  Cancel a subscription by its id
+    # Cancel a subscription by its id
     # @param tr_organization [String] The organization id to use for the request
     # @param subscription_id [String] id of the subscription you want to cancel
     # @param [Hash] opts the optional parameters
@@ -31,7 +31,7 @@ module TrieveRubyClient
     end
 
     # Cancel Subscription
-    # Cancel Subscription  Cancel a subscription by its id
+    # Cancel a subscription by its id
     # @param tr_organization [String] The organization id to use for the request
     # @param subscription_id [String] id of the subscription you want to cancel
     # @param [Hash] opts the optional parameters
@@ -89,8 +89,71 @@ module TrieveRubyClient
       return data, status_code, headers
     end
 
+    # Create checkout session setup
+    # Create a checkout session (setup)
+    # @param organization_id [String] The id of the organization to create setup checkout session for.
+    # @param [Hash] opts the optional parameters
+    # @return [CreateSetupCheckoutSessionResPayload]
+    def create_setup_checkout_session(organization_id, opts = {})
+      data, _status_code, _headers = create_setup_checkout_session_with_http_info(organization_id, opts)
+      data
+    end
+
+    # Create checkout session setup
+    # Create a checkout session (setup)
+    # @param organization_id [String] The id of the organization to create setup checkout session for.
+    # @param [Hash] opts the optional parameters
+    # @return [Array<(CreateSetupCheckoutSessionResPayload, Integer, Hash)>] CreateSetupCheckoutSessionResPayload data, response status code and response headers
+    def create_setup_checkout_session_with_http_info(organization_id, opts = {})
+      if @api_client.config.debugging
+        @api_client.config.logger.debug 'Calling API: StripeApi.create_setup_checkout_session ...'
+      end
+      # verify the required parameter 'organization_id' is set
+      if @api_client.config.client_side_validation && organization_id.nil?
+        fail ArgumentError, "Missing the required parameter 'organization_id' when calling StripeApi.create_setup_checkout_session"
+      end
+      # resource path
+      local_var_path = '/api/stripe/checkout/setup/{organization_id}'.sub('{' + 'organization_id' + '}', CGI.escape(organization_id.to_s))
+
+      # 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'])
+
+      # form parameters
+      form_params = opts[:form_params] || {}
+
+      # http body (model)
+      post_body = opts[:debug_body]
+
+      # return_type
+      return_type = opts[:debug_return_type] || 'CreateSetupCheckoutSessionResPayload'
+
+      # auth_names
+      auth_names = opts[:debug_auth_names] || []
+
+      new_options = opts.merge(
+        :operation => :"StripeApi.create_setup_checkout_session",
+        :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: StripeApi#create_setup_checkout_session\nData: #{data.inspect}\nStatus code: #{status_code}\nHeaders: #{headers}"
+      end
+      return data, status_code, headers
+    end
+
     # Checkout
-    # Checkout  Get a direct link to the stripe checkout page for the plan and organization
+    # Get a 303 SeeOther redirect link to the stripe checkout page for the plan and organization
     # @param plan_id [String] id of the plan you want to subscribe to
     # @param organization_id [String] id of the organization you want to subscribe to the plan
     # @param [Hash] opts the optional parameters
@@ -101,7 +164,7 @@ module TrieveRubyClient
     end
 
     # Checkout
-    # Checkout  Get a direct link to the stripe checkout page for the plan and organization
+    # Get a 303 SeeOther redirect link to the stripe checkout page for the plan and organization
     # @param plan_id [String] id of the plan you want to subscribe to
     # @param organization_id [String] id of the organization you want to subscribe to the plan
     # @param [Hash] opts the optional parameters
@@ -158,8 +221,71 @@ module TrieveRubyClient
       return data, status_code, headers
     end
 
+    # Get All Invoices
+    # Get a list of all invoices
+    # @param organization_id [String] The id of the organization to get invoices for.
+    # @param [Hash] opts the optional parameters
+    # @return [Array<StripeInvoice>]
+    def get_all_invoices(organization_id, opts = {})
+      data, _status_code, _headers = get_all_invoices_with_http_info(organization_id, opts)
+      data
+    end
+
+    # Get All Invoices
+    # Get a list of all invoices
+    # @param organization_id [String] The id of the organization to get invoices for.
+    # @param [Hash] opts the optional parameters
+    # @return [Array<(Array<StripeInvoice>, Integer, Hash)>] Array<StripeInvoice> data, response status code and response headers
+    def get_all_invoices_with_http_info(organization_id, opts = {})
+      if @api_client.config.debugging
+        @api_client.config.logger.debug 'Calling API: StripeApi.get_all_invoices ...'
+      end
+      # verify the required parameter 'organization_id' is set
+      if @api_client.config.client_side_validation && organization_id.nil?
+        fail ArgumentError, "Missing the required parameter 'organization_id' when calling StripeApi.get_all_invoices"
+      end
+      # resource path
+      local_var_path = '/api/stripe/invoices/{organization_id}'.sub('{' + 'organization_id' + '}', CGI.escape(organization_id.to_s))
+
+      # 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'])
+
+      # form parameters
+      form_params = opts[:form_params] || {}
+
+      # http body (model)
+      post_body = opts[:debug_body]
+
+      # return_type
+      return_type = opts[:debug_return_type] || 'Array<StripeInvoice>'
+
+      # auth_names
+      auth_names = opts[:debug_auth_names] || []
+
+      new_options = opts.merge(
+        :operation => :"StripeApi.get_all_invoices",
+        :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(:GET, local_var_path, new_options)
+      if @api_client.config.debugging
+        @api_client.config.logger.debug "API called: StripeApi#get_all_invoices\nData: #{data.inspect}\nStatus code: #{status_code}\nHeaders: #{headers}"
+      end
+      return data, status_code, headers
+    end
+
     # Get All Plans
-    # Get All Plans  Get a list of all plans
+    # Get a list of all plans
     # @param [Hash] opts the optional parameters
     # @return [Array<StripePlan>]
     def get_all_plans(opts = {})
@@ -168,7 +294,7 @@ module TrieveRubyClient
     end
 
     # Get All Plans
-    # Get All Plans  Get a list of all plans
+    # Get a list of all plans
     # @param [Hash] opts the optional parameters
     # @return [Array<(Array<StripePlan>, Integer, Hash)>] Array<StripePlan> data, response status code and response headers
     def get_all_plans_with_http_info(opts = {})
@@ -216,7 +342,7 @@ module TrieveRubyClient
     end
 
     # Update Subscription Plan
-    # Update Subscription Plan  Update a subscription to a new plan
+    # Update a subscription to a new plan
     # @param tr_organization [String] The organization id to use for the request
     # @param subscription_id [String] id of the subscription you want to update
     # @param plan_id [String] id of the plan you want to subscribe to
@@ -228,7 +354,7 @@ module TrieveRubyClient
     end
 
     # Update Subscription Plan
-    # Update Subscription Plan  Update a subscription to a new plan
+    # Update a subscription to a new plan
     # @param tr_organization [String] The organization id to use for the request
     # @param subscription_id [String] id of the subscription you want to update
     # @param plan_id [String] id of the plan you want to subscribe to

data/lib/trieve_ruby_client/api/topic_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
     # Create Topic
-    # Create Topic  Create a new chat topic. Topics are attached to a owner_id's and act as a coordinator for conversation message history of gen-AI chat sessions. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
+    # Create a new chat topic. Topics are attached to a owner_id's and act as a coordinator for conversation message history of gen-AI chat sessions. 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_topic_req_payload [CreateTopicReqPayload] JSON request payload to create chat topic
     # @param [Hash] opts the optional parameters
@@ -31,7 +31,7 @@ module TrieveRubyClient
     end
 
     # Create Topic
-    # Create Topic  Create a new chat topic. Topics are attached to a owner_id's and act as a coordinator for conversation message history of gen-AI chat sessions. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
+    # Create a new chat topic. Topics are attached to a owner_id's and act as a coordinator for conversation message history of gen-AI chat sessions. 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_topic_req_payload [CreateTopicReqPayload] JSON request payload to create chat topic
     # @param [Hash] opts the optional parameters
@@ -95,7 +95,7 @@ module TrieveRubyClient
     end
 
     # Delete Topic
-    # Delete Topic  Delete an existing chat topic. When a topic is deleted, all associated chat messages are also deleted. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
+    # Delete an existing chat topic. When a topic is deleted, all associated chat messages are also deleted. 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 topic_id [String] The id of the topic you want to delete.
     # @param [Hash] opts the optional parameters
@@ -106,7 +106,7 @@ module TrieveRubyClient
     end
 
     # Delete Topic
-    # Delete Topic  Delete an existing chat topic. When a topic is deleted, all associated chat messages are also deleted. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
+    # Delete an existing chat topic. When a topic is deleted, all associated chat messages are also deleted. 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 topic_id [String] The id of the topic you want to delete.
     # @param [Hash] opts the optional parameters
@@ -165,7 +165,7 @@ module TrieveRubyClient
     end
 
     # Get All Topics for Owner ID
-    # Get All Topics for Owner ID  Get all topics belonging to an arbitary owner_id. This is useful for managing message history and chat sessions. It is common to use a browser fingerprint or your user's id as the owner_id. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
+    # Get all topics belonging to an arbitary owner_id. This is useful for managing message history and chat sessions. It is common to use a browser fingerprint or your user's id as the owner_id. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
     # @param owner_id [String] The owner_id to get topics of; A common approach is to use a browser fingerprint or your user's id
     # @param tr_dataset [String] The dataset id to use for the request
     # @param [Hash] opts the optional parameters
@@ -176,7 +176,7 @@ module TrieveRubyClient
     end
 
     # Get All Topics for Owner ID
-    # Get All Topics for Owner ID  Get all topics belonging to an arbitary owner_id. This is useful for managing message history and chat sessions. It is common to use a browser fingerprint or your user's id as the owner_id. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
+    # Get all topics belonging to an arbitary owner_id. This is useful for managing message history and chat sessions. It is common to use a browser fingerprint or your user's id as the owner_id. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
     # @param owner_id [String] The owner_id to get topics of; A common approach is to use a browser fingerprint or your user's id
     # @param tr_dataset [String] The dataset id to use for the request
     # @param [Hash] opts the optional parameters
@@ -235,7 +235,7 @@ module TrieveRubyClient
     end
 
     # Update Topic
-    # Update Topic  Update an existing chat topic. Currently, only the name of the topic can be updated. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
+    # Update an existing chat topic. Currently, only the name of the topic can be updated. 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_topic_req_payload [UpdateTopicReqPayload] JSON request payload to update a chat topic
     # @param [Hash] opts the optional parameters
@@ -246,7 +246,7 @@ module TrieveRubyClient
     end
 
     # Update Topic
-    # Update Topic  Update an existing chat topic. Currently, only the name of the topic can be updated. Auth'ed user or api key must have an admin or owner role for the specified dataset's organization.
+    # Update an existing chat topic. Currently, only the name of the topic can be updated. 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_topic_req_payload [UpdateTopicReqPayload] JSON request payload to update a chat topic
     # @param [Hash] opts the optional parameters

data/lib/trieve_ruby_client/api/user_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
     # Delete User Api Key
-    # Delete User Api Key  Delete an api key for the auth'ed user.
+    # Delete an api key for the auth'ed user.
     # @param api_key_id [String] The id of the api key to delete
     # @param [Hash] opts the optional parameters
     # @return [nil]
@@ -30,7 +30,7 @@ module TrieveRubyClient
     end
 
     # Delete User Api Key
-    # Delete User Api Key  Delete an api key for the auth'ed user.
+    # Delete an api key for the auth'ed user.
     # @param api_key_id [String] The id of the api key to delete
     # @param [Hash] opts the optional parameters
     # @return [Array<(nil, Integer, Hash)>] nil, response status code and response headers
@@ -83,7 +83,7 @@ module TrieveRubyClient
     end
 
     # Set User Api Key
-    # Set User Api Key  Create a new api key for the auth'ed user. Successful response will contain the newly created api key. If a write role is assigned the api key will have permission level of the auth'ed user who calls this endpoint.
+    # Create a new api key for the auth'ed user. Successful response will contain the newly created api key. If a write role is assigned the api key will have permission level of the auth'ed user who calls this endpoint.
     # @param set_user_api_key_request [SetUserApiKeyRequest] JSON request payload to create a new user api key
     # @param [Hash] opts the optional parameters
     # @return [SetUserApiKeyResponse]
@@ -93,7 +93,7 @@ module TrieveRubyClient
     end
 
     # Set User Api Key
-    # Set User Api Key  Create a new api key for the auth&#39;ed user. Successful response will contain the newly created api key. If a write role is assigned the api key will have permission level of the auth&#39;ed user who calls this endpoint.
+    # Create a new api key for the auth&#39;ed user. Successful response will contain the newly created api key. If a write role is assigned the api key will have permission level of the auth&#39;ed user who calls this endpoint.
     # @param set_user_api_key_request [SetUserApiKeyRequest] JSON request payload to create a new user api key
     # @param [Hash] opts the optional parameters
     # @return [Array<(SetUserApiKeyResponse, Integer, Hash)>] SetUserApiKeyResponse data, response status code and response headers
@@ -151,7 +151,7 @@ module TrieveRubyClient
     end
 
     # Update User
-    # Update User  Update a user's information. If the user_id is not provided, the auth'ed user will be updated. If the user_id is provided, the role of the auth'ed user or api key must be an admin (1) or owner (2) of the organization.
+    # Update a user's information. If the user_id is not provided, the auth'ed user will be updated. If the user_id is provided, the role of the auth'ed user or api key must be an admin (1) or owner (2) of the organization.
     # @param update_user_org_role_data [UpdateUserOrgRoleData] JSON request payload to update user information for the auth&#39;ed user
     # @param [Hash] opts the optional parameters
     # @return [nil]
@@ -161,7 +161,7 @@ module TrieveRubyClient
     end
 
     # Update User
-    # Update User  Update a user&#39;s information. If the user_id is not provided, the auth&#39;ed user will be updated. If the user_id is provided, the role of the auth&#39;ed user or api key must be an admin (1) or owner (2) of the organization.
+    # Update a user&#39;s information. If the user_id is not provided, the auth&#39;ed user will be updated. If the user_id is provided, the role of the auth&#39;ed user or api key must be an admin (1) or owner (2) of the organization.
     # @param update_user_org_role_data [UpdateUserOrgRoleData] JSON request payload to update user information for the auth&#39;ed user
     # @param [Hash] opts the optional parameters
     # @return [Array<(nil, Integer, Hash)>] nil, response status code and response headers

data/lib/trieve_ruby_client/api_client.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

data/lib/trieve_ruby_client/api_error.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

data/lib/trieve_ruby_client/configuration.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
@@ -241,6 +241,13 @@ module TrieveRubyClient
             key: 'Authorization',
             value: api_key_with_prefix('ApiKey')
           },
+        'X-API-KEY' =>
+          {
+            type: 'api_key',
+            in: 'header',
+            key: 'X-API-KEY',
+            value: api_key_with_prefix('X-API-KEY')
+          },
       }
     end
 

data/lib/trieve_ruby_client/models/add_chunk_to_group_data.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

data/lib/trieve_ruby_client/models/api_key_resp_body.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

data/lib/trieve_ruby_client/models/api_version.rb

@@ -0,0 +1,40 @@
+=begin
+#Trieve API
+
+#Trieve OpenAPI Specification. This document describes all of the operations available through the Trieve API.
+
+The version of the OpenAPI document: 0.11.0
+Contact: developers@trieve.ai
+Generated by: https://openapi-generator.tech
+Generator version: 7.4.0
+
+=end
+
+require 'date'
+require 'time'
+
+module TrieveRubyClient
+  class APIVersion
+    V1 = "V1".freeze
+    V2 = "V2".freeze
+
+    def self.all_vars
+      @all_vars ||= [V1, V2].freeze
+    end
+
+    # Builds the enum from string
+    # @param [String] The enum value in the form of the string
+    # @return [String] The enum value
+    def self.build_from_hash(value)
+      new.build_from_hash(value)
+    end
+
+    # Builds the enum from string
+    # @param [String] The enum value in the form of the string
+    # @return [String] The enum value
+    def build_from_hash(value)
+      return value if APIVersion.all_vars.include?(value)
+      raise "Invalid ENUM value #{value} for class #APIVersion"
+    end
+  end
+end

data/lib/trieve_ruby_client/models/auth_query.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

data/lib/trieve_ruby_client/models/autocomplete_req_payload.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
@@ -41,6 +41,8 @@ module TrieveRubyClient
     # Set highlight_window to a number to control the amount of words that are returned around the matched phrases. If not specified, this defaults to 0. This is useful for when you want to show more context around the matched words. When specified, window/2 whitespace separated words are added before and after each highlight in the response's highlights array. If an extended highlight overlaps with another highlight, the overlapping words are only included once.
     attr_accessor :highlight_window
 
+    attr_accessor :location_bias
+
     # Page size is the number of chunks to fetch. This can be used to fetch more than 10 chunks at a time.
     attr_accessor :page_size
 
@@ -50,10 +52,9 @@ module TrieveRubyClient
     # Recency Bias lets you determine how much of an effect the recency of chunks will have on the search results. If not specified, this defaults to 0.0. We recommend setting this to 1.0 for a gentle reranking of the results, >3.0 for a strong reranking of the results.
     attr_accessor :recency_bias
 
-    # Set score_threshold to a float to filter out chunks with a score below the threshold.
+    # Set score_threshold to a float to filter out chunks with a score below the threshold. This threshold applies before weight and bias modifications. If not specified, this defaults to 0.0.
     attr_accessor :score_threshold
 
-    # Can be either \"semantic\", or \"fulltext\". \"semantic\" will pull in one page_size of the nearest cosine distant vectors. \"fulltext\" will pull in one page_size of full-text results based on SPLADE.
     attr_accessor :search_type
 
     # 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 (typically 10-50ms). Default is false.
@@ -62,9 +63,34 @@ module TrieveRubyClient
     # Tag weights is a JSON object which can be used to boost the ranking of chunks with certain tags. This is useful for when you want to be able to bias towards chunks with a certain tag on the fly. The keys are the tag names and the values are the weights.
     attr_accessor :tag_weights
 
+    # If true, chunks will be reranked using scores from a cross encoder model. \"hybrid\" search will always use the reranker regardless of this setting.
+    attr_accessor :use_reranker
+
     # Set use_weights to true to use the weights of the chunks in the result set in order to sort them. If not specified, this defaults to true.
     attr_accessor :use_weights
 
+    class EnumAttributeValidator
+      attr_reader :datatype
+      attr_reader :allowable_values
+
+      def initialize(datatype, allowable_values)
+        @allowable_values = allowable_values.map do |value|
+          case datatype.to_s
+          when /Integer/i
+            value.to_i
+          when /Float/i
+            value.to_f
+          else
+            value
+          end
+        end
+      end
+
+      def valid?(value)
+        !value || allowable_values.include?(value)
+      end
+    end
+
     # Attribute mapping from ruby-style variable name to JSON key.
     def self.attribute_map
       {
@@ -77,6 +103,7 @@ module TrieveRubyClient
         :'highlight_results' => :'highlight_results',
         :'highlight_threshold' => :'highlight_threshold',
         :'highlight_window' => :'highlight_window',
+        :'location_bias' => :'location_bias',
         :'page_size' => :'page_size',
         :'query' => :'query',
         :'recency_bias' => :'recency_bias',
@@ -84,6 +111,7 @@ module TrieveRubyClient
         :'search_type' => :'search_type',
         :'slim_chunks' => :'slim_chunks',
         :'tag_weights' => :'tag_weights',
+        :'use_reranker' => :'use_reranker',
         :'use_weights' => :'use_weights'
       }
     end
@@ -105,13 +133,15 @@ module TrieveRubyClient
         :'highlight_results' => :'Boolean',
         :'highlight_threshold' => :'Float',
         :'highlight_window' => :'Integer',
+        :'location_bias' => :'GeoInfoWithBias',
         :'page_size' => :'Integer',
         :'query' => :'String',
         :'recency_bias' => :'Float',
         :'score_threshold' => :'Float',
-        :'search_type' => :'String',
+        :'search_type' => :'SearchMethod',
         :'slim_chunks' => :'Boolean',
         :'tag_weights' => :'Hash<String, Float>',
+        :'use_reranker' => :'Boolean',
         :'use_weights' => :'Boolean'
       }
     end
@@ -128,11 +158,13 @@ module TrieveRubyClient
         :'highlight_results',
         :'highlight_threshold',
         :'highlight_window',
+        :'location_bias',
         :'page_size',
         :'recency_bias',
         :'score_threshold',
         :'slim_chunks',
         :'tag_weights',
+        :'use_reranker',
         :'use_weights'
       ])
     end
@@ -190,6 +222,10 @@ module TrieveRubyClient
         self.highlight_window = attributes[:'highlight_window']
       end
 
+      if attributes.key?(:'location_bias')
+        self.location_bias = attributes[:'location_bias']
+      end
+
       if attributes.key?(:'page_size')
         self.page_size = attributes[:'page_size']
       end
@@ -224,6 +260,10 @@ module TrieveRubyClient
         end
       end
 
+      if attributes.key?(:'use_reranker')
+        self.use_reranker = attributes[:'use_reranker']
+      end
+
       if attributes.key?(:'use_weights')
         self.use_weights = attributes[:'use_weights']
       end
@@ -328,6 +368,7 @@ module TrieveRubyClient
           highlight_results == o.highlight_results &&
           highlight_threshold == o.highlight_threshold &&
           highlight_window == o.highlight_window &&
+          location_bias == o.location_bias &&
           page_size == o.page_size &&
           query == o.query &&
           recency_bias == o.recency_bias &&
@@ -335,6 +376,7 @@ module TrieveRubyClient
           search_type == o.search_type &&
           slim_chunks == o.slim_chunks &&
           tag_weights == o.tag_weights &&
+          use_reranker == o.use_reranker &&
           use_weights == o.use_weights
     end
 
@@ -347,7 +389,7 @@ module TrieveRubyClient
     # Calculates hash code according to all attributes.
     # @return [Integer] Hash code
     def hash
-      [content_only, extend_results, filters, highlight_delimiters, highlight_max_length, highlight_max_num, highlight_results, highlight_threshold, highlight_window, page_size, query, recency_bias, score_threshold, search_type, slim_chunks, tag_weights, use_weights].hash
+      [content_only, extend_results, filters, highlight_delimiters, highlight_max_length, highlight_max_num, highlight_results, highlight_threshold, highlight_window, location_bias, page_size, query, recency_bias, score_threshold, search_type, slim_chunks, tag_weights, use_reranker, use_weights].hash
     end
 
     # Builds the object from hash

data/lib/trieve_ruby_client/models/batch_queued_chunk_response.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