Trait rig::completion::CompletionModel

pub trait CompletionModel: Clone + Send + Sync {
    type T: Send + Sync;

    // Required method
    fn completion(
        &self,
        request: CompletionRequest,
    ) -> impl Future<Output = Result<CompletionResponse<Self::T>, CompletionError>> + Send;

    // Provided methods
    fn completion_request(&self, prompt: &str) -> CompletionRequestBuilder<Self> { ... }
    fn simple_completion(
        &self,
        prompt: &str,
        chat_history: Vec<Message>,
    ) -> impl Future<Output = Result<CompletionResponse<Self::T>, CompletionError>> + Send { ... }
}

Required Associated Types

type T: Send + Sync

Required Methods

fn completion( &self, request: CompletionRequest, ) -> impl Future<Output = Result<CompletionResponse<Self::T>, CompletionError>> + Send

Provided Methods

fn completion_request(&self, prompt: &str) -> CompletionRequestBuilder<Self>

fn simple_completion( &self, prompt: &str, chat_history: Vec<Message>, ) -> impl Future<Output = Result<CompletionResponse<Self::T>, CompletionError>> + Send

Object Safety

This trait is not object safe.

Implementors

impl CompletionModel for rig::providers::cohere::CompletionModel

type T = CompletionResponse

impl CompletionModel for rig::providers::openai::CompletionModel

type T = CompletionResponse