Update src/agents/db_executor/db_executor.py (#2)

- Update src/agents/db_executor/db_executor.py (4aea5afe5798e66ef823976068b38410e4e65361)


Co-authored-by: Sebastian Wefers <Basti-1995@users.noreply.huggingface.co>

src/agents/db_executor/db_executor.py

@@ -11,6 +11,7 @@ from langchain_core.tools import tool
 from typing import Dict, Any
 from src.prompts import get_prompt
 from src.database.candidates import evaluate_cv_screening_decision
+from src.state.candidate import CandidateStatus, InterviewStatus, DecisionStatus
 
 
 SYSTEM_PROMPT = get_prompt(
@@ -40,6 +41,9 @@ def db_executor(query: str) -> str:
         "VoiceScreeningResult": VoiceScreeningResult,
         "InterviewScheduling": InterviewScheduling,
         "FinalDecision": FinalDecision,
+        "CandidateStatus": CandidateStatus,
+        "InterviewStatus": InterviewStatus,
+        "DecisionStatus": DecisionStatus,
     }
 
     try:
@@ -77,19 +81,3 @@ def db_executor(query: str) -> str:
 
 
 
-if __name__ == "__main__":
-    from rich.console import Console
-    from rich.panel import Panel
-
-    console = Console()
-    query = "Fetch all candidates and their status."
-    
-    console.rule("[bold magenta]DB Executor Test Run[/bold magenta]")
-    console.print(f"[cyan]Query:[/] {query}\n")
-
-    result = db_executor(query)
-
-    # 🧠 Show model result nicely
-    console.print(Panel.fit(result, title="🧠 Model Output", border_style="blue"))
-
-    console.rule("[bold green]End of Execution[/bold green]")