Class FoamEmbeddings

packages/foam-vscode/src/ai/model/embeddings.ts:36–382

 * Manages embeddings for all resources in the workspace
 */
export class FoamEmbeddings implements IDisposable {
  /**
   * Maps resource URIs to their embeddings
   */
  private embeddings: Map<string, Embedding> = new Map();

  private onDidUpdateEmitter = new Emitter<void>();
  onDidUpdate = this.onDidUpdateEmitter.event;

  /**
   * List of disposables to destroy with the embeddings
   */
  private disposables: IDisposable[] = [];

  constructor(
    private readonly workspace: FoamWorkspace,
    private readonly provider: EmbeddingProvider,
    private readonly cache?: EmbeddingCache
  ) {}

  /**
   * Get the embedding for a resource
   * @param uri The URI of the resource
   * @returns The embedding vector, or null if not found
   */
  public getEmbedding(uri: URI): number[] | null {
    const embedding = this.embeddings.get(uri.path);
    return embedding ? embedding.vector : null;
  }

  /**
   * Check if embeddings are available
   * @returns true if at least one embedding exists
   */
  public hasEmbeddings(): boolean {
    return this.embeddings.size > 0;
  }

  /**
   * Get the number of embeddings
   * @returns The count of embeddings
   */
  public size(): number {
    return this.embeddings.size;
  }

  /**
   * Find similar resources to a given resource
   * @param uri The URI of the target resource
   * @param topK The number of similar resources to return
   * @returns Array of similar resources sorted by similarity (highest first)
   */
  public getSimilar(uri: URI, topK: number = 10): SimilarResource[] {
    const targetEmbedding = this.getEmbedding(uri);
    if (!targetEmbedding) {
      return [];
    }