diff options
| author | Eddie Santos <[email protected]> | 2025-06-07 13:38:05 -0700 |
|---|---|---|
| committer | GitHub <[email protected]> | 2025-06-07 13:38:05 -0700 |
| commit | 27fdd1b6e6e50882ee9a17c85c5f6b845d4973ce (patch) | |
| tree | 1a75c9c755c8ce02e39e0da79342cbd79b359acd /packages/cli/src/ui/App.test.tsx | |
| parent | 51cd5ffd91b1a061ee1da2b048c22cd05ca3e836 (diff) | |
Add embedder (#818)
Diffstat (limited to 'packages/cli/src/ui/App.test.tsx')
| -rw-r--r-- | packages/cli/src/ui/App.test.tsx | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/packages/cli/src/ui/App.test.tsx b/packages/cli/src/ui/App.test.tsx index 98d82be8..f4ada985 100644 --- a/packages/cli/src/ui/App.test.tsx +++ b/packages/cli/src/ui/App.test.tsx @@ -38,6 +38,7 @@ interface MockServerConfig { vertexai?: boolean; showMemoryUsage?: boolean; accessibility?: AccessibilitySettings; + embeddingModel: string; getApiKey: Mock<() => string>; getModel: Mock<() => string>; @@ -92,6 +93,7 @@ vi.mock('@gemini-code/core', async (importOriginal) => { vertexai: opts.vertexai, showMemoryUsage: opts.showMemoryUsage ?? false, accessibility: opts.accessibility ?? {}, + embeddingModel: opts.embeddingModel || 'test-embedding-model', getApiKey: vi.fn(() => opts.apiKey || 'test-key'), getModel: vi.fn(() => opts.model || 'test-model-in-mock-factory'), @@ -178,7 +180,8 @@ describe('App UI', () => { const ServerConfigMocked = vi.mocked(ServerConfig, true); mockConfig = new ServerConfigMocked({ apiKey: 'test-key', - model: 'test-model-in-options', + model: 'test-model', + embeddingModel: 'test-embedding-model', sandbox: false, targetDir: '/test/dir', debugMode: false, |