Squashed 'packages/tools/' content from commit 78317b9c

git-subtree-dir: packages/tools
git-subtree-split: 78317b9c127f18bd040c1d77e3c0840cdc9a5b38

crewai_tools/adapters/pdf_embedchain_adapter.py

@@ -0,0 +1,41 @@
+from typing import Any, Optional
+
+from crewai_tools.tools.rag.rag_tool import Adapter
+
+try:
+    from embedchain import App
+    EMBEDCHAIN_AVAILABLE = True
+except ImportError:
+    EMBEDCHAIN_AVAILABLE = False
+
+
+class PDFEmbedchainAdapter(Adapter):
+    embedchain_app: Any  # Will be App when embedchain is available
+    summarize: bool = False
+    src: Optional[str] = None
+
+    def __init__(self, **data):
+        if not EMBEDCHAIN_AVAILABLE:
+            raise ImportError("embedchain is not installed. Please install it with `pip install crewai-tools[embedchain]`")
+        super().__init__(**data)
+
+    def query(self, question: str) -> str:
+        where = (
+            {"app_id": self.embedchain_app.config.id, "source": self.src}
+            if self.src
+            else None
+        )
+        result, sources = self.embedchain_app.query(
+            question, citations=True, dry_run=(not self.summarize), where=where
+        )
+        if self.summarize:
+            return result
+        return "\n\n".join([source[0] for source in sources])
+
+    def add(
+        self,
+        *args: Any,
+        **kwargs: Any,
+    ) -> None:
+        self.src = args[0] if args else None
+        self.embedchain_app.add(*args, **kwargs)