mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-27 09:08:14 +00:00
fix: Initialize storage in StringKnowledgeSource
- Add storage initialization in model_post_init - Add test coverage for StringKnowledgeSource - Fixes #2150 Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
@@ -12,8 +12,12 @@ class StringKnowledgeSource(BaseKnowledgeSource):
|
||||
collection_name: Optional[str] = Field(default=None)
|
||||
|
||||
def model_post_init(self, _):
|
||||
"""Post-initialization method to validate content."""
|
||||
"""Post-initialization method to validate content and initialize storage."""
|
||||
self.validate_content()
|
||||
if self.storage is None:
|
||||
from crewai.knowledge.storage.knowledge_storage import KnowledgeStorage
|
||||
self.storage = KnowledgeStorage(collection_name=self.collection_name)
|
||||
self.storage.initialize_knowledge_storage()
|
||||
|
||||
def validate_content(self):
|
||||
"""Validate string content."""
|
||||
|
||||
Reference in New Issue
Block a user