models/__init__.py

# 聊天相关模型
from .chat import (
    FunctionCall,
    ToolCall,
    MessageContentItem,
    Message,
    ChatCompletionRequest
)

# 异常类
from .exceptions import ClientDisconnectedError

# 日志工具类
from .logging import (
    StreamToLogger,
    WebSocketConnectionManager,
    WebSocketLogHandler
)

__all__ = [
    # 聊天模型
    'FunctionCall',
    'ToolCall', 
    'MessageContentItem',
    'Message',
    'ChatCompletionRequest',
    
    # 异常
    'ClientDisconnectedError',
    
    # 日志工具
    'StreamToLogger',
    'WebSocketConnectionManager',
    'WebSocketLogHandler'
]