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

Add interactive Qdrant client installation prompt

Browse Source
This commit is contained in:
Lorenze Jay
2025-02-07 10:03:37 -08:00
parent 554bba8036
commit 837198ae08
1 changed files with 15 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/crewai_tools/tools/qdrant_vector_search_tool/qdrant_search_tool.py
View File

@@ -82,6 +82,21 @@ class QdrantVectorSearchTool(BaseTool):
url=self.qdrant_url, url=self.qdrant_url,
api_key=self.qdrant_api_key, api_key=self.qdrant_api_key,
) )
else:
import click
if click.confirm(
"The 'qdrant-client' package is required to use the QdrantVectorSearchTool. "
"Would you like to install it?"
):
import subprocess
subprocess.run(["uv", "add", "qdrant-client"], check=True)
else:
raise ImportError(
"The 'qdrant-client' package is required to use the QdrantVectorSearchTool. "
"Please install it with: uv add qdrant-client"
)
def _run( def _run(
self, self,
@@ -103,11 +118,6 @@ class QdrantVectorSearchTool(BaseTool):
ImportError: If qdrant-client is not installed ImportError: If qdrant-client is not installed
ValueError: If Qdrant credentials are missing ValueError: If Qdrant credentials are missing
""" """
if not QDRANT_AVAILABLE:
raise ImportError(
"The 'qdrant-client' package is required to use the QdrantVectorSearchTool. "
"Please install it with: pip install qdrant-client"
)
if not self.qdrant_url: if not self.qdrant_url:
raise ValueError("QDRANT_URL is not set") raise ValueError("QDRANT_URL is not set")