@mixedbread/sdk 0.5.0 → 0.6.0

package/src/resources/data-sources/data-sources.ts

@@ -0,0 +1,271 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+import { APIResource } from '../../core/resource';
+import * as ConnectorsAPI from './connectors';
+import {
+  ConnectorCreateParams,
+  ConnectorDeleteParams,
+  ConnectorDeleteResponse,
+  ConnectorListParams,
+  ConnectorRetrieveParams,
+  ConnectorUpdateParams,
+  Connectors,
+  DataSourceConnector,
+  DataSourceConnectorsLimitOffset,
+} from './connectors';
+import { APIPromise } from '../../core/api-promise';
+import { LimitOffset, type LimitOffsetParams, PagePromise } from '../../core/pagination';
+import { RequestOptions } from '../../internal/request-options';
+import { path } from '../../internal/utils/path';
+
+export class DataSources extends APIResource {
+  connectors: ConnectorsAPI.Connectors = new ConnectorsAPI.Connectors(this._client);
+
+  /**
+   * Create a new data source.
+   *
+   * Args: params: The data source to create.
+   *
+   * Returns: The created data source.
+   */
+  create(body: DataSourceCreateParams, options?: RequestOptions): APIPromise<DataSource> {
+    return this._client.post('/v1/data_sources/', { body, ...options });
+  }
+
+  /**
+   * Get a data source by ID.
+   *
+   * Args: data_source_id: The ID of the data source to fetch.
+   *
+   * Returns: The data source.
+   */
+  retrieve(dataSourceID: string, options?: RequestOptions): APIPromise<DataSource> {
+    return this._client.get(path`/v1/data_sources/${dataSourceID}`, options);
+  }
+
+  /**
+   * Update a data source.
+   *
+   * Args: data_source_id: The ID of the data source to update. params: The data
+   * source to update.
+   *
+   * Returns: The updated data source.
+   */
+  update(
+    dataSourceID: string,
+    body: DataSourceUpdateParams,
+    options?: RequestOptions,
+  ): APIPromise<DataSource> {
+    return this._client.put(path`/v1/data_sources/${dataSourceID}`, { body, ...options });
+  }
+
+  /**
+   * Get all data sources.
+   *
+   * Returns: The list of data sources.
+   */
+  list(
+    query: DataSourceListParams | null | undefined = {},
+    options?: RequestOptions,
+  ): PagePromise<DataSourcesLimitOffset, DataSource> {
+    return this._client.getAPIList('/v1/data_sources/', LimitOffset<DataSource>, { query, ...options });
+  }
+
+  /**
+   * Delete a data source.
+   *
+   * Args: data_source_id: The ID of the data source to delete.
+   */
+  delete(dataSourceID: string, options?: RequestOptions): APIPromise<DataSourceDeleteResponse> {
+    return this._client.delete(path`/v1/data_sources/${dataSourceID}`, options);
+  }
+}
+
+export type DataSourcesLimitOffset = LimitOffset<DataSource>;
+
+/**
+ * Service-level representation of a data source.
+ */
+export interface DataSource {
+  /**
+   * The ID of the data source
+   */
+  id: string;
+
+  /**
+   * The creation time of the data source
+   */
+  created_at: string;
+
+  /**
+   * The last update time of the data source
+   */
+  updated_at: string;
+
+  /**
+   * The type of data source
+   */
+  type: DataSourceType;
+
+  /**
+   * The name of the data source
+   */
+  name: string;
+
+  /**
+   * The metadata of the data source
+   */
+  metadata: unknown;
+
+  /**
+   * Authentication parameters for a OAuth data source.
+   */
+  auth_params: DataSourceOauth2Params | null;
+
+  /**
+   * The type of the object
+   */
+  object?: 'data_source';
+}
+
+/**
+ * Authentication parameters for a OAuth data source.
+ */
+export interface DataSourceOauth2Params {
+  type?: 'oauth2';
+
+  /**
+   * The timestamp when the OAuth2 credentials were created
+   */
+  created_at?: string;
+
+  /**
+   * The OAuth2 client ID
+   */
+  client_id: string;
+
+  /**
+   * The OAuth2 client secret
+   */
+  client_secret: string;
+
+  /**
+   * The OAuth2 redirect URI
+   */
+  redirect_uri: string;
+
+  /**
+   * The OAuth2 scope
+   */
+  scope: string;
+
+  /**
+   * The OAuth2 access token
+   */
+  access_token?: string | null;
+
+  /**
+   * The OAuth2 refresh token
+   */
+  refresh_token?: string | null;
+
+  /**
+   * The OAuth2 token type
+   */
+  token_type?: string | null;
+
+  /**
+   * The OAuth2 token expiration timestamp
+   */
+  expires_on?: string | null;
+}
+
+export type DataSourceType = 'notion' | 'linear';
+
+/**
+ * Deleted data source.
+ */
+export interface DataSourceDeleteResponse {
+  /**
+   * The ID of the data source
+   */
+  id: string;
+
+  /**
+   * Whether the data source was deleted
+   */
+  deleted?: boolean;
+
+  /**
+   * The type of the object
+   */
+  object?: 'data_source';
+}
+
+export interface DataSourceCreateParams {
+  /**
+   * The type of data source to create
+   */
+  type: DataSourceType;
+
+  /**
+   * The name of the data source
+   */
+  name: string;
+
+  /**
+   * The metadata of the data source
+   */
+  metadata?: unknown;
+
+  /**
+   * Authentication parameters for a OAuth data source.
+   */
+  auth_params?: DataSourceOauth2Params | null;
+}
+
+export interface DataSourceUpdateParams {
+  /**
+   * The name of the data source
+   */
+  name?: string | null;
+
+  /**
+   * The metadata of the data source
+   */
+  metadata?: unknown;
+
+  /**
+   * Authentication parameters for a OAuth data source.
+   */
+  auth_params?: DataSourceOauth2Params | null;
+}
+
+export interface DataSourceListParams extends LimitOffsetParams {}
+
+DataSources.Connectors = Connectors;
+
+export declare namespace DataSources {
+  export {
+    type DataSource as DataSource,
+    type DataSourceOauth2Params as DataSourceOauth2Params,
+    type DataSourceType as DataSourceType,
+    type DataSourceDeleteResponse as DataSourceDeleteResponse,
+    type DataSourcesLimitOffset as DataSourcesLimitOffset,
+    type DataSourceCreateParams as DataSourceCreateParams,
+    type DataSourceUpdateParams as DataSourceUpdateParams,
+    type DataSourceListParams as DataSourceListParams,
+  };
+
+  export {
+    Connectors as Connectors,
+    type DataSourceConnector as DataSourceConnector,
+    type ConnectorDeleteResponse as ConnectorDeleteResponse,
+    type DataSourceConnectorsLimitOffset as DataSourceConnectorsLimitOffset,
+    type ConnectorCreateParams as ConnectorCreateParams,
+    type ConnectorRetrieveParams as ConnectorRetrieveParams,
+    type ConnectorUpdateParams as ConnectorUpdateParams,
+    type ConnectorListParams as ConnectorListParams,
+    type ConnectorDeleteParams as ConnectorDeleteParams,
+  };
+}

package/src/resources/data-sources/index.ts

@@ -0,0 +1,24 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+export {
+  Connectors,
+  type DataSourceConnector,
+  type ConnectorDeleteResponse,
+  type ConnectorCreateParams,
+  type ConnectorRetrieveParams,
+  type ConnectorUpdateParams,
+  type ConnectorListParams,
+  type ConnectorDeleteParams,
+  type DataSourceConnectorsLimitOffset,
+} from './connectors';
+export {
+  DataSources,
+  type DataSource,
+  type DataSourceOauth2Params,
+  type DataSourceType,
+  type DataSourceDeleteResponse,
+  type DataSourceCreateParams,
+  type DataSourceUpdateParams,
+  type DataSourceListParams,
+  type DataSourcesLimitOffset,
+} from './data-sources';

package/src/resources/data-sources.ts

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

package/src/resources/embeddings.ts

@@ -13,15 +13,6 @@ export class Embeddings extends APIResource {
    * Args: params: The parameters for creating embeddings.
    *
    * Returns: EmbeddingCreateResponse: The response containing the embeddings.
-   *
-   * @example
-   * ```ts
-   * const embeddingCreateResponse =
-   *   await client.embeddings.create({
-   *     model: 'mixedbread-ai/mxbai-embed-large-v1',
-   *     input: 'x',
-   *   });
-   * ```
    */
   create(
     body: EmbeddingCreateParams,
@@ -31,6 +22,25 @@ export class Embeddings extends APIResource {
   }
 }
 
+/**
+ * Enumeration of encoding formats.
+ */
+export type EncodingFormat = 'float' | 'float16' | 'base64' | 'binary' | 'ubinary' | 'int8' | 'uint8';
+
+export type ObjectType =
+  | 'list'
+  | 'parsing_job'
+  | 'extraction_job'
+  | 'embedding'
+  | 'embedding_dict'
+  | 'rank_result'
+  | 'file'
+  | 'vector_store'
+  | 'vector_store.file'
+  | 'api_key'
+  | 'data_source'
+  | 'data_source.connector';
+
 export interface EmbeddingCreateParams {
   /**
    * The model to use for creating embeddings.
@@ -61,17 +71,13 @@ export interface EmbeddingCreateParams {
    * 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'>;
+  encoding_format?: EncodingFormat | Array<EncodingFormat>;
 }
 
 export declare namespace Embeddings {
-  export { type EmbeddingCreateParams as EmbeddingCreateParams };
+  export {
+    type EncodingFormat as EncodingFormat,
+    type ObjectType as ObjectType,
+    type EmbeddingCreateParams as EmbeddingCreateParams,
+  };
 }

package/src/resources/extractions/content.ts

@@ -26,70 +26,73 @@ export interface ExtractionResult {
   warnings: Array<string>;
 }
 
-export interface ContentCreateParams {
+/**
+ * Model for image input validation.
+ */
+export interface ImageURLInput {
   /**
-   * The content to extract from
+   * Input type identifier
    */
-  content: string | Array<string> | Array<ContentCreateParams.TextInput | ContentCreateParams.ImageURLInput>;
+  type?: 'image_url';
 
   /**
-   * The JSON schema to use for extraction
+   * The image input specification.
    */
-  json_schema: Record<string, unknown>;
+  image_url: ImageURLInput.ImageURL;
+}
 
+export namespace ImageURLInput {
   /**
-   * Additional instructions for the extraction
+   * The image input specification.
    */
-  instructions?: string | null;
+  export interface ImageURL {
+    /**
+     * The image URL. Can be either a URL or a Data URI.
+     */
+    url: string;
+  }
 }
 
-export namespace ContentCreateParams {
+/**
+ * Model for text input validation.
+ *
+ * Attributes: type: Input type identifier, always "text" text: The actual text
+ * content, with length and whitespace constraints
+ */
+export interface TextInput {
   /**
-   * Model for text input validation.
-   *
-   * Attributes: type: Input type identifier, always "text" text: The actual text
-   * content, with length and whitespace constraints
+   * Input type identifier
    */
-  export interface TextInput {
-    /**
-     * Input type identifier
-     */
-    type?: 'text';
+  type?: 'text';
 
-    /**
-     * Text content to process
-     */
-    text: string;
-  }
+  /**
+   * Text content to process
+   */
+  text: string;
+}
 
+export interface ContentCreateParams {
   /**
-   * Model for image input validation.
+   * The content to extract from
    */
-  export interface ImageURLInput {
-    /**
-     * Input type identifier
-     */
-    type?: 'image_url';
+  content: string | Array<string> | Array<TextInput | ImageURLInput>;
 
-    /**
-     * The image input specification.
-     */
-    image_url: ImageURLInput.ImageURL;
-  }
+  /**
+   * The JSON schema to use for extraction
+   */
+  json_schema: Record<string, unknown>;
 
-  export namespace ImageURLInput {
-    /**
-     * The image input specification.
-     */
-    export interface ImageURL {
-      /**
-       * The image URL. Can be either a URL or a Data URI.
-       */
-      url: string;
-    }
-  }
+  /**
+   * Additional instructions for the extraction
+   */
+  instructions?: string | null;
 }
 
 export declare namespace Content {
-  export { type ExtractionResult as ExtractionResult, type ContentCreateParams as ContentCreateParams };
+  export {
+    type ExtractionResult as ExtractionResult,
+    type ImageURLInput as ImageURLInput,
+    type TextInput as TextInput,
+    type ContentCreateParams as ContentCreateParams,
+  };
 }

package/src/resources/extractions/extractions.ts

@@ -2,7 +2,7 @@
 
 import { APIResource } from '../../core/resource';
 import * as ContentAPI from './content';
-import { Content, ContentCreateParams, ExtractionResult } from './content';
+import { Content, ContentCreateParams, ExtractionResult, ImageURLInput, TextInput } from './content';
 import * as JobsAPI from './jobs';
 import { ExtractionJob, JobCreateParams, Jobs } from './jobs';
 import * as SchemaAPI from './schema';
@@ -42,6 +42,8 @@ export declare namespace Extractions {
   export {
     Content as Content,
     type ExtractionResult as ExtractionResult,
+    type ImageURLInput as ImageURLInput,
+    type TextInput as TextInput,
     type ContentCreateParams as ContentCreateParams,
   };
 }

package/src/resources/extractions/index.ts

@@ -1,6 +1,12 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-export { Content, type ExtractionResult, type ContentCreateParams } from './content';
+export {
+  Content,
+  type ExtractionResult,
+  type ImageURLInput,
+  type TextInput,
+  type ContentCreateParams,
+} from './content';
 export { Extractions } from './extractions';
 export { Jobs, type ExtractionJob, type JobCreateParams } from './jobs';
 export {

package/src/resources/extractions/jobs.ts

@@ -2,6 +2,7 @@
 
 import { APIResource } from '../../core/resource';
 import * as ContentAPI from './content';
+import * as ParsingJobsAPI from '../parsing/jobs';
 import { APIPromise } from '../../core/api-promise';
 import { RequestOptions } from '../../internal/request-options';
 import { path } from '../../internal/utils/path';
@@ -72,7 +73,7 @@ export interface ExtractionJob {
   /**
    * Current status of the job
    */
-  status: 'pending' | 'in_progress' | 'cancelled' | 'completed' | 'failed';
+  status: ParsingJobsAPI.ParsingJobStatus;
 
   /**
    * The result of an extraction job.

package/src/resources/files.ts

@@ -132,6 +132,26 @@ export interface FileObject {
   updated_at: string;
 }
 
+/**
+ * Pagination model that includes total count of items.
+ */
+export interface PaginationWithTotal {
+  /**
+   * Maximum number of items to return per page
+   */
+  limit?: number;
+
+  /**
+   * Offset of the first item to return
+   */
+  offset?: number;
+
+  /**
+   * Total number of items available
+   */
+  total?: number;
+}
+
 export interface FileDeleteResponse {
   /**
    * The ID of the deleted file
@@ -168,6 +188,7 @@ export interface FileListParams extends LimitOffsetParams {}
 export declare namespace Files {
   export {
     type FileObject as FileObject,
+    type PaginationWithTotal as PaginationWithTotal,
     type FileDeleteResponse as FileDeleteResponse,
     type FileObjectsLimitOffset as FileObjectsLimitOffset,
     type FileCreateParams as FileCreateParams,

package/src/resources/index.ts

@@ -1,11 +1,33 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 export * from './shared';
-export { Embeddings, type EmbeddingCreateParams } from './embeddings';
+export {
+  APIKeys,
+  type APIKey,
+  type APIKeyCreated,
+  type APIKeyDeleteResponse,
+  type APIKeyCreateParams,
+  type APIKeyListParams,
+  type APIKeysLimitOffset,
+} from './api-keys';
+export { Chat, type ChatCreateCompletionResponse } from './chat';
+export {
+  DataSources,
+  type DataSource,
+  type DataSourceOauth2Params,
+  type DataSourceType,
+  type DataSourceDeleteResponse,
+  type DataSourceCreateParams,
+  type DataSourceUpdateParams,
+  type DataSourceListParams,
+  type DataSourcesLimitOffset,
+} from './data-sources/data-sources';
+export { Embeddings, type EncodingFormat, type ObjectType, type EmbeddingCreateParams } from './embeddings';
 export { Extractions } from './extractions/extractions';
 export {
   Files,
   type FileObject,
+  type PaginationWithTotal,
   type FileDeleteResponse,
   type FileCreateParams,
   type FileUpdateParams,
@@ -16,11 +38,12 @@ export { Parsing } from './parsing/parsing';
 export {
   VectorStores,
   type ExpiresAfter,
-  type FileCounts,
-  type ScoredVectorStoreChunk,
+  type ScoredAudioURLInputChunk,
+  type ScoredImageURLInputChunk,
+  type ScoredTextInputChunk,
+  type ScoredVideoURLInputChunk,
   type VectorStore,
   type VectorStoreChunkSearchOptions,
-  type VectorStoreFileSearchOptions,
   type VectorStoreDeleteResponse,
   type VectorStoreQuestionAnsweringResponse,
   type VectorStoreSearchResponse,

package/src/resources/parsing/index.ts

@@ -2,7 +2,11 @@
 
 export {
   Jobs,
+  type ChunkingStrategy,
+  type ElementType,
+  type ParsingJobStatus,
   type ParsingJob,
+  type ReturnFormat,
   type JobListResponse,
   type JobDeleteResponse,
   type JobCreateParams,