From 5b317f3eb314b48fad959710a7424a3d09cd86e7 Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Thu, 13 Feb 2025 08:28:12 +0000 Subject: [PATCH] fix: Allow extra fields in FAISSSearchTool model Co-Authored-By: Joe Moura --- src/crewai/tools/faiss_search_tool.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/crewai/tools/faiss_search_tool.py b/src/crewai/tools/faiss_search_tool.py index 401429684..6b5c01202 100644 --- a/src/crewai/tools/faiss_search_tool.py +++ b/src/crewai/tools/faiss_search_tool.py @@ -14,6 +14,8 @@ logger = logging.getLogger(__name__) class FAISSSearchTool(BaseTool): """FAISS vector similarity search tool for efficient document search.""" + model_config = {"extra": "allow"} + name: str = "FAISS Search Tool" description: str = "Search through documents using FAISS vector similarity search" embedder_config: Optional[Dict[str, Any]] = Field(default=None) @@ -21,6 +23,7 @@ class FAISSSearchTool(BaseTool): texts: List[str] = Field(default_factory=list) index_type: str = Field(default="L2") index: Any = Field(default=None) # FAISS index instance + embedder: Any = Field(default=None) # Embedder instance def __init__( self,