llmao 0.0.2

Large Language Model API Ops, an abstraction layer for interfacing with different LLM providers
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

use crate::Provider;

pub trait Error: std::fmt::Debug {
    fn kind(&self) -> ErrorKind;
}

/// Chat error kinds
///
/// This represents set of chat errors
#[derive(Debug, Copy, Clone, Eq, PartialEq)]
#[non_exhaustive]
pub enum ErrorKind {
    Other,
}

impl std::fmt::Display for ErrorKind {
    fn fmt(
        &self,
        f: &mut std::fmt::Formatter<'_>,
    ) -> std::fmt::Result {
        match self {
            ErrorKind::Other => write!(f, "Other"),
        }
    }
}

impl Error for ErrorKind {
    #[inline]
    fn kind(&self) -> ErrorKind {
        *self
    }
}

/// Chat trait
pub trait Chat: Provider
where
    Self::Error: Error,
{
    type Message;
    type Response;

    fn chat(
        &mut self,
        messages: &[Self::Message],
    ) -> Result<Self::Response, Self::Error>;
}