andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-10 00:28:31 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

string literal returns

Browse Source
This commit is contained in:
lorenzejay
2025-12-27 12:16:13 -08:00
parent b95f97fcaa
commit 0260fbb6f6
1 changed files with 13 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
lib/crewai/src/crewai/experimental/crew_agent_executor_flow.py
View File

@@ -232,7 +232,7 @@ class CrewAgentExecutorFlow(Flow[AgentReActState], CrewAgentExecutorMixin):
return "initialized" return "initialized"
@listen("force_final_answer") @listen("force_final_answer")
def force_final_answer(self) -> str: def force_final_answer(self) -> Literal["agent_finished"]:
"""Force agent to provide final answer when max iterations exceeded.""" """Force agent to provide final answer when max iterations exceeded."""
formatted_answer = handle_max_iterations_exceeded( formatted_answer = handle_max_iterations_exceeded(
formatted_answer=None, formatted_answer=None,
@@ -249,7 +249,7 @@ class CrewAgentExecutorFlow(Flow[AgentReActState], CrewAgentExecutorMixin):
return "agent_finished" return "agent_finished"
@listen("continue_reasoning") @listen("continue_reasoning")
def call_llm_and_parse(self) -> str: def call_llm_and_parse(self) -> Literal["parsed", "parser_error", "context_error"]:
"""Execute LLM call with hooks and parse the response. """Execute LLM call with hooks and parse the response.
Returns routing decision based on parsing result. Returns routing decision based on parsing result.
@@ -267,7 +267,6 @@ class CrewAgentExecutorFlow(Flow[AgentReActState], CrewAgentExecutorMixin):
response_model=self.response_model, response_model=self.response_model,
executor_context=self, executor_context=self,
) )
print(f"lorenze answer: {answer}")
# Parse the LLM response # Parse the LLM response
formatted_answer = process_llm_response(answer, self.use_stop_words) formatted_answer = process_llm_response(answer, self.use_stop_words)
@@ -305,14 +304,14 @@ class CrewAgentExecutorFlow(Flow[AgentReActState], CrewAgentExecutorMixin):
raise raise
@router(call_llm_and_parse) @router(call_llm_and_parse)
def route_by_answer_type(self) -> str: def route_by_answer_type(self) -> Literal["execute_tool", "agent_finished"]:
"""Route based on whether answer is AgentAction or AgentFinish.""" """Route based on whether answer is AgentAction or AgentFinish."""
if isinstance(self.state.current_answer, AgentAction): if isinstance(self.state.current_answer, AgentAction):
return "execute_tool" return "execute_tool"
return "agent_finished" return "agent_finished"
@listen("execute_tool") @listen("execute_tool")
def execute_tool_action(self) -> str: def execute_tool_action(self) -> Literal["tool_completed", "tool_result_is_final"]:
"""Execute the tool action and handle the result.""" """Execute the tool action and handle the result."""
try: try:
action = cast(AgentAction, self.state.current_answer) action = cast(AgentAction, self.state.current_answer)
@@ -369,19 +368,21 @@ class CrewAgentExecutorFlow(Flow[AgentReActState], CrewAgentExecutorMixin):
raise raise
@listen("initialized") @listen("initialized")
def continue_iteration(self) -> str: def continue_iteration(self) -> Literal["check_iteration"]:
"""Bridge listener that connects iteration loop back to iteration check.""" """Bridge listener that connects iteration loop back to iteration check."""
return "check_iteration" return "check_iteration"
@router(or_(initialize_reasoning, continue_iteration)) @router(or_(initialize_reasoning, continue_iteration))
def check_max_iterations(self) -> str: def check_max_iterations(
self,
) -> Literal["force_final_answer", "continue_reasoning"]:
"""Check if max iterations reached before proceeding with reasoning.""" """Check if max iterations reached before proceeding with reasoning."""
if has_reached_max_iterations(self.state.iterations, self.max_iter): if has_reached_max_iterations(self.state.iterations, self.max_iter):
return "force_final_answer" return "force_final_answer"
return "continue_reasoning" return "continue_reasoning"
@router(execute_tool_action) @router(execute_tool_action)
def increment_and_continue(self) -> str: def increment_and_continue(self) -> Literal["initialized"]:
"""Increment iteration counter and loop back for next iteration.""" """Increment iteration counter and loop back for next iteration."""
self.state.iterations += 1 self.state.iterations += 1
return "initialized" return "initialized"
@@ -415,7 +416,7 @@ class CrewAgentExecutorFlow(Flow[AgentReActState], CrewAgentExecutorMixin):
return "completed" return "completed"
@listen("parser_error") @listen("parser_error")
def recover_from_parser_error(self) -> str: def recover_from_parser_error(self) -> Literal["initialized"]:
"""Recover from output parser errors and retry.""" """Recover from output parser errors and retry."""
formatted_answer = handle_output_parser_exception( formatted_answer = handle_output_parser_exception(
e=self._last_parser_error, e=self._last_parser_error,
@@ -433,12 +434,12 @@ class CrewAgentExecutorFlow(Flow[AgentReActState], CrewAgentExecutorMixin):
return "initialized" return "initialized"
@listen("context_error") @listen("context_error")
def recover_from_context_length(self) -> str: def recover_from_context_length(self) -> Literal["initialized"]:
"""Recover from context length errors and retry.""" """Recover from context length errors and retry."""
handle_context_length( handle_context_length(
respect_context_window=self.respect_context_window, respect_context_window=self.respect_context_window,
printer=self._printer, printer=self._printer,
messages=list(self.state.messages), messages=self.state.messages,
llm=self.llm, llm=self.llm,
callbacks=self.callbacks, callbacks=self.callbacks,
i18n=self._i18n, i18n=self._i18n,
@@ -550,7 +551,7 @@ class CrewAgentExecutorFlow(Flow[AgentReActState], CrewAgentExecutorMixin):
return handle_agent_action_core( return handle_agent_action_core(
formatted_answer=formatted_answer, formatted_answer=formatted_answer,
tool_result=tool_result, tool_result=tool_result,
messages=list(self.state.messages), messages=self.state.messages,
step_callback=self.step_callback, step_callback=self.step_callback,
show_logs=self._show_logs, show_logs=self._show_logs,
) )