1 files changed, 5 insertions, 19 deletions
diff --git a/src/client/mod.rs b/src/client/mod.rs
index 5562eb6..dba049d 100644
--- a/src/client/mod.rs
+++ b/src/client/mod.rs
@@ -1,29 +1,15 @@
 #[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;
+use crate::{config::ModelInfo, repl::ReplyStreamHandler, utils::PromptKind};
 
-register_role!(
-    ("openai", OpenAI, OpenAIConfig, OpenAIClient),
-    ("localai", LocalAI, LocalAIConfig, LocalAIClient),
+register_client!(
+    (openai, "openai", OpenAI, OpenAIConfig, OpenAIClient),
+    (localai, "localai", LocalAI, LocalAIConfig, LocalAIClient),
     (
+        azure_openai,
         "azure-openai",
         AzureOpenAI,
         AzureOpenAIConfig,