@lov3kaizen/agentsea-embeddings 1.0.1 → 1.1.1

package/dist/stores/index.d.ts

@@ -138,6 +138,8 @@ interface DeleteOptions {
 }
 interface DeleteResult {
     deletedCount: number;
+    countExact: boolean;
+    requestedCount?: number;
     durationMs: number;
 }
 interface StoreQueryOptions extends SearchOptions {
@@ -293,6 +295,258 @@ declare class QdrantStore extends BaseStore {
 }
 declare function createQdrantStore(config: QdrantStoreConfig): QdrantStore;
 
+interface PgPoolLike {
+    query(text: string, params?: unknown[]): Promise<{
+        rows: Array<Record<string, unknown>>;
+        rowCount: number | null;
+    }>;
+    end(): Promise<void>;
+}
+interface PgVectorStoreOptions extends PgVectorStoreConfig {
+    pool?: PgPoolLike;
+}
+declare class PgVectorStore extends BaseStore {
+    readonly storeType: VectorStoreType;
+    private pool?;
+    private readonly injectedPool?;
+    private readonly table;
+    private readonly vectorColumn;
+    private readonly contentColumn;
+    private readonly metadataColumn;
+    private readonly pgConfig;
+    private initialized;
+    constructor(config: PgVectorStoreOptions);
+    private get distanceOperator();
+    init(): Promise<void>;
+    private ensureInitialized;
+    private ident;
+    private toVectorLiteral;
+    upsert(records: VectorRecord[], options?: UpsertOptions): Promise<UpsertResult>;
+    query(vector: EmbeddingVector, options?: StoreQueryOptions): Promise<StoreQueryResult>;
+    delete(ids: string[], _options?: DeleteOptions): Promise<DeleteResult>;
+    deleteAll(_options?: DeleteOptions): Promise<DeleteResult>;
+    getStats(): Promise<StoreStats>;
+    checkHealth(): Promise<StoreHealth>;
+    close(): Promise<void>;
+}
+declare function createPgVectorStore(config: PgVectorStoreOptions): PgVectorStore;
+
+interface WeaviateBackend {
+    ensureClass(className: string, dimensions?: number): Promise<void>;
+    upsert(className: string, objects: Array<{
+        id: string;
+        vector: number[];
+        properties: Record<string, unknown>;
+    }>): Promise<void>;
+    nearVector(className: string, vector: number[], limit: number, filter?: Record<string, unknown>): Promise<Array<{
+        id: string;
+        score: number;
+        properties: Record<string, unknown>;
+    }>>;
+    deleteByIds(className: string, ids: string[]): Promise<number>;
+    deleteAll(className: string): Promise<number>;
+    count(className: string): Promise<number>;
+    ping(): Promise<void>;
+}
+interface WeaviateStoreOptions extends WeaviateStoreConfig {
+    backend?: WeaviateBackend;
+}
+declare class WeaviateStore extends BaseStore {
+    readonly storeType: VectorStoreType;
+    private backend?;
+    private readonly injectedBackend?;
+    private readonly className;
+    private readonly url;
+    private readonly apiKey?;
+    private initialized;
+    constructor(config: WeaviateStoreOptions);
+    init(): Promise<void>;
+    private ensureInitialized;
+    private buildSdkBackend;
+    upsert(records: VectorRecord[], options?: UpsertOptions): Promise<UpsertResult>;
+    query(vector: EmbeddingVector, options?: StoreQueryOptions): Promise<StoreQueryResult>;
+    delete(ids: string[], _options?: DeleteOptions): Promise<DeleteResult>;
+    deleteAll(_options?: DeleteOptions): Promise<DeleteResult>;
+    getStats(): Promise<StoreStats>;
+    checkHealth(): Promise<StoreHealth>;
+    close(): Promise<void>;
+}
+interface WeaviateSdkClient {
+    schema: {
+        getter: () => {
+            do: () => Promise<{
+                classes?: Array<{
+                    class: string;
+                }>;
+            }>;
+        };
+        classCreator: () => {
+            withClass: (c: unknown) => {
+                do: () => Promise<unknown>;
+            };
+        };
+    };
+    batch: {
+        objectsBatcher: () => WeaviateBatcher;
+    };
+    data: {
+        deleter: () => {
+            withClassName: (c: string) => {
+                withId: (id: string) => {
+                    do: () => Promise<unknown>;
+                };
+            };
+        };
+    };
+    graphql: {
+        get: () => WeaviateGraphQLGet;
+        aggregate: () => WeaviateAggregate;
+    };
+    misc: {
+        liveChecker: () => {
+            do: () => Promise<unknown>;
+        };
+    };
+}
+interface WeaviateBatcher {
+    withObject: (o: unknown) => WeaviateBatcher;
+    do: () => Promise<unknown>;
+}
+interface WeaviateGraphQLGet {
+    withClassName: (c: string) => WeaviateGraphQLGet;
+    withFields: (f: string) => WeaviateGraphQLGet;
+    withNearVector: (v: unknown) => WeaviateGraphQLGet;
+    withWhere: (w: unknown) => WeaviateGraphQLGet;
+    withLimit: (n: number) => WeaviateGraphQLGet;
+    do: () => Promise<{
+        data?: Record<string, unknown>;
+    }>;
+}
+interface WeaviateAggregate {
+    withClassName: (c: string) => WeaviateAggregate;
+    withFields: (f: string) => WeaviateAggregate;
+    do: () => Promise<{
+        data?: Record<string, unknown>;
+    }>;
+}
+declare class WeaviateSdkBackend implements WeaviateBackend {
+    private readonly client;
+    constructor(client: WeaviateSdkClient);
+    ensureClass(className: string, _dimensions?: number): Promise<void>;
+    upsert(className: string, objects: Array<{
+        id: string;
+        vector: number[];
+        properties: Record<string, unknown>;
+    }>): Promise<void>;
+    nearVector(className: string, vector: number[], limit: number, filter?: Record<string, unknown>): Promise<Array<{
+        id: string;
+        score: number;
+        properties: Record<string, unknown>;
+    }>>;
+    deleteByIds(className: string, ids: string[]): Promise<number>;
+    deleteAll(className: string): Promise<number>;
+    count(className: string): Promise<number>;
+    ping(): Promise<void>;
+    private toWhere;
+}
+declare function createWeaviateStore(config: WeaviateStoreOptions): WeaviateStore;
+
+interface MilvusBackend {
+    ensureCollection(collection: string, dimensions: number, metric: string): Promise<void>;
+    upsert(collection: string, rows: Array<{
+        id: string;
+        vector: number[];
+        text: string;
+        metadata: Record<string, unknown>;
+    }>): Promise<void>;
+    search(collection: string, vector: number[], limit: number, filter?: string): Promise<Array<{
+        id: string;
+        score: number;
+        text: string;
+        metadata: Record<string, unknown>;
+    }>>;
+    deleteByIds(collection: string, ids: string[]): Promise<number>;
+    deleteAll(collection: string): Promise<number>;
+    count(collection: string): Promise<number>;
+    ping(): Promise<void>;
+}
+interface MilvusStoreOptions extends MilvusStoreConfig {
+    backend?: MilvusBackend;
+}
+declare class MilvusStore extends BaseStore {
+    readonly storeType: VectorStoreType;
+    private backend?;
+    private readonly injectedBackend?;
+    private readonly collection;
+    private readonly milvusConfig;
+    private initialized;
+    constructor(config: MilvusStoreOptions);
+    private get metricType();
+    init(): Promise<void>;
+    private ensureInitialized;
+    private buildSdkBackend;
+    upsert(records: VectorRecord[], options?: UpsertOptions): Promise<UpsertResult>;
+    query(vector: EmbeddingVector, options?: StoreQueryOptions): Promise<StoreQueryResult>;
+    delete(ids: string[], _options?: DeleteOptions): Promise<DeleteResult>;
+    deleteAll(_options?: DeleteOptions): Promise<DeleteResult>;
+    getStats(): Promise<StoreStats>;
+    checkHealth(): Promise<StoreHealth>;
+    close(): Promise<void>;
+    private toExpr;
+}
+interface MilvusSdkClient {
+    hasCollection(p: {
+        collection_name: string;
+    }): Promise<{
+        value: boolean;
+    }>;
+    createCollection(p: unknown): Promise<unknown>;
+    createIndex(p: unknown): Promise<unknown>;
+    loadCollectionSync(p: {
+        collection_name: string;
+    }): Promise<unknown>;
+    insert(p: {
+        collection_name: string;
+        data: unknown[];
+    }): Promise<unknown>;
+    search(p: unknown): Promise<{
+        results: Array<Record<string, unknown>>;
+    }>;
+    deleteEntities(p: {
+        collection_name: string;
+        expr: string;
+    }): Promise<unknown>;
+    getCollectionStatistics(p: {
+        collection_name: string;
+    }): Promise<{
+        data?: {
+            row_count?: string | number;
+        };
+    }>;
+}
+declare class MilvusSdkBackend implements MilvusBackend {
+    private readonly client;
+    constructor(client: MilvusSdkClient);
+    ensureCollection(collection: string, dimensions: number, metric: string): Promise<void>;
+    upsert(collection: string, rows: Array<{
+        id: string;
+        vector: number[];
+        text: string;
+        metadata: Record<string, unknown>;
+    }>): Promise<void>;
+    search(collection: string, vector: number[], limit: number, filter?: string): Promise<Array<{
+        id: string;
+        score: number;
+        text: string;
+        metadata: Record<string, unknown>;
+    }>>;
+    deleteByIds(collection: string, ids: string[]): Promise<number>;
+    deleteAll(collection: string): Promise<number>;
+    count(collection: string): Promise<number>;
+    ping(): Promise<void>;
+}
+declare function createMilvusStore(config: MilvusStoreOptions): MilvusStore;
+
 declare function createStore(type: VectorStoreType, config: StoreConfig): BaseStore;
 
-export { BaseStore, ChromaStore, type ChromaStoreConfig, type CollectionInfo, type DeleteOptions, type DeleteResult, type DistanceMetric, type IndexInfo, MemoryStore, type MemoryStoreConfig, type MilvusStoreConfig, type PgVectorStoreConfig, PineconeStore, type PineconeStoreConfig, QdrantStore, type QdrantStoreConfig, type StoreConfig, type StoreFactoryOptions, type StoreHealth, type StoreQueryOptions, type StoreQueryResult, type StoreStats, type StoredVector, type UpsertOptions, type UpsertResult, type VectorRecord, type VectorStoreType, type WeaviateStoreConfig, createChromaStore, createMemoryStore, createPineconeStore, createQdrantStore, createStore };
+export { BaseStore, ChromaStore, type ChromaStoreConfig, type CollectionInfo, type DeleteOptions, type DeleteResult, type DistanceMetric, type IndexInfo, MemoryStore, type MemoryStoreConfig, type MilvusBackend, MilvusSdkBackend, MilvusStore, type MilvusStoreConfig, type MilvusStoreOptions, type PgPoolLike, PgVectorStore, type PgVectorStoreConfig, type PgVectorStoreOptions, PineconeStore, type PineconeStoreConfig, QdrantStore, type QdrantStoreConfig, type StoreConfig, type StoreFactoryOptions, type StoreHealth, type StoreQueryOptions, type StoreQueryResult, type StoreStats, type StoredVector, type UpsertOptions, type UpsertResult, type VectorRecord, type VectorStoreType, type WeaviateBackend, WeaviateSdkBackend, WeaviateStore, type WeaviateStoreConfig, type WeaviateStoreOptions, createChromaStore, createMemoryStore, createMilvusStore, createPgVectorStore, createPineconeStore, createQdrantStore, createStore, createWeaviateStore };