summaryrefslogtreecommitdiffstats
path: root/src/client/mod.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-04-28 11:27:06 +0800
committerGitHub <noreply@github.com>2024-04-28 11:27:06 +0800
commit1f2b62670379a1a96bdc602cf38b38c71deb3f06 (patch)
treed2e4235b9097f73b9e4c77c9890f0f02eb375e63 /src/client/mod.rs
parent615bab215bdc7ba156b90d5bbfdd720e5ce430a2 (diff)
downloadaichat-1f2b62670379a1a96bdc602cf38b38c71deb3f06.tar.gz
feat: support bedrock client (#450)
Diffstat (limited to 'src/client/mod.rs')
-rw-r--r--src/client/mod.rs3
1 files changed, 3 insertions, 0 deletions
diff --git a/src/client/mod.rs b/src/client/mod.rs
index 07520c2..a5aaab1 100644
--- a/src/client/mod.rs
+++ b/src/client/mod.rs
@@ -2,11 +2,13 @@
mod common;
mod message;
mod model;
+mod prompt_format;
mod reply_handler;
pub use common::*;
pub use message::*;
pub use model::*;
+pub use prompt_format::*;
pub use reply_handler::*;
register_client!(
@@ -31,6 +33,7 @@ register_client!(
AzureOpenAIClient
),
(vertexai, "vertexai", VertexAIConfig, VertexAIClient),
+ (bedrock, "bedrock", BedrockConfig, BedrockClient),
(ernie, "ernie", ErnieConfig, ErnieClient),
(qianwen, "qianwen", QianwenConfig, QianwenClient),
(moonshot, "moonshot", MoonshotConfig, MoonshotClient),