ensure use of other knowledge storage works

This commit is contained in:
Lorenze Jay
2025-01-24 09:45:15 -08:00
parent c67f75d848
commit 591c4a511b

View File

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