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

Make RagTool processs-safe (#399)

Browse Source 
This commit adds a file lock to `RagTool`. As it uses Embedchain and
Chroma internally, this tool was not process-safe.
This commit is contained in:
Vini Brasil
2025-07-28 17:09:53 -03:00
committed by GitHub
parent e6ec6cc332
commit b8bd3000c6
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/crewai_tools/tools/rag/rag_tool.py
View File

@@ -1,8 +1,10 @@
import portalocker
from abc import ABC, abstractmethod from abc import ABC, abstractmethod
from typing import Any from typing import Any
from pydantic import BaseModel, ConfigDict, Field, model_validator
from crewai.tools import BaseTool from crewai.tools import BaseTool
from pydantic import BaseModel, ConfigDict, Field, model_validator
class Adapter(BaseModel, ABC): class Adapter(BaseModel, ABC):
@@ -39,10 +41,11 @@ class RagTool(BaseTool):
def _set_default_adapter(self): def _set_default_adapter(self):
if isinstance(self.adapter, RagTool._AdapterPlaceholder): if isinstance(self.adapter, RagTool._AdapterPlaceholder):
from embedchain import App from embedchain import App
from crewai_tools.adapters.embedchain_adapter import EmbedchainAdapter from crewai_tools.adapters.embedchain_adapter import EmbedchainAdapter
app = App.from_config(config=self.config) if self.config else App() with portalocker.Lock("crewai-rag-tool.lock", timeout=10):
app = App.from_config(config=self.config) if self.config else App()
self.adapter = EmbedchainAdapter( self.adapter = EmbedchainAdapter(
embedchain_app=app, summarize=self.summarize embedchain_app=app, summarize=self.summarize
) )