summaryrefslogtreecommitdiff
path: root/packages/cli/src/nonInteractiveCli.test.ts
diff options
context:
space:
mode:
Diffstat (limited to 'packages/cli/src/nonInteractiveCli.test.ts')
-rw-r--r--packages/cli/src/nonInteractiveCli.test.ts3
1 files changed, 1 insertions, 2 deletions
diff --git a/packages/cli/src/nonInteractiveCli.test.ts b/packages/cli/src/nonInteractiveCli.test.ts
index 389d35f2..c071dedc 100644
--- a/packages/cli/src/nonInteractiveCli.test.ts
+++ b/packages/cli/src/nonInteractiveCli.test.ts
@@ -39,7 +39,7 @@ describe('runNonInteractive', () => {
sendMessageStream: vi.fn(),
};
mockGeminiClient = {
- startChat: vi.fn().mockResolvedValue(mockChat),
+ getChat: vi.fn().mockResolvedValue(mockChat),
} as unknown as GeminiClient;
mockToolRegistry = {
getFunctionDeclarations: vi.fn().mockReturnValue([]),
@@ -80,7 +80,6 @@ describe('runNonInteractive', () => {
await runNonInteractive(mockConfig, 'Test input');
- expect(mockGeminiClient.startChat).toHaveBeenCalled();
expect(mockChat.sendMessageStream).toHaveBeenCalledWith({
message: [{ text: 'Test input' }],
config: {