From da3c541b681b140feb52b4297167ca233535fa90 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Wed, 1 Nov 2023 15:01:50 +0800
Subject: refactor: add/use openai_compatible_client macro

---
 src/client/mod.rs | 24 +++++-------------------
 1 file changed, 5 insertions(+), 19 deletions(-)

(limited to 'src/client/mod.rs')

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,
-- 
cgit v1.2.3