@retrivora-ai/rag-engine 0.1.2 → 0.1.4

package/src/vectordb/adapters/UniversalVectorDBAdapter.ts

@@ -1,177 +0,0 @@
-import axios, { AxiosInstance } from 'axios';
-import { IVectorDB, VectorMatch, UpsertDocument } from '../IVectorDB';
-import { VectorDBConfig } from '../../config/RagConfig';
-import { resolvePath, buildPayload } from '../../utils/templateUtils';
-import { VECTOR_PROFILES } from '../../config/UniversalProfiles';
-
-interface UniversalVectorDBOptions {
-  baseUrl?: string;
-  headers?: Record<string, string>;
-  queryPath?: string;
-  upsertPath?: string;
-  deletePath?: string;
-  deleteNamespacePath?: string;
-  pingPath?: string;
-  queryPayloadTemplate?: string;
-  upsertPayloadTemplate?: string;
-  batchUpsertPayloadTemplate?: string;
-  responseExtractPath?: string;
-  idPath?: string;
-  scorePath?: string;
-  contentPath?: string;
-  metadataPath?: string;
-  timeout?: number;
-}
-
-export class UniversalVectorDBAdapter implements IVectorDB {
-  private readonly http: AxiosInstance;
-  private readonly indexName: string;
-  private readonly opts: UniversalVectorDBOptions;
-
-  constructor(config: VectorDBConfig) {
-    this.indexName = config.indexName;
-    
-    // Merge profile defaults if specified
-    const options = (config.options as Record<string, unknown>) ?? {};
-    let profile: Record<string, unknown> = {};
-
-    if (typeof options.profile === 'string') {
-      profile = (VECTOR_PROFILES as Record<string, Record<string, unknown>>)[options.profile] || {};
-    } else if (typeof options.profile === 'object' && options.profile !== null) {
-      profile = options.profile as Record<string, unknown>;
-    }
-
-    this.opts = { ...profile, ...(config.options ?? {}) } as UniversalVectorDBOptions;
-
-    if (!this.opts.baseUrl) {
-      throw new Error('[UniversalVectorDBAdapter] options.baseUrl is required');
-    }
-
-    this.http = axios.create({
-      baseURL: this.opts.baseUrl,
-      headers: {
-        'Content-Type': 'application/json',
-        ...(this.opts.headers || {}),
-      },
-      timeout: this.opts.timeout ?? 30_000,
-    });
-  }
-
-  async initialize(): Promise<void> {
-    if (this.opts.pingPath) {
-      const ok = await this.ping();
-      if (!ok) {
-        throw new Error(`[UniversalVectorDBAdapter] Could not reach health endpoint: ${this.opts.pingPath}`);
-      }
-    }
-  }
-
-  async upsert(doc: UpsertDocument, namespace?: string): Promise<void> {
-    const path = this.opts.upsertPath ?? '/upsert';
-    
-    if (this.opts.upsertPayloadTemplate) {
-      const payload = buildPayload(this.opts.upsertPayloadTemplate, {
-        id: doc.id,
-        vector: doc.vector,
-        content: doc.content,
-        metadata: doc.metadata ?? {},
-        namespace: namespace ?? '',
-        index: this.indexName,
-      });
-      await this.http.post(path, payload);
-    } else {
-      // Fallback to standard generic shape
-      await this.http.post(path, {
-        index: this.indexName,
-        namespace,
-        ...doc,
-      });
-    }
-  }
-
-  async batchUpsert(docs: UpsertDocument[], namespace?: string): Promise<void> {
-    const path = this.opts.upsertPath ?? '/upsert';
-    
-    // For batch upsert with templating, if the user provided batchUpsertPayloadTemplate
-    if (this.opts.batchUpsertPayloadTemplate) {
-       const payload = buildPayload(this.opts.batchUpsertPayloadTemplate, {
-          vectors: docs,
-          namespace: namespace ?? '',
-          index: this.indexName,
-       });
-       await this.http.post(path, payload);
-    } else {
-       // Fallback to standard loop or generic shape
-       await Promise.all(docs.map((doc) => this.upsert(doc, namespace)));
-    }
-  }
-
-  async query(
-    vector: number[],
-    topK: number,
-    namespace?: string,
-    filter?: Record<string, unknown>
-  ): Promise<VectorMatch[]> {
-    const path = this.opts.queryPath ?? '/query';
-
-    let payload: unknown;
-    if (this.opts.queryPayloadTemplate) {
-      payload = buildPayload(this.opts.queryPayloadTemplate, {
-        vector,
-        topK,
-        namespace: namespace ?? '',
-        index: this.indexName,
-        filter: filter ?? {},
-      });
-    } else {
-      payload = { index: this.indexName, namespace, vector, topK, filter };
-    }
-
-    const { data } = await this.http.post(path, payload);
-
-    const extractPath = this.opts.responseExtractPath ?? 'matches';
-    const matchesRaw = resolvePath(data, extractPath);
-
-    if (!Array.isArray(matchesRaw)) {
-       throw new Error(`[UniversalVectorDBAdapter] Expected an array at responseExtractPath '${extractPath}', but got ${typeof matchesRaw}`);
-    }
-
-    const idPath = this.opts.idPath ?? 'id';
-    const scorePath = this.opts.scorePath ?? 'score';
-    const contentPath = this.opts.contentPath ?? 'content';
-    const metadataPath = this.opts.metadataPath ?? 'metadata';
-
-    return matchesRaw.map((match) => ({
-      id: String(resolvePath(match, idPath) || ''),
-      score: Number(resolvePath(match, scorePath) || 0),
-      content: String(resolvePath(match, contentPath) || ''),
-      metadata: (resolvePath(match, metadataPath) as Record<string, unknown> | undefined) || {},
-    }));
-  }
-
-  async delete(id: string, namespace?: string): Promise<void> {
-    const path = this.opts.deletePath ?? '/delete';
-    await this.http.post(path, { index: this.indexName, namespace, id });
-  }
-
-  async deleteNamespace(namespace: string): Promise<void> {
-    const path = this.opts.deleteNamespacePath ?? '/delete-namespace';
-    await this.http.post(path, { index: this.indexName, namespace });
-  }
-
-  async ping(): Promise<boolean> {
-    try {
-      if (this.opts.pingPath) {
-        await this.http.get(this.opts.pingPath);
-      }
-      return true;
-    } catch (err) {
-      console.error('[UniversalVectorDBAdapter] Ping failed:', err);
-      return false;
-    }
-  }
-
-  async disconnect(): Promise<void> {
-    // Axios based adapter is stateless per request.
-  }
-}