ollama-rs 0.3.5

A Rust library for interacting with the Ollama 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

use serde::{Deserialize, Serialize};

use crate::{error::OllamaError, Ollama};

use super::LocalModel;

impl Ollama {
    pub async fn list_local_models(&self) -> crate::error::Result<Vec<LocalModel>> {
        let url = format!("{}api/tags", self.url_str());
        let builder = self.reqwest_client.get(url);

        #[cfg(feature = "headers")]
        let builder = builder.headers(self.request_headers.clone());

        let res = builder.send().await?;

        if !res.status().is_success() {
            return Err(OllamaError::Other(res.text().await?));
        }

        let res = res.bytes().await?;
        let res = serde_json::from_slice::<ListLocalModelsResponse>(&res)?;

        Ok(res.models)
    }
}

/// A response from Ollama containing a list of local models.
#[derive(Debug, Clone, Serialize, Deserialize)]
struct ListLocalModelsResponse {
    models: Vec<LocalModel>,
}