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

fix: fix trainig_data error (#820)

Browse Source 
* fix: fix trainig_data error

* fix: fix lack crew on agent

* fix: fix lack crew on agent executor
This commit is contained in:
Eduardo Chiarotti
2024-06-27 12:58:20 -03:00
committed by GitHub
parent c005ec3f78
commit da9cc5f097
2 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/crewai/agent.py
View File

@@ -241,7 +241,7 @@ class Agent(BaseModel):
self.agent_executor.tools_description = render_text_description(parsed_tools) self.agent_executor.tools_description = render_text_description(parsed_tools)
self.agent_executor.tools_names = self.__tools_names(parsed_tools) self.agent_executor.tools_names = self.__tools_names(parsed_tools)
if self.crew._train: if self.crew and self.crew._train:
task_prompt = self._training_handler(task_prompt=task_prompt) task_prompt = self._training_handler(task_prompt=task_prompt)
else: else:
task_prompt = self._use_trained_data(task_prompt=task_prompt) task_prompt = self._use_trained_data(task_prompt=task_prompt)

7
src/crewai/agents/executor.py
View File

@@ -250,7 +250,7 @@ class CrewAgentExecutor(AgentExecutor):
if self.should_ask_for_human_input: if self.should_ask_for_human_input:
human_feedback = self._ask_human_input(output.return_values["output"]) human_feedback = self._ask_human_input(output.return_values["output"])
if self.crew._train: if self.crew and self.crew._train:
self._handle_crew_training_output(output, human_feedback) self._handle_crew_training_output(output, human_feedback)
# Making sure we only ask for it once, so disabling for the next thought loop # Making sure we only ask for it once, so disabling for the next thought loop
@@ -268,7 +268,7 @@ class CrewAgentExecutor(AgentExecutor):
return return
else: else:
if self.crew._train: if self.crew and self.crew._train:
self._handle_crew_training_output(output) self._handle_crew_training_output(output)
yield output yield output
@@ -323,9 +323,10 @@ class CrewAgentExecutor(AgentExecutor):
agent_id = str(self.crew_agent.id) agent_id = str(self.crew_agent.id)
if ( if (
training_data := CrewTrainingHandler(TRAINING_DATA_FILE).load() CrewTrainingHandler(TRAINING_DATA_FILE).load()
and not self.should_ask_for_human_input and not self.should_ask_for_human_input
): ):
training_data = CrewTrainingHandler(TRAINING_DATA_FILE).load()
if training_data.get(agent_id): if training_data.get(agent_id):
training_data[agent_id][self.crew._train_iteration][ training_data[agent_id][self.crew._train_iteration][
"improved_output" "improved_output"