llama-stack-client 0.2.22 → 0.2.23-rc1

package/resources/responses/responses.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"responses.mjs","sourceRoot":"","sources":["../../src/resources/responses/responses.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAAE,WAAW,EAAE;OACf,EAAE,gBAAgB,EAAE;OAIpB,KAAK,aAAa;OAClB,EAA8C,UAAU,EAAE;OAC1D,EAAE,gBAAgB,EAA+B;AAGxD,MAAM,OAAO,SAAU,SAAQ,WAAW;IAA1C;;QACE,eAAU,GAA6B,IAAI,aAAa,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAsDpF,CAAC;IAxCC,MAAM,CACJ,IAA0B,EAC1B,OAA6B;QAE7B,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,yBAAyB,EAAE;YAClD,IAAI;YACJ,GAAG,OAAO;YACV,MAAM,EAAE,IAAI,CAAC,MAAM,IAAI,KAAK;SAC7B,CAA0E,CAAC;IAC9E,CAAC;IAED;;OAEG;IACH,QAAQ,CAAC,UAAkB,EAAE,OAA6B;QACxD,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,2BAA2B,UAAU,EAAE,EAAE,OAAO,CAAC,CAAC;IAC5E,CAAC;IAYD,IAAI,CACF,QAAkD,EAAE,EACpD,OAA6B;QAE7B,IAAI,gBAAgB,CAAC,KAAK,CAAC,EAAE;YAC3B,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;SAC7B;QACD,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,yBAAyB,EAAE,qCAAqC,EAAE;YAC/F,KAAK;YACL,GAAG,OAAO;SACX,CAAC,CAAC;IACL,CAAC;CACF;AAED,MAAM,OAAO,qCAAsC,SAAQ,gBAAsC;CAAG;AA6gGpG,SAAS,CAAC,qCAAqC,GAAG,qCAAqC,CAAC;AACxF,SAAS,CAAC,UAAU,GAAG,UAAU,CAAC"}
+{"version":3,"file":"responses.mjs","sourceRoot":"","sources":["../../src/resources/responses/responses.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAAE,WAAW,EAAE;OACf,EAAE,gBAAgB,EAAE;OAIpB,KAAK,aAAa;OAClB,EAA8C,UAAU,EAAE;OAC1D,EAAE,gBAAgB,EAA+B;AAGxD,MAAM,OAAO,SAAU,SAAQ,WAAW;IAA1C;;QACE,eAAU,GAA6B,IAAI,aAAa,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IA6DpF,CAAC;IA/CC,MAAM,CACJ,IAA0B,EAC1B,OAA6B;QAE7B,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,yBAAyB,EAAE;YAClD,IAAI;YACJ,GAAG,OAAO;YACV,MAAM,EAAE,IAAI,CAAC,MAAM,IAAI,KAAK;SAC7B,CAA0E,CAAC;IAC9E,CAAC;IAED;;OAEG;IACH,QAAQ,CAAC,UAAkB,EAAE,OAA6B;QACxD,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,2BAA2B,UAAU,EAAE,EAAE,OAAO,CAAC,CAAC;IAC5E,CAAC;IAYD,IAAI,CACF,QAAkD,EAAE,EACpD,OAA6B;QAE7B,IAAI,gBAAgB,CAAC,KAAK,CAAC,EAAE;YAC3B,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;SAC7B;QACD,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,yBAAyB,EAAE,qCAAqC,EAAE;YAC/F,KAAK;YACL,GAAG,OAAO;SACX,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,UAAkB,EAAE,OAA6B;QACtD,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,2BAA2B,UAAU,EAAE,EAAE,OAAO,CAAC,CAAC;IAC/E,CAAC;CACF;AAED,MAAM,OAAO,qCAAsC,SAAQ,gBAAsC;CAAG;AAiiGpG,SAAS,CAAC,qCAAqC,GAAG,qCAAqC,CAAC;AACxF,SAAS,CAAC,UAAU,GAAG,UAAU,CAAC"}

package/resources/shields.d.ts

@@ -9,6 +9,10 @@ export declare class Shields extends APIResource {
      * List all shields.
      */
     list(options?: Core.RequestOptions): Core.APIPromise<ShieldListResponse>;
+    /**
+     * Unregister a shield.
+     */
+    delete(identifier: string, options?: Core.RequestOptions): Core.APIPromise<void>;
     /**
      * Register a shield.
      */

package/resources/shields.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"shields.d.ts","sourceRoot":"","sources":["../src/resources/shields.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,KAAK,IAAI,MAAM,SAAS,CAAC;AAEhC,qBAAa,OAAQ,SAAQ,WAAW;IACtC;;OAEG;IACH,QAAQ,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC;IAIpF;;OAEG;IACH,IAAI,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,kBAAkB,CAAC;IAMxE;;OAEG;IACH,QAAQ,CAAC,IAAI,EAAE,oBAAoB,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC;CAG7F;AAED,MAAM,WAAW,mBAAmB;IAClC,IAAI,EAAE,kBAAkB,CAAC;CAC1B;AAED;;GAEG;AACH,MAAM,WAAW,MAAM;IACrB,UAAU,EAAE,MAAM,CAAC;IAEnB,WAAW,EAAE,MAAM,CAAC;IAEpB;;OAEG;IACH,IAAI,EAAE,QAAQ,CAAC;IAEf;;OAEG;IACH,MAAM,CAAC,EAAE;QAAE,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,GAAG,IAAI,CAAA;KAAE,CAAC;IAExF,oBAAoB,CAAC,EAAE,MAAM,CAAC;CAC/B;AAED,MAAM,MAAM,kBAAkB,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC;AAE/C,MAAM,WAAW,oBAAoB;IACnC;;OAEG;IACH,SAAS,EAAE,MAAM,CAAC;IAElB;;OAEG;IACH,MAAM,CAAC,EAAE;QAAE,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,GAAG,IAAI,CAAA;KAAE,CAAC;IAExF;;OAEG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IAErB;;OAEG;IACH,kBAAkB,CAAC,EAAE,MAAM,CAAC;CAC7B;AAED,MAAM,CAAC,OAAO,WAAW,OAAO,CAAC;IAC/B,OAAO,EACL,KAAK,mBAAmB,IAAI,mBAAmB,EAC/C,KAAK,MAAM,IAAI,MAAM,EACrB,KAAK,kBAAkB,IAAI,kBAAkB,EAC7C,KAAK,oBAAoB,IAAI,oBAAoB,GAClD,CAAC;CACH"}
+{"version":3,"file":"shields.d.ts","sourceRoot":"","sources":["../src/resources/shields.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,KAAK,IAAI,MAAM,SAAS,CAAC;AAEhC,qBAAa,OAAQ,SAAQ,WAAW;IACtC;;OAEG;IACH,QAAQ,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC;IAIpF;;OAEG;IACH,IAAI,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,kBAAkB,CAAC;IAMxE;;OAEG;IACH,MAAM,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;IAOhF;;OAEG;IACH,QAAQ,CAAC,IAAI,EAAE,oBAAoB,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC;CAG7F;AAED,MAAM,WAAW,mBAAmB;IAClC,IAAI,EAAE,kBAAkB,CAAC;CAC1B;AAED;;GAEG;AACH,MAAM,WAAW,MAAM;IACrB,UAAU,EAAE,MAAM,CAAC;IAEnB,WAAW,EAAE,MAAM,CAAC;IAEpB;;OAEG;IACH,IAAI,EAAE,QAAQ,CAAC;IAEf;;OAEG;IACH,MAAM,CAAC,EAAE;QAAE,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,GAAG,IAAI,CAAA;KAAE,CAAC;IAExF,oBAAoB,CAAC,EAAE,MAAM,CAAC;CAC/B;AAED,MAAM,MAAM,kBAAkB,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC;AAE/C,MAAM,WAAW,oBAAoB;IACnC;;OAEG;IACH,SAAS,EAAE,MAAM,CAAC;IAElB;;OAEG;IACH,MAAM,CAAC,EAAE;QAAE,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,GAAG,IAAI,CAAA;KAAE,CAAC;IAExF;;OAEG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IAErB;;OAEG;IACH,kBAAkB,CAAC,EAAE,MAAM,CAAC;CAC7B;AAED,MAAM,CAAC,OAAO,WAAW,OAAO,CAAC;IAC/B,OAAO,EACL,KAAK,mBAAmB,IAAI,mBAAmB,EAC/C,KAAK,MAAM,IAAI,MAAM,EACrB,KAAK,kBAAkB,IAAI,kBAAkB,EAC7C,KAAK,oBAAoB,IAAI,oBAAoB,GAClD,CAAC;CACH"}

package/resources/shields.js

@@ -16,6 +16,15 @@ class Shields extends resource_1.APIResource {
     list(options) {
         return this._client.get('/v1/shields', options)._thenUnwrap((obj) => obj.data);
     }
+    /**
+     * Unregister a shield.
+     */
+    delete(identifier, options) {
+        return this._client.delete(`/v1/shields/${identifier}`, {
+            ...options,
+            headers: { Accept: '*/*', ...options?.headers },
+        });
+    }
     /**
      * Register a shield.
      */

package/resources/shields.js.map

@@ -1 +1 @@
-{"version":3,"file":"shields.js","sourceRoot":"","sources":["../src/resources/shields.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;AAEtF,6CAA0C;AAG1C,MAAa,OAAQ,SAAQ,sBAAW;IACtC;;OAEG;IACH,QAAQ,CAAC,UAAkB,EAAE,OAA6B;QACxD,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,eAAe,UAAU,EAAE,EAAE,OAAO,CAAC,CAAC;IAChE,CAAC;IAED;;OAEG;IACH,IAAI,CAAC,OAA6B;QAChC,OACE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,OAAO,CACxC,CAAC,WAAW,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAED;;OAEG;IACH,QAAQ,CAAC,IAA0B,EAAE,OAA6B;QAChE,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,aAAa,EAAE,EAAE,IAAI,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IAChE,CAAC;CACF;AAvBD,0BAuBC"}
+{"version":3,"file":"shields.js","sourceRoot":"","sources":["../src/resources/shields.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;AAEtF,6CAA0C;AAG1C,MAAa,OAAQ,SAAQ,sBAAW;IACtC;;OAEG;IACH,QAAQ,CAAC,UAAkB,EAAE,OAA6B;QACxD,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,eAAe,UAAU,EAAE,EAAE,OAAO,CAAC,CAAC;IAChE,CAAC;IAED;;OAEG;IACH,IAAI,CAAC,OAA6B;QAChC,OACE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,OAAO,CACxC,CAAC,WAAW,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,UAAkB,EAAE,OAA6B;QACtD,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,eAAe,UAAU,EAAE,EAAE;YACtD,GAAG,OAAO;YACV,OAAO,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,OAAO,EAAE;SAChD,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,QAAQ,CAAC,IAA0B,EAAE,OAA6B;QAChE,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,aAAa,EAAE,EAAE,IAAI,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IAChE,CAAC;CACF;AAjCD,0BAiCC"}

package/resources/shields.mjs

@@ -13,6 +13,15 @@ export class Shields extends APIResource {
     list(options) {
         return this._client.get('/v1/shields', options)._thenUnwrap((obj) => obj.data);
     }
+    /**
+     * Unregister a shield.
+     */
+    delete(identifier, options) {
+        return this._client.delete(`/v1/shields/${identifier}`, {
+            ...options,
+            headers: { Accept: '*/*', ...options?.headers },
+        });
+    }
     /**
      * Register a shield.
      */

package/resources/shields.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"shields.mjs","sourceRoot":"","sources":["../src/resources/shields.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAAE,WAAW,EAAE;AAGtB,MAAM,OAAO,OAAQ,SAAQ,WAAW;IACtC;;OAEG;IACH,QAAQ,CAAC,UAAkB,EAAE,OAA6B;QACxD,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,eAAe,UAAU,EAAE,EAAE,OAAO,CAAC,CAAC;IAChE,CAAC;IAED;;OAEG;IACH,IAAI,CAAC,OAA6B;QAChC,OACE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,OAAO,CACxC,CAAC,WAAW,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAED;;OAEG;IACH,QAAQ,CAAC,IAA0B,EAAE,OAA6B;QAChE,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,aAAa,EAAE,EAAE,IAAI,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IAChE,CAAC;CACF"}
+{"version":3,"file":"shields.mjs","sourceRoot":"","sources":["../src/resources/shields.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAAE,WAAW,EAAE;AAGtB,MAAM,OAAO,OAAQ,SAAQ,WAAW;IACtC;;OAEG;IACH,QAAQ,CAAC,UAAkB,EAAE,OAA6B;QACxD,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,eAAe,UAAU,EAAE,EAAE,OAAO,CAAC,CAAC;IAChE,CAAC;IAED;;OAEG;IACH,IAAI,CAAC,OAA6B;QAChC,OACE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,OAAO,CACxC,CAAC,WAAW,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,UAAkB,EAAE,OAA6B;QACtD,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,eAAe,UAAU,EAAE,EAAE;YACtD,GAAG,OAAO;YACV,OAAO,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,OAAO,EAAE;SAChD,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,QAAQ,CAAC,IAA0B,EAAE,OAA6B;QAChE,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,aAAa,EAAE,EAAE,IAAI,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IAChE,CAAC;CACF"}

package/src/core.ts

@@ -1104,21 +1104,21 @@ export const coerceBoolean = (value: unknown): boolean => {
 };
 
 export const maybeCoerceInteger = (value: unknown): number | undefined => {
-  if (value === undefined) {
+  if (value == null) {
     return undefined;
   }
   return coerceInteger(value);
 };
 
 export const maybeCoerceFloat = (value: unknown): number | undefined => {
-  if (value === undefined) {
+  if (value == null) {
     return undefined;
   }
   return coerceFloat(value);
 };
 
 export const maybeCoerceBoolean = (value: unknown): boolean | undefined => {
-  if (value === undefined) {
+  if (value == null) {
     return undefined;
   }
   return coerceBoolean(value);

package/src/index.ts

@@ -64,16 +64,11 @@ import {
   InferenceCompletionParamsNonStreaming,
   InferenceCompletionParamsStreaming,
   InferenceEmbeddingsParams,
+  InferenceRerankParams,
+  InferenceRerankResponse,
   TokenLogProbs,
 } from './resources/inference';
 import { HealthInfo, Inspect, ProviderInfo, RouteInfo, VersionInfo } from './resources/inspect';
-import {
-  ListModelsResponse,
-  Model,
-  ModelListResponse,
-  ModelRegisterParams,
-  Models,
-} from './resources/models';
 import { CreateResponse, ModerationCreateParams, Moderations } from './resources/moderations';
 import { ListProvidersResponse, ProviderListResponse, Providers } from './resources/providers';
 import { ListRoutesResponse, RouteListResponse, Routes } from './resources/routes';
@@ -171,6 +166,13 @@ import {
   EvaluateResponse,
   Job,
 } from './resources/eval/eval';
+import {
+  ListModelsResponse,
+  Model,
+  ModelListResponse,
+  ModelRegisterParams,
+  Models,
+} from './resources/models/models';
 import {
   AlgorithmConfig,
   ListPostTrainingJobsResponse,
@@ -183,6 +185,7 @@ import {
   ResponseCreateParams,
   ResponseCreateParamsNonStreaming,
   ResponseCreateParamsStreaming,
+  ResponseDeleteResponse,
   ResponseListParams,
   ResponseListResponse,
   ResponseListResponsesOpenAICursorPage,
@@ -477,6 +480,7 @@ export declare namespace LlamaStackClient {
     type ResponseObject as ResponseObject,
     type ResponseObjectStream as ResponseObjectStream,
     type ResponseListResponse as ResponseListResponse,
+    type ResponseDeleteResponse as ResponseDeleteResponse,
     ResponseListResponsesOpenAICursorPage as ResponseListResponsesOpenAICursorPage,
     type ResponseCreateParams as ResponseCreateParams,
     type ResponseCreateParamsNonStreaming as ResponseCreateParamsNonStreaming,
@@ -537,6 +541,7 @@ export declare namespace LlamaStackClient {
     type EmbeddingsResponse as EmbeddingsResponse,
     type TokenLogProbs as TokenLogProbs,
     type InferenceBatchChatCompletionResponse as InferenceBatchChatCompletionResponse,
+    type InferenceRerankResponse as InferenceRerankResponse,
     type InferenceBatchChatCompletionParams as InferenceBatchChatCompletionParams,
     type InferenceBatchCompletionParams as InferenceBatchCompletionParams,
     type InferenceChatCompletionParams as InferenceChatCompletionParams,
@@ -546,6 +551,7 @@ export declare namespace LlamaStackClient {
     type InferenceCompletionParamsNonStreaming as InferenceCompletionParamsNonStreaming,
     type InferenceCompletionParamsStreaming as InferenceCompletionParamsStreaming,
     type InferenceEmbeddingsParams as InferenceEmbeddingsParams,
+    type InferenceRerankParams as InferenceRerankParams,
   };
 
   export {

package/src/pagination.ts

@@ -48,16 +48,15 @@ export class DatasetsIterrows<Item> extends AbstractPage<Item> implements Datase
   }
 
   nextPageInfo(): PageInfo | null {
-    const cursor = this.next_index;
-    if (!cursor) {
+    const offset = this.next_index;
+    if (!offset) {
       return null;
     }
 
-    return {
-      params: {
-        start_index: cursor,
-      },
-    };
+    const length = this.getPaginatedItems().length;
+    const currentCount = offset + length;
+
+    return { params: { start_index: currentCount } };
   }
 }
 

package/src/resources/index.ts

@@ -69,6 +69,7 @@ export {
   type EmbeddingsResponse,
   type TokenLogProbs,
   type InferenceBatchChatCompletionResponse,
+  type InferenceRerankResponse,
   type InferenceBatchChatCompletionParams,
   type InferenceBatchCompletionParams,
   type InferenceChatCompletionParams,
@@ -78,6 +79,7 @@ export {
   type InferenceCompletionParamsNonStreaming,
   type InferenceCompletionParamsStreaming,
   type InferenceEmbeddingsParams,
+  type InferenceRerankParams,
 } from './inference';
 export { Inspect, type HealthInfo, type ProviderInfo, type RouteInfo, type VersionInfo } from './inspect';
 export {
@@ -86,7 +88,7 @@ export {
   type Model,
   type ModelListResponse,
   type ModelRegisterParams,
-} from './models';
+} from './models/models';
 export { Moderations, type CreateResponse, type ModerationCreateParams } from './moderations';
 export {
   PostTraining,
@@ -103,6 +105,7 @@ export {
   type ResponseObject,
   type ResponseObjectStream,
   type ResponseListResponse,
+  type ResponseDeleteResponse,
   type ResponseCreateParams,
   type ResponseCreateParamsNonStreaming,
   type ResponseCreateParamsStreaming,

package/src/resources/inference.ts

@@ -95,6 +95,20 @@ export class Inference extends APIResource {
   ): Core.APIPromise<EmbeddingsResponse> {
     return this._client.post('/v1/inference/embeddings', { body, ...options });
   }
+
+  /**
+   * Rerank a list of documents based on their relevance to a query.
+   */
+  rerank(
+    body: InferenceRerankParams,
+    options?: Core.RequestOptions,
+  ): Core.APIPromise<InferenceRerankResponse> {
+    return (
+      this._client.post('/v1/inference/rerank', { body, ...options }) as Core.APIPromise<{
+        data: InferenceRerankResponse;
+      }>
+    )._thenUnwrap((obj) => obj.data);
+  }
 }
 
 /**
@@ -197,6 +211,29 @@ export interface InferenceBatchChatCompletionResponse {
   batch: Array<Shared.ChatCompletionResponse>;
 }
 
+/**
+ * List of rerank result objects, sorted by relevance score (descending)
+ */
+export type InferenceRerankResponse = Array<InferenceRerankResponse.InferenceRerankResponseItem>;
+
+export namespace InferenceRerankResponse {
+  /**
+   * A single rerank result from a reranking response.
+   */
+  export interface InferenceRerankResponseItem {
+    /**
+     * The original index of the document in the input list
+     */
+    index: number;
+
+    /**
+     * The relevance score from the model output. Values are inverted when applicable
+     * so that higher scores indicate greater relevance.
+     */
+    relevance_score: number;
+  }
+}
+
 export interface InferenceBatchChatCompletionParams {
   /**
    * The messages to generate completions for.
@@ -575,6 +612,134 @@ export interface InferenceEmbeddingsParams {
   text_truncation?: 'none' | 'start' | 'end';
 }
 
+export interface InferenceRerankParams {
+  /**
+   * List of items to rerank. Each item can be a string, text content part, or image
+   * content part. Each input must not exceed the model's max input token length.
+   */
+  items: Array<
+    | string
+    | InferenceRerankParams.OpenAIChatCompletionContentPartTextParam
+    | InferenceRerankParams.OpenAIChatCompletionContentPartImageParam
+  >;
+
+  /**
+   * The identifier of the reranking model to use.
+   */
+  model: string;
+
+  /**
+   * The search query to rank items against. Can be a string, text content part, or
+   * image content part. The input must not exceed the model's max input token
+   * length.
+   */
+  query:
+    | string
+    | InferenceRerankParams.OpenAIChatCompletionContentPartTextParam
+    | InferenceRerankParams.OpenAIChatCompletionContentPartImageParam;
+
+  /**
+   * (Optional) Maximum number of results to return. Default: returns all.
+   */
+  max_num_results?: number;
+}
+
+export namespace InferenceRerankParams {
+  /**
+   * Text content part for OpenAI-compatible chat completion messages.
+   */
+  export interface OpenAIChatCompletionContentPartTextParam {
+    /**
+     * The text content of the message
+     */
+    text: string;
+
+    /**
+     * Must be "text" to identify this as text content
+     */
+    type: 'text';
+  }
+
+  /**
+   * Image content part for OpenAI-compatible chat completion messages.
+   */
+  export interface OpenAIChatCompletionContentPartImageParam {
+    /**
+     * Image URL specification and processing details
+     */
+    image_url: OpenAIChatCompletionContentPartImageParam.ImageURL;
+
+    /**
+     * Must be "image_url" to identify this as image content
+     */
+    type: 'image_url';
+  }
+
+  export namespace OpenAIChatCompletionContentPartImageParam {
+    /**
+     * Image URL specification and processing details
+     */
+    export interface ImageURL {
+      /**
+       * URL of the image to include in the message
+       */
+      url: string;
+
+      /**
+       * (Optional) Level of detail for image processing. Can be "low", "high", or "auto"
+       */
+      detail?: string;
+    }
+  }
+
+  /**
+   * Text content part for OpenAI-compatible chat completion messages.
+   */
+  export interface OpenAIChatCompletionContentPartTextParam {
+    /**
+     * The text content of the message
+     */
+    text: string;
+
+    /**
+     * Must be "text" to identify this as text content
+     */
+    type: 'text';
+  }
+
+  /**
+   * Image content part for OpenAI-compatible chat completion messages.
+   */
+  export interface OpenAIChatCompletionContentPartImageParam {
+    /**
+     * Image URL specification and processing details
+     */
+    image_url: OpenAIChatCompletionContentPartImageParam.ImageURL;
+
+    /**
+     * Must be "image_url" to identify this as image content
+     */
+    type: 'image_url';
+  }
+
+  export namespace OpenAIChatCompletionContentPartImageParam {
+    /**
+     * Image URL specification and processing details
+     */
+    export interface ImageURL {
+      /**
+       * URL of the image to include in the message
+       */
+      url: string;
+
+      /**
+       * (Optional) Level of detail for image processing. Can be "low", "high", or "auto"
+       */
+      detail?: string;
+    }
+  }
+}
+
 export declare namespace Inference {
   export {
     type ChatCompletionResponseStreamChunk as ChatCompletionResponseStreamChunk,
@@ -582,6 +747,7 @@ export declare namespace Inference {
     type EmbeddingsResponse as EmbeddingsResponse,
     type TokenLogProbs as TokenLogProbs,
     type InferenceBatchChatCompletionResponse as InferenceBatchChatCompletionResponse,
+    type InferenceRerankResponse as InferenceRerankResponse,
     type InferenceBatchChatCompletionParams as InferenceBatchChatCompletionParams,
     type InferenceBatchCompletionParams as InferenceBatchCompletionParams,
     type InferenceChatCompletionParams as InferenceChatCompletionParams,
@@ -591,5 +757,6 @@ export declare namespace Inference {
     type InferenceCompletionParamsNonStreaming as InferenceCompletionParamsNonStreaming,
     type InferenceCompletionParamsStreaming as InferenceCompletionParamsStreaming,
     type InferenceEmbeddingsParams as InferenceEmbeddingsParams,
+    type InferenceRerankParams as InferenceRerankParams,
   };
 }

package/src/resources/models/index.ts

@@ -0,0 +1,10 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+export {
+  Models,
+  type ListModelsResponse,
+  type Model,
+  type ModelListResponse,
+  type ModelRegisterParams,
+} from './models';
+export { OpenAI, type OpenAIListResponse } from './openai';

package/src/resources/models/models.ts

@@ -0,0 +1,124 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+import { APIResource } from '../../resource';
+import * as Core from '../../core';
+import * as OpenAIAPI from './openai';
+import { OpenAI, OpenAIListResponse } from './openai';
+
+export class Models extends APIResource {
+  openai: OpenAIAPI.OpenAI = new OpenAIAPI.OpenAI(this._client);
+
+  /**
+   * Get a model by its identifier.
+   */
+  retrieve(modelId: string, options?: Core.RequestOptions): Core.APIPromise<Model> {
+    return this._client.get(`/v1/models/${modelId}`, options);
+  }
+
+  /**
+   * List all models.
+   */
+  list(options?: Core.RequestOptions): Core.APIPromise<ModelListResponse> {
+    return (
+      this._client.get('/v1/models', options) as Core.APIPromise<{ data: ModelListResponse }>
+    )._thenUnwrap((obj) => obj.data);
+  }
+
+  /**
+   * Register a model.
+   */
+  register(body: ModelRegisterParams, options?: Core.RequestOptions): Core.APIPromise<Model> {
+    return this._client.post('/v1/models', { body, ...options });
+  }
+
+  /**
+   * Unregister a model.
+   */
+  unregister(modelId: string, options?: Core.RequestOptions): Core.APIPromise<void> {
+    return this._client.delete(`/v1/models/${modelId}`, {
+      ...options,
+      headers: { Accept: '*/*', ...options?.headers },
+    });
+  }
+}
+
+export interface ListModelsResponse {
+  data: ModelListResponse;
+}
+
+/**
+ * A model resource representing an AI model registered in Llama Stack.
+ */
+export interface Model {
+  /**
+   * Unique identifier for this resource in llama stack
+   */
+  identifier: string;
+
+  /**
+   * Any additional metadata for this model
+   */
+  metadata: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
+
+  /**
+   * The type of model (LLM or embedding model)
+   */
+  model_type: 'llm' | 'embedding';
+
+  /**
+   * ID of the provider that owns this resource
+   */
+  provider_id: string;
+
+  /**
+   * The resource type, always 'model' for model resources
+   */
+  type: 'model';
+
+  /**
+   * Unique identifier for this resource in the provider
+   */
+  provider_resource_id?: string;
+}
+
+export type ModelListResponse = Array<Model>;
+
+export interface ModelRegisterParams {
+  /**
+   * The identifier of the model to register.
+   */
+  model_id: string;
+
+  /**
+   * Any additional metadata for this model.
+   */
+  metadata?: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
+
+  /**
+   * The type of model to register.
+   */
+  model_type?: 'llm' | 'embedding';
+
+  /**
+   * The identifier of the provider.
+   */
+  provider_id?: string;
+
+  /**
+   * The identifier of the model in the provider.
+   */
+  provider_model_id?: string;
+}
+
+Models.OpenAI = OpenAI;
+
+export declare namespace Models {
+  export {
+    type ListModelsResponse as ListModelsResponse,
+    type Model as Model,
+    type ModelListResponse as ModelListResponse,
+    type ModelRegisterParams as ModelRegisterParams,
+  };
+
+  export { OpenAI as OpenAI, type OpenAIListResponse as OpenAIListResponse };
+}

package/src/resources/models/openai.ts

@@ -0,0 +1,36 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+import { APIResource } from '../../resource';
+import * as Core from '../../core';
+
+export class OpenAI extends APIResource {
+  /**
+   * List models using the OpenAI API.
+   */
+  list(options?: Core.RequestOptions): Core.APIPromise<OpenAIListResponse> {
+    return (
+      this._client.get('/v1/openai/v1/models', options) as Core.APIPromise<{ data: OpenAIListResponse }>
+    )._thenUnwrap((obj) => obj.data);
+  }
+}
+
+export type OpenAIListResponse = Array<OpenAIListResponse.OpenAIListResponseItem>;
+
+export namespace OpenAIListResponse {
+  /**
+   * A model from OpenAI.
+   */
+  export interface OpenAIListResponseItem {
+    id: string;
+
+    created: number;
+
+    object: 'model';
+
+    owned_by: string;
+  }
+}
+
+export declare namespace OpenAI {
+  export { type OpenAIListResponse as OpenAIListResponse };
+}