mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-09 08:08:32 +00:00
refactor: Improve training feedback handling
- Add comprehensive docstrings - Add error handling for training feedback - Add proper type hints Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
@@ -537,16 +537,35 @@ class CrewAgentExecutor(CrewAgentExecutorMixin):
|
|||||||
def _handle_training_feedback(
|
def _handle_training_feedback(
|
||||||
self, initial_answer: AgentFinish, feedback: str
|
self, initial_answer: AgentFinish, feedback: str
|
||||||
) -> AgentFinish:
|
) -> AgentFinish:
|
||||||
"""Process feedback for training scenarios with single iteration."""
|
"""Process feedback for training scenarios with single iteration.
|
||||||
self._printer.print(
|
|
||||||
content="\nProcessing training feedback.\n",
|
Args:
|
||||||
color="yellow",
|
initial_answer (AgentFinish): The initial answer to improve
|
||||||
)
|
feedback (str): The feedback to process
|
||||||
self._handle_crew_training_output(initial_answer, feedback)
|
|
||||||
improved_answer = self._process_feedback_iteration(feedback)
|
Returns:
|
||||||
self._handle_crew_training_output(improved_answer)
|
AgentFinish: The improved answer after processing feedback
|
||||||
self.ask_for_human_input = False
|
|
||||||
return improved_answer
|
Raises:
|
||||||
|
FeedbackProcessingError: If feedback processing fails
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
self._printer.print(
|
||||||
|
content="\nProcessing training feedback.\n",
|
||||||
|
color="yellow",
|
||||||
|
)
|
||||||
|
self._handle_crew_training_output(initial_answer, feedback)
|
||||||
|
improved_answer = self._process_feedback_iteration(feedback)
|
||||||
|
self._handle_crew_training_output(improved_answer)
|
||||||
|
self.ask_for_human_input = False
|
||||||
|
return improved_answer
|
||||||
|
except Exception as e:
|
||||||
|
error_msg = f"Failed to process training feedback: {str(e)}"
|
||||||
|
self._printer.print(
|
||||||
|
content=error_msg,
|
||||||
|
color="red"
|
||||||
|
)
|
||||||
|
raise FeedbackProcessingError(error_msg, original_error=e)
|
||||||
|
|
||||||
def _handle_regular_feedback(
|
def _handle_regular_feedback(
|
||||||
self,
|
self,
|
||||||
|
|||||||
Reference in New Issue
Block a user