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

Squashed 'packages/tools/' changes from 78317b9c..0b3f00e6

Browse Source 
0b3f00e6 chore: update project version to 0.73.0 and revise uv.lock dependencies (#455)
ad19b074 feat: replace embedchain with native crewai adapter (#451)

git-subtree-dir: packages/tools
git-subtree-split: 0b3f00e67c0dae24d188c292dc99759fd1c841f7
This commit is contained in:
Greyson LaLonde
2025-09-18 23:38:08 -04:00
parent e16606672a
commit c960f26601
35 changed files with 4897 additions and 3951 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
crewai_tools/tools/csv_search_tool/csv_search_tool.py
View File

@@ -1,14 +1,10 @@
from typing import Optional, Type
try:
from embedchain.models.data_type import DataType
EMBEDCHAIN_AVAILABLE = True
except ImportError:
EMBEDCHAIN_AVAILABLE = False
from pydantic import BaseModel, Field
from ..rag.rag_tool import RagTool
from crewai_tools.rag.data_types import DataType
class FixedCSVSearchToolSchema(BaseModel):
@@ -42,15 +38,16 @@ class CSVSearchTool(RagTool):
self._generate_description()
def add(self, csv: str) -> None:
if not EMBEDCHAIN_AVAILABLE:
raise ImportError("embedchain is not installed. Please install it with `pip install crewai-tools[embedchain]`")
super().add(csv, data_type=DataType.CSV)
def _run(
self,
search_query: str,
csv: Optional[str] = None,
similarity_threshold: float | None = None,
limit: int | None = None,
) -> str:
if csv is not None:
self.add(csv)
return super()._run(query=search_query)
return super()._run(query=search_query, similarity_threshold=similarity_threshold, limit=limit)