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

Improve extract thought

Browse Source
This commit is contained in:
Brandon Hancock
2025-02-25 12:02:18 -05:00
parent 5235442a5b
commit 302b6528d6
1 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/crewai/agents/parser.py
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."""