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

Adding new test for crew memory

Browse Source
This commit is contained in:
João Moura
2024-04-04 15:26:12 -03:00
parent 0d4c062487
commit 0362449fe4
5 changed files with 3440 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/crewai/agent.py
View File

@@ -198,7 +198,8 @@ class Agent(BaseModel):
self.crew._entity_memory,
)
memory = contextual_memory.build_context_for_task(task, context)
task_prompt += self.i18n.slice("memory").format(memory=memory)
if memory.strip() != "":
task_prompt += self.i18n.slice("memory").format(memory=memory)
tools = tools or self.tools
parsed_tools = self._parse_tools(tools)

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

@@ -54,7 +54,7 @@ class CrewAgentExecutor(AgentExecutor):
def _create_short_term_memory(self, output) -> None:
if (
self.crew
and self.crew_agent.memory
and self.crew.memory
and "Action: Delegate work to co-worker" not in output.log
):
memory = ShortTermMemoryItem(
@@ -67,7 +67,7 @@ class CrewAgentExecutor(AgentExecutor):
self.crew._short_term_memory.save(memory)
def _create_long_term_memory(self, output) -> None:
if self.crew and self.crew_agent.memory:
if self.crew and self.crew.memory:
ltm_agent = TaskEvaluator(self.crew_agent)
evaluation = ltm_agent.evaluate(self.task, output.log)