feat: add support for limit query parameter

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 11
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/turbopuffer-benesch%2Fturbopuffer-8f6363d984eadd23f3dd12e8607be864813cfe08a3a86df083bcabc07e5f9553.yml
-openapi_spec_hash: 4dc391a593c54bc32ae5f0a73a45e2b6
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/turbopuffer-benesch%2Fturbopuffer-1c3d2e045078b0ae0c1b6e233f8b437002bfcc0cfc5476a55e7363a9b41772c5.yml
+openapi_spec_hash: 56a385c701bc60fab06857b4bcdeee2a
 config_hash: f55839cc66fb5661bc3292f5f3ac48cf

src/resources/namespaces.ts

@@ -473,6 +473,11 @@ export interface Query {
    */
   include_attributes?: IncludeAttributes;
 
+  /**
+   * Limit configuration for query results.
+   */
+  limit?: number | Query.Limit;
+
   /**
    * How to rank the documents in the namespace.
    */
@@ -484,6 +489,25 @@ export interface Query {
   top_k?: number;
 }
 
+export namespace Query {
+  export interface Limit {
+    /**
+     * The total number of results to return.
+     */
+    total: number;
+
+    per?: Limit.Per;
+  }
+
+  export namespace Limit {
+    export interface Per {
+      attributes: Array<string>;
+
+      limit: number;
+    }
+  }
+}
+
 /**
  * The billing information for a query.
  */
@@ -836,6 +860,11 @@ export interface NamespaceExplainQueryParams {
    */
   include_attributes?: IncludeAttributes;
 
+  /**
+   * Body param: Limit configuration for query results.
+   */
+  limit?: number | NamespaceExplainQueryParams.Limit;
+
   /**
    * Body param: How to rank the documents in the namespace.
    */
@@ -867,6 +896,23 @@ export namespace NamespaceExplainQueryParams {
      */
     level?: 'strong' | 'eventual';
   }
+
+  export interface Limit {
+    /**
+     * The total number of results to return.
+     */
+    total: number;
+
+    per?: Limit.Per;
+  }
+
+  export namespace Limit {
+    export interface Per {
+      attributes: Array<string>;
+
+      limit: number;
+    }
+  }
 }
 
 export interface NamespaceHintCacheWarmParams {
@@ -967,6 +1013,11 @@ export interface NamespaceQueryParams {
    */
   include_attributes?: IncludeAttributes;
 
+  /**
+   * Body param: Limit configuration for query results.
+   */
+  limit?: number | NamespaceQueryParams.Limit;
+
   /**
    * Body param: How to rank the documents in the namespace.
    */
@@ -998,6 +1049,23 @@ export namespace NamespaceQueryParams {
      */
     level?: 'strong' | 'eventual';
   }
+
+  export interface Limit {
+    /**
+     * The total number of results to return.
+     */
+    total: number;
+
+    per?: Limit.Per;
+  }
+
+  export namespace Limit {
+    export interface Per {
+      attributes: Array<string>;
+
+      limit: number;
+    }
+  }
 }
 
 export interface NamespaceRecallParams {

tests/api-resources/namespaces.test.ts

@@ -46,6 +46,7 @@ describe('resource namespaces', () => {
       exclude_attributes: ['string'],
       group_by: ['string'],
       include_attributes: true,
+      limit: 0,
       top_k: 0,
       vector_encoding: 'float',
     });
@@ -109,6 +110,7 @@ describe('resource namespaces', () => {
           exclude_attributes: ['string'],
           group_by: ['string'],
           include_attributes: true,
+          limit: 0,
           rank_by: ['id', 'asc'],
           top_k: 0,
         },
@@ -139,6 +141,7 @@ describe('resource namespaces', () => {
       exclude_attributes: ['string'],
       group_by: ['string'],
       include_attributes: true,
+      limit: 0,
       rank_by: ['id', 'asc'],
       top_k: 0,
       vector_encoding: 'float',