Detect vision capable models

simonw · Nov 27, 2024 · 518b5f8 · 518b5f8
1 parent b01ab71
commit 518b5f8
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/llm_openrouter.py b/llm_openrouter.py
@@ -29,12 +29,16 @@ def register_models(register):
     if not key:
         return
     for model_definition in get_openrouter_models():
+        vision = False
+        if "image" in model_definition["architecture"]["modality"].split("->")[0]:
+            vision = True
         register(
             OpenRouterChat(
                 model_id="openrouter/{}".format(model_definition["id"]),
                 model_name=model_definition["id"],
                 api_base="https://openrouter.ai/api/v1",
                 headers={"HTTP-Referer": "https://llm.datasette.io/", "X-Title": "LLM"},
+                vision=vision
             )
         )