Fix ws error receive and send

This commit is contained in:
Xin Wang
2026-02-09 15:49:00 +08:00
parent fe05cf5d74
commit 8fd6daaed1
3 changed files with 34 additions and 4 deletions

View File

@@ -1,5 +1,6 @@
"""Session management for active calls."""
import asyncio
import uuid
import json
import time
@@ -78,6 +79,8 @@ class Session:
self._history_turn_index: int = 0
self._history_call_started_mono: Optional[float] = None
self._history_finalized: bool = False
self._cleanup_lock = asyncio.Lock()
self._cleaned_up = False
self.pipeline.conversation.on_turn_complete(self._on_turn_complete)
@@ -288,10 +291,15 @@ class Session:
async def cleanup(self) -> None:
"""Cleanup session resources."""
logger.info(f"Session {self.id} cleaning up")
await self._finalize_history(status="connected")
await self.pipeline.cleanup()
await self.transport.close()
async with self._cleanup_lock:
if self._cleaned_up:
return
self._cleaned_up = True
logger.info(f"Session {self.id} cleaning up")
await self._finalize_history(status="connected")
await self.pipeline.cleanup()
await self.transport.close()
async def _start_history_bridge(self, metadata: Dict[str, Any]) -> None:
"""Initialize backend history call record for this session."""