An unofficial Rust library for the OpenAI API.
Warning
There may be breaking changes between versions while in alpha. See Implementation Progress.
Examples can be found in the examples
directory.
Please note that examples are not available for all the crate's functionality, PRs are appreciated to expand the coverage.
Currently, there are examples for the completions
module and the chat
module.
For other modules, refer to the tests
submodules for some reference.
// Relies on OPENAI_KEY and optionally OPENAI_BASE_URL.
let credentials = Credentials::from_env();
let messages = vec![
ChatCompletionMessage {
role: ChatCompletionMessageRole::System,
content: Some("You are a helpful assistant.".to_string()),
name: None,
function_call: None,
},
ChatCompletionMessage {
role: ChatCompletionMessageRole::User,
content: Some("Tell me a random crab fact".to_string()),
name: None,
function_call: None,
},
];
let chat_completion = ChatCompletion::builder("gpt-4o", messages.clone())
.credentials(credentials.clone())
.create()
.await
.unwrap();
let returned_message = chat_completion.choices.first().unwrap().message.clone();
// Assistant: Sure! Here's a random crab fact: ...
println!(
"{:#?}: {}",
returned_message.role,
returned_message.content.unwrap().trim()
);
██████████
Models
████████░░
Completions (Function calling is supported)
████████░░
Chat
██████████
Edits
░░░░░░░░░░
Images
█████████░
Embeddings
░░░░░░░░░░
Audio
███████░░░
Files
░░░░░░░░░░
Fine-tunes
██████████
Moderations
All contributions are welcome. Unit tests are encouraged.
Fork Notice
This package was initially developed by Valentine Briese. As the original repo was archived, this is a fork and continuation of the project.