From ecb739ed07176ed48835a916afdeceab5cde2dbc Mon Sep 17 00:00:00 2001
From: Paul Gauthier <aider@paulg.org>
Date: Sun, 2 Jul 2023 06:36:42 -0700
Subject: [PATCH] validate all jsonschemas

---
 aider/coders/editblock_func_coder.py        | 4 ++++
 aider/coders/single_wholefile_func_coder.py | 4 ++++
 aider/coders/wholefile_func_coder.py        | 4 ++++
 requirements.txt                            | 2 +-
 4 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/aider/coders/editblock_func_coder.py b/aider/coders/editblock_func_coder.py
index 676952938..565460b75 100644
--- a/aider/coders/editblock_func_coder.py
+++ b/aider/coders/editblock_func_coder.py
@@ -1,5 +1,7 @@
 import json
 
+from jsonschema import Draft7Validator
+
 from ..dump import dump  # noqa: F401
 from .base_coder import Coder
 from .editblock_coder import do_replace
@@ -80,6 +82,8 @@ class EditBlockFunctionCoder(Coder):
                 "updated_lines"
             ] = updated_lines
 
+        Draft7Validator.check_schema(self.functions[0])
+
         self.gpt_prompts = EditBlockFunctionPrompts()
         super().__init__(*args, **kwargs)
 
diff --git a/aider/coders/single_wholefile_func_coder.py b/aider/coders/single_wholefile_func_coder.py
index b6368311a..f6bdd47d9 100644
--- a/aider/coders/single_wholefile_func_coder.py
+++ b/aider/coders/single_wholefile_func_coder.py
@@ -1,5 +1,7 @@
 import os
 
+from jsonschema import Draft7Validator
+
 from aider import diffs
 
 from ..dump import dump  # noqa: F401
@@ -33,6 +35,8 @@ class SingleWholeFileFunctionCoder(Coder):
     ]
 
     def __init__(self, *args, **kwargs):
+        Draft7Validator.check_schema(self.functions[0])
+
         self.gpt_prompts = SingleWholeFileFunctionPrompts()
         super().__init__(*args, **kwargs)
 
diff --git a/aider/coders/wholefile_func_coder.py b/aider/coders/wholefile_func_coder.py
index f45ef48a7..26785a87c 100644
--- a/aider/coders/wholefile_func_coder.py
+++ b/aider/coders/wholefile_func_coder.py
@@ -1,5 +1,7 @@
 import os
 
+from jsonschema import Draft7Validator
+
 from aider import diffs
 
 from ..dump import dump  # noqa: F401
@@ -46,6 +48,8 @@ class WholeFileFunctionCoder(Coder):
     ]
 
     def __init__(self, *args, **kwargs):
+        Draft7Validator.check_schema(self.functions[0])
+
         self.gpt_prompts = WholeFileFunctionPrompts()
         super().__init__(*args, **kwargs)
 
diff --git a/requirements.txt b/requirements.txt
index 8fb6723ae..8ef45c37d 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -30,4 +30,4 @@ networkx==3.1
 diskcache==5.6.1
 numpy==1.24.3
 scipy==1.10.1
-
+jsonschema==4.17.3