109 lines
2.8 KiB
Go
109 lines
2.8 KiB
Go
|
package models
|
||
|
|
|
||
|
|
import "time"
|
||
|
|
|
||
|
|
// Session 会话。
|
||
|
|
type Session struct {
|
||
|
|
ID string `json:"session_id"`
|
||
|
|
CreatedAt time.Time `json:"created_at"`
|
||
|
|
Config SessionConfig `json:"config"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// SessionConfig 会话配置。
|
||
|
|
type SessionConfig struct {
|
||
|
|
TTSEnabled bool `json:"tts_enabled"`
|
||
|
|
DetailLevel string `json:"detail_level"` // "low" | "high"
|
||
|
|
Language string `json:"language"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// DefaultConfig 默认会话配置。
|
||
|
|
func DefaultConfig() SessionConfig {
|
||
|
|
return SessionConfig{TTSEnabled: true, DetailLevel: "low", Language: "zh-CN"}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Message 对话消息。
|
||
|
|
type Message struct {
|
||
|
|
Role string `json:"role"` // "user" | "assistant"
|
||
|
|
Content string `json:"content"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// --- WebSocket 消息 ---
|
||
|
|
|
||
|
|
// WsQuery 客户端 query 消息。
|
||
|
|
type WsQuery struct {
|
||
|
|
Type string `json:"type"`
|
||
|
|
RequestID string `json:"request_id"`
|
||
|
|
Image string `json:"image"` // base64
|
||
|
|
Audio string `json:"audio"` // base64
|
||
|
|
MimeType string `json:"mime_type"` // 默认 "audio/pcm"
|
||
|
|
}
|
||
|
|
|
||
|
|
// WsConfig 客户端 config 消息。
|
||
|
|
type WsConfig struct {
|
||
|
|
Type string `json:"type"`
|
||
|
|
Payload struct {
|
||
|
|
TTSEnabled *bool `json:"tts_enabled,omitempty"`
|
||
|
|
DetailLevel *string `json:"detail_level,omitempty"`
|
||
|
|
Language *string `json:"language,omitempty"`
|
||
|
|
} `json:"payload"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// WsConnected 服务端 connected 消息。
|
||
|
|
type WsConnected struct {
|
||
|
|
Type string `json:"type"`
|
||
|
|
SessionID string `json:"session_id"`
|
||
|
|
ServerVersion string `json:"server_version"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// WsSTTResult 服务端 stt_result 消息。
|
||
|
|
type WsSTTResult struct {
|
||
|
|
Type string `json:"type"`
|
||
|
|
RequestID string `json:"request_id"`
|
||
|
|
Text string `json:"text"`
|
||
|
|
IsFinal bool `json:"is_final"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// WsLLMChunk 服务端 llm_chunk 消息。
|
||
|
|
type WsLLMChunk struct {
|
||
|
|
Type string `json:"type"`
|
||
|
|
RequestID string `json:"request_id"`
|
||
|
|
Delta string `json:"delta"`
|
||
|
|
Role string `json:"role"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// WsLLMDone 服务端 llm_done 消息。
|
||
|
|
type WsLLMDone struct {
|
||
|
|
Type string `json:"type"`
|
||
|
|
RequestID string `json:"request_id"`
|
||
|
|
FullText string `json:"full_text"`
|
||
|
|
TokensUsed struct {
|
||
|
|
Prompt int `json:"prompt"`
|
||
|
|
Completion int `json:"completion"`
|
||
|
|
Total int `json:"total"`
|
||
|
|
} `json:"tokens_used"`
|
||
|
|
Model string `json:"model"`
|
||
|
|
LatencyMs int64 `json:"latency_ms"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// WsTTSAudio 服务端 tts_audio 消息。
|
||
|
|
type WsTTSAudio struct {
|
||
|
|
Type string `json:"type"`
|
||
|
|
RequestID string `json:"request_id"`
|
||
|
|
Audio string `json:"audio"` // base64
|
||
|
|
MimeType string `json:"mime_type"` // "audio/mp3" 或 "audio/pcm"
|
||
|
|
IsLast bool `json:"is_last"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// WsError 服务端 error 消息。
|
||
|
|
type WsError struct {
|
||
|
|
Type string `json:"type"`
|
||
|
|
RequestID string `json:"request_id,omitempty"`
|
||
|
|
Code string `json:"code"`
|
||
|
|
Message string `json:"message"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// WsPong 服务端 pong 消息。
|
||
|
|
type WsPong struct {
|
||
|
|
Type string `json:"type"`
|
||
|
|
}
|