From 389907ce6537bd8caeb10b3d4dacaa8ece059500 Mon Sep 17 00:00:00 2001 From: Tommaso Sciortino Date: Sat, 7 Jun 2025 16:17:27 -0700 Subject: Introduce ContentGeneratorConfig (#826) --- packages/core/src/tools/tool-registry.test.ts | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) (limited to 'packages/core/src/tools') diff --git a/packages/core/src/tools/tool-registry.test.ts b/packages/core/src/tools/tool-registry.test.ts index 0c23a74e..44e04a9d 100644 --- a/packages/core/src/tools/tool-registry.test.ts +++ b/packages/core/src/tools/tool-registry.test.ts @@ -124,17 +124,19 @@ class MockTool extends BaseTool<{ param: string }, ToolResult> { } const baseConfigParams: ConfigParameters = { - apiKey: 'test-api-key', - model: 'test-model', + contentGeneratorConfig: { + model: 'test-model', + apiKey: 'test-api-key', + vertexai: false, + userAgent: 'TestAgent/1.0', + }, embeddingModel: 'test-embedding-model', sandbox: false, targetDir: '/test/dir', debugMode: false, - userAgent: 'TestAgent/1.0', userMemory: '', geminiMdFileCount: 0, approvalMode: ApprovalMode.DEFAULT, - vertexai: false, }; describe('ToolRegistry', () => { -- cgit v1.2.3