did a little refactoring

src/lib.rs

@@ -231,23 +231,14 @@ impl ChatResponse {
 }
 
 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 {
-        if [Models::ModerationStable, Models::ModerationLatest].contains(model) {
-            return true;
-        } else {
-            return false;
-        }
-    }
 
-    false
+    match endpoint {
+        Endpoints::ChatCompletion => chat_models.contains(model),
+        Endpoints::Moderation => moderation_models.contains(model),
+        _ => false
+    }
 }
 
 fn key_is_valid(api_key: &str) -> bool {