Add compat filter in list models

1 files changed, 5 insertions, 1 deletions

diff --git a/src/gpt_chat_cli/openai_wrappers.py b/src/gpt_chat_cli/openai_wrappers.py
index 6eeba4d..3e1ec06 100644
--- a/src/gpt_chat_cli/openai_wrappers.py
+++ b/src/gpt_chat_cli/openai_wrappers.py
@@ -103,6 +103,9 @@ def create_chat_completion(hist : ChatHistory, args: CompletionArguments) \
         for update in response
     )
 
+def is_compatible_model(_id : str):
+    ''' FIXME: There seems no better way to do this currently ... '''
+    return 'gpt' in _id
 
 def list_models() -> List[str]:
 
@@ -111,7 +114,8 @@ def list_models() -> List[str]:
     models = []
 
     for model in model_data["data"]:
-        models.append(model["id"])
+        if is_compatible_model(model["id"]):
+            models.append(model["id"])
 
     models.sort()