Struct openai_gpt_rs::args::CompletionArgs

pub struct CompletionArgs {
    pub prompt: String,
    pub model: String,
    pub suffix: String,
    pub max_tokens: u32,
    pub n: usize,
    pub temperature: f64,
}

Fields

prompt: String

model: String

suffix: String

max_tokens: u32

n: usize

temperature: f64

Implementations

impl CompletionArgs

pub fn prompt<T>(&mut self, prompt: T) -> &mut Selfwhere T: ToString,

Set the prompt for the completion

Examples found in repository

examples/completion.rs (line 19)

async fn main() {
    let mut prompt = String::new();

    print!("Enter a prompt: ");
    let _ = stdout().flush();

    stdin().read_line(&mut prompt).unwrap();

    let client = Client::new(env::var("OPENAI_API_KEY").unwrap().as_str());

    let resp = client
        .create_completion(|args| {
            args.prompt(prompt)
                .model(CompletionModels::TextDavinci3)
                .max_tokens(32)
                .n(1)
                .temperature(1.0)
        })
        .await
        .unwrap();

    let completion = resp.get_content(0).unwrap();

    println!("{}", completion);
}

pub fn model(&mut self, model: CompletionModels) -> &mut Self

Set the model to use for the completion

Examples found in repository

examples/completion.rs (line 20)

async fn main() {
    let mut prompt = String::new();

    print!("Enter a prompt: ");
    let _ = stdout().flush();

    stdin().read_line(&mut prompt).unwrap();

    let client = Client::new(env::var("OPENAI_API_KEY").unwrap().as_str());

    let resp = client
        .create_completion(|args| {
            args.prompt(prompt)
                .model(CompletionModels::TextDavinci3)
                .max_tokens(32)
                .n(1)
                .temperature(1.0)
        })
        .await
        .unwrap();

    let completion = resp.get_content(0).unwrap();

    println!("{}", completion);
}

pub fn suffix<T>(&mut self, suffix: T) -> &mut Selfwhere T: ToString,

Set the suffix for the completion

pub fn max_tokens(&mut self, max_tokens: u32) -> &mut Self

Set the maximum number of tokens for the completion

Examples found in repository

examples/completion.rs (line 21)

async fn main() {
    let mut prompt = String::new();

    print!("Enter a prompt: ");
    let _ = stdout().flush();

    stdin().read_line(&mut prompt).unwrap();

    let client = Client::new(env::var("OPENAI_API_KEY").unwrap().as_str());

    let resp = client
        .create_completion(|args| {
            args.prompt(prompt)
                .model(CompletionModels::TextDavinci3)
                .max_tokens(32)
                .n(1)
                .temperature(1.0)
        })
        .await
        .unwrap();

    let completion = resp.get_content(0).unwrap();

    println!("{}", completion);
}

pub fn n(&mut self, n: usize) -> &mut Self

Set the number of completions to return

Examples found in repository

examples/completion.rs (line 22)

async fn main() {
    let mut prompt = String::new();

    print!("Enter a prompt: ");
    let _ = stdout().flush();

    stdin().read_line(&mut prompt).unwrap();

    let client = Client::new(env::var("OPENAI_API_KEY").unwrap().as_str());

    let resp = client
        .create_completion(|args| {
            args.prompt(prompt)
                .model(CompletionModels::TextDavinci3)
                .max_tokens(32)
                .n(1)
                .temperature(1.0)
        })
        .await
        .unwrap();

    let completion = resp.get_content(0).unwrap();

    println!("{}", completion);
}

pub fn temperature(&mut self, temperature: f64) -> &mut Self

Set the temperature for the completion

Examples found in repository

examples/completion.rs (line 23)

async fn main() {
    let mut prompt = String::new();

    print!("Enter a prompt: ");
    let _ = stdout().flush();

    stdin().read_line(&mut prompt).unwrap();

    let client = Client::new(env::var("OPENAI_API_KEY").unwrap().as_str());

    let resp = client
        .create_completion(|args| {
            args.prompt(prompt)
                .model(CompletionModels::TextDavinci3)
                .max_tokens(32)
                .n(1)
                .temperature(1.0)
        })
        .await
        .unwrap();

    let completion = resp.get_content(0).unwrap();

    println!("{}", completion);
}

Trait Implementations

impl Debug for CompletionArgs

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for CompletionArgs

fn default() -> Self

Create a new CompletionArgs struct with default values