polish

Author: linyuan.yang

packages/chat-ui/src/WebSocketTransport.ts

@@ -37,28 +37,26 @@ export class WebSocketTransport implements IChatTransport {
 
   connect(): void {
     if (this.ws?.readyState === WebSocket.CONNECTING || this.ws?.readyState === WebSocket.OPEN) return
+    if (this.reconnectTimer) { clearTimeout(this.reconnectTimer); this.reconnectTimer = null }
     const url = this._baseUrl ? new URL(this._baseUrl) : location
     const proto = url.protocol === 'https:' ? 'wss:' : 'ws:'
-    const host = url.host
-    const ws = new WebSocket(`${proto}//${host}/ws/chat`)
+    const ws = new WebSocket(`${proto}//${url.host}/ws/chat`)
     this.ws = ws
     ws.onopen = () => {
       this.emit({ type: ChatEventType.ConnectionStatus, online: true } as ChatEvent)
     }
     ws.onmessage = (e: MessageEvent) => {
       let msg: any
       try { msg = JSON.parse(e.data as string) } catch { return }
-      const type = msg.type as string
-      if (SESSION_EVENT_TYPES.has(type)) {
-        this.emit({ type, sessionId: msg.sessionId, data: msg.data } as ChatEvent)
+      if (msg && SESSION_EVENT_TYPES.has(msg.type)) {
+        this.emit({ type: msg.type, sessionId: msg.sessionId, data: msg.data } as ChatEvent)
       }
     }
     ws.onclose = () => {
       if (this.ws === ws) this.ws = null
       this.emit({ type: ChatEventType.ConnectionStatus, online: false } as ChatEvent)
       this.reconnectTimer = setTimeout(() => this.connect(), 3000)
     }
-    ws.onerror = () => {}
   }
 
   disconnect(): void {

packages/chat-ui/src/components/RichInput.vue

@@ -28,7 +28,7 @@ function fileToDataUrl(file: File): Promise<string> {
   })
 }
 
-function handleImageDrop(view: any, event: DragEvent, _slice: any, moved: boolean): boolean {
+function handleImageDrop(_view: any, event: DragEvent, _slice: any, moved: boolean): boolean {
   if (moved || !event.dataTransfer?.files?.length) return false
   const allFiles = Array.from(event.dataTransfer.files)
   const images = allFiles.filter(f => f.type.startsWith('image/'))
@@ -44,7 +44,7 @@ function handleImageDrop(view: any, event: DragEvent, _slice: any, moved: boolea
   return true
 }
 
-function handleImagePaste(view: any, event: ClipboardEvent): boolean {
+function handleImagePaste(_view: any, event: ClipboardEvent): boolean {
   const items = Array.from(event.clipboardData?.items ?? [])
   const fileItems = items.filter(i => i.kind === 'file')
   if (fileItems.length === 0) return false

packages/chat-ui/src/components/ThinkDrawer.vue

@@ -55,10 +55,6 @@ function close() {
   thinkStack.value = []
 }
 
-function onNestedThink(thinkId: string) {
-  open(thinkId)
-}
-
 defineExpose({ open })
 </script>
 
@@ -94,7 +90,7 @@ defineExpose({ open })
                 :thinks-url-prefix="thinksUrlPrefix"
                 :labels="labels"
                 :fetch-fn="fetchFn"
-                :on-think-click="onNestedThink"
+                :on-think-click="open"
               />
             </template>
           </div>

packages/chat-ui/src/components/ToolApprovalBar.vue

@@ -18,23 +18,23 @@ const countdown = ref(props.initialCountdown)
 const argsExpanded = ref(false)
 let timer: ReturnType<typeof setInterval> | null = null
 
+function stopTimer() {
+  if (timer) { clearInterval(timer); timer = null }
+}
+
 function startTimer() {
   stopTimer()
   countdown.value = props.initialCountdown
   timer = setInterval(() => { if (countdown.value > 0) countdown.value-- }, 1000)
 }
 
-function stopTimer() {
-  if (timer !== null) { clearInterval(timer); timer = null }
-}
-
 function approve(type: ToolApprovalType) {
   stopTimer()
   emit('approve', { approvalId: props.toolCall.approvalId, approval: type })
 }
 
 function formatArgVal(v: unknown): string {
-  if (v === null || v === undefined) return 'null'
+  if (v == null) return 'null'
   if (typeof v === 'string') return v.length > 80 ? v.slice(0, 80) + '…' : v
   if (typeof v === 'number' || typeof v === 'boolean') return String(v)
   if (Array.isArray(v)) return `[${v.length} items]`

packages/chat-ui/src/types.ts

@@ -22,6 +22,9 @@ export interface MemoryOption { id: string; name: string }
 
 // ── Message types ──
 
+/** Multimodal message content: a plain string or a list of content parts (text/image_url/…). */
+export type DisplayContent = string | any[]
+
 export enum MessageRole {
   Human = 'human',
   AI    = 'ai',
@@ -36,7 +39,7 @@ export interface ToolCall {
 
 export interface ChatMessage {
   role: MessageRole | string
-  content?: string | any[]
+  content?: DisplayContent
   tool_calls?: ToolCall[]
   tool_call_id?: string
   name?: string
@@ -232,9 +235,6 @@ export interface AskAnswerPayload {
   answers: Record<string, string | string[]>
 }
 
-// ── Queued messages ──
-export type DisplayContent = string | any[]
-
 // ── Directory browser ──
 export interface DirListResult {
   path: string