summaryrefslogtreecommitdiffstats
path: root/src/client/mod.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-06-25 07:39:35 +0800
committerGitHub <noreply@github.com>2024-06-25 07:39:35 +0800
commit2fbb5271af27eee7ba3ae6d2b3c1aa0818648c8f (patch)
tree5e72e72d592025c00a46cebd0eb7012ccd765702 /src/client/mod.rs
parented71901611247d41daed8112a5106b42eb12395b (diff)
downloadaichat-2fbb5271af27eee7ba3ae6d2b3c1aa0818648c8f.tar.gz
feat: support rag-dedicated clients (jina and voyageai) (#645)
Diffstat (limited to 'src/client/mod.rs')
-rw-r--r--src/client/mod.rs11
1 files changed, 11 insertions, 0 deletions
diff --git a/src/client/mod.rs b/src/client/mod.rs
index 579c2a3..6878541 100644
--- a/src/client/mod.rs
+++ b/src/client/mod.rs
@@ -21,6 +21,12 @@ register_client!(
OpenAICompatibleConfig,
OpenAICompatibleClient
),
+ (
+ rag_dedicated,
+ "rag-dedicated",
+ RagDedicatedConfig,
+ RagDedicatedClient
+ ),
(gemini, "gemini", GeminiConfig, GeminiClient),
(claude, "claude", ClaudeConfig, ClaudeClient),
(cohere, "cohere", CohereConfig, CohereClient),
@@ -60,3 +66,8 @@ pub const OPENAI_COMPATIBLE_PLATFORMS: [(&str, &str); 13] = [
("zhipuai", "https://open.bigmodel.cn/api/paas/v4"),
("lingyiwanwu", "https://api.lingyiwanwu.com/v1"),
];
+
+pub const RAG_DEDICATED_PLATFORMS: [(&str, &str); 2] = [
+ ("jina", "https://api.jina.ai/v1"),
+ ("voyageai", "https://api.voyageai.com/v1"),
+];