feat: 新增maim_message配置,支持wss,tcp模式,支持token

This commit is contained in:
tcmofashi
2025-05-23 18:10:37 +08:00
parent d2a3f83bd6
commit 5536c5bf76
4 changed files with 76 additions and 4 deletions

View File

@@ -355,6 +355,35 @@ class ExperimentalConfig(ConfigBase):
"""是否启用PFC"""
@dataclass
class MaimMessageConfig(ConfigBase):
"""maim_message配置类"""
use_custom: bool = False
"""是否使用自定义的maim_message配置"""
host: str = "127.0.0.1"
"""主机地址"""
port: int = 8090
""""端口号"""
mode: str = "ws"
"""连接模式支持ws和tcp"""
use_wss: bool = False
"""是否使用WSS安全连接"""
cert_file: str = ""
"""SSL证书文件路径仅在use_wss=True时有效"""
key_file: str = ""
"""SSL密钥文件路径仅在use_wss=True时有效"""
auth_token: list[str] = field(default_factory=lambda: [])
"""认证令牌用于API验证为空则不启用验证"""
@dataclass
class ModelConfig(ConfigBase):
"""模型配置类"""