updated models to have consistent json shcema file checks

Author: Will Kukkamalla

ai_feedback/models/CodeLlamaModel.py

@@ -49,7 +49,10 @@ def generate_response(
                                        or None if no valid response is returned.
         """
         if json_schema:
-            with open(json_schema, "r", encoding="utf-8") as f:
+            schema_path = Path(json_schema)
+            if not schema_path.exists():
+                raise FileNotFoundError(f"JSON schema file not found: {schema_path}")
+            with open(schema_path, "r", encoding="utf-8") as f:
                 schema = json.load(f)
         else:
             schema = None

ai_feedback/models/DeepSeekModel.py

@@ -46,7 +46,10 @@ def generate_response(
                                        or None if the response was invalid.
         """
         if json_schema:
-            with open(json_schema, "r", encoding="utf-8") as f:
+            schema_path = Path(json_schema)
+            if not schema_path.exists():
+                raise FileNotFoundError(f"JSON schema file not found: {schema_path}")
+            with open(schema_path, "r", encoding="utf-8") as f:
                 schema = json.load(f)
         else:
             schema = None

ai_feedback/models/DeepSeekV3Model.py

@@ -52,10 +52,14 @@ def generate_response(
             Optional[Tuple[str, str]]: A tuple containing the prompt and the model's response,
                                        or None if the response was invalid.
         """
-        schema = None
         if json_schema:
-            with open(json_schema) as f:
+            schema_path = Path(json_schema)
+            if not schema_path.exists():
+                raise FileNotFoundError(f"JSON schema file not found: {schema_path}")
+            with open(schema_path, "r", encoding="utf-8") as f:
                 schema = json.load(f)
+        else:
+            schema = None
 
         prompt = f"{system_instructions}\n{prompt}"
         if llama_mode == 'server':

ai_feedback/models/OpenAIModelVector.py

@@ -67,10 +67,14 @@ def generate_response(
         if not self.model:
             raise RuntimeError("Model was not created successfully.")
 
-        schema = None
         if json_schema:
-            with open(json_schema, "r") as f:
+            schema_path = Path(json_schema)
+            if not schema_path.exists():
+                raise FileNotFoundError(f"JSON schema file not found: {schema_path}")
+            with open(schema_path, "r", encoding="utf-8") as f:
                 schema = json.load(f)
+        else:
+            schema = None
 
         request = "Uploaded Files: "
         file_ids: List[str] = []