11import * as bigquery from '@codebuff/bigquery'
22import * as analytics from '@codebuff/common/analytics'
33import { TEST_USER_ID } from '@codebuff/common/constants'
4+ import {
5+ clearMockedModules ,
6+ mockModule ,
7+ } from '@codebuff/common/testing/mock-modules'
8+ import {
9+ getToolCallString ,
10+ renderToolResults ,
11+ } from '@codebuff/common/tools/utils'
412import { getInitialSessionState } from '@codebuff/common/types/session-state'
513import {
614 afterAll ,
@@ -13,28 +21,21 @@ import {
1321 mock ,
1422 spyOn ,
1523} from 'bun:test'
16- import { WebSocket } from 'ws'
24+
1725
1826// Mock imports
1927import * as checkTerminalCommandModule from '../check-terminal-command'
2028import * as requestFilesPrompt from '../find-files/request-files-prompt'
29+ import * as getDocumentationForQueryModule from '../get-documentation-for-query'
2130import * as liveUserInputs from '../live-user-inputs'
2231import * as aisdk from '../llm-apis/vercel-ai-sdk/ai-sdk'
2332import { mainPrompt } from '../main-prompt'
2433import * as processFileBlockModule from '../process-file-block'
25-
26- import {
27- clearMockedModules ,
28- mockModule ,
29- } from '@codebuff/common/testing/mock-modules'
30- import {
31- getToolCallString ,
32- renderToolResults ,
33- } from '@codebuff/common/tools/utils'
34- import { ProjectFileContext } from '@codebuff/common/util/file'
35- import * as getDocumentationForQueryModule from '../get-documentation-for-query'
3634import * as websocketAction from '../websockets/websocket-action'
3735
36+ import type { ProjectFileContext } from '@codebuff/common/util/file'
37+ import type { WebSocket } from 'ws'
38+
3839const mockAgentStream = ( streamOutput : string ) => {
3940 spyOn ( aisdk , 'promptAiSdkStream' ) . mockImplementation ( async function * ( ) {
4041 yield streamOutput
0 commit comments