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

checking crew before using memory

Browse Source
This commit is contained in:
João Moura
2024-04-03 05:03:43 -03:00
parent 33dfcc700b
commit 2d97177eff
2 changed files with 3 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

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

7
src/crewai/crew.py
View File

@@ -1,8 +1,5 @@
import json import json
import subprocess
import sys
import uuid import uuid
from pathlib import Path
from typing import Any, Dict, List, Optional, Union from typing import Any, Dict, List, Optional, Union
from langchain_core.callbacks import BaseCallbackHandler from langchain_core.callbacks import BaseCallbackHandler
@@ -158,10 +155,6 @@ class Crew(BaseModel):
def create_crew_memory(self) -> "Crew": def create_crew_memory(self) -> "Crew":
"""Set private attributes.""" """Set private attributes."""
if self.memory: if self.memory:
storage_dir = Path(".db")
storage_dir.mkdir(exist_ok=True)
if sys.platform.startswith("win"):
subprocess.call(["attrib", "+H", str(storage_dir)])
self._long_term_memory = LongTermMemory() self._long_term_memory = LongTermMemory()
self._short_term_memory = ShortTermMemory(embedder_config=self.embedder) self._short_term_memory = ShortTermMemory(embedder_config=self.embedder)
self._entity_memory = EntityMemory(embedder_config=self.embedder) self._entity_memory = EntityMemory(embedder_config=self.embedder)