Update text model check for refresh command

ar-jan · Nov 26, 2024 · d22bd72 · d22bd72
1 parent 3b47dd7
commit d22bd72
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/llm_venice.py b/llm_venice.py
@@ -41,12 +41,7 @@ def refresh():
         )
         response.raise_for_status()
         models = response.json()["data"]
-        text_models = [
-            model["id"]
-            for model in models
-            if model.get("type", {})
-            == "text"
-        ]
+        text_models = [model["id"] for model in models if model.get("type") == "text"]
         if not text_models:
             raise click.ClickException("No text generation models found")
         path = llm.user_dir() / "llm-venice.json"