mirrors/lumen
1
0
Fork 0
mirror of https://github.com/jnsahaj/lumen.git synced 2025-07-07 21:35:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(provider): add get_model method to DeepSeekProvider

Browse source
This commit is contained in:
Sahaj Jain 2025-02-26 21:34:32 +05:30
parent 5ced59291a
commit 7c7029dc14
2 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/provider/deepseek.rs
View file

@ -93,4 +93,8 @@ impl AIProvider for DeepSeekProvider {
async fn complete(&self, prompt: AIPrompt) -> Result<String, ProviderError> {
self.complete(prompt).await
}
fn get_model(&self) -> String {
self.config.model.clone()
}
}

1
src/provider/mod.rs
View file

@ -146,6 +146,7 @@ impl std::fmt::Display for LumenProvider {
LumenProvider::Claude(p) => write!(f, "Claude ({})", p.get_model()),
LumenProvider::Ollama(p) => write!(f, "Ollama ({})", p.get_model()),
LumenProvider::OpenRouter(p) => write!(f, "OpenRouter ({})", p.get_model()),
LumenProvider::DeepSeek(p) => write!(f, "DeepSeek ({})", p.get_model()),
}
}
}