edgequake-llm 0.10.0

Multi-provider LLM abstraction library with caching, rate limiting, and cost tracking
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

//! xAI Grok discovery — STATIC strategy.
//!
//! xAI's API (`GET /v1/models`) returns minimal metadata,
//! so we rely on the static registry with cited documentation.
//!
//! Source: <https://docs.x.ai/docs/models>

use async_trait::async_trait;

use crate::discovery::registry::xai_models;
use crate::discovery::traits::ModelDiscoveryProvider;
use crate::discovery::types::{DiscoveredModel, DiscoveryStrategy};

pub struct XAIDiscovery;

impl Default for XAIDiscovery {
    fn default() -> Self {
        Self::new()
    }
}

impl XAIDiscovery {
    pub fn new() -> Self {
        Self
    }
}

#[async_trait]
impl ModelDiscoveryProvider for XAIDiscovery {
    fn provider_id(&self) -> &str {
        "xai"
    }

    fn discovery_strategy(&self) -> DiscoveryStrategy {
        DiscoveryStrategy::Static
    }

    async fn discover_models(&self) -> crate::error::Result<Vec<DiscoveredModel>> {
        Ok(xai_models())
    }
}