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

fix: Update metadata type handling in KnowledgeStorage

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-02-09 19:44:22 +00:00
parent 86844ff3df
commit c149b75874
1 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/crewai/knowledge/storage/knowledge_storage.py
View File

@@ -154,13 +154,15 @@ class KnowledgeStorage(BaseKnowledgeStorage):
filtered_ids.append(doc_id) filtered_ids.append(doc_id)
# If we have no metadata at all, set it to None # If we have no metadata at all, set it to None
final_metadata: Optional[List[Dict[str, Union[str, int, float, bool]]]] = ( final_metadata: Optional[List[Dict[str, Union[str, int, float, bool]]]] = None
None if all(m is None for m in filtered_metadata) else [ if not all(m is None for m in filtered_metadata):
{k: v for k, v in m.items() if isinstance(v, (str, int, float, bool))} final_metadata = []
if m is not None else None for m in filtered_metadata:
for m in filtered_metadata if m is not None:
] filtered_m = {k: v for k, v in m.items() if isinstance(v, (str, int, float, bool))}
) final_metadata.append(filtered_m)
else:
final_metadata.append({"empty": True})
self.collection.upsert( self.collection.upsert(
documents=filtered_docs, documents=filtered_docs,