chromadb 1.7.2 → 1.7.3-beta2

package/src/generated/configuration.ts

@@ -2,10 +2,10 @@
 // tslint:disable
 /**
  * FastAPI
- * 
+ *
  *
  * OpenAPI spec version: 0.1.0
- * 
+ *
  *
  * NOTE: This class is auto generated by OpenAPI Generator+.
  * https://github.com/karlvr/openapi-generator-plus
@@ -29,14 +29,14 @@ export class Configuration {
 	apiKey?: string | ((name: string) => string | null);
 	/**
 	 * parameter for basic security
-	 * 
+	 *
 	 * @type {string}
 	 * @memberof Configuration
 	 */
 	username?: string;
 	/**
 	 * parameter for basic security
-	 * 
+	 *
 	 * @type {string}
 	 * @memberof Configuration
 	 */
@@ -50,7 +50,7 @@ export class Configuration {
 	authorization?: string | ((name: string, scopes?: string[]) => string | null);
 	/**
 	 * override base path
-	 * 
+	 *
 	 * @type {string}
 	 * @memberof Configuration
 	 */

package/src/generated/index.ts

@@ -2,10 +2,10 @@
 // tslint:disable
 /**
  * FastAPI
- * 
+ *
  *
  * OpenAPI spec version: 0.1.0
- * 
+ *
  *
  * NOTE: This class is auto generated by OpenAPI Generator+.
  * https://github.com/karlvr/openapi-generator-plus

package/src/generated/models.ts

@@ -20,7 +20,6 @@ export namespace Api {
 		embeddings?: Api.AddEmbedding.Embedding[];
 		metadatas?: Api.AddEmbedding.Metadata[];
 		documents?: string[];
-		uris?: string[];
 		ids: string[];
 	}
 
@@ -31,10 +30,10 @@ export namespace Api {
 	export namespace AddEmbedding {
 		export interface Embedding {
 		}
-	
+
 		export interface Metadata {
 		}
-	
+
 	}
 
 	export interface ADelete200Response {
@@ -46,15 +45,12 @@ export namespace Api {
 	export interface Count200Response {
 	}
 
-	export interface CountCollections200Response {
-	}
-
 	export interface CreateCollection {
 		name: string;
 		metadata?: Api.CreateCollection.Metadata;
 		'get_or_create'?: boolean;
 	}
-	
+
 	/**
 	 * @export
 	 * @namespace CreateCollection
@@ -62,24 +58,10 @@ export namespace Api {
 	export namespace CreateCollection {
 		export interface Metadata {
 		}
-	
-	}
-
-	export interface CreateCollection200Response {
-	}
 
-	export interface CreateDatabase {
-		name: string;
 	}
 
-	export interface CreateDatabase200Response {
-	}
-
-	export interface CreateTenant {
-		name: string;
-	}
-
-	export interface CreateTenant200Response {
+	export interface CreateCollection200Response {
 	}
 
 	export interface DeleteCollection200Response {
@@ -90,7 +72,7 @@ export namespace Api {
 		where?: Api.DeleteEmbedding.Where;
 		'where_document'?: Api.DeleteEmbedding.WhereDocument;
 	}
-	
+
 	/**
 	 * @export
 	 * @namespace DeleteEmbedding
@@ -98,16 +80,13 @@ export namespace Api {
 	export namespace DeleteEmbedding {
 		export interface Where {
 		}
-	
+
 		export interface WhereDocument {
 		}
-	
-	}
 
-	export interface GetCollection200Response {
 	}
 
-	export interface GetDatabase200Response {
+	export interface GetCollection200Response {
 	}
 
 	export interface GetEmbedding {
@@ -125,9 +104,9 @@ export namespace Api {
 		 * @memberof GetEmbedding
 		 */
 		offset?: number;
-		include?: (Api.GetEmbedding.Include.EnumValueEnum | Api.GetEmbedding.Include.EnumValueEnum2 | Api.GetEmbedding.Include.EnumValueEnum3 | Api.GetEmbedding.Include.EnumValueEnum4 | Api.GetEmbedding.Include.EnumValueEnum5 | Api.GetEmbedding.Include.EnumValueEnum6)[];
+		include?: (Api.GetEmbedding.Include.EnumValueEnum | Api.GetEmbedding.Include.EnumValueEnum2 | Api.GetEmbedding.Include.EnumValueEnum3 | Api.GetEmbedding.Include.EnumValueEnum4)[];
 	}
-	
+
 	/**
 	 * @export
 	 * @namespace GetEmbedding
@@ -135,12 +114,12 @@ export namespace Api {
 	export namespace GetEmbedding {
 		export interface Where {
 		}
-	
+
 		export interface WhereDocument {
 		}
-	
-		export type Include = Api.GetEmbedding.Include.EnumValueEnum | Api.GetEmbedding.Include.EnumValueEnum2 | Api.GetEmbedding.Include.EnumValueEnum3 | Api.GetEmbedding.Include.EnumValueEnum4 | Api.GetEmbedding.Include.EnumValueEnum5 | Api.GetEmbedding.Include.EnumValueEnum6;
-		
+
+		export type Include = Api.GetEmbedding.Include.EnumValueEnum | Api.GetEmbedding.Include.EnumValueEnum2 | Api.GetEmbedding.Include.EnumValueEnum3 | Api.GetEmbedding.Include.EnumValueEnum4;
+
 		/**
 		 * @export
 		 * @namespace Include
@@ -149,35 +128,24 @@ export namespace Api {
 			export enum EnumValueEnum {
 				Documents = 'documents'
 			}
-		
+
 			export enum EnumValueEnum2 {
 				Embeddings = 'embeddings'
 			}
-		
+
 			export enum EnumValueEnum3 {
 				Metadatas = 'metadatas'
 			}
-		
+
 			export enum EnumValueEnum4 {
 				Distances = 'distances'
 			}
-		
-			export enum EnumValueEnum5 {
-				Uris = 'uris'
-			}
-		
-			export enum EnumValueEnum6 {
-				Data = 'data'
-			}
-		
+
 		}
-	
-	}
 
-	export interface GetNearestNeighbors200Response {
 	}
 
-	export interface GetTenant200Response {
+	export interface GetNearestNeighbors200Response {
 	}
 
 	export interface HTTPValidationError {
@@ -187,9 +155,6 @@ export namespace Api {
 	export interface ListCollections200Response {
 	}
 
-	export interface PreFlightChecks200Response {
-	}
-
 	export interface QueryEmbedding {
 		where?: Api.QueryEmbedding.Where;
 		'where_document'?: Api.QueryEmbedding.WhereDocument;
@@ -199,9 +164,9 @@ export namespace Api {
 		 * @memberof QueryEmbedding
 		 */
 		'n_results'?: number;
-		include?: (Api.QueryEmbedding.Include.EnumValueEnum | Api.QueryEmbedding.Include.EnumValueEnum2 | Api.QueryEmbedding.Include.EnumValueEnum3 | Api.QueryEmbedding.Include.EnumValueEnum4 | Api.QueryEmbedding.Include.EnumValueEnum5 | Api.QueryEmbedding.Include.EnumValueEnum6)[];
+		include?: (Api.QueryEmbedding.Include.EnumValueEnum | Api.QueryEmbedding.Include.EnumValueEnum2 | Api.QueryEmbedding.Include.EnumValueEnum3 | Api.QueryEmbedding.Include.EnumValueEnum4)[];
 	}
-	
+
 	/**
 	 * @export
 	 * @namespace QueryEmbedding
@@ -209,15 +174,15 @@ export namespace Api {
 	export namespace QueryEmbedding {
 		export interface Where {
 		}
-	
+
 		export interface WhereDocument {
 		}
-	
+
 		export interface QueryEmbedding2 {
 		}
-	
-		export type Include = Api.QueryEmbedding.Include.EnumValueEnum | Api.QueryEmbedding.Include.EnumValueEnum2 | Api.QueryEmbedding.Include.EnumValueEnum3 | Api.QueryEmbedding.Include.EnumValueEnum4 | Api.QueryEmbedding.Include.EnumValueEnum5 | Api.QueryEmbedding.Include.EnumValueEnum6;
-		
+
+		export type Include = Api.QueryEmbedding.Include.EnumValueEnum | Api.QueryEmbedding.Include.EnumValueEnum2 | Api.QueryEmbedding.Include.EnumValueEnum3 | Api.QueryEmbedding.Include.EnumValueEnum4;
+
 		/**
 		 * @export
 		 * @namespace Include
@@ -226,29 +191,21 @@ export namespace Api {
 			export enum EnumValueEnum {
 				Documents = 'documents'
 			}
-		
+
 			export enum EnumValueEnum2 {
 				Embeddings = 'embeddings'
 			}
-		
+
 			export enum EnumValueEnum3 {
 				Metadatas = 'metadatas'
 			}
-		
+
 			export enum EnumValueEnum4 {
 				Distances = 'distances'
 			}
-		
-			export enum EnumValueEnum5 {
-				Uris = 'uris'
-			}
-		
-			export enum EnumValueEnum6 {
-				Data = 'data'
-			}
-		
+
 		}
-	
+
 	}
 
 	export interface Update200Response {
@@ -258,7 +215,7 @@ export namespace Api {
 		'new_name'?: string;
 		'new_metadata'?: Api.UpdateCollection.NewMetadata;
 	}
-	
+
 	/**
 	 * @export
 	 * @namespace UpdateCollection
@@ -266,7 +223,7 @@ export namespace Api {
 	export namespace UpdateCollection {
 		export interface NewMetadata {
 		}
-	
+
 	}
 
 	export interface UpdateCollection200Response {
@@ -276,10 +233,9 @@ export namespace Api {
 		embeddings?: Api.UpdateEmbedding.Embedding[];
 		metadatas?: Api.UpdateEmbedding.Metadata[];
 		documents?: string[];
-		uris?: string[];
 		ids: string[];
 	}
-	
+
 	/**
 	 * @export
 	 * @namespace UpdateEmbedding
@@ -287,10 +243,10 @@ export namespace Api {
 	export namespace UpdateEmbedding {
 		export interface Embedding {
 		}
-	
+
 		export interface Metadata {
 		}
-	
+
 	}
 
 	export interface Upsert200Response {

package/src/generated/runtime.ts

@@ -3,10 +3,10 @@ import 'isomorphic-fetch';
 // tslint:disable
 /**
  * FastAPI
- * 
+ *
  *
  * OpenAPI spec version: 0.1.0
- * 
+ *
  *
  * NOTE: This class is auto generated by OpenAPI Generator+.
  * https://github.com/karlvr/openapi-generator-plus
@@ -37,7 +37,7 @@ export const COLLECTION_FORMATS = {
 export type FetchAPI = typeof defaultFetch;
 
 /**
- *  
+ *
  * @export
  * @interface FetchArgs
  */
@@ -47,7 +47,7 @@ export interface FetchArgs {
 }
 
 /**
- * 
+ *
  * @export
  * @class BaseAPI
  */
@@ -63,7 +63,7 @@ export class BaseAPI {
 };
 
 /**
- * 
+ *
  * @export
  * @class RequiredError
  * @extends {Error}

package/src/index.ts

@@ -1,41 +1,8 @@
 export { ChromaClient } from './ChromaClient';
-export { AdminClient } from './AdminClient';
-export { CloudClient } from './CloudClient';
 export { Collection } from './Collection';
 export { IEmbeddingFunction } from './embeddings/IEmbeddingFunction';
 export { OpenAIEmbeddingFunction } from './embeddings/OpenAIEmbeddingFunction';
 export { CohereEmbeddingFunction } from './embeddings/CohereEmbeddingFunction';
-export { GoogleGenerativeAiEmbeddingFunction } from './embeddings/GoogleGeminiEmbeddingFunction';
-export {
-    IncludeEnum,
-    GetParams,
-    CollectionType,
-    CollectionMetadata,
-    Embedding,
-    Embeddings,
-    Metadata,
-    Metadatas,
-    Document,
-    Documents,
-    ID,
-    IDs,
-    Where,
-    WhereDocument,
-    GetResponse,
-    QueryResponse,
-    ListCollectionsParams,
-    ChromaClientParams,
-    CreateCollectionParams,
-    GetOrCreateCollectionParams,
-    GetCollectionParams,
-    DeleteCollectionParams,
-    AddParams,
-    UpsertParams,
-    UpdateParams,
-    ModifyCollectionParams,
-    QueryParams,
-    PeekParams,
-    DeleteParams
-} from './types';
-export { HuggingFaceEmbeddingServerFunction } from './embeddings/HuggingFaceEmbeddingServerFunction';
-export { JinaEmbeddingFunction } from './embeddings/JinaEmbeddingFunction';
+export { TransformersEmbeddingFunction } from './embeddings/TransformersEmbeddingFunction';
+export { DefaultEmbeddingFunction } from './embeddings/DefaultEmbeddingFunction';
+export { IncludeEnum } from './types';

package/src/types.ts

@@ -1,6 +1,3 @@
-import { AuthOptions } from "./auth";
-import { IEmbeddingFunction } from "./embeddings/IEmbeddingFunction";
-
 export enum IncludeEnum {
   Documents = 'documents',
   Embeddings = 'embeddings',
@@ -83,75 +80,3 @@ export type CollectionMetadata = Record<string, unknown>;
 export type ConfigOptions = {
   options?: RequestInit;
 };
-
-export type GetParams = {
-  ids?: ID | IDs,
-  where?: Where,
-  limit?: PositiveInteger,
-  offset?: PositiveInteger,
-  include?: IncludeEnum[],
-  whereDocument?: WhereDocument
-}
-
-export type ListCollectionsParams = {
-  limit?: PositiveInteger,
-  offset?: PositiveInteger,
-}
-
-export type ChromaClientParams = {
-  path?: string,
-  fetchOptions?: RequestInit,
-  auth?: AuthOptions,
-  tenant?: string,
-  database?: string,
-}
-
-export type CreateCollectionParams = {
-  name: string,
-  metadata?: CollectionMetadata,
-  embeddingFunction?: IEmbeddingFunction
-}
-
-export type GetOrCreateCollectionParams = CreateCollectionParams
-
-export type GetCollectionParams = {
-  name: string;
-  embeddingFunction?: IEmbeddingFunction
-}
-
-export type DeleteCollectionParams = {
-  name: string
-}
-
-export type AddParams = {
-  ids: ID | IDs,
-  embeddings?: Embedding | Embeddings,
-  metadatas?: Metadata | Metadatas,
-  documents?: Document | Documents,
-}
-
-export type UpsertParams = AddParams;
-export type UpdateParams = AddParams;
-
-export type ModifyCollectionParams = {
-  name?: string,
-  metadata?: CollectionMetadata
-}
-
-export type QueryParams = {
-    queryEmbeddings?: Embedding | Embeddings,
-    nResults?: PositiveInteger,
-    where?: Where,
-    queryTexts?: string | string[],
-    whereDocument?: WhereDocument, // {"$contains":"search_string"}
-    include?: IncludeEnum[] // ["metadata", "document"]
-}
-
-export type PeekParams = { limit?: PositiveInteger }
-
-export type DeleteParams = {
-    ids?: ID | IDs,
-    where?: Where,
-    whereDocument?: WhereDocument
-}
-

package/src/utils.ts

@@ -1,6 +1,5 @@
 import { Api } from "./generated";
 import Count200Response = Api.Count200Response;
-import { AdminClient } from "./AdminClient";
 
 // a function to convert a non-Array object to an Array
 export function toArray<T>(obj: T | Array<T>): Array<T> {
@@ -39,12 +38,14 @@ export function repack(value: unknown): any {
 export async function handleError(error: unknown) {
   if (error instanceof Response) {
     try {
-      const res = await (error as Response).json();
+      // @ts-ignore
+      const res = await error.json();
       if ("error" in res) {
         return { error: res.error };
       }
     } catch (e: unknown) {
       return {
+        //@ts-ignore
         error:
           e && typeof e === "object" && "message" in e
             ? e.message
@@ -82,18 +83,3 @@ export async function handleSuccess(
 export async function importOptionalModule(moduleName: string) {
   return Function(`return import("${moduleName}")`)();
 }
-
-
-export async function validateTenantDatabase(adminClient: AdminClient, tenant: string, database: string): Promise<void> {
-    try {
-        await adminClient.getTenant({name: tenant});
-    } catch (error) {
-        throw new Error(`Error: ${error}, Could not connect to tenant ${tenant}. Are you sure it exists?`);
-    }
-
-    try {
-        await adminClient.getDatabase({name: database, tenantName: tenant});
-    } catch (error) {
-        throw new Error(`Error: ${error}, Could not connect to database ${database} for tenant ${tenant}. Are you sure it exists?`);
-    }
-}