agent-v1

// tools/docFileRegistry.ts
import { type FileHandle } from "@lmstudio/sdk";

const registry = new Map<string, Map<string, FileHandle>>(); // workingDir -> fileName -> handle

export function registerDocFile(workingDir: string, file: FileHandle) {
    if (!registry.has(workingDir)) {
        registry.set(workingDir, new Map());
    }
    registry.get(workingDir)!.set(file.name, file);
}

export function getRegisteredDocFiles(workingDir: string): FileHandle[] {
    return Array.from(registry.get(workingDir)?.values() ?? []);
}

export function lookupDocFile(workingDir: string, fileName: string): FileHandle | undefined {
    return registry.get(workingDir)?.get(fileName);
}