From 3224f82cffa0033ef44e2acaadf610fed8ee4f56 Mon Sep 17 00:00:00 2001
From: Simon Willison <swillison@gmail.com>
Date: Sun, 3 Sep 2023 07:37:59 -0700
Subject: [PATCH] Ensure cog runs against docs/*.md as well

---
 .github/workflows/test.yml | 2 +-
 Justfile                   | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 9b5193e..da79431 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -31,7 +31,7 @@ jobs:
       run: |
         cog --check \
           -p "import sys, os; sys._called_from_test=True; os.environ['LLM_USER_PATH'] = '/tmp'" \
-          docs/**/*.md
+          docs/**/*.md docs/*.md
     - name: Run Black
       run: |
         black --check .
diff --git a/Justfile b/Justfile
index f3f4b5e..6eab0e5 100644
--- a/Justfile
+++ b/Justfile
@@ -25,7 +25,7 @@
 
 # Rebuild docs with cog
 @cog:
-  pipenv run cog -r -p "import sys, os; sys._called_from_test=True; os.environ['LLM_USER_PATH'] = '/tmp'" docs/**/*.md
+  pipenv run cog -r -p "import sys, os; sys._called_from_test=True; os.environ['LLM_USER_PATH'] = '/tmp'" docs/**/*.md docs/*.md
 
 # Serve live docs on localhost:8000
 @docs: cog