diff options
| -rw-r--r-- | packages/core/src/core/modelCheck.ts | 2 | ||||
| -rw-r--r-- | packages/core/src/telemetry/loggers.test.ts | 11 | ||||
| -rw-r--r-- | packages/core/src/telemetry/loggers.ts | 2 |
3 files changed, 4 insertions, 11 deletions
diff --git a/packages/core/src/core/modelCheck.ts b/packages/core/src/core/modelCheck.ts index 7d7a3b7d..6caea54d 100644 --- a/packages/core/src/core/modelCheck.ts +++ b/packages/core/src/core/modelCheck.ts @@ -7,7 +7,7 @@ import { DEFAULT_GEMINI_MODEL, DEFAULT_GEMINI_FLASH_MODEL, -} from '@gemini-cli/core'; +} from '../config/models.js'; /** * Checks if the default "pro" model is rate-limited and returns a fallback "flash" diff --git a/packages/core/src/telemetry/loggers.test.ts b/packages/core/src/telemetry/loggers.test.ts index 3f61530f..3fa9ad1c 100644 --- a/packages/core/src/telemetry/loggers.test.ts +++ b/packages/core/src/telemetry/loggers.test.ts @@ -4,7 +4,8 @@ * SPDX-License-Identifier: Apache-2.0 */ -import { AuthType, ToolConfirmationOutcome } from '../index.js'; +import { ToolConfirmationOutcome } from '../tools/tools.js'; +import { AuthType } from '../core/contentGenerator.js'; import { logs } from '@opentelemetry/api-logs'; import { SemanticAttributes } from '@opentelemetry/semantic-conventions'; import { Config } from '../config/config.js'; @@ -27,14 +28,6 @@ import * as sdk from './sdk.js'; import { vi, describe, beforeEach, it, expect } from 'vitest'; import { GenerateContentResponse } from '@google/genai'; -vi.mock('@gemini-cli/cli/dist/src/utils/version', () => ({ - getCliVersion: () => 'test-version', -})); - -vi.mock('@gemini-cli/cli/dist/src/config/settings', () => ({ - getTheme: () => 'test-theme', -})); - describe('loggers', () => { const mockLogger = { emit: vi.fn(), diff --git a/packages/core/src/telemetry/loggers.ts b/packages/core/src/telemetry/loggers.ts index f7edca5e..0ecf130f 100644 --- a/packages/core/src/telemetry/loggers.ts +++ b/packages/core/src/telemetry/loggers.ts @@ -30,7 +30,7 @@ import { recordToolCallMetrics, } from './metrics.js'; import { isTelemetrySdkInitialized } from './sdk.js'; -import { ToolConfirmationOutcome } from '../index.js'; +import { ToolConfirmationOutcome } from '../tools/tools.js'; import { GenerateContentResponse, GenerateContentResponseUsageMetadata, |
