alex/nova-python
1
0
Fork 0
mirror of https://github.com/NovaOSS/nova-python.git synced 2024-11-25 20:13:58 +01:00
Code Issues Packages Projects Releases Wiki Activity

did a little refactoring

Browse source
This commit is contained in:
henceiusegentoo 2023-08-11 19:40:06 +02:00
parent ccbaa4e247
commit 3a210d7b5b
1 changed files with 7 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
src/lib.rs
View file

@ -231,25 +231,16 @@ impl ChatResponse {
} }
fn model_is_compatible(endpoint: &Endpoints, model: &Models) -> bool { fn model_is_compatible(endpoint: &Endpoints, model: &Models) -> bool {
if endpoint == &Endpoints::ChatCompletion { let chat_models = [Models::Gpt3, Models::Gpt4];
if [Models::Gpt3, Models::Gpt4].contains(model) { let moderation_models = [Models::ModerationStable, Models::ModerationLatest];
return true;
} else {
return false;
}
}
else if endpoint == &Endpoints::Moderation { match endpoint {
if [Models::ModerationStable, Models::ModerationLatest].contains(model) { Endpoints::ChatCompletion => chat_models.contains(model),
return true; Endpoints::Moderation => moderation_models.contains(model),
} else { _ => false
return false;
} }
} }
false
}
fn key_is_valid(api_key: &str) -> bool { fn key_is_valid(api_key: &str) -> bool {
if !api_key.starts_with("nv-") { if !api_key.starts_with("nv-") {
return false; return false;