From 2fbb5271af27eee7ba3ae6d2b3c1aa0818648c8f Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Tue, 25 Jun 2024 07:39:35 +0800
Subject: feat: support rag-dedicated clients (jina and voyageai) (#645)

---
 src/client/mod.rs | 11 +++++++++++
 1 file changed, 11 insertions(+)

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

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"),
+];
-- 
cgit v1.2.3