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: Add index field to FAISSSearchTool model

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

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

@@ -20,6 +20,7 @@ class FAISSSearchTool(BaseTool):
dimension: int = Field(default=384) # Default for BAAI/bge-small-en-v1.5 dimension: int = Field(default=384) # Default for BAAI/bge-small-en-v1.5
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
def __init__( def __init__(
self, self,