diff options
| author | sigoden <sigoden@gmail.com> | 2024-06-25 07:39:35 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-06-25 07:39:35 +0800 |
| commit | 2fbb5271af27eee7ba3ae6d2b3c1aa0818648c8f (patch) | |
| tree | 5e72e72d592025c00a46cebd0eb7012ccd765702 /src/client/mod.rs | |
| parent | ed71901611247d41daed8112a5106b42eb12395b (diff) | |
| download | aichat-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.rs | 11 |
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"), +]; |
