Class MixedbreadStore

src/lib/store.ts:150–283

 * Mixedbread implementation of the Store interface
 */
export class MixedbreadStore implements Store {
  private client: Mixedbread;
  private clientFactory: () => Promise<Mixedbread>;

  constructor(client: Mixedbread, clientFactory: () => Promise<Mixedbread>) {
    this.client = client;
    this.clientFactory = clientFactory;
  }

  async refreshClient(): Promise<void> {
    this.client = await this.clientFactory();
  }

  async *listFiles(
    storeId: string,
    options?: ListFilesOptions,
  ): AsyncGenerator<StoreFile> {
    let after: string | undefined;
    do {
      const response = await this.client.stores.files.list(storeId, {
        limit: 100,
        after,
        metadata_filter: options?.pathPrefix
          ? { key: "path", operator: "starts_with", value: options.pathPrefix }
          : null,
      });

      for (const f of response.data) {
        yield {
          external_id: f.external_id ?? null,
          metadata: (f.metadata || null) as FileMetadata | null,
        };
      }

      after = response.pagination?.has_more
        ? (response.pagination?.last_cursor ?? undefined)
        : undefined;
    } while (after);
  }

  async uploadFile(
    storeId: string,
    file: File | ReadableStream,
    options: UploadFileOptions,
  ): Promise<void> {
    await (
      this.client.stores.files.upload as (
        storeIdentifier: string,
        file: Uploadable,
        body?: {
          external_id?: string | null;
          overwrite?: boolean;
          metadata?: unknown;
        },
      ) => Promise<unknown>
    )(storeId, file as Uploadable, {
      external_id: options.external_id,
      overwrite: options.overwrite ?? true,