@mixedbread/sdk 0.1.0-alpha.8 → 0.1.0

package/resources/vector-stores.d.ts

@@ -0,0 +1,2 @@
+export * from "./vector-stores/index.js";
+//# sourceMappingURL=vector-stores.d.ts.map

package/resources/vector-stores.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"vector-stores.d.ts","sourceRoot":"","sources":["../src/resources/vector-stores.ts"],"names":[],"mappings":"AAEA,cAAc,uBAAuB,CAAC"}

package/resources/vector-stores.js

@@ -0,0 +1,19 @@
+"use strict";
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./vector-stores/index.js"), exports);
+//# sourceMappingURL=vector-stores.js.map

package/resources/vector-stores.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vector-stores.js","sourceRoot":"","sources":["../src/resources/vector-stores.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;;;;;;;;;;;;;;AAEtF,2DAAsC"}

package/resources/vector-stores.mjs

@@ -0,0 +1,3 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+export * from "./vector-stores/index.mjs";
+//# sourceMappingURL=vector-stores.mjs.map

package/resources/vector-stores.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"vector-stores.mjs","sourceRoot":"","sources":["../src/resources/vector-stores.ts"],"names":[],"mappings":"AAAA,sFAAsF"}

package/resources.d.ts

@@ -0,0 +1,2 @@
+export * from "./resources/index.js";
+//# sourceMappingURL=resources.d.ts.map

package/resources.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"resources.d.ts","sourceRoot":"","sources":["src/resources.ts"],"names":[],"mappings":"AAAA,cAAc,mBAAmB,CAAC"}

package/resources.js

@@ -0,0 +1,18 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./resources/index.js"), exports);
+//# sourceMappingURL=resources.js.map

package/resources.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"resources.js","sourceRoot":"","sources":["src/resources.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,uDAAkC"}

package/resources.mjs

@@ -0,0 +1,2 @@
+export * from "./resources/index.mjs";
+//# sourceMappingURL=resources.mjs.map

package/resources.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"resources.mjs","sourceRoot":"","sources":["src/resources.ts"],"names":[],"mappings":""}

package/src/_shims/index.d.ts

@@ -79,3 +79,5 @@ export function fileFromPath(path: string, options?: FileFromPathOptions): Promi
 export function fileFromPath(path: string, filename?: string, options?: FileFromPathOptions): Promise<File>;
 
 export function isFsReadStream(value: any): value is FsReadStream;
+
+export const init: () => void;

package/src/_shims/index.js

@@ -3,7 +3,9 @@
  */
 const shims = require('./registry');
 const auto = require('@mixedbread/sdk/_shims/auto/runtime');
-if (!shims.kind) shims.setShims(auto.getRuntime(), { auto: true });
+exports.init = () => {
+  if (!shims.kind) shims.setShims(auto.getRuntime(), { auto: true });
+};
 for (const property of Object.keys(shims)) {
   Object.defineProperty(exports, property, {
     get() {
@@ -11,3 +13,5 @@ for (const property of Object.keys(shims)) {
     },
   });
 }
+
+exports.init();

package/src/_shims/index.mjs

@@ -3,5 +3,9 @@
  */
 import * as shims from './registry.mjs';
 import * as auto from "./auto/runtime";
-if (!shims.kind) shims.setShims(auto.getRuntime(), { auto: true });
+export const init = () => {
+  if (!shims.kind) shims.setShims(auto.getRuntime(), { auto: true });
+};
 export * from './registry.mjs';
+
+init();

package/src/core.ts

@@ -16,7 +16,12 @@ import {
   type RequestInit,
   type Response,
   type HeadersInit,
+  init,
 } from './_shims/index';
+
+// try running side effects outside of _shims/index to workaround https://github.com/vercel/next.js/issues/76881
+init();
+
 export { type Response };
 import { BlobLike, isBlobLike, isMultipartBody } from './uploads';
 export {
@@ -28,6 +33,20 @@ export {
 
 export type Fetch = (url: RequestInfo, init?: RequestInit) => Promise<Response>;
 
+/**
+ * An alias to the builtin `Array` type so we can
+ * easily alias it in import statements if there are name clashes.
+ */
+type _Array<T> = Array<T>;
+
+/**
+ * An alias to the builtin `Record` type so we can
+ * easily alias it in import statements if there are name clashes.
+ */
+type _Record<K extends keyof any, T> = Record<K, T>;
+
+export type { _Array as Array, _Record as Record };
+
 type PromiseOrValue<T> = T | Promise<T>;
 
 type APIResponseProps = {
@@ -48,8 +67,8 @@ async function defaultParseResponse<T>(props: APIResponseProps): Promise<T> {
   }
 
   const contentType = response.headers.get('content-type');
-  const isJSON =
-    contentType?.includes('application/json') || contentType?.includes('application/vnd.api+json');
+  const mediaType = contentType?.split(';')[0]?.trim();
+  const isJSON = mediaType?.includes('application/json') || mediaType?.endsWith('+json');
   if (isJSON) {
     const json = await response.json();
 
@@ -366,7 +385,7 @@ export abstract class APIClient {
       getHeader(headers, 'x-stainless-timeout') === undefined &&
       options.timeout
     ) {
-      reqHeaders['x-stainless-timeout'] = String(options.timeout);
+      reqHeaders['x-stainless-timeout'] = String(Math.trunc(options.timeout / 1000));
     }
 
     this.validateHeaders(reqHeaders, headers);
@@ -395,7 +414,7 @@ export abstract class APIClient {
       !headers ? {}
       : Symbol.iterator in headers ?
         Object.fromEntries(Array.from(headers as Iterable<string[]>).map((header) => [...header]))
-      : { ...headers }
+      : { ...(headers as any as Record<string, string>) }
     );
   }
 

package/src/index.ts

@@ -8,7 +8,16 @@ import { type LimitOffsetParams, LimitOffsetResponse } from './pagination';
 import * as Uploads from './uploads';
 import * as API from './resources/index';
 import * as TopLevelAPI from './resources/top-level';
-import { InfoResponse } from './resources/top-level';
+import {
+  EmbedParams,
+  Embedding,
+  EmbeddingCreateResponse,
+  InfoResponse,
+  MultiEncodingEmbedding,
+  RerankParams,
+  RerankResponse,
+} from './resources/top-level';
+import { EmbeddingCreateParams, Embeddings } from './resources/embeddings';
 import {
   FileCreateParams,
   FileDeleteResponse,
@@ -25,12 +34,13 @@ import {
   FileCounts,
   ScoredVectorStoreChunk,
   VectorStore,
+  VectorStoreChunkSearchOptions,
   VectorStoreCreateParams,
   VectorStoreDeleteResponse,
+  VectorStoreFileSearchOptions,
   VectorStoreListParams,
   VectorStoreQuestionAnsweringParams,
   VectorStoreQuestionAnsweringResponse,
-  VectorStoreSearchOptions,
   VectorStoreSearchParams,
   VectorStoreSearchResponse,
   VectorStoreUpdateParams,
@@ -39,7 +49,7 @@ import {
 } from './resources/vector-stores/vector-stores';
 
 const environments = {
-  production: 'https://api.mixedbread.ai',
+  production: 'https://api.mixedbread.com',
   local: 'http://127.0.0.1:8000',
 };
 type Environment = keyof typeof environments;
@@ -54,7 +64,7 @@ export interface ClientOptions {
    * Specifies the environment to use for the API.
    *
    * Each environment maps to a different base URL:
-   * - `production` corresponds to `https://api.mixedbread.ai`
+   * - `production` corresponds to `https://api.mixedbread.com`
    * - `local` corresponds to `http://127.0.0.1:8000`
    */
   environment?: Environment | undefined;
@@ -127,9 +137,9 @@ export class Mixedbread extends Core.APIClient {
   /**
    * API Client for interfacing with the Mixedbread API.
    *
-   * @param {string | undefined} [opts.apiKey=process.env['MXBAI_API_KEY'] ?? undefined]
+   * @param {string | undefined} [opts.apiKey=process.env['MIXEDBREAD_API_KEY'] ?? undefined]
    * @param {Environment} [opts.environment=production] - Specifies the environment URL to use for the API.
-   * @param {string} [opts.baseURL=process.env['MIXEDBREAD_BASE_URL'] ?? https://api.mixedbread.ai] - Override the default base URL for the API.
+   * @param {string} [opts.baseURL=process.env['MIXEDBREAD_BASE_URL'] ?? https://api.mixedbread.com] - Override the default base URL for the API.
    * @param {number} [opts.timeout=1 minute] - The maximum amount of time (in milliseconds) the client will wait for a response before timing out.
    * @param {number} [opts.httpAgent] - An HTTP agent used to manage HTTP(s) connections.
    * @param {Core.Fetch} [opts.fetch] - Specify a custom `fetch` function implementation.
@@ -139,12 +149,12 @@ export class Mixedbread extends Core.APIClient {
    */
   constructor({
     baseURL = Core.readEnv('MIXEDBREAD_BASE_URL'),
-    apiKey = Core.readEnv('MXBAI_API_KEY'),
+    apiKey = Core.readEnv('MIXEDBREAD_API_KEY'),
     ...opts
   }: ClientOptions = {}) {
     if (apiKey === undefined) {
       throw new Errors.MixedbreadError(
-        "The MXBAI_API_KEY environment variable is missing or empty; either provide it, or instantiate the Mixedbread client with an apiKey option, like new Mixedbread({ apiKey: 'My API Key' }).",
+        "The MIXEDBREAD_API_KEY environment variable is missing or empty; either provide it, or instantiate the Mixedbread client with an apiKey option, like new Mixedbread({ apiKey: 'My API Key' }).",
       );
     }
 
@@ -174,10 +184,26 @@ export class Mixedbread extends Core.APIClient {
     this.apiKey = apiKey;
   }
 
+  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,
+   * and normalization.
+   *
+   * Args: params: The parameters for creating embeddings.
+   *
+   * Returns: EmbeddingCreateResponse: The response containing the embeddings.
+   */
+  embed(
+    body: TopLevelAPI.EmbedParams,
+    options?: Core.RequestOptions,
+  ): Core.APIPromise<TopLevelAPI.EmbeddingCreateResponse> {
+    return this.post('/v1/embeddings', { body, ...options });
+  }
 
   /**
    * Returns service information, including name and version.
@@ -188,6 +214,20 @@ export class Mixedbread extends Core.APIClient {
     return this.get('/', options);
   }
 
+  /**
+   * Rerank different kind of documents for a given query.
+   *
+   * Args: params: RerankParams: The parameters for reranking.
+   *
+   * Returns: RerankResponse: The reranked documents for the input query.
+   */
+  rerank(
+    body: TopLevelAPI.RerankParams,
+    options?: Core.RequestOptions,
+  ): Core.APIPromise<TopLevelAPI.RerankResponse> {
+    return this.post('/v1/reranking', { body, ...options });
+  }
+
   protected override defaultQuery(): Core.DefaultQuery | undefined {
     return this._options.defaultQuery;
   }
@@ -224,30 +264,27 @@ export class Mixedbread extends Core.APIClient {
   static fileFromPath = Uploads.fileFromPath;
 }
 
+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;
 
   export import LimitOffset = Pagination.LimitOffset;
   export { type LimitOffsetParams as LimitOffsetParams, type LimitOffsetResponse as LimitOffsetResponse };
 
-  export { type InfoResponse as InfoResponse };
-
-  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,
+    type Embedding as Embedding,
+    type EmbeddingCreateResponse as EmbeddingCreateResponse,
+    type MultiEncodingEmbedding as MultiEncodingEmbedding,
+    type InfoResponse as InfoResponse,
+    type RerankResponse as RerankResponse,
+    type EmbedParams as EmbedParams,
+    type RerankParams as RerankParams,
   };
 
   export {
@@ -256,7 +293,8 @@ export declare namespace Mixedbread {
     type FileCounts as FileCounts,
     type ScoredVectorStoreChunk as ScoredVectorStoreChunk,
     type VectorStore as VectorStore,
-    type VectorStoreSearchOptions as VectorStoreSearchOptions,
+    type VectorStoreChunkSearchOptions as VectorStoreChunkSearchOptions,
+    type VectorStoreFileSearchOptions as VectorStoreFileSearchOptions,
     type VectorStoreDeleteResponse as VectorStoreDeleteResponse,
     type VectorStoreQuestionAnsweringResponse as VectorStoreQuestionAnsweringResponse,
     type VectorStoreSearchResponse as VectorStoreSearchResponse,
@@ -268,10 +306,25 @@ 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;
+  export type Usage = API.Usage;
 }
 
 export { toFile, fileFromPath } from './uploads';

package/src/resources/embeddings.ts

@@ -0,0 +1,67 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+import { APIResource } from '../resource';
+import * as Core from '../core';
+import * as TopLevelAPI from './top-level';
+
+export class Embeddings extends APIResource {
+  /**
+   * Create embeddings for text or images using the specified model, encoding format,
+   * and normalization.
+   *
+   * Args: params: The parameters for creating embeddings.
+   *
+   * Returns: EmbeddingCreateResponse: The response containing the embeddings.
+   */
+  create(
+    body: EmbeddingCreateParams,
+    options?: Core.RequestOptions,
+  ): Core.APIPromise<TopLevelAPI.EmbeddingCreateResponse> {
+    return this._client.post('/v1/embeddings', { body, ...options });
+  }
+}
+
+export interface EmbeddingCreateParams {
+  /**
+   * The model to use for creating embeddings.
+   */
+  model: string;
+
+  /**
+   * The input to create embeddings for.
+   */
+  input: Array<string>;
+
+  /**
+   * The number of dimensions to use for the embeddings.
+   */
+  dimensions?: number | null;
+
+  /**
+   * The prompt to use for the embedding creation.
+   */
+  prompt?: string | null;
+
+  /**
+   * Whether to normalize the embeddings.
+   */
+  normalized?: boolean;
+
+  /**
+   * The encoding format(s) of the embeddings. Can be a single format or a list of
+   * formats.
+   */
+  encoding_format?:
+    | 'float'
+    | 'float16'
+    | 'base64'
+    | 'binary'
+    | 'ubinary'
+    | 'int8'
+    | 'uint8'
+    | Array<'float' | 'float16' | 'base64' | 'binary' | 'ubinary' | 'int8' | 'uint8'>;
+}
+
+export declare namespace Embeddings {
+  export { type EmbeddingCreateParams as EmbeddingCreateParams };
+}

package/src/resources/extractions/content.ts

@@ -17,13 +17,12 @@ export class Content extends APIResource {
 }
 
 /**
- * Result of an extraction operation.
+ * The result of an extraction job.
  */
 export interface ExtractionResult {
-  /**
-   * The extracted data
-   */
-  data: unknown;
+  data: Record<string, unknown>;
+
+  warnings: Array<string>;
 }
 
 export interface ContentCreateParams {
@@ -35,7 +34,7 @@ export interface ContentCreateParams {
   /**
    * The JSON schema to use for extraction
    */
-  json_schema: unknown;
+  json_schema: Record<string, unknown>;
 }
 
 export declare namespace Content {

package/src/resources/extractions/jobs.ts

@@ -33,9 +33,59 @@ export class Jobs extends APIResource {
  */
 export interface ExtractionJob {
   /**
-   * Result of an extraction operation.
+   * Unique identifier for the extraction job
+   */
+  id: string;
+
+  /**
+   * ID of the organization that owns this job
+   */
+  organization_id: string;
+
+  /**
+   * ID of the file being extracted
+   */
+  file_id: string;
+
+  /**
+   * When the job was created
+   */
+  created_at: string;
+
+  /**
+   * When the job was last updated
+   */
+  updated_at: string;
+
+  /**
+   * When the job started processing
+   */
+  started_at: string | null;
+
+  /**
+   * When the job finished processing
+   */
+  finished_at: string | null;
+
+  /**
+   * Current status of the job
+   */
+  status: 'pending' | 'in_progress' | 'cancelled' | 'completed' | 'failed';
+
+  /**
+   * The result of an extraction job.
    */
   result: ContentAPI.ExtractionResult | null;
+
+  /**
+   * Error information if failed
+   */
+  error: Record<string, unknown> | null;
+
+  /**
+   * The JSON schema used for extraction
+   */
+  json_schema: Record<string, unknown>;
 }
 
 export interface JobCreateParams {
@@ -47,7 +97,7 @@ export interface JobCreateParams {
   /**
    * The JSON schema to use for extraction
    */
-  json_schema: unknown;
+  json_schema: Record<string, unknown>;
 }
 
 export declare namespace Jobs {

package/src/resources/extractions/schema.ts

@@ -45,7 +45,7 @@ export interface CreatedJsonSchema {
   /**
    * The created JSON schema
    */
-  json_schema: unknown;
+  json_schema: Record<string, unknown>;
 }
 
 /**
@@ -55,7 +55,7 @@ export interface EnhancedJsonSchema {
   /**
    * The enhanced JSON schema
    */
-  json_schema: unknown;
+  json_schema: Record<string, unknown>;
 }
 
 /**
@@ -75,7 +75,7 @@ export interface ValidatedJsonSchema {
   /**
    * The validated JSON schema
    */
-  json_schema: unknown;
+  json_schema: Record<string, unknown>;
 }
 
 export interface SchemaCreateParams {
@@ -89,14 +89,14 @@ export interface SchemaEnhanceParams {
   /**
    * The JSON schema to enhance
    */
-  json_schema: unknown;
+  json_schema: Record<string, unknown>;
 }
 
 export interface SchemaValidateParams {
   /**
    * The JSON schema to validate
    */
-  json_schema: unknown;
+  json_schema: Record<string, unknown>;
 }
 
 export declare namespace Schema {

package/src/resources/extractions.ts

@@ -0,0 +1,3 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+export * from './extractions/index';

package/src/resources/index.ts

@@ -1,6 +1,7 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 export * from './shared';
+export { Embeddings, type EmbeddingCreateParams } from './embeddings';
 export { Extractions } from './extractions/extractions';
 export {
   FileObjectsLimitOffset,
@@ -19,7 +20,8 @@ export {
   type FileCounts,
   type ScoredVectorStoreChunk,
   type VectorStore,
-  type VectorStoreSearchOptions,
+  type VectorStoreChunkSearchOptions,
+  type VectorStoreFileSearchOptions,
   type VectorStoreDeleteResponse,
   type VectorStoreQuestionAnsweringResponse,
   type VectorStoreSearchResponse,
@@ -29,4 +31,12 @@ export {
   type VectorStoreQuestionAnsweringParams,
   type VectorStoreSearchParams,
 } from './vector-stores/vector-stores';
-export { type InfoResponse } from './top-level';
+export {
+  type Embedding,
+  type EmbeddingCreateResponse,
+  type MultiEncodingEmbedding,
+  type InfoResponse,
+  type RerankResponse,
+  type EmbedParams,
+  type RerankParams,
+} from './top-level';

package/src/resources/parsing/jobs.ts

@@ -178,6 +178,11 @@ export interface ParsingJob {
    */
   id: string;
 
+  /**
+   * The ID of the file to parse
+   */
+  file_id: string;
+
   /**
    * The status of the job
    */
@@ -255,6 +260,11 @@ export namespace ParsingJob {
      * List of extracted chunks from the document
      */
     chunks: Array<Result.Chunk>;
+
+    /**
+     * List of (width, height) tuples for each page
+     */
+    page_sizes?: Array<Array<unknown>>;
   }
 
   export namespace Result {
@@ -337,6 +347,11 @@ export interface JobListResponse {
    */
   id: string;
 
+  /**
+   * The ID of the file to parse
+   */
+  file_id: string;
+
   /**
    * The status of the job
    */

package/src/resources/parsing.ts

@@ -0,0 +1,3 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+export * from './parsing/index';