andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-02 07:42:40 +00:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

fix: Allow extra fields in FAISSSearchTool model

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-02-13 08:28:12 +00:00
parent c453a65a0a
commit 5b317f3eb3
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/crewai/tools/faiss_search_tool.py
View File

@@ -14,6 +14,8 @@ logger = logging.getLogger(__name__)
class FAISSSearchTool(BaseTool): class FAISSSearchTool(BaseTool):
"""FAISS vector similarity search tool for efficient document search.""" """FAISS vector similarity search tool for efficient document search."""
model_config = {"extra": "allow"}
name: str = "FAISS Search Tool" name: str = "FAISS Search Tool"
description: str = "Search through documents using FAISS vector similarity search" description: str = "Search through documents using FAISS vector similarity search"
embedder_config: Optional[Dict[str, Any]] = Field(default=None) embedder_config: Optional[Dict[str, Any]] = Field(default=None)
@@ -21,6 +23,7 @@ class FAISSSearchTool(BaseTool):
texts: List[str] = Field(default_factory=list) texts: List[str] = Field(default_factory=list)
index_type: str = Field(default="L2") index_type: str = Field(default="L2")
index: Any = Field(default=None) # FAISS index instance index: Any = Field(default=None) # FAISS index instance
embedder: Any = Field(default=None) # Embedder instance
def __init__( def __init__(
self, self,