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

cleaner refactor

Browse Source
This commit is contained in:
Lorenze Jay
2024-12-03 12:18:07 -08:00
parent 7e93285df1
commit 8b1aef9e2d
6 changed files with 24 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/crewai/knowledge/source/excel_knowledge_source.py
View File

@@ -10,14 +10,13 @@ class ExcelKnowledgeSource(BaseFileKnowledgeSource):
"""Load and preprocess Excel file content."""
pd = self._import_dependencies()
if isinstance(self.file_path, list):
file_path = self.convert_to_path(self.file_path[0])
else:
file_path = self.convert_to_path(self.file_path)
df = pd.read_excel(file_path)
content = df.to_csv(index=False)
return {file_path: content}
content_dict = {}
for file_path in self.safe_file_paths:
file_path = self.convert_to_path(file_path)
df = pd.read_excel(file_path)
content = df.to_csv(index=False)
content_dict[file_path] = content
return content_dict
def _import_dependencies(self):
"""Dynamically import dependencies."""