mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-09 16:18:30 +00:00
Weaviate hybrid search (#428)
* Update tool to use hybrid search * Set default alpha to 0.75
This commit is contained in:
@@ -41,6 +41,7 @@ class WeaviateVectorSearchTool(BaseTool):
|
||||
collection_name: Optional[str] = None
|
||||
limit: Optional[int] = Field(default=3)
|
||||
headers: Optional[dict] = None
|
||||
alpha: Optional[int] = Field(default=0.75)
|
||||
env_vars: List[EnvVar] = [
|
||||
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,
|
||||
)
|
||||
|
||||
response = internal_docs.query.near_text(
|
||||
response = internal_docs.query.hybrid(
|
||||
query=query,
|
||||
limit=self.limit,
|
||||
alpha=self.alpha
|
||||
)
|
||||
json_response = ""
|
||||
for obj in response.objects:
|
||||
|
||||
Reference in New Issue
Block a user