Class VectorMock

src/vector-mock.ts:16–230

import { flattenHeaders, readBody } from "./helpers.js";

export class VectorMock implements Mountable {
  private collections: Map<string, VectorCollection> = new Map();
  private queryHandlers: Map<string, QueryHandler> = new Map();
  private server: http.Server | null = null;
  private journal: Journal | null = null;
  private registry: MetricsRegistry | null = null;
  private options: VectorMockOptions;
  private requestHandler: ReturnType<typeof createVectorRequestHandler>;

  constructor(options?: VectorMockOptions) {
    this.options = options ?? {};
    this.requestHandler = this.buildHandler();
  }

  // ---- Configuration ----

  addCollection(name: string, opts: { dimension: number }): this {
    const collection: VectorCollection = {
      name,
      dimension: opts.dimension,
      vectors: new Map(),
    };
    this.collections.set(name, collection);
    this.requestHandler = this.buildHandler();
    return this;
  }

  upsert(collection: string, vectors: VectorEntry[]): this {
    let col = this.collections.get(collection);
    if (!col) {
      const dim = vectors.length > 0 ? vectors[0].values.length : 0;
      col = { name: collection, dimension: dim, vectors: new Map() };
      this.collections.set(collection, col);
    }
    for (const v of vectors) {
      col.vectors.set(v.id, v);
    }
    this.requestHandler = this.buildHandler();
    return this;
  }

  onQuery(
    collection: string,
    results: QueryResult[] | ((query: VectorQuery) => QueryResult[]),
  ): this {
    this.queryHandlers.set(collection, results);
    this.requestHandler = this.buildHandler();
    return this;
  }

  deleteCollection(name: string): this {
    this.collections.delete(name);
    this.queryHandlers.delete(name);
    this.requestHandler = this.buildHandler();
    return this;
  }

  // ---- Mountable interface ----