diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/gpt_chat_cli/openai_wrappers.py | 6 |
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() |