@mondaydotcomorg/atp-runtime/dist/embedding/vector-store.d.ts

Runtime SDK injected into sandbox for Agent Tool Protocol

/**
 */
import type { EmbeddingRecord, SearchOptions, SearchResult } from './types.js';
/**
 * In-memory vector store for the current execution
 * Cleared after each execution completes
 */
export declare class VectorStore {
    private records;
    private queryEmbedding;
    /**
     * Store embeddings from client response
     */
    store(id: string, text: string, embedding: number[], metadata?: Record<string, unknown>): void;
    /**
     * Store multiple embeddings
     */
    storeBatch(records: EmbeddingRecord[]): void;
    /**
     * Set the query embedding for search
     */
    setQueryEmbedding(embedding: number[]): void;
    /**
     * Search stored embeddings by similarity to query
     */
    search(options: SearchOptions): SearchResult[];
    /**
     * Get all stored embeddings
     */
    getAll(): EmbeddingRecord[];
    /**
     * Get embedding by ID
     */
    get(id: string): EmbeddingRecord | undefined;
    /**
     * Clear all stored embeddings
     */
    clear(): void;
    /**
     * Get count of stored embeddings
     */
    count(): number;
}
/**
 * Set the current execution ID for vector store operations
 */
export declare function setVectorStoreExecutionId(executionId: string): void;
/**
 * Clear the current execution ID
 */
export declare function clearVectorStoreExecutionId(): void;
/**
 * Initialize a new vector store for a new execution
 */
export declare function initializeVectorStore(executionId?: string): void;
/**
 * Clear the vector store after execution completes
 */
export declare function clearVectorStore(executionId?: string): void;
/**
 * Get the current vector store (for executor to manage)
 */
export declare function getVectorStore(executionId?: string): VectorStore;
//# sourceMappingURL=vector-store.d.ts.map