summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--packages/core/src/core/modelCheck.ts2
-rw-r--r--packages/core/src/telemetry/loggers.test.ts11
-rw-r--r--packages/core/src/telemetry/loggers.ts2
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,