@livekit/agents-plugin-livekit 0.1.2 → 1.0.0-next.0

package/dist/hf_utils.test.js

@@ -0,0 +1,307 @@
+import { initializeLogger } from "@livekit/agents";
+import { existsSync, rmSync, statSync } from "fs";
+import { homedir } from "os";
+import { join } from "path";
+import { afterAll, beforeAll, describe, expect, it } from "vitest";
+import { downloadFileToCacheDir } from "./hf_utils.js";
+function getCachePath(repo, cacheDir) {
+  const baseCacheDir = cacheDir || join(homedir(), ".cache", "huggingface", "hub");
+  return join(baseCacheDir, `models--${repo.replace(/\//g, "--")}`);
+}
+function clearCache(repo, cacheDir) {
+  const repoPath = getCachePath(repo, cacheDir);
+  if (existsSync(repoPath)) {
+    rmSync(repoPath, { recursive: true, force: true });
+  }
+}
+describe("HuggingFace Download Fixed Implementation", () => {
+  initializeLogger({ pretty: true, level: "debug" });
+  const TEST_REPO = "livekit/turn-detector";
+  const TEST_CACHE_DIR = join(process.cwd(), ".test-cache");
+  beforeAll(() => {
+    clearCache(TEST_REPO, TEST_CACHE_DIR);
+  });
+  afterAll(() => {
+    if (existsSync(TEST_CACHE_DIR)) {
+      rmSync(TEST_CACHE_DIR, { recursive: true, force: true });
+    }
+  });
+  describe("Basic Downloads", () => {
+    it("should download a standard file in subdirectory", async () => {
+      const result = await downloadFileToCacheDir({
+        repo: TEST_REPO,
+        path: "onnx/model_q8.onnx",
+        revision: "v1.2.2-en",
+        cacheDir: TEST_CACHE_DIR
+      });
+      expect(result).toBeTruthy();
+      expect(existsSync(result)).toBe(true);
+      const stats = statSync(result);
+      const sizeMB = stats.size / 1024 / 1024;
+      expect(sizeMB).toBeCloseTo(62.67, 1);
+    });
+    it("should download a large file with retry logic", async () => {
+      const result = await downloadFileToCacheDir({
+        repo: TEST_REPO,
+        path: "onnx/model.onnx",
+        revision: "v1.2.2-en",
+        cacheDir: TEST_CACHE_DIR
+      });
+      expect(result).toBeTruthy();
+      expect(existsSync(result)).toBe(true);
+      const stats = statSync(result);
+      const sizeMB = stats.size / 1024 / 1024;
+      expect(sizeMB).toBeCloseTo(249.96, 1);
+    });
+    it("should download a very small file", async () => {
+      const result = await downloadFileToCacheDir({
+        repo: TEST_REPO,
+        path: "languages.json",
+        revision: "v1.2.2-en",
+        cacheDir: TEST_CACHE_DIR
+      });
+      expect(result).toBeTruthy();
+      expect(existsSync(result)).toBe(true);
+      const stats = statSync(result);
+      expect(stats.size).toBeLessThan(200);
+    });
+    it("should download from different revision", async () => {
+      const result = await downloadFileToCacheDir({
+        repo: TEST_REPO,
+        path: "tokenizer.json",
+        revision: "v0.2.0-intl",
+        cacheDir: TEST_CACHE_DIR
+      });
+      expect(result).toBeTruthy();
+      expect(existsSync(result)).toBe(true);
+      const stats = statSync(result);
+      const sizeMB = stats.size / 1024 / 1024;
+      expect(sizeMB).toBeGreaterThan(3);
+    });
+  });
+  describe("Cache Behavior", () => {
+    it("should use cache on second download", async () => {
+      const firstPath = await downloadFileToCacheDir({
+        repo: TEST_REPO,
+        path: "onnx/model_q8.onnx",
+        revision: "v1.2.2-en",
+        cacheDir: TEST_CACHE_DIR
+      });
+      const startTime = Date.now();
+      const secondPath = await downloadFileToCacheDir({
+        repo: TEST_REPO,
+        path: "onnx/model_q8.onnx",
+        revision: "v1.2.2-en",
+        cacheDir: TEST_CACHE_DIR
+      });
+      const cacheTime = Date.now() - startTime;
+      expect(secondPath).toBe(firstPath);
+      expect(cacheTime).toBeLessThan(500);
+    });
+    it("should respect localFileOnly flag when file is cached", async () => {
+      await downloadFileToCacheDir({
+        repo: TEST_REPO,
+        path: "tokenizer.json",
+        revision: "v0.2.0-intl",
+        cacheDir: TEST_CACHE_DIR
+      });
+      const cachedPath = await downloadFileToCacheDir({
+        repo: TEST_REPO,
+        path: "tokenizer.json",
+        revision: "v0.2.0-intl",
+        cacheDir: TEST_CACHE_DIR,
+        localFileOnly: true
+      });
+      expect(cachedPath).toBeTruthy();
+      expect(existsSync(cachedPath)).toBe(true);
+    });
+    it("should throw error with localFileOnly when file is not cached", async () => {
+      await expect(
+        downloadFileToCacheDir({
+          repo: TEST_REPO,
+          path: "non-existent-file.txt",
+          revision: "main",
+          cacheDir: TEST_CACHE_DIR,
+          localFileOnly: true
+        })
+      ).rejects.toThrow(/File not found in cache/);
+    });
+    it("should save revision-to-commit mappings", async () => {
+      await downloadFileToCacheDir({
+        repo: TEST_REPO,
+        path: "languages.json",
+        revision: "v1.2.2-en",
+        cacheDir: TEST_CACHE_DIR
+      });
+      const refsPath = join(getCachePath(TEST_REPO, TEST_CACHE_DIR), "refs", "v1.2.2-en");
+      expect(existsSync(refsPath)).toBe(true);
+    });
+    it("should handle multiple files from same revision without overwriting refs", async () => {
+      const file1Path = await downloadFileToCacheDir({
+        repo: TEST_REPO,
+        path: "onnx/model_q8.onnx",
+        revision: "v1.2.2-en",
+        cacheDir: TEST_CACHE_DIR
+      });
+      const file2Path = await downloadFileToCacheDir({
+        repo: TEST_REPO,
+        path: "languages.json",
+        revision: "v1.2.2-en",
+        cacheDir: TEST_CACHE_DIR
+      });
+      expect(existsSync(file1Path)).toBe(true);
+      expect(existsSync(file2Path)).toBe(true);
+      const cachedFile1 = await downloadFileToCacheDir({
+        repo: TEST_REPO,
+        path: "onnx/model_q8.onnx",
+        revision: "v1.2.2-en",
+        cacheDir: TEST_CACHE_DIR,
+        localFileOnly: true
+      });
+      const cachedFile2 = await downloadFileToCacheDir({
+        repo: TEST_REPO,
+        path: "languages.json",
+        revision: "v1.2.2-en",
+        cacheDir: TEST_CACHE_DIR,
+        localFileOnly: true
+      });
+      expect(cachedFile1).toBe(file1Path);
+      expect(cachedFile2).toBe(file2Path);
+      const match1 = file1Path.match(/snapshots\/([a-f0-9]{40})\//);
+      const match2 = file2Path.match(/snapshots\/([a-f0-9]{40})\//);
+      expect(match1).toBeTruthy();
+      expect(match2).toBeTruthy();
+      const commitHash1 = match1[1];
+      const commitHash2 = match2[1];
+      expect(commitHash1).toBe(commitHash2);
+      const { readFileSync } = await import("fs");
+      const refsPath = join(getCachePath(TEST_REPO, TEST_CACHE_DIR), "refs", "v1.2.2-en");
+      const refsContent = readFileSync(refsPath, "utf-8").trim();
+      expect(refsContent).toMatch(/^[a-f0-9]{40}$/);
+      expect(refsContent).toBe(commitHash1);
+    });
+  });
+  describe("Error Handling", () => {
+    it("should handle invalid repository gracefully", async () => {
+      await expect(
+        downloadFileToCacheDir({
+          repo: "non-existent-org/non-existent-repo",
+          path: "file.txt",
+          revision: "main",
+          cacheDir: TEST_CACHE_DIR
+        })
+      ).rejects.toThrow();
+    });
+    it("should handle invalid file path gracefully", async () => {
+      await expect(
+        downloadFileToCacheDir({
+          repo: TEST_REPO,
+          path: "non-existent-file-path.xyz",
+          revision: "v1.2.2-en",
+          cacheDir: TEST_CACHE_DIR
+        })
+      ).rejects.toThrow();
+    });
+    it("should handle invalid revision gracefully", async () => {
+      await expect(
+        downloadFileToCacheDir({
+          repo: TEST_REPO,
+          path: "tokenizer.json",
+          revision: "non-existent-revision",
+          cacheDir: TEST_CACHE_DIR
+        })
+      ).rejects.toThrow();
+    });
+  });
+  describe("Cache Structure", () => {
+    it("should create proper cache directory structure", async () => {
+      await downloadFileToCacheDir({
+        repo: TEST_REPO,
+        path: "onnx/model_q8.onnx",
+        revision: "v1.2.2-en",
+        cacheDir: TEST_CACHE_DIR
+      });
+      const cachePath = getCachePath(TEST_REPO, TEST_CACHE_DIR);
+      expect(existsSync(join(cachePath, "blobs"))).toBe(true);
+      expect(existsSync(join(cachePath, "snapshots"))).toBe(true);
+      expect(existsSync(join(cachePath, "refs"))).toBe(true);
+    });
+    it("should handle commit hash revisions", async () => {
+      const tagResult = await downloadFileToCacheDir({
+        repo: TEST_REPO,
+        path: "onnx/model_q8.onnx",
+        revision: "v1.2.2-en",
+        cacheDir: TEST_CACHE_DIR
+      });
+      const match = tagResult.match(/snapshots\/([a-f0-9]{40})\//);
+      expect(match).toBeTruthy();
+      const commitHash = match[1];
+      const result = await downloadFileToCacheDir({
+        repo: TEST_REPO,
+        path: "onnx/model_q8.onnx",
+        revision: commitHash,
+        cacheDir: TEST_CACHE_DIR
+      });
+      expect(result).toBeTruthy();
+      expect(result).toContain(commitHash);
+    });
+    it("should store files as content-addressed blobs", async () => {
+      await downloadFileToCacheDir({
+        repo: TEST_REPO,
+        path: "languages.json",
+        revision: "v1.2.2-en",
+        cacheDir: TEST_CACHE_DIR
+      });
+      const blobsPath = join(getCachePath(TEST_REPO, TEST_CACHE_DIR), "blobs");
+      const { readdirSync } = await import("fs");
+      const blobs = readdirSync(blobsPath);
+      expect(blobs.length).toBeGreaterThan(0);
+      expect(blobs[0]).toMatch(/^[a-f0-9]{64}$/);
+    });
+  });
+  describe("Performance", () => {
+    it("should download files in parallel efficiently", async () => {
+      const startTime = Date.now();
+      const promises = [
+        downloadFileToCacheDir({
+          repo: TEST_REPO,
+          path: "onnx/model_q8.onnx",
+          revision: "v1.2.2-en",
+          cacheDir: TEST_CACHE_DIR
+        }),
+        downloadFileToCacheDir({
+          repo: TEST_REPO,
+          path: "languages.json",
+          revision: "v1.2.2-en",
+          cacheDir: TEST_CACHE_DIR
+        }),
+        downloadFileToCacheDir({
+          repo: TEST_REPO,
+          path: "tokenizer.json",
+          revision: "v0.2.0-intl",
+          cacheDir: TEST_CACHE_DIR
+        })
+      ];
+      const results = await Promise.all(promises);
+      const totalTime = Date.now() - startTime;
+      results.forEach((result) => {
+        expect(result).toBeTruthy();
+        expect(existsSync(result)).toBe(true);
+      });
+      console.log(`Parallel download took ${totalTime}ms`);
+    });
+  });
+  describe("Failures", () => {
+    it("should handle non-existent file", async () => {
+      await expect(
+        downloadFileToCacheDir({
+          repo: TEST_REPO,
+          path: "onnx/model_non_existent.onnx",
+          revision: "v1.2.2-en",
+          cacheDir: TEST_CACHE_DIR
+        })
+      ).rejects.toThrow("cannot get path info for onnx/model_non_existent.onnx");
+    });
+  });
+});
+//# sourceMappingURL=hf_utils.test.js.map

package/dist/hf_utils.test.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/hf_utils.test.ts"],"sourcesContent":["// SPDX-FileCopyrightText: 2025 LiveKit, Inc.\n//\n// SPDX-License-Identifier: Apache-2.0\nimport { initializeLogger } from '@livekit/agents';\nimport { existsSync, rmSync, statSync } from 'fs';\nimport { homedir } from 'os';\nimport { join } from 'path';\nimport { afterAll, beforeAll, describe, expect, it } from 'vitest';\nimport { downloadFileToCacheDir } from './hf_utils.js';\n\nfunction getCachePath(repo: string, cacheDir?: string): string {\n  const baseCacheDir = cacheDir || join(homedir(), '.cache', 'huggingface', 'hub');\n  return join(baseCacheDir, `models--${repo.replace(/\\//g, '--')}`);\n}\n\nfunction clearCache(repo: string, cacheDir?: string): void {\n  const repoPath = getCachePath(repo, cacheDir);\n  if (existsSync(repoPath)) {\n    rmSync(repoPath, { recursive: true, force: true });\n  }\n}\n\ndescribe('HuggingFace Download Fixed Implementation', () => {\n  initializeLogger({ pretty: true, level: 'debug' });\n\n  const TEST_REPO = 'livekit/turn-detector';\n  const TEST_CACHE_DIR = join(process.cwd(), '.test-cache');\n\n  beforeAll(() => {\n    // Clear test cache before all tests\n    clearCache(TEST_REPO, TEST_CACHE_DIR);\n  });\n\n  afterAll(() => {\n    // Clean up test cache after all tests\n    if (existsSync(TEST_CACHE_DIR)) {\n      rmSync(TEST_CACHE_DIR, { recursive: true, force: true });\n    }\n  });\n\n  describe('Basic Downloads', () => {\n    it('should download a standard file in subdirectory', async () => {\n      const result = await downloadFileToCacheDir({\n        repo: TEST_REPO,\n        path: 'onnx/model_q8.onnx',\n        revision: 'v1.2.2-en',\n        cacheDir: TEST_CACHE_DIR,\n      });\n\n      expect(result).toBeTruthy();\n      expect(existsSync(result)).toBe(true);\n\n      const stats = statSync(result);\n      const sizeMB = stats.size / 1024 / 1024;\n      expect(sizeMB).toBeCloseTo(62.67, 1); // ~62.67 MB\n    });\n\n    it('should download a large file with retry logic', async () => {\n      const result = await downloadFileToCacheDir({\n        repo: TEST_REPO,\n        path: 'onnx/model.onnx',\n        revision: 'v1.2.2-en',\n        cacheDir: TEST_CACHE_DIR,\n      });\n\n      expect(result).toBeTruthy();\n      expect(existsSync(result)).toBe(true);\n\n      const stats = statSync(result);\n      const sizeMB = stats.size / 1024 / 1024;\n      expect(sizeMB).toBeCloseTo(249.96, 1); // ~250 MB\n    });\n\n    it('should download a very small file', async () => {\n      const result = await downloadFileToCacheDir({\n        repo: TEST_REPO,\n        path: 'languages.json',\n        revision: 'v1.2.2-en',\n        cacheDir: TEST_CACHE_DIR,\n      });\n\n      expect(result).toBeTruthy();\n      expect(existsSync(result)).toBe(true);\n\n      const stats = statSync(result);\n      expect(stats.size).toBeLessThan(200); // Very small file (102 bytes)\n    });\n\n    it('should download from different revision', async () => {\n      const result = await downloadFileToCacheDir({\n        repo: TEST_REPO,\n        path: 'tokenizer.json',\n        revision: 'v0.2.0-intl',\n        cacheDir: TEST_CACHE_DIR,\n      });\n\n      expect(result).toBeTruthy();\n      expect(existsSync(result)).toBe(true);\n\n      const stats = statSync(result);\n      const sizeMB = stats.size / 1024 / 1024;\n      expect(sizeMB).toBeGreaterThan(3); // Should be around 3.36 MB or more\n    });\n  });\n\n  describe('Cache Behavior', () => {\n    it('should use cache on second download', async () => {\n      // First download\n      const firstPath = await downloadFileToCacheDir({\n        repo: TEST_REPO,\n        path: 'onnx/model_q8.onnx',\n        revision: 'v1.2.2-en',\n        cacheDir: TEST_CACHE_DIR,\n      });\n\n      // Second download (should be from cache)\n      const startTime = Date.now();\n      const secondPath = await downloadFileToCacheDir({\n        repo: TEST_REPO,\n        path: 'onnx/model_q8.onnx',\n        revision: 'v1.2.2-en',\n        cacheDir: TEST_CACHE_DIR,\n      });\n      const cacheTime = Date.now() - startTime;\n\n      expect(secondPath).toBe(firstPath);\n      expect(cacheTime).toBeLessThan(500); // Should be very fast if from cache\n    });\n\n    it('should respect localFileOnly flag when file is cached', async () => {\n      // Ensure file is cached first\n      await downloadFileToCacheDir({\n        repo: TEST_REPO,\n        path: 'tokenizer.json',\n        revision: 'v0.2.0-intl',\n        cacheDir: TEST_CACHE_DIR,\n      });\n\n      // Now try with localFileOnly\n      const cachedPath = await downloadFileToCacheDir({\n        repo: TEST_REPO,\n        path: 'tokenizer.json',\n        revision: 'v0.2.0-intl',\n        cacheDir: TEST_CACHE_DIR,\n        localFileOnly: true,\n      });\n\n      expect(cachedPath).toBeTruthy();\n      expect(existsSync(cachedPath)).toBe(true);\n    });\n\n    it('should throw error with localFileOnly when file is not cached', async () => {\n      await expect(\n        downloadFileToCacheDir({\n          repo: TEST_REPO,\n          path: 'non-existent-file.txt',\n          revision: 'main',\n          cacheDir: TEST_CACHE_DIR,\n          localFileOnly: true,\n        }),\n      ).rejects.toThrow(/File not found in cache/);\n    });\n\n    it('should save revision-to-commit mappings', async () => {\n      await downloadFileToCacheDir({\n        repo: TEST_REPO,\n        path: 'languages.json',\n        revision: 'v1.2.2-en',\n        cacheDir: TEST_CACHE_DIR,\n      });\n\n      // Check if refs file was created\n      const refsPath = join(getCachePath(TEST_REPO, TEST_CACHE_DIR), 'refs', 'v1.2.2-en');\n      expect(existsSync(refsPath)).toBe(true);\n    });\n\n    it('should handle multiple files from same revision without overwriting refs', async () => {\n      // Download two different files from the same revision\n      const file1Path = await downloadFileToCacheDir({\n        repo: TEST_REPO,\n        path: 'onnx/model_q8.onnx',\n        revision: 'v1.2.2-en',\n        cacheDir: TEST_CACHE_DIR,\n      });\n\n      const file2Path = await downloadFileToCacheDir({\n        repo: TEST_REPO,\n        path: 'languages.json',\n        revision: 'v1.2.2-en',\n        cacheDir: TEST_CACHE_DIR,\n      });\n\n      // Both files should exist\n      expect(existsSync(file1Path)).toBe(true);\n      expect(existsSync(file2Path)).toBe(true);\n\n      // Now test that both files can be retrieved with localFileOnly\n      const cachedFile1 = await downloadFileToCacheDir({\n        repo: TEST_REPO,\n        path: 'onnx/model_q8.onnx',\n        revision: 'v1.2.2-en',\n        cacheDir: TEST_CACHE_DIR,\n        localFileOnly: true,\n      });\n\n      const cachedFile2 = await downloadFileToCacheDir({\n        repo: TEST_REPO,\n        path: 'languages.json',\n        revision: 'v1.2.2-en',\n        cacheDir: TEST_CACHE_DIR,\n        localFileOnly: true,\n      });\n\n      // Both should be found in cache\n      expect(cachedFile1).toBe(file1Path);\n      expect(cachedFile2).toBe(file2Path);\n\n      // Check that both files are in the same snapshot folder (same commit hash)\n      // Extract commit hash from paths\n      const match1 = file1Path.match(/snapshots\\/([a-f0-9]{40})\\//);\n      const match2 = file2Path.match(/snapshots\\/([a-f0-9]{40})\\//);\n\n      expect(match1).toBeTruthy();\n      expect(match2).toBeTruthy();\n\n      const commitHash1 = match1![1];\n      const commitHash2 = match2![1];\n\n      // FIXED: All files from the same revision should use the same HEAD commit\n      expect(commitHash1).toBe(commitHash2);\n\n      // Check that the refs file contains the single HEAD commit\n      const { readFileSync } = await import('fs');\n      const refsPath = join(getCachePath(TEST_REPO, TEST_CACHE_DIR), 'refs', 'v1.2.2-en');\n      const refsContent = readFileSync(refsPath, 'utf-8').trim();\n\n      // The refs file should contain just the commit hash, not a JSON mapping\n      expect(refsContent).toMatch(/^[a-f0-9]{40}$/);\n      expect(refsContent).toBe(commitHash1);\n    });\n  });\n\n  describe('Error Handling', () => {\n    it('should handle invalid repository gracefully', async () => {\n      await expect(\n        downloadFileToCacheDir({\n          repo: 'non-existent-org/non-existent-repo',\n          path: 'file.txt',\n          revision: 'main',\n          cacheDir: TEST_CACHE_DIR,\n        }),\n      ).rejects.toThrow();\n    });\n\n    it('should handle invalid file path gracefully', async () => {\n      await expect(\n        downloadFileToCacheDir({\n          repo: TEST_REPO,\n          path: 'non-existent-file-path.xyz',\n          revision: 'v1.2.2-en',\n          cacheDir: TEST_CACHE_DIR,\n        }),\n      ).rejects.toThrow();\n    });\n\n    it('should handle invalid revision gracefully', async () => {\n      await expect(\n        downloadFileToCacheDir({\n          repo: TEST_REPO,\n          path: 'tokenizer.json',\n          revision: 'non-existent-revision',\n          cacheDir: TEST_CACHE_DIR,\n        }),\n      ).rejects.toThrow();\n    });\n  });\n\n  describe('Cache Structure', () => {\n    it('should create proper cache directory structure', async () => {\n      await downloadFileToCacheDir({\n        repo: TEST_REPO,\n        path: 'onnx/model_q8.onnx',\n        revision: 'v1.2.2-en',\n        cacheDir: TEST_CACHE_DIR,\n      });\n\n      const cachePath = getCachePath(TEST_REPO, TEST_CACHE_DIR);\n\n      // Check expected directories exist\n      expect(existsSync(join(cachePath, 'blobs'))).toBe(true);\n      expect(existsSync(join(cachePath, 'snapshots'))).toBe(true);\n      expect(existsSync(join(cachePath, 'refs'))).toBe(true);\n    });\n\n    it('should handle commit hash revisions', async () => {\n      // We'll use the actual commit hash from the v1.2.2-en tag\n      // First, download to ensure we have the commit hash mapping\n      const tagResult = await downloadFileToCacheDir({\n        repo: TEST_REPO,\n        path: 'onnx/model_q8.onnx',\n        revision: 'v1.2.2-en',\n        cacheDir: TEST_CACHE_DIR,\n      });\n\n      // Extract the commit hash from the path\n      const match = tagResult.match(/snapshots\\/([a-f0-9]{40})\\//);\n      expect(match).toBeTruthy();\n      const commitHash = match![1];\n\n      // Now download with commit hash directly\n      const result = await downloadFileToCacheDir({\n        repo: TEST_REPO,\n        path: 'onnx/model_q8.onnx',\n        revision: commitHash,\n        cacheDir: TEST_CACHE_DIR,\n      });\n\n      expect(result).toBeTruthy();\n      expect(result).toContain(commitHash);\n    });\n\n    it('should store files as content-addressed blobs', async () => {\n      await downloadFileToCacheDir({\n        repo: TEST_REPO,\n        path: 'languages.json',\n        revision: 'v1.2.2-en',\n        cacheDir: TEST_CACHE_DIR,\n      });\n\n      const blobsPath = join(getCachePath(TEST_REPO, TEST_CACHE_DIR), 'blobs');\n      const { readdirSync } = await import('fs');\n      const blobs = readdirSync(blobsPath);\n\n      // Should have at least one blob with SHA256 hash name\n      expect(blobs.length).toBeGreaterThan(0);\n      expect(blobs[0]).toMatch(/^[a-f0-9]{64}$/);\n    });\n  });\n\n  describe('Performance', () => {\n    it('should download files in parallel efficiently', async () => {\n      const startTime = Date.now();\n\n      // Download multiple files in parallel\n      const promises = [\n        downloadFileToCacheDir({\n          repo: TEST_REPO,\n          path: 'onnx/model_q8.onnx',\n          revision: 'v1.2.2-en',\n          cacheDir: TEST_CACHE_DIR,\n        }),\n        downloadFileToCacheDir({\n          repo: TEST_REPO,\n          path: 'languages.json',\n          revision: 'v1.2.2-en',\n          cacheDir: TEST_CACHE_DIR,\n        }),\n        downloadFileToCacheDir({\n          repo: TEST_REPO,\n          path: 'tokenizer.json',\n          revision: 'v0.2.0-intl',\n          cacheDir: TEST_CACHE_DIR,\n        }),\n      ];\n\n      const results = await Promise.all(promises);\n      const totalTime = Date.now() - startTime;\n\n      // All should succeed\n      results.forEach((result) => {\n        expect(result).toBeTruthy();\n        expect(existsSync(result)).toBe(true);\n      });\n\n      // Should be faster than downloading sequentially\n      console.log(`Parallel download took ${totalTime}ms`);\n    });\n  });\n\n  describe('Failures', () => {\n    it('should handle non-existent file', async () => {\n      await expect(\n        downloadFileToCacheDir({\n          repo: TEST_REPO,\n          path: 'onnx/model_non_existent.onnx',\n          revision: 'v1.2.2-en',\n          cacheDir: TEST_CACHE_DIR,\n        }),\n      ).rejects.toThrow('cannot get path info for onnx/model_non_existent.onnx');\n    });\n  });\n});\n"],"mappings":"AAGA,SAAS,wBAAwB;AACjC,SAAS,YAAY,QAAQ,gBAAgB;AAC7C,SAAS,eAAe;AACxB,SAAS,YAAY;AACrB,SAAS,UAAU,WAAW,UAAU,QAAQ,UAAU;AAC1D,SAAS,8BAA8B;AAEvC,SAAS,aAAa,MAAc,UAA2B;AAC7D,QAAM,eAAe,YAAY,KAAK,QAAQ,GAAG,UAAU,eAAe,KAAK;AAC/E,SAAO,KAAK,cAAc,WAAW,KAAK,QAAQ,OAAO,IAAI,CAAC,EAAE;AAClE;AAEA,SAAS,WAAW,MAAc,UAAyB;AACzD,QAAM,WAAW,aAAa,MAAM,QAAQ;AAC5C,MAAI,WAAW,QAAQ,GAAG;AACxB,WAAO,UAAU,EAAE,WAAW,MAAM,OAAO,KAAK,CAAC;AAAA,EACnD;AACF;AAEA,SAAS,6CAA6C,MAAM;AAC1D,mBAAiB,EAAE,QAAQ,MAAM,OAAO,QAAQ,CAAC;AAEjD,QAAM,YAAY;AAClB,QAAM,iBAAiB,KAAK,QAAQ,IAAI,GAAG,aAAa;AAExD,YAAU,MAAM;AAEd,eAAW,WAAW,cAAc;AAAA,EACtC,CAAC;AAED,WAAS,MAAM;AAEb,QAAI,WAAW,cAAc,GAAG;AAC9B,aAAO,gBAAgB,EAAE,WAAW,MAAM,OAAO,KAAK,CAAC;AAAA,IACzD;AAAA,EACF,CAAC;AAED,WAAS,mBAAmB,MAAM;AAChC,OAAG,mDAAmD,YAAY;AAChE,YAAM,SAAS,MAAM,uBAAuB;AAAA,QAC1C,MAAM;AAAA,QACN,MAAM;AAAA,QACN,UAAU;AAAA,QACV,UAAU;AAAA,MACZ,CAAC;AAED,aAAO,MAAM,EAAE,WAAW;AAC1B,aAAO,WAAW,MAAM,CAAC,EAAE,KAAK,IAAI;AAEpC,YAAM,QAAQ,SAAS,MAAM;AAC7B,YAAM,SAAS,MAAM,OAAO,OAAO;AACnC,aAAO,MAAM,EAAE,YAAY,OAAO,CAAC;AAAA,IACrC,CAAC;AAED,OAAG,iDAAiD,YAAY;AAC9D,YAAM,SAAS,MAAM,uBAAuB;AAAA,QAC1C,MAAM;AAAA,QACN,MAAM;AAAA,QACN,UAAU;AAAA,QACV,UAAU;AAAA,MACZ,CAAC;AAED,aAAO,MAAM,EAAE,WAAW;AAC1B,aAAO,WAAW,MAAM,CAAC,EAAE,KAAK,IAAI;AAEpC,YAAM,QAAQ,SAAS,MAAM;AAC7B,YAAM,SAAS,MAAM,OAAO,OAAO;AACnC,aAAO,MAAM,EAAE,YAAY,QAAQ,CAAC;AAAA,IACtC,CAAC;AAED,OAAG,qCAAqC,YAAY;AAClD,YAAM,SAAS,MAAM,uBAAuB;AAAA,QAC1C,MAAM;AAAA,QACN,MAAM;AAAA,QACN,UAAU;AAAA,QACV,UAAU;AAAA,MACZ,CAAC;AAED,aAAO,MAAM,EAAE,WAAW;AAC1B,aAAO,WAAW,MAAM,CAAC,EAAE,KAAK,IAAI;AAEpC,YAAM,QAAQ,SAAS,MAAM;AAC7B,aAAO,MAAM,IAAI,EAAE,aAAa,GAAG;AAAA,IACrC,CAAC;AAED,OAAG,2CAA2C,YAAY;AACxD,YAAM,SAAS,MAAM,uBAAuB;AAAA,QAC1C,MAAM;AAAA,QACN,MAAM;AAAA,QACN,UAAU;AAAA,QACV,UAAU;AAAA,MACZ,CAAC;AAED,aAAO,MAAM,EAAE,WAAW;AAC1B,aAAO,WAAW,MAAM,CAAC,EAAE,KAAK,IAAI;AAEpC,YAAM,QAAQ,SAAS,MAAM;AAC7B,YAAM,SAAS,MAAM,OAAO,OAAO;AACnC,aAAO,MAAM,EAAE,gBAAgB,CAAC;AAAA,IAClC,CAAC;AAAA,EACH,CAAC;AAED,WAAS,kBAAkB,MAAM;AAC/B,OAAG,uCAAuC,YAAY;AAEpD,YAAM,YAAY,MAAM,uBAAuB;AAAA,QAC7C,MAAM;AAAA,QACN,MAAM;AAAA,QACN,UAAU;AAAA,QACV,UAAU;AAAA,MACZ,CAAC;AAGD,YAAM,YAAY,KAAK,IAAI;AAC3B,YAAM,aAAa,MAAM,uBAAuB;AAAA,QAC9C,MAAM;AAAA,QACN,MAAM;AAAA,QACN,UAAU;AAAA,QACV,UAAU;AAAA,MACZ,CAAC;AACD,YAAM,YAAY,KAAK,IAAI,IAAI;AAE/B,aAAO,UAAU,EAAE,KAAK,SAAS;AACjC,aAAO,SAAS,EAAE,aAAa,GAAG;AAAA,IACpC,CAAC;AAED,OAAG,yDAAyD,YAAY;AAEtE,YAAM,uBAAuB;AAAA,QAC3B,MAAM;AAAA,QACN,MAAM;AAAA,QACN,UAAU;AAAA,QACV,UAAU;AAAA,MACZ,CAAC;AAGD,YAAM,aAAa,MAAM,uBAAuB;AAAA,QAC9C,MAAM;AAAA,QACN,MAAM;AAAA,QACN,UAAU;AAAA,QACV,UAAU;AAAA,QACV,eAAe;AAAA,MACjB,CAAC;AAED,aAAO,UAAU,EAAE,WAAW;AAC9B,aAAO,WAAW,UAAU,CAAC,EAAE,KAAK,IAAI;AAAA,IAC1C,CAAC;AAED,OAAG,iEAAiE,YAAY;AAC9E,YAAM;AAAA,QACJ,uBAAuB;AAAA,UACrB,MAAM;AAAA,UACN,MAAM;AAAA,UACN,UAAU;AAAA,UACV,UAAU;AAAA,UACV,eAAe;AAAA,QACjB,CAAC;AAAA,MACH,EAAE,QAAQ,QAAQ,yBAAyB;AAAA,IAC7C,CAAC;AAED,OAAG,2CAA2C,YAAY;AACxD,YAAM,uBAAuB;AAAA,QAC3B,MAAM;AAAA,QACN,MAAM;AAAA,QACN,UAAU;AAAA,QACV,UAAU;AAAA,MACZ,CAAC;AAGD,YAAM,WAAW,KAAK,aAAa,WAAW,cAAc,GAAG,QAAQ,WAAW;AAClF,aAAO,WAAW,QAAQ,CAAC,EAAE,KAAK,IAAI;AAAA,IACxC,CAAC;AAED,OAAG,4EAA4E,YAAY;AAEzF,YAAM,YAAY,MAAM,uBAAuB;AAAA,QAC7C,MAAM;AAAA,QACN,MAAM;AAAA,QACN,UAAU;AAAA,QACV,UAAU;AAAA,MACZ,CAAC;AAED,YAAM,YAAY,MAAM,uBAAuB;AAAA,QAC7C,MAAM;AAAA,QACN,MAAM;AAAA,QACN,UAAU;AAAA,QACV,UAAU;AAAA,MACZ,CAAC;AAGD,aAAO,WAAW,SAAS,CAAC,EAAE,KAAK,IAAI;AACvC,aAAO,WAAW,SAAS,CAAC,EAAE,KAAK,IAAI;AAGvC,YAAM,cAAc,MAAM,uBAAuB;AAAA,QAC/C,MAAM;AAAA,QACN,MAAM;AAAA,QACN,UAAU;AAAA,QACV,UAAU;AAAA,QACV,eAAe;AAAA,MACjB,CAAC;AAED,YAAM,cAAc,MAAM,uBAAuB;AAAA,QAC/C,MAAM;AAAA,QACN,MAAM;AAAA,QACN,UAAU;AAAA,QACV,UAAU;AAAA,QACV,eAAe;AAAA,MACjB,CAAC;AAGD,aAAO,WAAW,EAAE,KAAK,SAAS;AAClC,aAAO,WAAW,EAAE,KAAK,SAAS;AAIlC,YAAM,SAAS,UAAU,MAAM,6BAA6B;AAC5D,YAAM,SAAS,UAAU,MAAM,6BAA6B;AAE5D,aAAO,MAAM,EAAE,WAAW;AAC1B,aAAO,MAAM,EAAE,WAAW;AAE1B,YAAM,cAAc,OAAQ,CAAC;AAC7B,YAAM,cAAc,OAAQ,CAAC;AAG7B,aAAO,WAAW,EAAE,KAAK,WAAW;AAGpC,YAAM,EAAE,aAAa,IAAI,MAAM,OAAO,IAAI;AAC1C,YAAM,WAAW,KAAK,aAAa,WAAW,cAAc,GAAG,QAAQ,WAAW;AAClF,YAAM,cAAc,aAAa,UAAU,OAAO,EAAE,KAAK;AAGzD,aAAO,WAAW,EAAE,QAAQ,gBAAgB;AAC5C,aAAO,WAAW,EAAE,KAAK,WAAW;AAAA,IACtC,CAAC;AAAA,EACH,CAAC;AAED,WAAS,kBAAkB,MAAM;AAC/B,OAAG,+CAA+C,YAAY;AAC5D,YAAM;AAAA,QACJ,uBAAuB;AAAA,UACrB,MAAM;AAAA,UACN,MAAM;AAAA,UACN,UAAU;AAAA,UACV,UAAU;AAAA,QACZ,CAAC;AAAA,MACH,EAAE,QAAQ,QAAQ;AAAA,IACpB,CAAC;AAED,OAAG,8CAA8C,YAAY;AAC3D,YAAM;AAAA,QACJ,uBAAuB;AAAA,UACrB,MAAM;AAAA,UACN,MAAM;AAAA,UACN,UAAU;AAAA,UACV,UAAU;AAAA,QACZ,CAAC;AAAA,MACH,EAAE,QAAQ,QAAQ;AAAA,IACpB,CAAC;AAED,OAAG,6CAA6C,YAAY;AAC1D,YAAM;AAAA,QACJ,uBAAuB;AAAA,UACrB,MAAM;AAAA,UACN,MAAM;AAAA,UACN,UAAU;AAAA,UACV,UAAU;AAAA,QACZ,CAAC;AAAA,MACH,EAAE,QAAQ,QAAQ;AAAA,IACpB,CAAC;AAAA,EACH,CAAC;AAED,WAAS,mBAAmB,MAAM;AAChC,OAAG,kDAAkD,YAAY;AAC/D,YAAM,uBAAuB;AAAA,QAC3B,MAAM;AAAA,QACN,MAAM;AAAA,QACN,UAAU;AAAA,QACV,UAAU;AAAA,MACZ,CAAC;AAED,YAAM,YAAY,aAAa,WAAW,cAAc;AAGxD,aAAO,WAAW,KAAK,WAAW,OAAO,CAAC,CAAC,EAAE,KAAK,IAAI;AACtD,aAAO,WAAW,KAAK,WAAW,WAAW,CAAC,CAAC,EAAE,KAAK,IAAI;AAC1D,aAAO,WAAW,KAAK,WAAW,MAAM,CAAC,CAAC,EAAE,KAAK,IAAI;AAAA,IACvD,CAAC;AAED,OAAG,uCAAuC,YAAY;AAGpD,YAAM,YAAY,MAAM,uBAAuB;AAAA,QAC7C,MAAM;AAAA,QACN,MAAM;AAAA,QACN,UAAU;AAAA,QACV,UAAU;AAAA,MACZ,CAAC;AAGD,YAAM,QAAQ,UAAU,MAAM,6BAA6B;AAC3D,aAAO,KAAK,EAAE,WAAW;AACzB,YAAM,aAAa,MAAO,CAAC;AAG3B,YAAM,SAAS,MAAM,uBAAuB;AAAA,QAC1C,MAAM;AAAA,QACN,MAAM;AAAA,QACN,UAAU;AAAA,QACV,UAAU;AAAA,MACZ,CAAC;AAED,aAAO,MAAM,EAAE,WAAW;AAC1B,aAAO,MAAM,EAAE,UAAU,UAAU;AAAA,IACrC,CAAC;AAED,OAAG,iDAAiD,YAAY;AAC9D,YAAM,uBAAuB;AAAA,QAC3B,MAAM;AAAA,QACN,MAAM;AAAA,QACN,UAAU;AAAA,QACV,UAAU;AAAA,MACZ,CAAC;AAED,YAAM,YAAY,KAAK,aAAa,WAAW,cAAc,GAAG,OAAO;AACvE,YAAM,EAAE,YAAY,IAAI,MAAM,OAAO,IAAI;AACzC,YAAM,QAAQ,YAAY,SAAS;AAGnC,aAAO,MAAM,MAAM,EAAE,gBAAgB,CAAC;AACtC,aAAO,MAAM,CAAC,CAAC,EAAE,QAAQ,gBAAgB;AAAA,IAC3C,CAAC;AAAA,EACH,CAAC;AAED,WAAS,eAAe,MAAM;AAC5B,OAAG,iDAAiD,YAAY;AAC9D,YAAM,YAAY,KAAK,IAAI;AAG3B,YAAM,WAAW;AAAA,QACf,uBAAuB;AAAA,UACrB,MAAM;AAAA,UACN,MAAM;AAAA,UACN,UAAU;AAAA,UACV,UAAU;AAAA,QACZ,CAAC;AAAA,QACD,uBAAuB;AAAA,UACrB,MAAM;AAAA,UACN,MAAM;AAAA,UACN,UAAU;AAAA,UACV,UAAU;AAAA,QACZ,CAAC;AAAA,QACD,uBAAuB;AAAA,UACrB,MAAM;AAAA,UACN,MAAM;AAAA,UACN,UAAU;AAAA,UACV,UAAU;AAAA,QACZ,CAAC;AAAA,MACH;AAEA,YAAM,UAAU,MAAM,QAAQ,IAAI,QAAQ;AAC1C,YAAM,YAAY,KAAK,IAAI,IAAI;AAG/B,cAAQ,QAAQ,CAAC,WAAW;AAC1B,eAAO,MAAM,EAAE,WAAW;AAC1B,eAAO,WAAW,MAAM,CAAC,EAAE,KAAK,IAAI;AAAA,MACtC,CAAC;AAGD,cAAQ,IAAI,0BAA0B,SAAS,IAAI;AAAA,IACrD,CAAC;AAAA,EACH,CAAC;AAED,WAAS,YAAY,MAAM;AACzB,OAAG,mCAAmC,YAAY;AAChD,YAAM;AAAA,QACJ,uBAAuB;AAAA,UACrB,MAAM;AAAA,UACN,MAAM;AAAA,UACN,UAAU;AAAA,UACV,UAAU;AAAA,QACZ,CAAC;AAAA,MACH,EAAE,QAAQ,QAAQ,uDAAuD;AAAA,IAC3E,CAAC;AAAA,EACH,CAAC;AACH,CAAC;","names":[]}

package/dist/index.cjs

@@ -26,21 +26,38 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
   mod
 ));
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
-var src_exports = {};
-__export(src_exports, {
+var index_exports = {};
+__export(index_exports, {
+  downloadFileToCacheDir: () => import_hf_utils2.downloadFileToCacheDir,
   turnDetector: () => turnDetector
 });
-module.exports = __toCommonJS(src_exports);
-var getImportMetaUrl = () => typeof document === "undefined" ? new URL(`file:${__filename}`).href : document.currentScript && document.currentScript.src || new URL("main.js", document.baseURI).href;
-var importMetaUrl = /* @__PURE__ */ getImportMetaUrl();
+module.exports = __toCommonJS(index_exports);
 var import_agents = require("@livekit/agents");
-var turnDetector = __toESM(require("./turn_detector.cjs"), 1);
-import_agents.InferenceRunner.registerRunner(
-  turnDetector.EOURunner.INFERENCE_METHOD,
-  new URL("./turn_detector.js", importMetaUrl).toString()
-);
+var import_hf_utils = require("./hf_utils.cjs");
+var import_constants = require("./turn_detector/constants.cjs");
+var import_hf_utils2 = require("./hf_utils.cjs");
+var turnDetector = __toESM(require("./turn_detector/index.cjs"), 1);
+class EOUPlugin extends import_agents.Plugin {
+  constructor() {
+    super({
+      title: "turn-detector",
+      version: "0.1.1",
+      package: "@livekit/agents-plugin-livekit"
+    });
+  }
+  async downloadFiles() {
+    const { AutoTokenizer } = await import("@huggingface/transformers");
+    for (const revision of Object.values(import_constants.MODEL_REVISIONS)) {
+      await AutoTokenizer.from_pretrained(import_constants.HG_MODEL_REPO, { revision });
+      await (0, import_hf_utils.downloadFileToCacheDir)({ repo: import_constants.HG_MODEL_REPO, path: import_constants.ONNX_FILEPATH, revision });
+      await (0, import_hf_utils.downloadFileToCacheDir)({ repo: import_constants.HG_MODEL_REPO, path: "languages.json", revision });
+    }
+  }
+}
+import_agents.Plugin.registerPlugin(new EOUPlugin());
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  downloadFileToCacheDir,
   turnDetector
 });
 //# sourceMappingURL=index.cjs.map

package/dist/index.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/index.ts","../../../node_modules/.pnpm/tsup@8.3.5_@microsoft+api-extractor@7.43.7_@types+node@22.5.5__postcss@8.4.38_tsx@4.19.2_typescript@5.4.5/node_modules/tsup/assets/cjs_shims.js"],"sourcesContent":["// SPDX-FileCopyrightText: 2024 LiveKit, Inc.\n//\n// SPDX-License-Identifier: Apache-2.0\nimport { InferenceRunner } from '@livekit/agents';\nimport * as turnDetector from './turn_detector.js';\n\nInferenceRunner.registerRunner(\n  turnDetector.EOURunner.INFERENCE_METHOD,\n  new URL('./turn_detector.js', import.meta.url).toString(),\n);\nexport { turnDetector };\n","// Shim globals in cjs bundle\n// There's a weird bug that esbuild will always inject importMetaUrl\n// if we export it as `const importMetaUrl = ... __filename ...`\n// But using a function will not cause this issue\n\nconst getImportMetaUrl = () =>\n  typeof document === 'undefined'\n    ? new URL(`file:${__filename}`).href\n    : (document.currentScript && document.currentScript.src) ||\n      new URL('main.js', document.baseURI).href\n\nexport const importMetaUrl = /* @__PURE__ */ getImportMetaUrl()\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;ACKA,IAAM,mBAAmB,MACvB,OAAO,aAAa,cAChB,IAAI,IAAI,QAAQ,UAAU,EAAE,EAAE,OAC7B,SAAS,iBAAiB,SAAS,cAAc,OAClD,IAAI,IAAI,WAAW,SAAS,OAAO,EAAE;AAEpC,IAAM,gBAAgC,iCAAiB;ADR9D,oBAAgC;AAChC,mBAA8B;AAE9B,8BAAgB;AAAA,EACd,aAAa,UAAU;AAAA,EACvB,IAAI,IAAI,sBAAsB,aAAe,EAAE,SAAS;AAC1D;","names":[]}
+{"version":3,"sources":["../src/index.ts"],"sourcesContent":["// SPDX-FileCopyrightText: 2025 LiveKit, Inc.\n//\n// SPDX-License-Identifier: Apache-2.0\nimport { Plugin } from '@livekit/agents';\nimport { downloadFileToCacheDir as hfDownload } from './hf_utils.js';\nimport { HG_MODEL_REPO, MODEL_REVISIONS, ONNX_FILEPATH } from './turn_detector/constants.js';\n\nexport { downloadFileToCacheDir } from './hf_utils.js';\nexport * as turnDetector from './turn_detector/index.js';\n\nclass EOUPlugin extends Plugin {\n  constructor() {\n    super({\n      title: 'turn-detector',\n      version: '0.1.1',\n      package: '@livekit/agents-plugin-livekit',\n    });\n  }\n\n  async downloadFiles(): Promise<void> {\n    const { AutoTokenizer } = await import('@huggingface/transformers');\n\n    for (const revision of Object.values(MODEL_REVISIONS)) {\n      // Ensure tokenizer is cached\n      await AutoTokenizer.from_pretrained(HG_MODEL_REPO, { revision });\n\n      // Ensure ONNX model and language data are cached\n      await hfDownload({ repo: HG_MODEL_REPO, path: ONNX_FILEPATH, revision });\n      await hfDownload({ repo: HG_MODEL_REPO, path: 'languages.json', revision });\n    }\n  }\n}\n\nPlugin.registerPlugin(new EOUPlugin());\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAGA,oBAAuB;AACvB,sBAAqD;AACrD,uBAA8D;AAE9D,IAAAA,mBAAuC;AACvC,mBAA8B;AAE9B,MAAM,kBAAkB,qBAAO;AAAA,EAC7B,cAAc;AACZ,UAAM;AAAA,MACJ,OAAO;AAAA,MACP,SAAS;AAAA,MACT,SAAS;AAAA,IACX,CAAC;AAAA,EACH;AAAA,EAEA,MAAM,gBAA+B;AACnC,UAAM,EAAE,cAAc,IAAI,MAAM,OAAO,2BAA2B;AAElE,eAAW,YAAY,OAAO,OAAO,gCAAe,GAAG;AAErD,YAAM,cAAc,gBAAgB,gCAAe,EAAE,SAAS,CAAC;AAG/D,gBAAM,gBAAAC,wBAAW,EAAE,MAAM,gCAAe,MAAM,gCAAe,SAAS,CAAC;AACvE,gBAAM,gBAAAA,wBAAW,EAAE,MAAM,gCAAe,MAAM,kBAAkB,SAAS,CAAC;AAAA,IAC5E;AAAA,EACF;AACF;AAEA,qBAAO,eAAe,IAAI,UAAU,CAAC;","names":["import_hf_utils","hfDownload"]}

package/dist/index.d.cts

@@ -1,3 +1,3 @@
-import * as turnDetector from './turn_detector.js';
-export { turnDetector };
+export { downloadFileToCacheDir } from './hf_utils.js';
+export * as turnDetector from './turn_detector/index.js';
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts

@@ -1,3 +1,3 @@
-import * as turnDetector from './turn_detector.js';
-export { turnDetector };
+export { downloadFileToCacheDir } from './hf_utils.js';
+export * as turnDetector from './turn_detector/index.js';
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,YAAY,MAAM,oBAAoB,CAAC;AAMnD,OAAO,EAAE,YAAY,EAAE,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,sBAAsB,EAAE,MAAM,eAAe,CAAC;AACvD,OAAO,KAAK,YAAY,MAAM,0BAA0B,CAAC"}

package/dist/index.js

@@ -1,10 +1,28 @@
-import { InferenceRunner } from "@livekit/agents";
-import * as turnDetector from "./turn_detector.js";
-InferenceRunner.registerRunner(
-  turnDetector.EOURunner.INFERENCE_METHOD,
-  new URL("./turn_detector.js", import.meta.url).toString()
-);
+import { Plugin } from "@livekit/agents";
+import { downloadFileToCacheDir as hfDownload } from "./hf_utils.js";
+import { HG_MODEL_REPO, MODEL_REVISIONS, ONNX_FILEPATH } from "./turn_detector/constants.js";
+import { downloadFileToCacheDir } from "./hf_utils.js";
+import * as turnDetector from "./turn_detector/index.js";
+class EOUPlugin extends Plugin {
+  constructor() {
+    super({
+      title: "turn-detector",
+      version: "0.1.1",
+      package: "@livekit/agents-plugin-livekit"
+    });
+  }
+  async downloadFiles() {
+    const { AutoTokenizer } = await import("@huggingface/transformers");
+    for (const revision of Object.values(MODEL_REVISIONS)) {
+      await AutoTokenizer.from_pretrained(HG_MODEL_REPO, { revision });
+      await hfDownload({ repo: HG_MODEL_REPO, path: ONNX_FILEPATH, revision });
+      await hfDownload({ repo: HG_MODEL_REPO, path: "languages.json", revision });
+    }
+  }
+}
+Plugin.registerPlugin(new EOUPlugin());
 export {
+  downloadFileToCacheDir,
   turnDetector
 };
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/index.ts"],"sourcesContent":["// SPDX-FileCopyrightText: 2024 LiveKit, Inc.\n//\n// SPDX-License-Identifier: Apache-2.0\nimport { InferenceRunner } from '@livekit/agents';\nimport * as turnDetector from './turn_detector.js';\n\nInferenceRunner.registerRunner(\n  turnDetector.EOURunner.INFERENCE_METHOD,\n  new URL('./turn_detector.js', import.meta.url).toString(),\n);\nexport { turnDetector };\n"],"mappings":"AAGA,SAAS,uBAAuB;AAChC,YAAY,kBAAkB;AAE9B,gBAAgB;AAAA,EACd,aAAa,UAAU;AAAA,EACvB,IAAI,IAAI,sBAAsB,YAAY,GAAG,EAAE,SAAS;AAC1D;","names":[]}
+{"version":3,"sources":["../src/index.ts"],"sourcesContent":["// SPDX-FileCopyrightText: 2025 LiveKit, Inc.\n//\n// SPDX-License-Identifier: Apache-2.0\nimport { Plugin } from '@livekit/agents';\nimport { downloadFileToCacheDir as hfDownload } from './hf_utils.js';\nimport { HG_MODEL_REPO, MODEL_REVISIONS, ONNX_FILEPATH } from './turn_detector/constants.js';\n\nexport { downloadFileToCacheDir } from './hf_utils.js';\nexport * as turnDetector from './turn_detector/index.js';\n\nclass EOUPlugin extends Plugin {\n  constructor() {\n    super({\n      title: 'turn-detector',\n      version: '0.1.1',\n      package: '@livekit/agents-plugin-livekit',\n    });\n  }\n\n  async downloadFiles(): Promise<void> {\n    const { AutoTokenizer } = await import('@huggingface/transformers');\n\n    for (const revision of Object.values(MODEL_REVISIONS)) {\n      // Ensure tokenizer is cached\n      await AutoTokenizer.from_pretrained(HG_MODEL_REPO, { revision });\n\n      // Ensure ONNX model and language data are cached\n      await hfDownload({ repo: HG_MODEL_REPO, path: ONNX_FILEPATH, revision });\n      await hfDownload({ repo: HG_MODEL_REPO, path: 'languages.json', revision });\n    }\n  }\n}\n\nPlugin.registerPlugin(new EOUPlugin());\n"],"mappings":"AAGA,SAAS,cAAc;AACvB,SAAS,0BAA0B,kBAAkB;AACrD,SAAS,eAAe,iBAAiB,qBAAqB;AAE9D,SAAS,8BAA8B;AACvC,YAAY,kBAAkB;AAE9B,MAAM,kBAAkB,OAAO;AAAA,EAC7B,cAAc;AACZ,UAAM;AAAA,MACJ,OAAO;AAAA,MACP,SAAS;AAAA,MACT,SAAS;AAAA,IACX,CAAC;AAAA,EACH;AAAA,EAEA,MAAM,gBAA+B;AACnC,UAAM,EAAE,cAAc,IAAI,MAAM,OAAO,2BAA2B;AAElE,eAAW,YAAY,OAAO,OAAO,eAAe,GAAG;AAErD,YAAM,cAAc,gBAAgB,eAAe,EAAE,SAAS,CAAC;AAG/D,YAAM,WAAW,EAAE,MAAM,eAAe,MAAM,eAAe,SAAS,CAAC;AACvE,YAAM,WAAW,EAAE,MAAM,eAAe,MAAM,kBAAkB,SAAS,CAAC;AAAA,IAC5E;AAAA,EACF;AACF;AAEA,OAAO,eAAe,IAAI,UAAU,CAAC;","names":[]}