mirror of https://github.com/sigoden/aichat
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
61 lines
1.8 KiB
Rust
61 lines
1.8 KiB
Rust
#[macro_use]
|
|
mod common;
|
|
mod access_token;
|
|
mod message;
|
|
mod model;
|
|
mod prompt_format;
|
|
mod sse_handler;
|
|
|
|
pub use crate::utils::PromptKind;
|
|
pub use common::*;
|
|
pub use message::*;
|
|
pub use model::*;
|
|
pub use prompt_format::*;
|
|
pub use sse_handler::*;
|
|
|
|
register_client!(
|
|
(openai, "openai", OpenAIConfig, OpenAIClient),
|
|
(
|
|
openai_compatible,
|
|
"openai-compatible",
|
|
OpenAICompatibleConfig,
|
|
OpenAICompatibleClient
|
|
),
|
|
(gemini, "gemini", GeminiConfig, GeminiClient),
|
|
(claude, "claude", ClaudeConfig, ClaudeClient),
|
|
(cohere, "cohere", CohereConfig, CohereClient),
|
|
(ollama, "ollama", OllamaConfig, OllamaClient),
|
|
(
|
|
azure_openai,
|
|
"azure-openai",
|
|
AzureOpenAIConfig,
|
|
AzureOpenAIClient
|
|
),
|
|
(vertexai, "vertexai", VertexAIConfig, VertexAIClient),
|
|
(
|
|
vertexai_claude,
|
|
"vertexai-claude",
|
|
VertexAIClaudeConfig,
|
|
VertexAIClaudeClient
|
|
),
|
|
(bedrock, "bedrock", BedrockConfig, BedrockClient),
|
|
(cloudflare, "cloudflare", CloudflareConfig, CloudflareClient),
|
|
(replicate, "replicate", ReplicateConfig, ReplicateClient),
|
|
(ernie, "ernie", ErnieConfig, ErnieClient),
|
|
(qianwen, "qianwen", QianwenConfig, QianwenClient),
|
|
);
|
|
|
|
pub const OPENAI_COMPATIBLE_PLATFORMS: [(&str, &str); 11] = [
|
|
("anyscale", "https://api.endpoints.anyscale.com/v1"),
|
|
("deepinfra", "https://api.deepinfra.com/v1/openai"),
|
|
("deepseek", "https://api.deepseek.com"),
|
|
("fireworks", "https://api.fireworks.ai/inference/v1"),
|
|
("groq", "https://api.groq.com/openai/v1"),
|
|
("mistral", "https://api.mistral.ai/v1"),
|
|
("moonshot", "https://api.moonshot.cn/v1"),
|
|
("openrouter", "https://openrouter.ai/api/v1"),
|
|
("octoai", "https://text.octoai.run/v1"),
|
|
("perplexity", "https://api.perplexity.ai"),
|
|
("together", "https://api.together.xyz/v1"),
|
|
];
|