From c67f75d848c1821d351b9b6bb7fdbcebc117e565 Mon Sep 17 00:00:00 2001
From: Lorenze Jay <lorenzejaytech@gmail.com>
Date: Fri, 24 Jan 2025 09:42:09 -0800
Subject: [PATCH] better

---
 src/crewai/agents/agent_builder/base_agent.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/crewai/agents/agent_builder/base_agent.py b/src/crewai/agents/agent_builder/base_agent.py
index 87922834e..60b5717df 100644
--- a/src/crewai/agents/agent_builder/base_agent.py
+++ b/src/crewai/agents/agent_builder/base_agent.py
@@ -135,6 +135,10 @@ class BaseAgent(ABC, BaseModel):
         default=None,
         description="Knowledge sources for the agent.",
     )
+    custom_knowledge_storage: Optional[Any] = Field(
+        default=None,
+        description="Custom knowledge storage for the agent.",
+    )
 
     @model_validator(mode="before")
     @classmethod
@@ -272,9 +276,7 @@ class BaseAgent(ABC, BaseModel):
             **copied_data,
             llm=existing_llm,
             tools=self.tools,
-            knowledge_sources=self.knowledge_sources
-            if hasattr(self, "knowledge_sources")
-            else None,
+            knowledge_sources=getattr(self, "knowledge_sources", None),
         )
 
         return copied_agent