fix: reuse existing schemas for the output

Author: HongleiGu

evaluation_function/evaluation.py

@@ -19,7 +19,7 @@
     ParseResult,
     SectionFeedback,
     SpaceComplexityResult,
-    TestCaseFeedback,
+    # TestCaseFeedback,
     TestCaseResult,
     TimeComplexityResult,
 )
@@ -424,14 +424,14 @@ def _generate_feedback(
     #     )
 
     # Test cases section
-    test_case_feedback = [
-        TestCaseFeedback(
-            index=i + 1,
-            passed=tc.passed,
-            error_message=tc.error_message if tc.error_message else ""
-        )
-        for i, tc in enumerate(test_case_results)
-    ] if test_case_results else []
+    # test_case_feedback = [
+    #     TestCaseFeedback(
+    #         index=i + 1,
+    #         passed=tc.passed,
+    #         error_message=tc.error_message if tc.error_message else ""
+    #     )
+    #     for i, tc in enumerate(test_case_results)
+    # ] if test_case_results else []
 
     # Detailed sections
     detailed_sections = [
@@ -448,6 +448,6 @@ def _generate_feedback(
         overall_message=overall_message,
         time_complexity=time_result,
         space_complexity=space_result,
-        test_cases=test_case_feedback,
+        test_cases=test_case_results,
         detailed_sections=detailed_sections
     )

evaluation_function/schemas/output_schema.py

@@ -194,11 +194,6 @@ class ComplexityAnalysis(BaseModel):
 from typing import List, Optional
 
 
-class TestCaseFeedback(BaseModel):
-    index: int
-    passed: bool
-    error_message: str = Field(default="")
-
 
 class SectionFeedback(BaseModel):
     importance: str
@@ -211,5 +206,5 @@ class EvaluationResult(BaseModel):
     overall_message: str
     time_complexity: Optional[TimeComplexityResult] = Field(default=None)
     space_complexity: Optional[SpaceComplexityResult] = Field(default=None)
-    test_cases: List[TestCaseFeedback] = Field(default_factory=list)
+    test_cases: List[TestCaseResult] = Field(default_factory=list)
     detailed_sections: List[SectionFeedback] = Field(default_factory=list)