26 lines
846 B
Rust
26 lines
846 B
Rust
|
use async_trait::async_trait;
|
||
|
|
|
||
|
|
use crate::ai::{ChatResponse, Message, ProviderInfo};
|
||
|
|
use crate::state::ProviderConfig;
|
||
|
|
|
||
|
|
#[async_trait]
|
||
|
|
pub trait Provider: Send + Sync {
|
||
|
|
fn name(&self) -> &str;
|
||
|
|
fn info(&self) -> ProviderInfo;
|
||
|
|
async fn chat(
|
||
|
|
&self,
|
||
|
|
messages: Vec<Message>,
|
||
|
|
config: &ProviderConfig,
|
||
|
|
) -> anyhow::Result<ChatResponse>;
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn create_provider(config: &ProviderConfig) -> Box<dyn Provider> {
|
||
|
|
match config.name.as_str() {
|
||
|
|
"anthropic" => Box::new(crate::ai::anthropic::AnthropicProvider),
|
||
|
|
"gemini" => Box::new(crate::ai::gemini::GeminiProvider),
|
||
|
|
"mistral" => Box::new(crate::ai::mistral::MistralProvider),
|
||
|
|
"ollama" => Box::new(crate::ai::ollama::OllamaProvider),
|
||
|
|
_ => Box::new(crate::ai::openai::OpenAiProvider), // default: OpenAI-compatible
|
||
|
|
}
|
||
|
|
}
|