remove StreamChunk from ai-sdk.ts

Author: charleslien

backend/src/llm-apis/vercel-ai-sdk/ai-sdk.ts

@@ -31,17 +31,6 @@ import type {
 import type { LanguageModel } from 'ai'
 import type { z } from 'zod/v4'
 
-export type StreamChunk =
-  | {
-      type: 'text'
-      text: string
-    }
-  | {
-      type: 'reasoning'
-      text: string
-    }
-  | { type: 'error'; message: string }
-
 // TODO: We'll want to add all our models here!
 const modelToAiSDKModel = (model: Model): LanguageModel => {
   if (

backend/src/tools/stream-parser.ts

@@ -16,10 +16,12 @@ import { executeCustomToolCall, executeToolCall } from './tool-executor'
 
 import type { BatchStrReplaceState } from './batch-str-replace'
 import type { CustomToolCall, ExecuteToolCallParams } from './tool-executor'
-import type { StreamChunk } from '../llm-apis/vercel-ai-sdk/ai-sdk'
 import type { AgentTemplate } from '../templates/types'
 import type { ToolName } from '@codebuff/common/tools/constants'
 import type { CodebuffToolCall } from '@codebuff/common/tools/list'
+import type { StreamChunk } from '@codebuff/common/types/contracts/llm'
+import type { Logger } from '@codebuff/common/types/contracts/logger'
+import type { ParamsExcluding } from '@codebuff/common/types/function-params'
 import type {
   Message,
   ToolMessage,
@@ -28,8 +30,6 @@ import type { ToolResultPart } from '@codebuff/common/types/messages/content-par
 import type { PrintModeEvent } from '@codebuff/common/types/print-mode'
 import type { AgentState, Subgoal } from '@codebuff/common/types/session-state'
 import type { ProjectFileContext } from '@codebuff/common/util/file'
-import type { ParamsExcluding } from '@codebuff/common/types/function-params'
-import type { Logger } from '@codebuff/common/types/contracts/logger'
 import type { ToolCallPart } from 'ai'
 import type { WebSocket } from 'ws'
 

backend/src/xml-stream-parser.ts

@@ -7,14 +7,14 @@ import {
   toolNameParam,
 } from '@codebuff/common/tools/constants'
 
-import type { StreamChunk } from './llm-apis/vercel-ai-sdk/ai-sdk'
 import type { Model } from '@codebuff/common/old-constants'
+import type { StreamChunk } from '@codebuff/common/types/contracts/llm'
+import type { Logger } from '@codebuff/common/types/contracts/logger'
 import type {
   PrintModeError,
   PrintModeText,
   PrintModeToolCall,
 } from '@codebuff/common/types/print-mode'
-import type { Logger } from '@codebuff/common/types/contracts/logger'
 
 const toolExtractionPattern = new RegExp(
   `${startToolTag}(.*?)${endToolTag}`,