andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-08 15:48:29 +00:00
Code Issues Actions 4 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 a912ed33aa
commit 4e1506f68e
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_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)
else:
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:
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)
# Making sure we only ask for it once, so disabling for the next thought loop
@@ -268,7 +268,7 @@ class CrewAgentExecutor(AgentExecutor):
return
else:
if self.crew._train:
if self.crew and self.crew._train:
self._handle_crew_training_output(output)
yield output
@@ -323,9 +323,10 @@ class CrewAgentExecutor(AgentExecutor):
agent_id = str(self.crew_agent.id)
if (
training_data := CrewTrainingHandler(TRAINING_DATA_FILE).load()
CrewTrainingHandler(TRAINING_DATA_FILE).load()
and not self.should_ask_for_human_input
):
training_data = CrewTrainingHandler(TRAINING_DATA_FILE).load()
if training_data.get(agent_id):
training_data[agent_id][self.crew._train_iteration][
"improved_output"