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
33
34
35
36
37

use serde::{Deserialize, Serialize};

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

use super::ModelInfo;

impl Ollama {
    /// Show details about a model including modelfile, template, parameters, license, and system prompt.
    pub async fn show_model_info(&self, model_name: String) -> crate::error::Result<ModelInfo> {
        let url = format!("{}api/show", self.url_str());
        let builder = self.reqwest_client.post(url);

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

        let res = builder
            .json(&ModelInfoRequest { model_name })
            .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::<ModelInfo>(&res)?;

        Ok(res)
    }
}

/// A show model info request to Ollama.
#[derive(Serialize, Deserialize)]
struct ModelInfoRequest {
    #[serde(rename = "name")]
    model_name: String,
}