A8072846/searchbox
0
0
Fork 0
forked from innovacion/searchbox
Code Pull Requests Activity

Rename package to searchbox

Browse Source 
This commit renames the package from vector-search-mcp to searchbox. The
package imports and executable name are updated accordingly.
This commit is contained in:
Anibal Angulo
2025-09-27 01:44:46 +00:00
parent e162b0613b
commit cda334a45d
18 changed files with 107 additions and 107 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
tests/test_client/test_client.py
View File

@@ -4,9 +4,9 @@ from unittest.mock import AsyncMock, MagicMock
import pytest
from vector_search_mcp.client import Client
from vector_search_mcp.engine import Backend
from vector_search_mcp.models import Chunk, ChunkData, Condition, Match
from searchbox.client import Client
from searchbox.engine import Backend
from searchbox.models import Chunk, ChunkData, Match
@pytest.fixture
@@ -43,7 +43,7 @@ class TestClient:
"""Test that Client initializes correctly with backend and collection."""
# Mock the get_engine function
mock_get_engine = MagicMock(return_value=mock_engine)
monkeypatch.setattr("vector_search_mcp.client.get_engine", mock_get_engine)
monkeypatch.setattr("searchbox.client.get_engine", mock_get_engine)
client = Client(backend=Backend.QDRANT, collection="test_collection")
@@ -55,7 +55,7 @@ class TestClient:
async def test_create_index(self, mock_engine, monkeypatch):
"""Test create_index method delegates to engine."""
mock_get_engine = MagicMock(return_value=mock_engine)
monkeypatch.setattr("vector_search_mcp.client.get_engine", mock_get_engine)
monkeypatch.setattr("searchbox.client.get_engine", mock_get_engine)
client = Client(backend=Backend.QDRANT, collection="test_collection")
result = await client.create_index(size=512)
@@ -68,7 +68,7 @@ class TestClient:
"""Test create_index method handles failure."""
mock_engine.create_index.return_value = False
mock_get_engine = MagicMock(return_value=mock_engine)
monkeypatch.setattr("vector_search_mcp.client.get_engine", mock_get_engine)
monkeypatch.setattr("searchbox.client.get_engine", mock_get_engine)
client = Client(backend=Backend.QDRANT, collection="test_collection")
result = await client.create_index(size=256)
@@ -80,7 +80,7 @@ class TestClient:
async def test_upload_chunk(self, mock_engine, monkeypatch, sample_chunk):
"""Test upload_chunk method delegates to engine."""
mock_get_engine = MagicMock(return_value=mock_engine)
monkeypatch.setattr("vector_search_mcp.client.get_engine", mock_get_engine)
monkeypatch.setattr("searchbox.client.get_engine", mock_get_engine)
client = Client(backend=Backend.QDRANT, collection="documents")
result = await client.upload_chunk(sample_chunk)
@@ -93,7 +93,7 @@ class TestClient:
"""Test upload_chunk method handles failure."""
mock_engine.upload_chunk.return_value = False
mock_get_engine = MagicMock(return_value=mock_engine)
monkeypatch.setattr("vector_search_mcp.client.get_engine", mock_get_engine)
monkeypatch.setattr("searchbox.client.get_engine", mock_get_engine)
client = Client(backend=Backend.QDRANT, collection="documents")
result = await client.upload_chunk(sample_chunk)
@@ -105,7 +105,7 @@ class TestClient:
async def test_semantic_search_default_parameters(self, mock_engine, monkeypatch):
"""Test semantic_search with default parameters."""
mock_get_engine = MagicMock(return_value=mock_engine)
monkeypatch.setattr("vector_search_mcp.client.get_engine", mock_get_engine)
monkeypatch.setattr("searchbox.client.get_engine", mock_get_engine)
client = Client(backend=Backend.QDRANT, collection="search_collection")
embedding = [0.1, 0.2, 0.3, 0.4, 0.5]
@@ -121,7 +121,7 @@ class TestClient:
async def test_semantic_search_with_limit(self, mock_engine, monkeypatch):
"""Test semantic_search with custom limit."""
mock_get_engine = MagicMock(return_value=mock_engine)
monkeypatch.setattr("vector_search_mcp.client.get_engine", mock_get_engine)
monkeypatch.setattr("searchbox.client.get_engine", mock_get_engine)
client = Client(backend=Backend.QDRANT, collection="search_collection")
embedding = [0.1, 0.2, 0.3]
@@ -137,7 +137,7 @@ class TestClient:
async def test_semantic_search_with_conditions(self, mock_engine, monkeypatch):
"""Test semantic_search with filter conditions."""
mock_get_engine = MagicMock(return_value=mock_engine)
monkeypatch.setattr("vector_search_mcp.client.get_engine", mock_get_engine)
monkeypatch.setattr("searchbox.client.get_engine", mock_get_engine)
client = Client(backend=Backend.QDRANT, collection="filtered_collection")
embedding = [0.1, 0.2, 0.3, 0.4]
@@ -154,7 +154,7 @@ class TestClient:
async def test_semantic_search_with_threshold(self, mock_engine, monkeypatch):
"""Test semantic_search with similarity threshold."""
mock_get_engine = MagicMock(return_value=mock_engine)
monkeypatch.setattr("vector_search_mcp.client.get_engine", mock_get_engine)
monkeypatch.setattr("searchbox.client.get_engine", mock_get_engine)
client = Client(backend=Backend.QDRANT, collection="threshold_collection")
embedding = [0.5, 0.4, 0.3, 0.2, 0.1]
@@ -170,7 +170,7 @@ class TestClient:
async def test_semantic_search_all_parameters(self, mock_engine, monkeypatch):
"""Test semantic_search with all parameters specified."""
mock_get_engine = MagicMock(return_value=mock_engine)
monkeypatch.setattr("vector_search_mcp.client.get_engine", mock_get_engine)
monkeypatch.setattr("searchbox.client.get_engine", mock_get_engine)
client = Client(backend=Backend.QDRANT, collection="full_params_collection")
embedding = [0.2, 0.4, 0.6, 0.8, 1.0]
@@ -202,7 +202,7 @@ class TestClient:
async def test_client_integration_workflow(self, mock_engine, monkeypatch, sample_chunk):
"""Test a complete workflow: create index, upload chunk, search."""
mock_get_engine = MagicMock(return_value=mock_engine)
monkeypatch.setattr("vector_search_mcp.client.get_engine", mock_get_engine)
monkeypatch.setattr("searchbox.client.get_engine", mock_get_engine)
client = Client(backend=Backend.QDRANT, collection="workflow_test")
@@ -231,7 +231,7 @@ class TestClient:
"""Test semantic_search when no results are found."""
mock_engine.semantic_search.return_value = []
mock_get_engine = MagicMock(return_value=mock_engine)
monkeypatch.setattr("vector_search_mcp.client.get_engine", mock_get_engine)
monkeypatch.setattr("searchbox.client.get_engine", mock_get_engine)
client = Client(backend=Backend.QDRANT, collection="empty_results")
embedding = [0.1, 0.2, 0.3]
@@ -246,7 +246,7 @@ class TestClient:
def test_client_attributes_after_init(self, mock_engine, monkeypatch):
"""Test that client has the expected attributes after initialization."""
mock_get_engine = MagicMock(return_value=mock_engine)
monkeypatch.setattr("vector_search_mcp.client.get_engine", mock_get_engine)
monkeypatch.setattr("searchbox.client.get_engine", mock_get_engine)
client = Client(backend=Backend.QDRANT, collection="attr_test")