sdk: remove usage action

Author: charleslien

sdk/src/client.ts

@@ -3,18 +3,17 @@ import os from 'os'
 
 import { CODEBUFF_BINARY } from './constants'
 import { changeFile } from './tools/change-file'
+import { getFiles } from './tools/read-files'
 import { WebSocketHandler } from './websocket-client'
-import { API_KEY_ENV_VAR } from '../../common/src/constants'
 import {
   PromptResponseSchema,
   type ServerAction,
 } from '../../common/src/actions'
-import {
-  getInitialSessionState,
-  SessionState,
-} from '../../common/src/types/session-state'
-import { PrintModeEvent } from '../../common/src/types/print-mode'
-import { getFiles } from './tools/read-files'
+import { API_KEY_ENV_VAR } from '../../common/src/constants'
+import { getInitialSessionState } from '../../common/src/types/session-state'
+
+import type { PrintModeEvent } from '../../common/src/types/print-mode'
+import type { SessionState } from '../../common/src/types/session-state'
 
 type ClientToolName = 'write_file' | 'run_terminal_command'
 
@@ -93,7 +92,6 @@ export class CodebuffClient {
       readFiles: this.readFiles.bind(this),
       handleToolCall: this.handleToolCall.bind(this),
       onCostResponse: async () => {},
-      onUsageResponse: async () => {},
 
       onResponseChunk: async (action) => {
         const { userInputId, chunk } = action

sdk/src/websocket-client.ts

@@ -25,9 +25,6 @@ export type WebSocketHandlerOptions = {
   onCostResponse?: (
     action: Extract<ServerAction, { type: 'message-cost-response' }>,
   ) => Promise<void>
-  onUsageResponse?: (
-    action: Extract<ServerAction, { type: 'usage-response' }>,
-  ) => Promise<void>
 
   onResponseChunk?: (
     action: Extract<ServerAction, { type: 'response-chunk' }>,
@@ -53,7 +50,6 @@ export class WebSocketHandler {
   private readFiles: WebSocketHandlerOptionsWithDefaults['readFiles']
   private handleToolCall: WebSocketHandlerOptionsWithDefaults['handleToolCall']
   private onCostResponse: WebSocketHandlerOptionsWithDefaults['onCostResponse']
-  private onUsageResponse: WebSocketHandlerOptionsWithDefaults['onUsageResponse']
   private onResponseChunk: WebSocketHandlerOptionsWithDefaults['onResponseChunk']
   private onSubagentResponseChunk: WebSocketHandlerOptionsWithDefaults['onSubagentResponseChunk']
   private onPromptResponse: WebSocketHandlerOptionsWithDefaults['onPromptResponse']
@@ -68,7 +64,6 @@ export class WebSocketHandler {
     readFiles,
     handleToolCall,
     onCostResponse = async () => {},
-    onUsageResponse = async () => {},
 
     onResponseChunk = async () => {},
     onSubagentResponseChunk = async () => {},
@@ -88,7 +83,6 @@ export class WebSocketHandler {
     this.readFiles = readFiles
     this.handleToolCall = handleToolCall
     this.onCostResponse = onCostResponse
-    this.onUsageResponse = onUsageResponse
 
     this.onResponseChunk = onResponseChunk
     this.onSubagentResponseChunk = onSubagentResponseChunk
@@ -141,8 +135,6 @@ export class WebSocketHandler {
 
     this.cbWebSocket.subscribe('message-cost-response', this.onCostResponse)
 
-    this.cbWebSocket.subscribe('usage-response', this.onUsageResponse)
-
     // Used to handle server restarts gracefully
     this.cbWebSocket.subscribe('request-reconnect', this.onRequestReconnect)