edgequake-sdk 0.3.0

Rust SDK for the EdgeQuake RAG platform
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

//! Models resource.
//!
//! WHY: Models endpoints verified against routes.rs.

use crate::client::EdgeQuakeClient;
use crate::error::Result;
use crate::types::operations::*;

pub struct ModelsResource<'a> {
    pub(crate) client: &'a EdgeQuakeClient,
}

impl<'a> ModelsResource<'a> {
    /// `GET /api/v1/models` — returns provider catalog.
    pub async fn list(&self) -> Result<ProviderCatalog> {
        self.client.get("/api/v1/models").await
    }

    /// `GET /api/v1/models/health` — returns bare array of provider health.
    pub async fn providers_health(&self) -> Result<Vec<ProviderHealthInfo>> {
        self.client.get("/api/v1/models/health").await
    }

    /// `GET /api/v1/settings/provider/status`
    pub async fn current_provider(&self) -> Result<ProviderStatus> {
        self.client.get("/api/v1/settings/provider/status").await
    }

    /// `PUT /api/v1/settings/provider`
    pub async fn set_provider(&self, provider: &str) -> Result<ProviderStatus> {
        let body = serde_json::json!({ "provider": provider });
        self.client
            .put("/api/v1/settings/provider", Some(&body))
            .await
    }

    /// `GET /api/v1/models/llm` — List LLM models only.
    pub async fn list_llm(&self) -> Result<serde_json::Value> {
        self.client.get("/api/v1/models/llm").await
    }

    /// `GET /api/v1/models/embedding` — List embedding models only.
    pub async fn list_embedding(&self) -> Result<serde_json::Value> {
        self.client.get("/api/v1/models/embedding").await
    }

    /// `GET /api/v1/models/{provider}` — Get models for a specific provider.
    pub async fn get_provider(&self, provider: &str) -> Result<serde_json::Value> {
        self.client
            .get(&format!("/api/v1/models/{provider}"))
            .await
    }

    /// `GET /api/v1/models/{provider}/{model}` — Get specific model details.
    pub async fn get_model(
        &self,
        provider: &str,
        model: &str,
    ) -> Result<serde_json::Value> {
        self.client
            .get(&format!("/api/v1/models/{provider}/{model}"))
            .await
    }
}