zai-rs 0.1.14

一个 Rust SDK, 用于调用 智普AI API
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130

//! Real-time API types

use serde::{Deserialize, Serialize};

/// Real-time API event types
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(tag = "type")]
pub enum RealTimeEvent {
    /// Audio data event
    #[serde(rename = "audio")]
    Audio { data: Vec<u8>, format: AudioFormat },

    /// Text transcription event
    #[serde(rename = "text")]
    Text { content: String, is_final: bool },

    /// Session started event
    #[serde(rename = "session_started")]
    SessionStarted { session_id: String },

    /// Session ended event
    #[serde(rename = "session_ended")]
    SessionEnded { reason: String },

    /// Error event
    #[serde(rename = "error")]
    Error { code: u16, message: String },

    /// Status update event
    #[serde(rename = "status")]
    Status { state: SessionState },
}

/// Audio format enumeration
#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
pub enum AudioFormat {
    #[serde(rename = "wav")]
    Wav,

    #[serde(rename = "mp3")]
    Mp3,

    #[serde(rename = "pcm")]
    Pcm,

    #[serde(rename = "opus")]
    Opus,
}

/// Session state
#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
pub enum SessionState {
    #[serde(rename = "connecting")]
    Connecting,

    #[serde(rename = "connected")]
    Connected,

    #[serde(rename = "listening")]
    Listening,

    #[serde(rename = "processing")]
    Processing,

    #[serde(rename = "speaking")]
    Speaking,

    #[serde(rename = "disconnected")]
    Disconnected,

    #[serde(rename = "error")]
    Error,
}

/// Session configuration
#[derive(Debug, Clone, Serialize)]
pub struct SessionConfig {
    /// Audio format for input/output
    pub audio_format: AudioFormat,

    /// Sample rate in Hz
    pub sample_rate: u32,

    /// Number of audio channels
    pub channels: u8,

    /// Enable automatic transcription
    pub enable_transcription: bool,

    /// Enable voice activity detection
    pub enable_vad: bool,

    /// Session timeout in seconds
    pub timeout_seconds: u32,
}

impl Default for SessionConfig {
    fn default() -> Self {
        Self {
            audio_format: AudioFormat::Pcm,
            sample_rate: 16000,
            channels: 1,
            enable_transcription: true,
            enable_vad: true,
            timeout_seconds: 300,
        }
    }
}

/// Session statistics
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct SessionStats {
    /// Session duration in seconds
    pub duration_seconds: u64,

    /// Number of audio packets sent
    pub packets_sent: u64,

    /// Number of audio packets received
    pub packets_received: u64,

    /// Total bytes sent
    pub bytes_sent: u64,

    /// Total bytes received
    pub bytes_received: u64,

    /// Number of transcriptions
    pub transcription_count: u64,
}