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
|
#[macro_use]
mod common;
pub mod azure_openai;
pub mod localai;
pub mod openai;
pub use common::*;
use self::azure_openai::AzureOpenAIConfig;
use self::localai::LocalAIConfig;
use self::openai::OpenAIConfig;
use crate::{
config::{Config, ModelInfo, SharedConfig},
utils::PromptKind,
};
use anyhow::{anyhow, bail, Result};
use serde::Deserialize;
use serde_json::Value;
register_role!(
("openai", OpenAI, OpenAIConfig, OpenAIClient),
("localai", LocalAI, LocalAIConfig, LocalAIClient),
(
"azure-openai",
AzureOpenAI,
AzureOpenAIConfig,
AzureOpenAIClient
),
);
|
