andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-28 01:28:14 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

consolodation and improvements

Browse Source
This commit is contained in:
Lorenze Jay
2024-11-18 13:52:33 -08:00
parent b2c06d5b7a
commit cbfcde73ec
11 changed files with 248 additions and 320 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/crewai/knowledge/knowledge.py
View File

@@ -15,12 +15,13 @@ class Knowledge(BaseModel):
model_config = ConfigDict(arbitrary_types_allowed=True)
agents: List[str] = Field(default_factory=list)
storage: KnowledgeStorage = Field(default_factory=KnowledgeStorage)
embedder_config: Optional[Dict[str, Any]] = Field(default_factory=None)
def __init__(self, **data):
super().__init__(**data)
# Call add on all sources during initialization
for source in self.sources:
source.add(self.embedder)
embedder_config = data.get("embedder_config", None)
if embedder_config:
self.storage = KnowledgeStorage(embedder_config=embedder_config)
def query(
self, query: List[str], limit: int = 3, preference: Optional[str] = None