Update src/agent.py

src/agent.py

@@ -16,7 +16,11 @@ from fastapi.exceptions import RequestValidationError
 from sqlalchemy.orm import Session
 
 from .database import get_db, create_tables, CandidateEvaluation, Feedback
-from .models import *
+from .models import (
+    FitScoreRequest, CandidateRequest, FeedbackRequest,
+    FitScoreResponse, FeedbackResponse, RecalculateResponse, 
+    ComparisonResponse, HealthResponse, RootResponse
+)
 from .feedback_system import AdaptiveFeedbackSystem
 from .reinforcement_learning import ReinforcementLearningSystem
 from .advanced_learning import AdvancedLearningSystem
@@ -295,7 +299,7 @@ class FitScoreFeedbackAgent:
     def _register_endpoints(self):
         """Register all API endpoints"""
         
-        @self.app.get("/")
+        @self.app.get("/", response_model=RootResponse)
         async def root():
             """Root endpoint"""
             return {
@@ -317,7 +321,7 @@ class FitScoreFeedbackAgent:
                 ]
             }
         
-        @self.app.get("/health")
+        @self.app.get("/health", response_model=HealthResponse)
         async def health_check():
             """Health check endpoint"""
             return {
@@ -329,7 +333,7 @@ class FitScoreFeedbackAgent:
         
 
         
-        @self.app.post("/fitscore/simple")
+        @self.app.post("/fitscore/simple", response_model=FitScoreResponse)
         async def calculate_fitscore_simple(
             request: FitScoreRequest,
             db: Session = Depends(get_db)
@@ -378,7 +382,7 @@ class FitScoreFeedbackAgent:
             except Exception as e:
                 raise HTTPException(status_code=500, detail=f"Error calculating FitScore: {str(e)}")
         
-        @self.app.post("/fitscore/calculate")
+        @self.app.post("/fitscore/calculate", response_model=FitScoreResponse)
         async def calculate_fitscore(
             request: CandidateRequest,
             db: Session = Depends(get_db)
@@ -434,7 +438,7 @@ class FitScoreFeedbackAgent:
             except Exception as e:
                 raise HTTPException(status_code=500, detail=f"Error calculating FitScore: {str(e)}")
         
-        @self.app.post("/fitscore/feedback")
+        @self.app.post("/fitscore/feedback", response_model=FeedbackResponse)
         async def submit_feedback(
             request: FeedbackRequest,
             db: Session = Depends(get_db)
@@ -467,7 +471,7 @@ class FitScoreFeedbackAgent:
             except Exception as e:
                 raise HTTPException(status_code=500, detail=f"Error submitting feedback: {str(e)}")
         
-        @self.app.post("/fitscore/recalculate")
+        @self.app.post("/fitscore/recalculate", response_model=RecalculateResponse)
         async def recalculate_fitscore(
             candidate_id: str,
             job_id: str,
@@ -534,7 +538,7 @@ class FitScoreFeedbackAgent:
             except Exception as e:
                 raise HTTPException(status_code=500, detail=f"Error recalculating FitScore: {str(e)}")
         
-        @self.app.get("/fitscore/compare/{candidate_id}/{job_id}")
+        @self.app.get("/fitscore/compare/{candidate_id}/{job_id}", response_model=ComparisonResponse)
         async def compare_results(
             candidate_id: str,
             job_id: str,