alex/nova-python
1
0
Fork 0
mirror of https://github.com/NovaOSS/nova-python.git synced 2024-11-25 18:23:59 +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 {
if endpoint == &Endpoints::ChatCompletion {
if [Models::Gpt3, Models::Gpt4].contains(model) {
return true;
} else {
return false;
}
}
let chat_models = [Models::Gpt3, Models::Gpt4];
let moderation_models = [Models::ModerationStable, Models::ModerationLatest];
else if endpoint == &Endpoints::Moderation {
if [Models::ModerationStable, Models::ModerationLatest].contains(model) {
return true;
} else {
return false;
match endpoint {
Endpoints::ChatCompletion => chat_models.contains(model),
Endpoints::Moderation => moderation_models.contains(model),
_ => false
}
}
false
}
fn key_is_valid(api_key: &str) -> bool {
if !api_key.starts_with("nv-") {
return false;