remove a few instances of unused WebSocket

Author: charleslien

backend/src/main-prompt.ts

@@ -19,11 +19,9 @@ import type {
   AgentTemplateType,
   AgentOutput,
 } from '@codebuff/common/types/session-state'
-import type { WebSocket } from 'ws'
 
 export const mainPrompt = async (
   params: {
-    ws: WebSocket
     action: ClientAction<'prompt'>
 
     onResponseChunk: (chunk: string | PrintModeEvent) => void
@@ -51,7 +49,7 @@ export const mainPrompt = async (
   sessionState: SessionState
   output: AgentOutput
 }> => {
-  const { ws, action, localAgentTemplates, requestToolCall, logger } = params
+  const { action, localAgentTemplates, requestToolCall, logger } = params
 
   const {
     prompt,

backend/src/run-agent-step.ts

@@ -48,11 +48,9 @@ import type {
   AgentOutput,
 } from '@codebuff/common/types/session-state'
 import type { ProjectFileContext } from '@codebuff/common/util/file'
-import type { WebSocket } from 'ws'
 
 export const runAgentStep = async (
   params: {
-    ws: WebSocket
     userId: string | undefined
     userInputId: string
     clientSessionId: string
@@ -102,7 +100,6 @@ export const runAgentStep = async (
   messageId: string | null
 }> => {
   const {
-    ws,
     userId,
     userInputId,
     fingerprintId,
@@ -396,7 +393,6 @@ export const runAgentStep = async (
 
 export const loopAgentSteps = async (
   params: {
-    ws: WebSocket
     userInputId: string
     agentType: AgentTemplateType
     agentState: AgentState
@@ -444,7 +440,6 @@ export const loopAgentSteps = async (
   output: AgentOutput
 }> => {
   const {
-    ws,
     userInputId,
     agentType,
     agentState,
@@ -604,7 +599,6 @@ export const loopAgentSteps = async (
         } = await runProgrammaticStep({
           ...params,
           agentState: currentAgentState,
-          ws,
           template: agentTemplate,
           localAgentTemplates,
           prompt: currentPrompt,
@@ -669,7 +663,6 @@ export const loopAgentSteps = async (
         messageId,
       } = await runAgentStep({
         ...params,
-        ws,
         userId,
         userInputId,
         clientSessionId,

backend/src/tools/batch-str-replace.ts

@@ -20,7 +20,6 @@ import type {
 } from '@codebuff/common/types/function-params'
 import type { ToolResultPart } from '@codebuff/common/types/messages/content-part'
 import type { PrintModeEvent } from '@codebuff/common/types/print-mode'
-import type { WebSocket } from 'ws'
 
 export type DeferredStrReplace = {
   toolCall: CodebuffToolCall<'str_replace'>
@@ -80,7 +79,6 @@ export function getBenchifyClient(params: { logger: Logger }): Benchify | null {
 }
 
 type BatchContext = {
-  ws: WebSocket
   userInputId: string
   onResponseChunk: (chunk: string | PrintModeEvent) => void
   state: Record<string, any>
@@ -380,11 +378,10 @@ async function executeSingleStrReplace(
  * Creates a typed requestClientToolCall function for batch mode
  */
 function createRequestClientToolCall(params: {
-  ws: WebSocket
   requestToolCall: RequestToolCallFn
   userInputId: string
 }) {
-  const { ws, requestToolCall, userInputId } = params
+  const { requestToolCall, userInputId } = params
   return async (
     clientToolCall: any,
   ): Promise<CodebuffToolOutput<'str_replace'>> => {
@@ -776,7 +773,6 @@ async function applyBenchifyResultsGracefully(
 async function applyBenchifyResultSafely(params: {
   benchifyFile: { path: string; contents: string }
   benchifyDiff: string
-  ws: WebSocket
   onResponseChunk: (chunk: string | PrintModeEvent) => void
   state: Record<string, any>
   toolResults: ToolResultPart[]

backend/src/tools/handlers/tool/find-files.ts

@@ -58,7 +58,6 @@ export const handleFindFiles = ((
   > &
     ParamsExcluding<
       typeof uploadExpandedFileContextForTraining,
-      | 'ws'
       | 'messages'
       | 'system'
       | 'assistantPrompt'
@@ -79,11 +78,8 @@ export const handleFindFiles = ((
     state,
   } = params
   const { prompt } = toolCall.input
-  const { ws, fingerprintId, userId, repoId, messages } = state
+  const { fingerprintId, userId, repoId, messages } = state
 
-  if (!ws) {
-    throw new Error('Internal error for find_files: Missing WebSocket in state')
-  }
   if (!messages) {
     throw new Error('Internal error for find_files: Missing messages in state')
   }
@@ -129,7 +125,6 @@ export const handleFindFiles = ((
       if (COLLECT_FULL_FILE_CONTEXT && addedFiles.length > 0) {
         uploadExpandedFileContextForTraining({
           ...params,
-          ws,
           messages,
           system,
           assistantPrompt: prompt,
@@ -186,7 +181,6 @@ export const handleFindFiles = ((
 
 async function uploadExpandedFileContextForTraining(
   params: {
-    ws: WebSocket
     agentStepId: string
     clientSessionId: string
     fingerprintId: string
@@ -197,7 +191,6 @@ async function uploadExpandedFileContextForTraining(
   } & ParamsOf<typeof requestRelevantFilesForTraining>,
 ) {
   const {
-    ws,
     agentStepId,
     clientSessionId,
     fingerprintId,