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

Sanitize agent roles to ensure valid directory names (#1037)

Browse Source
This commit is contained in:
Joshua Harper
2024-08-10 08:50:38 -04:00
committed by GitHub
parent dbce944934
commit 638a8f03f0
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/crewai/memory/storage/rag_storage.py
View File

@@ -47,7 +47,7 @@ class RAGStorage(Storage):
os.environ["OPENAI_API_KEY"] = "fake" os.environ["OPENAI_API_KEY"] = "fake"
agents = crew.agents if crew else [] agents = crew.agents if crew else []
agents = [agent.role for agent in agents] agents = [self._sanitize_role(agent.role) for agent in agents]
agents = "_".join(agents) agents = "_".join(agents)
config = { config = {
@@ -77,6 +77,12 @@ class RAGStorage(Storage):
self.app.llm = FakeLLM() self.app.llm = FakeLLM()
if allow_reset: if allow_reset:
self.app.reset() self.app.reset()
def _sanitize_role(self, role: str) -> str:
"""
Sanitizes agent roles to ensure valid directory names.
"""
return role.replace('\n', '').replace(' ', '_').replace('/', '_')
def save(self, value: Any, metadata: Dict[str, Any]) -> None: def save(self, value: Any, metadata: Dict[str, Any]) -> None:
self._generate_embedding(value, metadata) self._generate_embedding(value, metadata)