andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-24 07:38:14 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Drop print statements

Browse Source
This commit is contained in:
Brandon Hancock
2024-12-04 13:05:17 -05:00
parent 1c7700e5e2
commit cb45b971d7
1 changed files with 0 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/crewai/agents/crew_agent_executor.py
View File

@@ -87,33 +87,24 @@ class CrewAgentExecutor(CrewAgentExecutorMixin):
self.llm.stop = self.stop self.llm.stop = self.stop
def invoke(self, inputs: Dict[str, str]) -> Dict[str, Any]: def invoke(self, inputs: Dict[str, str]) -> Dict[str, Any]:
print("Invoke method called with inputs:", inputs)
if "system" in self.prompt: if "system" in self.prompt:
system_prompt = self._format_prompt(self.prompt.get("system", ""), inputs) system_prompt = self._format_prompt(self.prompt.get("system", ""), inputs)
user_prompt = self._format_prompt(self.prompt.get("user", ""), inputs) user_prompt = self._format_prompt(self.prompt.get("user", ""), inputs)
print("System prompt:", system_prompt)
print("User prompt:", user_prompt)
self.messages.append(self._format_msg(system_prompt, role="system")) self.messages.append(self._format_msg(system_prompt, role="system"))
self.messages.append(self._format_msg(user_prompt)) self.messages.append(self._format_msg(user_prompt))
else: else:
user_prompt = self._format_prompt(self.prompt.get("prompt", ""), inputs) user_prompt = self._format_prompt(self.prompt.get("prompt", ""), inputs)
print("User prompt:", user_prompt)
self.messages.append(self._format_msg(user_prompt)) self.messages.append(self._format_msg(user_prompt))
self._show_start_logs() self._show_start_logs()
self.ask_for_human_input = bool(inputs.get("ask_for_human_input", False)) self.ask_for_human_input = bool(inputs.get("ask_for_human_input", False))
print("Initial ask_for_human_input:", self.ask_for_human_input)
formatted_answer = self._invoke_loop() formatted_answer = self._invoke_loop()
print("Initial formatted answer:", formatted_answer)
while self.ask_for_human_input: while self.ask_for_human_input:
print("Entering feedback loop")
human_feedback = self._ask_human_input(formatted_answer.output) human_feedback = self._ask_human_input(formatted_answer.output)
print("Human feedback received:", human_feedback)
if self.crew and self.crew._train: if self.crew and self.crew._train:
self._handle_crew_training_output(formatted_answer, human_feedback) self._handle_crew_training_output(formatted_answer, human_feedback)
@@ -122,10 +113,6 @@ class CrewAgentExecutor(CrewAgentExecutorMixin):
additional_changes_requested_prompt = self._i18n.slice( additional_changes_requested_prompt = self._i18n.slice(
"human_feedback_classification" "human_feedback_classification"
).format(feedback=human_feedback) ).format(feedback=human_feedback)
print(
"Additional changes prompt for LLM:",
additional_changes_requested_prompt,
)
additional_changes_requested_response = ( additional_changes_requested_response = (
self.llm.call( self.llm.call(
@@ -139,27 +126,19 @@ class CrewAgentExecutor(CrewAgentExecutorMixin):
.strip() .strip()
.lower() .lower()
) )
print(
"Additional changes response from LLM:",
additional_changes_requested_response,
)
if additional_changes_requested_response == "false": if additional_changes_requested_response == "false":
print("Human is satisfied, exiting loop")
self.ask_for_human_input = False self.ask_for_human_input = False
else: else:
print("Human wants changes, continuing loop")
self.ask_for_human_input = True self.ask_for_human_input = True
self.messages.append(self._format_msg(f"Feedback: {human_feedback}")) self.messages.append(self._format_msg(f"Feedback: {human_feedback}"))
formatted_answer = self._invoke_loop() formatted_answer = self._invoke_loop()
print("Updated formatted answer:", formatted_answer)
if self.crew and self.crew._train: if self.crew and self.crew._train:
self._handle_crew_training_output(formatted_answer) self._handle_crew_training_output(formatted_answer)
self._create_short_term_memory(formatted_answer) self._create_short_term_memory(formatted_answer)
self._create_long_term_memory(formatted_answer) self._create_long_term_memory(formatted_answer)
print("Final output:", formatted_answer.output)
return {"output": formatted_answer.output} return {"output": formatted_answer.output}
def _invoke_loop(self, formatted_answer=None): def _invoke_loop(self, formatted_answer=None):