seher-sdk 0.0.32

Seher SDK: agent resolution, rate-limit checks, and provider clients
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

use std::time::Duration;

use super::types::GlmUsageResponse;

const QUOTA_URL: &str = "https://open.bigmodel.cn/api/monitor/usage/quota/limit";

pub struct GlmClient;

impl GlmClient {
    /// # Errors
    ///
    /// Returns an error if the API request fails or the response cannot be parsed.
    pub async fn fetch_quota(
        api_key: &str,
    ) -> Result<GlmUsageResponse, Box<dyn std::error::Error>> {
        let client = Self::build_client()?;
        let response = client
            .get(QUOTA_URL)
            .header("Authorization", format!("Bearer {api_key}"))
            .header("Accept", "application/json")
            .send()
            .await?;

        let status = response.status();
        if !status.is_success() {
            let body = response.text().await.unwrap_or_default();
            return Err(format!("GLM API error {status}: {body}").into());
        }

        let quota: GlmUsageResponse = response.json().await?;
        Ok(quota)
    }

    fn build_client() -> Result<reqwest::Client, reqwest::Error> {
        reqwest::Client::builder()
            .timeout(Duration::from_secs(30))
            .build()
    }
}