From 8c91e107e00d91899ddbbfdf9183d8d689fbb22f Mon Sep 17 00:00:00 2001
From: Paul Gauthier <aider@paulg.org>
Date: Wed, 14 Jun 2023 17:10:27 -0700
Subject: [PATCH] wip

---
 aider/models.py | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/aider/models.py b/aider/models.py
index 143073799..4aa501496 100644
--- a/aider/models.py
+++ b/aider/models.py
@@ -27,4 +27,15 @@ def get_model(name):
         if model.name == name:
             return model
 
-    raise ValueError(f"Unsupported model: {name}")
+    tokens = 0
+
+    model = Model(name, tokens)
+
+    if name.startswith("gpt-4-"):
+        GPT4_models.append(model)
+    elif name.startswith("gpt-3.5-"):
+        GPT35_models.append(model)
+    else:
+        raise ValueError(f"Unsupported model: {name}")
+
+    return model