diff options
author | flu0r1ne <flu0r1ne@flu0r1ne.net> | 2023-05-06 16:24:12 -0500 |
---|---|---|
committer | flu0r1ne <flu0r1ne@flu0r1ne.net> | 2023-05-06 16:44:21 -0500 |
commit | 53379e5c9b0b45f37cc45dfb14f76e3d9864bce7 (patch) | |
tree | 71b29826808d2e9a97bd5c91ba827de5d2208bfe /src/gpt_chat_cli | |
parent | 788f6de2e37c01920f9b5befe5825d4c9fb20be8 (diff) | |
download | gpt-chat-cli-53379e5c9b0b45f37cc45dfb14f76e3d9864bce7.tar.xz gpt-chat-cli-53379e5c9b0b45f37cc45dfb14f76e3d9864bce7.zip |
Add compat filter in list models
Diffstat (limited to 'src/gpt_chat_cli')
-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() |