From 51ba09691636fc2f86c809285a457539e357e803 Mon Sep 17 00:00:00 2001
From: "Paul Gauthier (aider)" <aider@paulg.org>
Date: Mon, 30 Sep 2024 21:15:09 -0700
Subject: [PATCH] feat: implement cog to fetch and process model data

---
 aider/website/docs/more/infinite-output.md | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/aider/website/docs/more/infinite-output.md b/aider/website/docs/more/infinite-output.md
index 53b6032e3..3392d4d39 100644
--- a/aider/website/docs/more/infinite-output.md
+++ b/aider/website/docs/more/infinite-output.md
@@ -39,8 +39,21 @@ Aider supports "infinite output" for models that support "prefill",
 such as:
 
 <!--[[[cog
-# TODO: list the models
-cog.out(text)
+import requests
+import json
+
+# Fetch the JSON data
+url = "https://raw.githubusercontent.com/BerriAI/litellm/refs/heads/main/model_prices_and_context_window.json"
+response = requests.get(url)
+data = json.loads(response.text)
+
+# Process the JSON to find models with supports_assistant_prefill=true
+prefill_models = [model for model, info in data.items() if info.get('supports_assistant_prefill') == True]
+
+# Generate the list of models
+model_list = "\n".join(f"- {model}" for model in prefill_models)
+
+cog.out(f"Models that support prefill:\n\n{model_list}")
 ]]]-->
 
 <!--[[[end]]]-->