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)}")