@mixedbread/sdk 0.1.0-alpha.9 → 0.2.0

package/src/resources/top-level.ts

@@ -1,5 +1,117 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
+import * as Shared from './shared';
+
+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: Shared.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 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
  */
@@ -9,6 +121,153 @@ export interface InfoResponse {
   version: string;
 }
 
+export interface RerankResponse {
+  /**
+   * The usage of the model
+   */
+  usage: Shared.Usage;
+
+  /**
+   * The model used
+   */
+  model: string;
+
+  /**
+   * The ranked documents.
+   */
+  data: Array<RerankResponse.Data>;
+
+  /**
+   * The object type of the response
+   */
+  object?:
+    | 'list'
+    | 'parsing_job'
+    | 'job'
+    | 'embedding'
+    | 'embedding_dict'
+    | 'rank_result'
+    | 'file'
+    | 'vector_store'
+    | 'vector_store.file'
+    | 'api_key';
+
+  /**
+   * The number of documents to return.
+   */
+  top_k: number;
+
+  /**
+   * Whether to return the documents.
+   */
+  return_input: boolean;
+}
+
+export namespace RerankResponse {
+  export interface Data {
+    /**
+     * The index of the document.
+     */
+    index: number;
+
+    /**
+     * The score of the document.
+     */
+    score: number;
+
+    /**
+     * The input document.
+     */
+    input?: unknown;
+
+    /**
+     * The object type.
+     */
+    object?: 'rank_result';
+  }
+}
+
+export interface EmbedParams {
+  /**
+   * The model to use for creating embeddings.
+   */
+  model: string;
+
+  /**
+   * The input to create embeddings for.
+   */
+  input: string | 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 interface RerankParams {
+  /**
+   * The model to use for reranking documents.
+   */
+  model?: string;
+
+  /**
+   * The query to rerank the documents.
+   */
+  query: string;
+
+  /**
+   * The input documents to rerank.
+   */
+  input: Array<string | unknown | Array<unknown>>;
+
+  /**
+   * The fields of the documents to rank.
+   */
+  rank_fields?: Array<string> | null;
+
+  /**
+   * The number of documents to return.
+   */
+  top_k?: number;
+
+  /**
+   * Whether to return the documents.
+   */
+  return_input?: boolean;
+}
+
 export declare namespace TopLevel {
-  export { type InfoResponse as InfoResponse };
+  export {
+    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,
+  };
 }

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

@@ -223,6 +223,11 @@ export interface ScoredVectorStoreFile {
    */
   id: string;
 
+  /**
+   * Name of the file
+   */
+  filename?: string;
+
   /**
    * Optional file metadata
    */
@@ -283,6 +288,11 @@ export interface VectorStoreFile {
    */
   id: string;
 
+  /**
+   * Name of the file
+   */
+  filename?: string;
+
   /**
    * Optional file metadata
    */
@@ -382,6 +392,11 @@ export namespace FileCreateParams {
      * Strategy for adding the file
      */
     parsing_strategy?: 'fast' | 'high_quality';
+
+    /**
+     * Whether to contextualize the file
+     */
+    contextualization?: boolean;
   }
 }
 

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

@@ -207,6 +207,11 @@ export interface ScoredVectorStoreChunk {
    */
   file_id: string;
 
+  /**
+   * filename
+   */
+  filename: string;
+
   /**
    * vector store id
    */

package/src/resources/vector-stores.ts

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

package/src/resources.ts

@@ -0,0 +1 @@
+export * from './resources/index';

package/src/version.ts

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

package/version.d.ts

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

package/version.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"version.d.ts","sourceRoot":"","sources":["src/version.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,OAAO,kBAAkB,CAAC"}
+{"version":3,"file":"version.d.ts","sourceRoot":"","sources":["src/version.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,OAAO,UAAU,CAAC"}

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.9'; // x-release-please-version
+exports.VERSION = '0.2.0'; // x-release-please-version
 //# sourceMappingURL=version.js.map

package/version.js.map

@@ -1 +1 @@
-{"version":3,"file":"version.js","sourceRoot":"","sources":["src/version.ts"],"names":[],"mappings":";;;AAAa,QAAA,OAAO,GAAG,eAAe,CAAC,CAAC,2BAA2B"}
+{"version":3,"file":"version.js","sourceRoot":"","sources":["src/version.ts"],"names":[],"mappings":";;;AAAa,QAAA,OAAO,GAAG,OAAO,CAAC,CAAC,2BAA2B"}

package/version.mjs

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

package/version.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"version.mjs","sourceRoot":"","sources":["src/version.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,OAAO,GAAG,eAAe,CAAC,CAAC,2BAA2B"}
+{"version":3,"file":"version.mjs","sourceRoot":"","sources":["src/version.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,OAAO,GAAG,OAAO,CAAC,CAAC,2BAA2B"}