rs_openai 0.5.0

The OpenAI Rust library provides convenient access to the OpenAI API from Rust applications.
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
38
39
40
41
42
43
44
45
46
47
48
49

//! Given a prompt, the model will return one or more predicted completions, and can also return the probabilities of alternative tokens at each position.

use crate::client::OpenAI;
use crate::interfaces::completions;
use crate::shared::response_wrapper::{OpenAIError, OpenAIResponse};
use crate::shared::utils::is_stream;
use futures::Stream;
use std::pin::Pin;

pub struct Completions<'a> {
    openai: &'a OpenAI,
}

impl<'a> Completions<'a> {
    pub fn new(openai: &'a OpenAI) -> Self {
        Self { openai }
    }

    /// Creates a completion for the provided prompt and parameters.
    pub async fn create(
        &self,
        req: &completions::CreateCompletionRequest,
    ) -> OpenAIResponse<completions::CompletionResponse> {
        if is_stream(req.stream) {
            return Err(OpenAIError::InvalidArgument(
                "When stream is true, use Completions::create_with_stream".into(),
            ));
        }

        self.openai.post("/completions", req).await
    }

    /// Creates a completion for the provided prompt and parameters.
    pub async fn create_with_stream(
        &self,
        req: &completions::CreateCompletionRequest,
    ) -> Result<
        Pin<Box<dyn Stream<Item = OpenAIResponse<completions::CompletionStreamResponse>> + Send>>,
        OpenAIError,
    > {
        if !is_stream(req.stream) {
            return Err(OpenAIError::InvalidArgument(
                "When stream is false, use Completions::create".into(),
            ));
        }

        Ok(self.openai.post_stream("/completions", req).await)
    }
}