src/client/mod.rs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

#[macro_use]
mod common;

pub use common::*;

use crate::{
    config::{ModelInfo, TokensCountFactors},
    repl::ReplyStreamHandler,
    utils::PromptKind,
};

register_client!(
    (openai, "openai", OpenAI, OpenAIConfig, OpenAIClient),
    (localai, "localai", LocalAI, LocalAIConfig, LocalAIClient),
    (
        azure_openai,
        "azure-openai",
        AzureOpenAI,
        AzureOpenAIConfig,
        AzureOpenAIClient
    ),
);