Merge pull request #22 from ericmjl:update-openai

refactor(llamabot-cli): change model list attribute access
ericmjl · Nov 11, 2023 · 40ed26a · 40ed26a
2 parents 1e7badf + 4deb93f
commit 40ed26a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/llamabot/cli/configure.py b/llamabot/cli/configure.py
@@ -46,8 +46,8 @@ def default_model(model_name=None):
 
     load_dotenv(llamabotrc_path)
 
-    model_list = client.models.list()["data"]
-    available_models = [x["id"] for x in model_list if "gpt" in x["id"]]
+    model_list = client.models.list()
+    available_models = [x.id for x in model_list if "gpt" in x.id]
     available_models.sort()
 
     if model_name in available_models: