From 591c4a511bc98b116da31fe98642cbf39714d3d3 Mon Sep 17 00:00:00 2001
From: Lorenze Jay <lorenzejaytech@gmail.com>
Date: Fri, 24 Jan 2025 09:45:15 -0800
Subject: [PATCH] ensure use of other knowledge storage works

---
 src/crewai/agent.py | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/src/crewai/agent.py b/src/crewai/agent.py
index b7f0b2896..7ec36562d 100644
--- a/src/crewai/agent.py
+++ b/src/crewai/agent.py
@@ -128,10 +128,6 @@ class Agent(BaseAgent):
         default=None,
         description="Embedder configuration for the agent.",
     )
-    knowledge_sources: Optional[List[BaseKnowledgeSource]] = Field(
-        default=None,
-        description="Knowledge sources for the agent.",
-    )
     _knowledge: Optional[Knowledge] = PrivateAttr(
         default=None,
     )
@@ -169,6 +165,7 @@ class Agent(BaseAgent):
                         sources=self.knowledge_sources,
                         embedder_config=self.embedder_config,
                         collection_name=knowledge_agent_name,
+                        storage=self.custom_knowledge_storage or None,
                     )
         except (TypeError, ValueError) as e:
             raise ValueError(f"Invalid Knowledge Configuration: {str(e)}")