Improve extract thought (#2223)

Co-authored-by: Lorenze Jay <63378463+lorenzejay@users.noreply.github.com>
This commit is contained in:
Brandon Hancock (bhancock_ai)
2025-02-26 14:51:46 -05:00
committed by GitHub
parent b58253cacc
commit b9b625a70d

View File

@@ -124,14 +124,15 @@ class CrewAgentParser:
) )
def _extract_thought(self, text: str) -> str: def _extract_thought(self, text: str) -> str:
regex = r"(.*?)(?:\n\nAction|\n\nFinal Answer)" thought_index = text.find("\n\nAction")
thought_match = re.search(regex, text, re.DOTALL) if thought_index == -1:
if thought_match: thought_index = text.find("\n\nFinal Answer")
thought = thought_match.group(1).strip() if thought_index == -1:
# Remove any triple backticks from the thought string return ""
thought = thought.replace("```", "").strip() thought = text[:thought_index].strip()
return thought # Remove any triple backticks from the thought string
return "" thought = thought.replace("```", "").strip()
return thought
def _clean_action(self, text: str) -> str: def _clean_action(self, text: str) -> str:
"""Clean action string by removing non-essential formatting characters.""" """Clean action string by removing non-essential formatting characters."""