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

Weaviate hybrid search (#428)

Browse Source 
* Update tool to use hybrid search

* Set default alpha to 0.75
This commit is contained in:
Erika Shorten
2025-08-27 10:33:16 -04:00
committed by GitHub
parent 403bb7e208
commit 992cd726c4
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/crewai_tools/tools/weaviate_tool/vector_search.py
View File

@@ -41,6 +41,7 @@ class WeaviateVectorSearchTool(BaseTool):
collection_name: Optional[str] = None collection_name: Optional[str] = None
limit: Optional[int] = Field(default=3) limit: Optional[int] = Field(default=3)
headers: Optional[dict] = None headers: Optional[dict] = None
alpha: Optional[int] = Field(default=0.75)
env_vars: List[EnvVar] = [ env_vars: List[EnvVar] = [
EnvVar(name="OPENAI_API_KEY", description="OpenAI API key for embedding generation and retrieval", required=True), EnvVar(name="OPENAI_API_KEY", description="OpenAI API key for embedding generation and retrieval", required=True),
] ]
@@ -110,9 +111,10 @@ class WeaviateVectorSearchTool(BaseTool):
generative_config=self.generative_model, generative_config=self.generative_model,
) )
response = internal_docs.query.near_text( response = internal_docs.query.hybrid(
query=query, query=query,
limit=self.limit, limit=self.limit,
alpha=self.alpha
) )
json_response = "" json_response = ""
for obj in response.objects: for obj in response.objects: