diff options
author | flu0r1ne <flu0r1ne@flu0r1ne.net> | 2023-05-05 02:35:50 -0500 |
---|---|---|
committer | flu0r1ne <flu0r1ne@flu0r1ne.net> | 2023-05-05 02:35:50 -0500 |
commit | f6cd64a7870dafa80cc81e52ad9a8f7fe6529154 (patch) | |
tree | 9641d75b23a1d714bd4683058db24d6338e58a3a /src/gpt_chat_cli/gcli.py | |
parent | 07699e070fb44586a98926867b0168f64cef0f64 (diff) | |
download | gpt-chat-cli-f6cd64a7870dafa80cc81e52ad9a8f7fe6529154.tar.xz gpt-chat-cli-f6cd64a7870dafa80cc81e52ad9a8f7fe6529154.zip |
Add command to list models
Diffstat (limited to 'src/gpt_chat_cli/gcli.py')
-rw-r--r-- | src/gpt_chat_cli/gcli.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/src/gpt_chat_cli/gcli.py b/src/gpt_chat_cli/gcli.py index f0b754e..a9d36e4 100644 --- a/src/gpt_chat_cli/gcli.py +++ b/src/gpt_chat_cli/gcli.py @@ -10,6 +10,7 @@ from typing import Tuple from .openai_wrappers import ( create_chat_completion, + list_models, OpenAIChatResponse, OpenAIChatResponseStream, FinishReason, @@ -138,6 +139,13 @@ def main(): print(f'version {VERSION}') sys.exit(0) + if args.list_models: + + for model in list_models(): + print(model) + + sys.exit(0) + completion_args = args.completion_args COLOR_CODE = get_color_codes(no_color = not args.display_args.color) |