From 1f2b62670379a1a96bdc602cf38b38c71deb3f06 Mon Sep 17 00:00:00 2001 From: sigoden Date: Sun, 28 Apr 2024 11:27:06 +0800 Subject: feat: support bedrock client (#450) --- src/client/mod.rs | 3 +++ 1 file changed, 3 insertions(+) (limited to 'src/client/mod.rs') 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), -- cgit v1.2.3