@mixedbread/sdk 0.1.0-alpha.19 → 0.1.0-alpha.20

package/resources/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/resources/index.ts"],"names":[],"mappings":"AAEA,cAAc,UAAU,CAAC;AACzB,OAAO,EACL,UAAU,EACV,KAAK,SAAS,EACd,KAAK,uBAAuB,EAC5B,KAAK,qBAAqB,GAC3B,MAAM,cAAc,CAAC;AACtB,OAAO,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AACxD,OAAO,EACL,sBAAsB,EACtB,KAAK,EACL,KAAK,UAAU,EACf,KAAK,kBAAkB,EACvB,KAAK,gBAAgB,EACrB,KAAK,gBAAgB,EACrB,KAAK,cAAc,GACpB,MAAM,SAAS,CAAC;AACjB,OAAO,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAC5C,OAAO,EACL,uBAAuB,EACvB,YAAY,EACZ,KAAK,YAAY,EACjB,KAAK,UAAU,EACf,KAAK,sBAAsB,EAC3B,KAAK,WAAW,EAChB,KAAK,6BAA6B,EAClC,KAAK,4BAA4B,EACjC,KAAK,yBAAyB,EAC9B,KAAK,oCAAoC,EACzC,KAAK,yBAAyB,EAC9B,KAAK,uBAAuB,EAC5B,KAAK,uBAAuB,EAC5B,KAAK,qBAAqB,EAC1B,KAAK,kCAAkC,EACvC,KAAK,uBAAuB,GAC7B,MAAM,+BAA+B,CAAC;AACvC,OAAO,EAAE,KAAK,YAAY,EAAE,KAAK,cAAc,EAAE,KAAK,WAAW,EAAE,KAAK,YAAY,EAAE,MAAM,aAAa,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/resources/index.ts"],"names":[],"mappings":"AAEA,cAAc,UAAU,CAAC;AACzB,OAAO,EAAE,UAAU,EAAE,KAAK,qBAAqB,EAAE,MAAM,cAAc,CAAC;AACtE,OAAO,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AACxD,OAAO,EACL,sBAAsB,EACtB,KAAK,EACL,KAAK,UAAU,EACf,KAAK,kBAAkB,EACvB,KAAK,gBAAgB,EACrB,KAAK,gBAAgB,EACrB,KAAK,cAAc,GACpB,MAAM,SAAS,CAAC;AACjB,OAAO,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAC5C,OAAO,EACL,uBAAuB,EACvB,YAAY,EACZ,KAAK,YAAY,EACjB,KAAK,UAAU,EACf,KAAK,sBAAsB,EAC3B,KAAK,WAAW,EAChB,KAAK,6BAA6B,EAClC,KAAK,4BAA4B,EACjC,KAAK,yBAAyB,EAC9B,KAAK,oCAAoC,EACzC,KAAK,yBAAyB,EAC9B,KAAK,uBAAuB,EAC5B,KAAK,uBAAuB,EAC5B,KAAK,qBAAqB,EAC1B,KAAK,kCAAkC,EACvC,KAAK,uBAAuB,GAC7B,MAAM,+BAA+B,CAAC;AACvC,OAAO,EACL,KAAK,EAAE,EACP,KAAK,SAAS,EACd,KAAK,sBAAsB,EAC3B,KAAK,YAAY,EACjB,KAAK,cAAc,EACnB,KAAK,WAAW,EAChB,KAAK,YAAY,GAClB,MAAM,aAAa,CAAC"}

package/resources/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/resources/index.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;;;;;;;;;;;;;;;AAEtF,8CAAyB;AACzB,8CAKsB;AAJpB,wGAAA,UAAU,OAAA;AAKZ,4DAAwD;AAA/C,0GAAA,WAAW,OAAA;AACpB,oCAQiB;AAPf,+GAAA,sBAAsB,OAAA;AACtB,8FAAA,KAAK,OAAA;AAOP,gDAA4C;AAAnC,kGAAA,OAAO,OAAA;AAChB,kEAiBuC;AAhBrC,wHAAA,uBAAuB,OAAA;AACvB,6GAAA,YAAY,OAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/resources/index.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;;;;;;;;;;;;;;;AAEtF,8CAAyB;AACzB,8CAAsE;AAA7D,wGAAA,UAAU,OAAA;AACnB,4DAAwD;AAA/C,0GAAA,WAAW,OAAA;AACpB,oCAQiB;AAPf,+GAAA,sBAAsB,OAAA;AACtB,8FAAA,KAAK,OAAA;AAOP,gDAA4C;AAAnC,kGAAA,OAAO,OAAA;AAChB,kEAiBuC;AAhBrC,wHAAA,uBAAuB,OAAA;AACvB,6GAAA,YAAY,OAAA"}

package/resources/index.mjs

@@ -1,6 +1,6 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 export * from "./shared.mjs";
-export { Embeddings, } from "./embeddings.mjs";
+export { Embeddings } from "./embeddings.mjs";
 export { Extractions } from "./extractions/extractions.mjs";
 export { FileObjectsLimitOffset, Files, } from "./files.mjs";
 export { Parsing } from "./parsing/parsing.mjs";

package/resources/index.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.mjs","sourceRoot":"","sources":["../src/resources/index.ts"],"names":[],"mappings":"AAAA,sFAAsF;;OAG/E,EACL,UAAU,GAIX;OACM,EAAE,WAAW,EAAE;OACf,EACL,sBAAsB,EACtB,KAAK,GAMN;OACM,EAAE,OAAO,EAAE;OACX,EACL,uBAAuB,EACvB,YAAY,GAeb"}
+{"version":3,"file":"index.mjs","sourceRoot":"","sources":["../src/resources/index.ts"],"names":[],"mappings":"AAAA,sFAAsF;;OAG/E,EAAE,UAAU,EAA8B;OAC1C,EAAE,WAAW,EAAE;OACf,EACL,sBAAsB,EACtB,KAAK,GAMN;OACM,EAAE,OAAO,EAAE;OACX,EACL,uBAAuB,EACvB,YAAY,GAeb"}

package/resources/top-level.d.ts

@@ -1,3 +1,95 @@
+export interface Em {
+    /**
+     * The usage of the model
+     */
+    usage: Em.Usage;
+    /**
+     * The model used
+     */
+    model: string;
+    /**
+     * The created embeddings.
+     */
+    data: Array<Embedding> | Array<MultiEncodingEmbedding>;
+    /**
+     * The object type of the response
+     */
+    object?: 'list' | 'parsing_job' | 'job' | 'embedding' | 'embedding_dict' | 'rank_result' | 'file' | 'vector_store' | 'vector_store.file' | 'api_key';
+    /**
+     * Whether the embeddings are normalized.
+     */
+    normalized: boolean;
+    /**
+     * The encoding formats of the embeddings.
+     */
+    encoding_format: 'float' | 'float16' | 'base64' | 'binary' | 'ubinary' | 'int8' | 'uint8' | Array<'float' | 'float16' | 'base64' | 'binary' | 'ubinary' | 'int8' | 'uint8'>;
+    /**
+     * The number of dimensions used for the embeddings.
+     */
+    dimensions: number | null;
+}
+export declare namespace Em {
+    /**
+     * The usage of the model
+     */
+    interface Usage {
+        /**
+         * The number of tokens used for the prompt
+         */
+        prompt_tokens: number;
+        /**
+         * The total number of tokens used
+         */
+        total_tokens: number;
+        /**
+         * The number of tokens used for the completion
+         */
+        completion_tokens?: number | null;
+    }
+}
+export interface Embedding {
+    /**
+     * The encoded embedding.
+     */
+    embedding: Array<number> | Array<number> | string;
+    /**
+     * The index of the embedding.
+     */
+    index: number;
+    /**
+     * The object type of the embedding.
+     */
+    object?: 'embedding';
+}
+export interface MultiEncodingEmbedding {
+    /**
+     * The encoded embedding data by encoding format.Returned, if more than one
+     * encoding format is used.
+     */
+    embedding: MultiEncodingEmbedding.Embedding;
+    /**
+     * The index of the embedding.
+     */
+    index: number;
+    /**
+     * The object type of the embedding.
+     */
+    object?: 'embedding_dict';
+}
+export declare namespace MultiEncodingEmbedding {
+    /**
+     * The encoded embedding data by encoding format.Returned, if more than one
+     * encoding format is used.
+     */
+    interface Embedding {
+        float?: Array<number>;
+        int8?: Array<number>;
+        uint8?: Array<number>;
+        binary?: Array<number>;
+        ubinary?: Array<number>;
+        base64?: string;
+    }
+}
 /**
  * Info Pydantic Response Service Message
  */
@@ -107,7 +199,7 @@ export interface RerankParams {
     /**
      * The input documents to rerank.
      */
-    input: Array<string | unknown>;
+    input: Array<string | unknown | Array<unknown>>;
     /**
      * The fields of the documents to rank.
      */
@@ -122,6 +214,6 @@ export interface RerankParams {
     return_input?: boolean;
 }
 export declare namespace TopLevel {
-    export { type InfoResponse as InfoResponse, type RerankResponse as RerankResponse, type EmbedParams as EmbedParams, type RerankParams as RerankParams, };
+    export { type Em as Em, type Embedding as Embedding, type MultiEncodingEmbedding as MultiEncodingEmbedding, type InfoResponse as InfoResponse, type RerankResponse as RerankResponse, type EmbedParams as EmbedParams, type RerankParams as RerankParams, };
 }
 //# sourceMappingURL=top-level.d.ts.map

package/resources/top-level.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"top-level.d.ts","sourceRoot":"","sources":["../src/resources/top-level.ts"],"names":[],"mappings":"AAEA;;GAEG;AACH,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,MAAM,CAAC;IAEb,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,cAAc;IAC7B;;OAEG;IACH,KAAK,EAAE,cAAc,CAAC,KAAK,CAAC;IAE5B;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IAEd;;OAEG;IACH,IAAI,EAAE,KAAK,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;IAEjC;;OAEG;IACH,MAAM,CAAC,EACH,MAAM,GACN,aAAa,GACb,KAAK,GACL,WAAW,GACX,gBAAgB,GAChB,aAAa,GACb,MAAM,GACN,cAAc,GACd,mBAAmB,GACnB,SAAS,CAAC;IAEd;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IAEd;;OAEG;IACH,YAAY,EAAE,OAAO,CAAC;CACvB;AAED,yBAAiB,cAAc,CAAC;IAC9B;;OAEG;IACH,UAAiB,KAAK;QACpB;;WAEG;QACH,aAAa,EAAE,MAAM,CAAC;QAEtB;;WAEG;QACH,YAAY,EAAE,MAAM,CAAC;QAErB;;WAEG;QACH,iBAAiB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;KACnC;IAED,UAAiB,IAAI;QACnB;;WAEG;QACH,KAAK,EAAE,MAAM,CAAC;QAEd;;WAEG;QACH,KAAK,EAAE,MAAM,CAAC;QAEd;;WAEG;QACH,KAAK,CAAC,EAAE,OAAO,CAAC;QAEhB;;WAEG;QACH,MAAM,CAAC,EAAE,aAAa,CAAC;KACxB;CACF;AAED,MAAM,WAAW,WAAW;IAC1B;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IAEd;;OAEG;IACH,KAAK,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;IAErB;;OAEG;IACH,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAE3B;;OAEG;IACH,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAEvB;;OAEG;IACH,UAAU,CAAC,EAAE,OAAO,CAAC;IAErB;;;OAGG;IACH,eAAe,CAAC,EACZ,OAAO,GACP,SAAS,GACT,QAAQ,GACR,QAAQ,GACR,SAAS,GACT,MAAM,GACN,OAAO,GACP,KAAK,CAAC,OAAO,GAAG,SAAS,GAAG,QAAQ,GAAG,QAAQ,GAAG,SAAS,GAAG,MAAM,GAAG,OAAO,CAAC,CAAC;CACrF;AAED,MAAM,WAAW,YAAY;IAC3B;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IAEf;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IAEd;;OAEG;IACH,KAAK,EAAE,KAAK,CAAC,MAAM,GAAG,OAAO,CAAC,CAAC;IAE/B;;OAEG;IACH,WAAW,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC;IAEnC;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IAEf;;OAEG;IACH,YAAY,CAAC,EAAE,OAAO,CAAC;CACxB;AAED,MAAM,CAAC,OAAO,WAAW,QAAQ,CAAC;IAChC,OAAO,EACL,KAAK,YAAY,IAAI,YAAY,EACjC,KAAK,cAAc,IAAI,cAAc,EACrC,KAAK,WAAW,IAAI,WAAW,EAC/B,KAAK,YAAY,IAAI,YAAY,GAClC,CAAC;CACH"}
+{"version":3,"file":"top-level.d.ts","sourceRoot":"","sources":["../src/resources/top-level.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,EAAE;IACjB;;OAEG;IACH,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC;IAEhB;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IAEd;;OAEG;IACH,IAAI,EAAE,KAAK,CAAC,SAAS,CAAC,GAAG,KAAK,CAAC,sBAAsB,CAAC,CAAC;IAEvD;;OAEG;IACH,MAAM,CAAC,EACH,MAAM,GACN,aAAa,GACb,KAAK,GACL,WAAW,GACX,gBAAgB,GAChB,aAAa,GACb,MAAM,GACN,cAAc,GACd,mBAAmB,GACnB,SAAS,CAAC;IAEd;;OAEG;IACH,UAAU,EAAE,OAAO,CAAC;IAEpB;;OAEG;IACH,eAAe,EACX,OAAO,GACP,SAAS,GACT,QAAQ,GACR,QAAQ,GACR,SAAS,GACT,MAAM,GACN,OAAO,GACP,KAAK,CAAC,OAAO,GAAG,SAAS,GAAG,QAAQ,GAAG,QAAQ,GAAG,SAAS,GAAG,MAAM,GAAG,OAAO,CAAC,CAAC;IAEpF;;OAEG;IACH,UAAU,EAAE,MAAM,GAAG,IAAI,CAAC;CAC3B;AAED,yBAAiB,EAAE,CAAC;IAClB;;OAEG;IACH,UAAiB,KAAK;QACpB;;WAEG;QACH,aAAa,EAAE,MAAM,CAAC;QAEtB;;WAEG;QACH,YAAY,EAAE,MAAM,CAAC;QAErB;;WAEG;QACH,iBAAiB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;KACnC;CACF;AAED,MAAM,WAAW,SAAS;IACxB;;OAEG;IACH,SAAS,EAAE,KAAK,CAAC,MAAM,CAAC,GAAG,KAAK,CAAC,MAAM,CAAC,GAAG,MAAM,CAAC;IAElD;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IAEd;;OAEG;IACH,MAAM,CAAC,EAAE,WAAW,CAAC;CACtB;AAED,MAAM,WAAW,sBAAsB;IACrC;;;OAGG;IACH,SAAS,EAAE,sBAAsB,CAAC,SAAS,CAAC;IAE5C;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IAEd;;OAEG;IACH,MAAM,CAAC,EAAE,gBAAgB,CAAC;CAC3B;AAED,yBAAiB,sBAAsB,CAAC;IACtC;;;OAGG;IACH,UAAiB,SAAS;QACxB,KAAK,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;QAEtB,IAAI,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;QAErB,KAAK,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;QAEtB,MAAM,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;QAEvB,OAAO,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;QAExB,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB;CACF;AAED;;GAEG;AACH,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,MAAM,CAAC;IAEb,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,cAAc;IAC7B;;OAEG;IACH,KAAK,EAAE,cAAc,CAAC,KAAK,CAAC;IAE5B;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IAEd;;OAEG;IACH,IAAI,EAAE,KAAK,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;IAEjC;;OAEG;IACH,MAAM,CAAC,EACH,MAAM,GACN,aAAa,GACb,KAAK,GACL,WAAW,GACX,gBAAgB,GAChB,aAAa,GACb,MAAM,GACN,cAAc,GACd,mBAAmB,GACnB,SAAS,CAAC;IAEd;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IAEd;;OAEG;IACH,YAAY,EAAE,OAAO,CAAC;CACvB;AAED,yBAAiB,cAAc,CAAC;IAC9B;;OAEG;IACH,UAAiB,KAAK;QACpB;;WAEG;QACH,aAAa,EAAE,MAAM,CAAC;QAEtB;;WAEG;QACH,YAAY,EAAE,MAAM,CAAC;QAErB;;WAEG;QACH,iBAAiB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;KACnC;IAED,UAAiB,IAAI;QACnB;;WAEG;QACH,KAAK,EAAE,MAAM,CAAC;QAEd;;WAEG;QACH,KAAK,EAAE,MAAM,CAAC;QAEd;;WAEG;QACH,KAAK,CAAC,EAAE,OAAO,CAAC;QAEhB;;WAEG;QACH,MAAM,CAAC,EAAE,aAAa,CAAC;KACxB;CACF;AAED,MAAM,WAAW,WAAW;IAC1B;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IAEd;;OAEG;IACH,KAAK,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;IAErB;;OAEG;IACH,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAE3B;;OAEG;IACH,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAEvB;;OAEG;IACH,UAAU,CAAC,EAAE,OAAO,CAAC;IAErB;;;OAGG;IACH,eAAe,CAAC,EACZ,OAAO,GACP,SAAS,GACT,QAAQ,GACR,QAAQ,GACR,SAAS,GACT,MAAM,GACN,OAAO,GACP,KAAK,CAAC,OAAO,GAAG,SAAS,GAAG,QAAQ,GAAG,QAAQ,GAAG,SAAS,GAAG,MAAM,GAAG,OAAO,CAAC,CAAC;CACrF;AAED,MAAM,WAAW,YAAY;IAC3B;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IAEf;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IAEd;;OAEG;IACH,KAAK,EAAE,KAAK,CAAC,MAAM,GAAG,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;IAEhD;;OAEG;IACH,WAAW,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC;IAEnC;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IAEf;;OAEG;IACH,YAAY,CAAC,EAAE,OAAO,CAAC;CACxB;AAED,MAAM,CAAC,OAAO,WAAW,QAAQ,CAAC;IAChC,OAAO,EACL,KAAK,EAAE,IAAI,EAAE,EACb,KAAK,SAAS,IAAI,SAAS,EAC3B,KAAK,sBAAsB,IAAI,sBAAsB,EACrD,KAAK,YAAY,IAAI,YAAY,EACjC,KAAK,cAAc,IAAI,cAAc,EACrC,KAAK,WAAW,IAAI,WAAW,EAC/B,KAAK,YAAY,IAAI,YAAY,GAClC,CAAC;CACH"}

package/src/index.ts

@@ -6,16 +6,18 @@ import * as Errors from './error';
 import * as Pagination from './pagination';
 import { type LimitOffsetParams, LimitOffsetResponse } from './pagination';
 import * as Uploads from './uploads';
-import * as EmbeddingsAPI from './resources/embeddings';
-import {
-  Embedding,
-  EmbeddingCreateParams,
-  EmbeddingCreateResponse,
-  Embeddings,
-} from './resources/embeddings';
 import * as API from './resources/index';
 import * as TopLevelAPI from './resources/top-level';
-import { EmbedParams, InfoResponse, RerankParams, RerankResponse } from './resources/top-level';
+import {
+  Em,
+  EmbedParams,
+  Embedding,
+  InfoResponse,
+  MultiEncodingEmbedding,
+  RerankParams,
+  RerankResponse,
+} from './resources/top-level';
+import { EmbeddingCreateParams, Embeddings } from './resources/embeddings';
 import {
   FileCreateParams,
   FileDeleteResponse,
@@ -182,11 +184,11 @@ export class Mixedbread extends Core.APIClient {
     this.apiKey = apiKey;
   }
 
-  embeddings: API.Embeddings = new API.Embeddings(this);
+  vectorStores: API.VectorStores = new API.VectorStores(this);
   parsing: API.Parsing = new API.Parsing(this);
   files: API.Files = new API.Files(this);
-  vectorStores: API.VectorStores = new API.VectorStores(this);
   extractions: API.Extractions = new API.Extractions(this);
+  embeddings: API.Embeddings = new API.Embeddings(this);
 
   /**
    * Create embeddings for text or images using the specified model, encoding format,
@@ -196,10 +198,7 @@ export class Mixedbread extends Core.APIClient {
    *
    * Returns: EmbeddingCreateResponse: The response containing the embeddings.
    */
-  embed(
-    body: TopLevelAPI.EmbedParams,
-    options?: Core.RequestOptions,
-  ): Core.APIPromise<EmbeddingsAPI.EmbeddingCreateResponse> {
+  embed(body: TopLevelAPI.EmbedParams, options?: Core.RequestOptions): Core.APIPromise<TopLevelAPI.Em> {
     return this.post('/v1/embeddings', { body, ...options });
   }
 
@@ -215,9 +214,9 @@ export class Mixedbread extends Core.APIClient {
   /**
    * Rerank different kind of documents for a given query.
    *
-   * Args: params: RerankingCreateParams: The parameters for reranking.
+   * Args: params: RerankParams: The parameters for reranking.
    *
-   * Returns: RerankingCreateResponse: The reranked documents for the input query.
+   * Returns: RerankResponse: The reranked documents for the input query.
    */
   rerank(
     body: TopLevelAPI.RerankParams,
@@ -262,13 +261,13 @@ export class Mixedbread extends Core.APIClient {
   static fileFromPath = Uploads.fileFromPath;
 }
 
-Mixedbread.Embeddings = Embeddings;
+Mixedbread.VectorStores = VectorStores;
+Mixedbread.VectorStoresLimitOffset = VectorStoresLimitOffset;
 Mixedbread.Parsing = Parsing;
 Mixedbread.Files = Files;
 Mixedbread.FileObjectsLimitOffset = FileObjectsLimitOffset;
-Mixedbread.VectorStores = VectorStores;
-Mixedbread.VectorStoresLimitOffset = VectorStoresLimitOffset;
 Mixedbread.Extractions = Extractions;
+Mixedbread.Embeddings = Embeddings;
 export declare namespace Mixedbread {
   export type RequestOptions = Core.RequestOptions;
 
@@ -276,31 +275,15 @@ export declare namespace Mixedbread {
   export { type LimitOffsetParams as LimitOffsetParams, type LimitOffsetResponse as LimitOffsetResponse };
 
   export {
+    type Em as Em,
+    type Embedding as Embedding,
+    type MultiEncodingEmbedding as MultiEncodingEmbedding,
     type InfoResponse as InfoResponse,
     type RerankResponse as RerankResponse,
     type EmbedParams as EmbedParams,
     type RerankParams as RerankParams,
   };
 
-  export {
-    Embeddings as Embeddings,
-    type Embedding as Embedding,
-    type EmbeddingCreateResponse as EmbeddingCreateResponse,
-    type EmbeddingCreateParams as EmbeddingCreateParams,
-  };
-
-  export { Parsing as Parsing };
-
-  export {
-    Files as Files,
-    type FileObject as FileObject,
-    type FileDeleteResponse as FileDeleteResponse,
-    FileObjectsLimitOffset as FileObjectsLimitOffset,
-    type FileCreateParams as FileCreateParams,
-    type FileUpdateParams as FileUpdateParams,
-    type FileListParams as FileListParams,
-  };
-
   export {
     VectorStores as VectorStores,
     type ExpiresAfter as ExpiresAfter,
@@ -320,8 +303,22 @@ export declare namespace Mixedbread {
     type VectorStoreSearchParams as VectorStoreSearchParams,
   };
 
+  export { Parsing as Parsing };
+
+  export {
+    Files as Files,
+    type FileObject as FileObject,
+    type FileDeleteResponse as FileDeleteResponse,
+    FileObjectsLimitOffset as FileObjectsLimitOffset,
+    type FileCreateParams as FileCreateParams,
+    type FileUpdateParams as FileUpdateParams,
+    type FileListParams as FileListParams,
+  };
+
   export { Extractions as Extractions };
 
+  export { Embeddings as Embeddings, type EmbeddingCreateParams as EmbeddingCreateParams };
+
   export type SearchFilter = API.SearchFilter;
   export type SearchFilterCondition = API.SearchFilterCondition;
 }

package/src/resources/embeddings.ts

@@ -2,6 +2,7 @@
 
 import { APIResource } from '../resource';
 import * as Core from '../core';
+import * as TopLevelAPI from './top-level';
 
 export class Embeddings extends APIResource {
   /**
@@ -12,146 +13,11 @@ export class Embeddings extends APIResource {
    *
    * Returns: EmbeddingCreateResponse: The response containing the embeddings.
    */
-  create(
-    body: EmbeddingCreateParams,
-    options?: Core.RequestOptions,
-  ): Core.APIPromise<EmbeddingCreateResponse> {
+  create(body: EmbeddingCreateParams, options?: Core.RequestOptions): Core.APIPromise<TopLevelAPI.Em> {
     return this._client.post('/v1/embeddings', { body, ...options });
   }
 }
 
-export interface Embedding {
-  /**
-   * The encoded embedding.
-   */
-  embedding: Array<number> | Array<number> | string;
-
-  /**
-   * The index of the embedding.
-   */
-  index: number;
-
-  /**
-   * The object type of the embedding.
-   */
-  object?: 'embedding';
-}
-
-export interface EmbeddingCreateResponse {
-  /**
-   * The usage of the model
-   */
-  usage: EmbeddingCreateResponse.Usage;
-
-  /**
-   * The model used
-   */
-  model: string;
-
-  /**
-   * The created embeddings.
-   */
-  data: Array<Embedding> | Array<EmbeddingCreateResponse.UnionMember1>;
-
-  /**
-   * The object type of the response
-   */
-  object?:
-    | 'list'
-    | 'parsing_job'
-    | 'job'
-    | 'embedding'
-    | 'embedding_dict'
-    | 'rank_result'
-    | 'file'
-    | 'vector_store'
-    | 'vector_store.file'
-    | 'api_key';
-
-  /**
-   * Whether the embeddings are normalized.
-   */
-  normalized: boolean;
-
-  /**
-   * The encoding formats of the embeddings.
-   */
-  encoding_format:
-    | 'float'
-    | 'float16'
-    | 'base64'
-    | 'binary'
-    | 'ubinary'
-    | 'int8'
-    | 'uint8'
-    | Array<'float' | 'float16' | 'base64' | 'binary' | 'ubinary' | 'int8' | 'uint8'>;
-
-  /**
-   * The number of dimensions used for the embeddings.
-   */
-  dimensions: number | null;
-}
-
-export namespace EmbeddingCreateResponse {
-  /**
-   * The usage of the model
-   */
-  export interface Usage {
-    /**
-     * The number of tokens used for the prompt
-     */
-    prompt_tokens: number;
-
-    /**
-     * The total number of tokens used
-     */
-    total_tokens: number;
-
-    /**
-     * The number of tokens used for the completion
-     */
-    completion_tokens?: number | null;
-  }
-
-  export interface UnionMember1 {
-    /**
-     * The encoded embedding data by encoding format.Returned, if more than one
-     * encoding format is used.
-     */
-    embedding: UnionMember1.Embedding;
-
-    /**
-     * The index of the embedding.
-     */
-    index: number;
-
-    /**
-     * The object type of the embedding.
-     */
-    object?: 'embedding_dict';
-  }
-
-  export namespace UnionMember1 {
-    /**
-     * The encoded embedding data by encoding format.Returned, if more than one
-     * encoding format is used.
-     */
-    export interface Embedding {
-      float?: Array<number>;
-
-      int8?: Array<number>;
-
-      uint8?: Array<number>;
-
-      binary?: Array<number>;
-
-      ubinary?: Array<number>;
-
-      base64?: string;
-    }
-  }
-}
-
 export interface EmbeddingCreateParams {
   /**
    * The model to use for creating embeddings.
@@ -194,9 +60,5 @@ export interface EmbeddingCreateParams {
 }
 
 export declare namespace Embeddings {
-  export {
-    type Embedding as Embedding,
-    type EmbeddingCreateResponse as EmbeddingCreateResponse,
-    type EmbeddingCreateParams as EmbeddingCreateParams,
-  };
+  export { type EmbeddingCreateParams as EmbeddingCreateParams };
 }

package/src/resources/index.ts

@@ -1,12 +1,7 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 export * from './shared';
-export {
-  Embeddings,
-  type Embedding,
-  type EmbeddingCreateResponse,
-  type EmbeddingCreateParams,
-} from './embeddings';
+export { Embeddings, type EmbeddingCreateParams } from './embeddings';
 export { Extractions } from './extractions/extractions';
 export {
   FileObjectsLimitOffset,
@@ -36,4 +31,12 @@ export {
   type VectorStoreQuestionAnsweringParams,
   type VectorStoreSearchParams,
 } from './vector-stores/vector-stores';
-export { type InfoResponse, type RerankResponse, type EmbedParams, type RerankParams } from './top-level';
+export {
+  type Em,
+  type Embedding,
+  type MultiEncodingEmbedding,
+  type InfoResponse,
+  type RerankResponse,
+  type EmbedParams,
+  type RerankParams,
+} from './top-level';

package/src/resources/top-level.ts

@@ -1,5 +1,137 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
+export interface Em {
+  /**
+   * The usage of the model
+   */
+  usage: Em.Usage;
+
+  /**
+   * The model used
+   */
+  model: string;
+
+  /**
+   * The created embeddings.
+   */
+  data: Array<Embedding> | Array<MultiEncodingEmbedding>;
+
+  /**
+   * The object type of the response
+   */
+  object?:
+    | 'list'
+    | 'parsing_job'
+    | 'job'
+    | 'embedding'
+    | 'embedding_dict'
+    | 'rank_result'
+    | 'file'
+    | 'vector_store'
+    | 'vector_store.file'
+    | 'api_key';
+
+  /**
+   * Whether the embeddings are normalized.
+   */
+  normalized: boolean;
+
+  /**
+   * The encoding formats of the embeddings.
+   */
+  encoding_format:
+    | 'float'
+    | 'float16'
+    | 'base64'
+    | 'binary'
+    | 'ubinary'
+    | 'int8'
+    | 'uint8'
+    | Array<'float' | 'float16' | 'base64' | 'binary' | 'ubinary' | 'int8' | 'uint8'>;
+
+  /**
+   * The number of dimensions used for the embeddings.
+   */
+  dimensions: number | null;
+}
+
+export namespace Em {
+  /**
+   * The usage of the model
+   */
+  export interface Usage {
+    /**
+     * The number of tokens used for the prompt
+     */
+    prompt_tokens: number;
+
+    /**
+     * The total number of tokens used
+     */
+    total_tokens: number;
+
+    /**
+     * The number of tokens used for the completion
+     */
+    completion_tokens?: number | null;
+  }
+}
+
+export interface Embedding {
+  /**
+   * The encoded embedding.
+   */
+  embedding: Array<number> | Array<number> | string;
+
+  /**
+   * The index of the embedding.
+   */
+  index: number;
+
+  /**
+   * The object type of the embedding.
+   */
+  object?: 'embedding';
+}
+
+export interface MultiEncodingEmbedding {
+  /**
+   * The encoded embedding data by encoding format.Returned, if more than one
+   * encoding format is used.
+   */
+  embedding: MultiEncodingEmbedding.Embedding;
+
+  /**
+   * The index of the embedding.
+   */
+  index: number;
+
+  /**
+   * The object type of the embedding.
+   */
+  object?: 'embedding_dict';
+}
+
+export namespace MultiEncodingEmbedding {
+  /**
+   * The encoded embedding data by encoding format.Returned, if more than one
+   * encoding format is used.
+   */
+  export interface Embedding {
+    float?: Array<number>;
+
+    int8?: Array<number>;
+
+    uint8?: Array<number>;
+
+    binary?: Array<number>;
+
+    ubinary?: Array<number>;
+
+    base64?: string;
+  }
+}
+
 /**
  * Info Pydantic Response Service Message
  */
@@ -150,7 +282,7 @@ export interface RerankParams {
   /**
    * The input documents to rerank.
    */
-  input: Array<string | unknown>;
+  input: Array<string | unknown | Array<unknown>>;
 
   /**
    * The fields of the documents to rank.
@@ -170,6 +302,9 @@ export interface RerankParams {
 
 export declare namespace TopLevel {
   export {
+    type Em as Em,
+    type Embedding as Embedding,
+    type MultiEncodingEmbedding as MultiEncodingEmbedding,
     type InfoResponse as InfoResponse,
     type RerankResponse as RerankResponse,
     type EmbedParams as EmbedParams,

package/src/version.ts

@@ -1 +1 @@
-export const VERSION = '0.1.0-alpha.19'; // x-release-please-version
+export const VERSION = '0.1.0-alpha.20'; // x-release-please-version

package/version.d.ts

@@ -1,2 +1,2 @@
-export declare const VERSION = "0.1.0-alpha.19";
+export declare const VERSION = "0.1.0-alpha.20";
 //# sourceMappingURL=version.d.ts.map

package/version.js

@@ -1,5 +1,5 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.VERSION = void 0;
-exports.VERSION = '0.1.0-alpha.19'; // x-release-please-version
+exports.VERSION = '0.1.0-alpha.20'; // x-release-please-version
 //# sourceMappingURL=version.js.map

package/version.mjs

@@ -1,2 +1,2 @@
-export const VERSION = '0.1.0-alpha.19'; // x-release-please-version
+export const VERSION = '0.1.0-alpha.20'; // x-release-please-version
 //# sourceMappingURL=version.mjs.map