src/client/mod.rs


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

#[macro_use]
mod common;

pub use common::*;

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

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