#llm #client #api-client #github #wrapper #runner #pantry

pantry-rs

Rust API for working with the pantry project, found in https://github.com/JuliaMerz/pantry

4 releases

0.0.4 Oct 13, 2023
0.0.3 Sep 6, 2023
0.0.2 Aug 30, 2023
0.0.1 Aug 9, 2023

#45 in #runner

MIT/Apache

95KB
1.5K SLoC

pantry.rs

Rust library for the pantry llm runner.

Read the documentation at docs.rs

Usage

The API connects to your local pantry installation, so you need to download pantry first. Client library for the Pantry LLM API.

Example

It's strongly recommended that you use PantryClient and LLMSession, which are a higher level wrapper around PantryAPI.

let perms = UserPermissions {
    perm_superuser: false,
    perm_load_llm: false,
    perm_unload_llm: false,
    perm_download_llm: false,
    perm_session: true, //this is for create_session AND prompt_session
    perm_request_download: true,
    perm_request_load: true,
    perm_request_unload: true,
    perm_view_llms: true,
};

let pantry = PantryClient::register("my project name".into(), perms).await.unwrap();

// Pause here and use the UI to accept the permission request.

// The empty hashmap means we just use default parameters.
// create_session just uses the best currently running LLM. use create_session_id or _flex for
// more finegrained control
let sess = pantry.create_session(HashMap::new()).await.unwrap();

let recv = ses.prompt_session("About me: ".into(), HashMap::new()).await.unwrap();

If you aren't already running an LLM from the ui, you can use

pantry.load_llm_flex(None, None).await.unwrap();

If you want to use your existing ggml infrastructure, you can get a bare model path

let (model, path) = pantry.bare_model_flex(None, None).await.unwrap();

Dependencies

~8–19MB
~272K SLoC