@mixedbread/sdk 0.5.0 → 0.6.0

package/src/resources/parsing/jobs.ts

@@ -1,6 +1,7 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 import { APIResource } from '../../core/resource';
+import * as JobsAPI from './jobs';
 import { APIPromise } from '../../core/api-promise';
 import { LimitOffset, type LimitOffsetParams, PagePromise } from '../../core/pagination';
 import { RequestOptions } from '../../internal/request-options';
@@ -162,6 +163,29 @@ export class Jobs extends APIResource {
 
 export type JobListResponsesLimitOffset = LimitOffset<JobListResponse>;
 
+/**
+ * Strategy used for chunking document content.
+ */
+export type ChunkingStrategy = 'page';
+
+/**
+ * Types of elements that can be extracted from a document.
+ */
+export type ElementType =
+  | 'caption'
+  | 'footnote'
+  | 'formula'
+  | 'list-item'
+  | 'page-footer'
+  | 'page-header'
+  | 'picture'
+  | 'section-header'
+  | 'table'
+  | 'text'
+  | 'title';
+
+export type ParsingJobStatus = 'pending' | 'in_progress' | 'cancelled' | 'completed' | 'failed';
+
 /**
  * A job for parsing documents with its current state and result.
  */
@@ -179,7 +203,7 @@ export interface ParsingJob {
   /**
    * The status of the job
    */
-  status: 'pending' | 'in_progress' | 'cancelled' | 'completed' | 'failed';
+  status: ParsingJobStatus;
 
   /**
    * The error of the job
@@ -225,29 +249,17 @@ export namespace ParsingJob {
     /**
      * The strategy used for chunking the document
      */
-    chunking_strategy: 'page';
+    chunking_strategy: JobsAPI.ChunkingStrategy;
 
     /**
      * The format of the returned content
      */
-    return_format: 'html' | 'markdown' | 'plain';
+    return_format: JobsAPI.ReturnFormat;
 
     /**
      * The types of elements extracted
      */
-    element_types: Array<
-      | 'caption'
-      | 'footnote'
-      | 'formula'
-      | 'list-item'
-      | 'page-footer'
-      | 'page-header'
-      | 'picture'
-      | 'section-header'
-      | 'table'
-      | 'text'
-      | 'title'
-    >;
+    element_types: Array<JobsAPI.ElementType>;
 
     /**
      * List of extracted chunks from the document
@@ -289,18 +301,7 @@ export namespace ParsingJob {
         /**
          * The type of the extracted element
          */
-        type:
-          | 'caption'
-          | 'footnote'
-          | 'formula'
-          | 'list-item'
-          | 'page-footer'
-          | 'page-header'
-          | 'picture'
-          | 'section-header'
-          | 'table'
-          | 'text'
-          | 'title';
+        type: JobsAPI.ElementType;
 
         /**
          * The confidence score of the extraction
@@ -331,6 +332,11 @@ export namespace ParsingJob {
   }
 }
 
+/**
+ * Format options for the returned document content.
+ */
+export type ReturnFormat = 'html' | 'markdown' | 'plain';
+
 /**
  * A parsing job item for list responses, omitting result and error fields.
  */
@@ -348,7 +354,7 @@ export interface JobListResponse {
   /**
    * The status of the job
    */
-  status: 'pending' | 'in_progress' | 'cancelled' | 'completed' | 'failed';
+  status: ParsingJobStatus;
 
   /**
    * The started time of the job
@@ -405,29 +411,17 @@ export interface JobCreateParams {
   /**
    * The elements to extract from the document
    */
-  element_types?: Array<
-    | 'caption'
-    | 'footnote'
-    | 'formula'
-    | 'list-item'
-    | 'page-footer'
-    | 'page-header'
-    | 'picture'
-    | 'section-header'
-    | 'table'
-    | 'text'
-    | 'title'
-  > | null;
+  element_types?: Array<ElementType> | null;
 
   /**
    * The strategy to use for chunking the content
    */
-  chunking_strategy?: 'page';
+  chunking_strategy?: ChunkingStrategy;
 
   /**
    * The format of the returned content
    */
-  return_format?: 'html' | 'markdown' | 'plain';
+  return_format?: ReturnFormat;
 
   /**
    * The strategy to use for OCR
@@ -439,7 +433,11 @@ export interface JobListParams extends LimitOffsetParams {}
 
 export declare namespace Jobs {
   export {
+    type ChunkingStrategy as ChunkingStrategy,
+    type ElementType as ElementType,
+    type ParsingJobStatus as ParsingJobStatus,
     type ParsingJob as ParsingJob,
+    type ReturnFormat as ReturnFormat,
     type JobListResponse as JobListResponse,
     type JobDeleteResponse as JobDeleteResponse,
     type JobListResponsesLimitOffset as JobListResponsesLimitOffset,

package/src/resources/parsing/parsing.ts

@@ -3,6 +3,8 @@
 import { APIResource } from '../../core/resource';
 import * as JobsAPI from './jobs';
 import {
+  ChunkingStrategy,
+  ElementType,
   JobCreateParams,
   JobDeleteResponse,
   JobListParams,
@@ -10,6 +12,8 @@ import {
   JobListResponsesLimitOffset,
   Jobs,
   ParsingJob,
+  ParsingJobStatus,
+  ReturnFormat,
 } from './jobs';
 
 export class Parsing extends APIResource {
@@ -21,7 +25,11 @@ Parsing.Jobs = Jobs;
 export declare namespace Parsing {
   export {
     Jobs as Jobs,
+    type ChunkingStrategy as ChunkingStrategy,
+    type ElementType as ElementType,
+    type ParsingJobStatus as ParsingJobStatus,
     type ParsingJob as ParsingJob,
+    type ReturnFormat as ReturnFormat,
     type JobListResponse as JobListResponse,
     type JobDeleteResponse as JobDeleteResponse,
     type JobListResponsesLimitOffset as JobListResponsesLimitOffset,

package/src/resources/top-level.ts

@@ -1,5 +1,6 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
+import * as EmbeddingsAPI from './embeddings';
 import * as Shared from './shared';
 
 export interface Embedding {
@@ -38,19 +39,7 @@ export interface EmbeddingCreateResponse {
   /**
    * The object type of the response
    */
-  object?:
-    | 'list'
-    | 'parsing_job'
-    | 'extraction_job'
-    | 'embedding'
-    | 'embedding_dict'
-    | 'rank_result'
-    | 'file'
-    | 'vector_store'
-    | 'vector_store.file'
-    | 'api_key'
-    | 'data_source'
-    | 'data_source.connector';
+  object?: EmbeddingsAPI.ObjectType;
 
   /**
    * Whether the embeddings are normalized.
@@ -60,15 +49,7 @@ export interface EmbeddingCreateResponse {
   /**
    * The encoding formats of the embeddings.
    */
-  encoding_format:
-    | 'float'
-    | 'float16'
-    | 'base64'
-    | 'binary'
-    | 'ubinary'
-    | 'int8'
-    | 'uint8'
-    | Array<'float' | 'float16' | 'base64' | 'binary' | 'ubinary' | 'int8' | 'uint8'>;
+  encoding_format: EmbeddingsAPI.EncodingFormat | Array<EmbeddingsAPI.EncodingFormat>;
 
   /**
    * The number of dimensions used for the embeddings.
@@ -142,19 +123,7 @@ export interface RerankResponse {
   /**
    * The object type of the response
    */
-  object?:
-    | 'list'
-    | 'parsing_job'
-    | 'extraction_job'
-    | 'embedding'
-    | 'embedding_dict'
-    | 'rank_result'
-    | 'file'
-    | 'vector_store'
-    | 'vector_store.file'
-    | 'api_key'
-    | 'data_source'
-    | 'data_source.connector';
+  object?: 'list';
 
   /**
    * The number of documents to return.
@@ -221,15 +190,7 @@ export interface EmbedParams {
    * 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?: EmbeddingsAPI.EncodingFormat | Array<EmbeddingsAPI.EncodingFormat>;
 }
 
 export interface RerankParams {

package/src/resources/vector-stores/files.ts

@@ -264,7 +264,12 @@ export interface ScoredVectorStoreFile {
   /**
    * chunks
    */
-  chunks: Array<VectorStoresAPI.ScoredVectorStoreChunk> | null;
+  chunks: Array<
+    | VectorStoresAPI.ScoredTextInputChunk
+    | VectorStoresAPI.ScoredImageURLInputChunk
+    | VectorStoresAPI.ScoredAudioURLInputChunk
+    | VectorStoresAPI.ScoredVideoURLInputChunk
+  > | null;
 }
 
 /**
@@ -432,7 +437,39 @@ export interface FileSearchParams {
   /**
    * Search configuration options
    */
-  search_options?: VectorStoresAPI.VectorStoreFileSearchOptions;
+  search_options?: FileSearchParams.SearchOptions;
+}
+
+export namespace FileSearchParams {
+  /**
+   * Search configuration options
+   */
+  export interface SearchOptions {
+    /**
+     * Minimum similarity score threshold
+     */
+    score_threshold?: number;
+
+    /**
+     * Whether to rewrite the query
+     */
+    rewrite_query?: boolean;
+
+    /**
+     * Whether to return file metadata
+     */
+    return_metadata?: boolean;
+
+    /**
+     * Whether to return matching text chunks
+     */
+    return_chunks?: boolean;
+
+    /**
+     * Number of chunks to return for each file
+     */
+    chunks_per_file?: number;
+  }
 }
 
 export declare namespace Files {

package/src/resources/vector-stores/index.ts

@@ -16,11 +16,12 @@ export {
 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,