validate all jsonschemas

This commit is contained in:
Paul Gauthier 2023-07-02 06:36:42 -07:00
parent 0d72ee1e45
commit ecb739ed07
4 changed files with 13 additions and 1 deletions

View file

@ -1,5 +1,7 @@
import json import json
from jsonschema import Draft7Validator
from ..dump import dump # noqa: F401 from ..dump import dump # noqa: F401
from .base_coder import Coder from .base_coder import Coder
from .editblock_coder import do_replace from .editblock_coder import do_replace
@ -80,6 +82,8 @@ class EditBlockFunctionCoder(Coder):
"updated_lines" "updated_lines"
] = updated_lines ] = updated_lines
Draft7Validator.check_schema(self.functions[0])
self.gpt_prompts = EditBlockFunctionPrompts() self.gpt_prompts = EditBlockFunctionPrompts()
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)

View file

@ -1,5 +1,7 @@
import os import os
from jsonschema import Draft7Validator
from aider import diffs from aider import diffs
from ..dump import dump # noqa: F401 from ..dump import dump # noqa: F401
@ -33,6 +35,8 @@ class SingleWholeFileFunctionCoder(Coder):
] ]
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
Draft7Validator.check_schema(self.functions[0])
self.gpt_prompts = SingleWholeFileFunctionPrompts() self.gpt_prompts = SingleWholeFileFunctionPrompts()
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)

View file

@ -1,5 +1,7 @@
import os import os
from jsonschema import Draft7Validator
from aider import diffs from aider import diffs
from ..dump import dump # noqa: F401 from ..dump import dump # noqa: F401
@ -46,6 +48,8 @@ class WholeFileFunctionCoder(Coder):
] ]
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
Draft7Validator.check_schema(self.functions[0])
self.gpt_prompts = WholeFileFunctionPrompts() self.gpt_prompts = WholeFileFunctionPrompts()
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)

View file

@ -30,4 +30,4 @@ networkx==3.1
diskcache==5.6.1 diskcache==5.6.1
numpy==1.24.3 numpy==1.24.3
scipy==1.10.1 scipy==1.10.1
jsonschema==4.17.3