forked from innovacion/searchbox
Rename package to searchbox
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:
@@ -4,9 +4,9 @@ import pytest
|
||||
from qdrant_client import models
|
||||
from qdrant_client.models import ScoredPoint
|
||||
|
||||
from vector_search_mcp.engine.base_engine import BaseEngine
|
||||
from vector_search_mcp.engine.qdrant_engine import QdrantEngine
|
||||
from vector_search_mcp.models import Match, MatchAny, MatchExclude, SearchRow
|
||||
from searchbox.engine.base_engine import BaseEngine
|
||||
from searchbox.engine.qdrant_engine import QdrantEngine
|
||||
from searchbox.models import Match, MatchAny, MatchExclude, SearchRow
|
||||
|
||||
|
||||
class TestQdrantEngine:
|
||||
@@ -30,10 +30,10 @@ class TestQdrantEngine:
|
||||
"""Create a QdrantEngine instance with mocked dependencies"""
|
||||
with (
|
||||
patch(
|
||||
"vector_search_mcp.engine.qdrant_engine.Settings"
|
||||
"searchbox.engine.qdrant_engine.Settings"
|
||||
) as mock_settings_class,
|
||||
patch(
|
||||
"vector_search_mcp.engine.qdrant_engine.AsyncQdrantClient"
|
||||
"searchbox.engine.qdrant_engine.AsyncQdrantClient"
|
||||
) as mock_client_class,
|
||||
):
|
||||
mock_settings_class.return_value = mock_settings
|
||||
@@ -159,7 +159,7 @@ class TestQdrantEngine:
|
||||
|
||||
def test_transform_chunk(self, qdrant_engine):
|
||||
"""Test transform_chunk method"""
|
||||
from vector_search_mcp.models import Chunk, ChunkData
|
||||
from searchbox.models import Chunk, ChunkData
|
||||
|
||||
chunk = Chunk(
|
||||
id="test-chunk-1",
|
||||
@@ -226,7 +226,7 @@ class TestQdrantEngine:
|
||||
@pytest.mark.asyncio
|
||||
async def test_upload_chunk_integration(self, qdrant_engine, mock_client):
|
||||
"""Test the complete upload_chunk workflow"""
|
||||
from vector_search_mcp.models import Chunk, ChunkData
|
||||
from searchbox.models import Chunk, ChunkData
|
||||
|
||||
# Setup mock response
|
||||
mock_response = MagicMock()
|
||||
@@ -468,10 +468,10 @@ class TestQdrantEngine:
|
||||
"""Test QdrantEngine initialization uses settings correctly"""
|
||||
with (
|
||||
patch(
|
||||
"vector_search_mcp.engine.qdrant_engine.Settings"
|
||||
"searchbox.engine.qdrant_engine.Settings"
|
||||
) as mock_settings_class,
|
||||
patch(
|
||||
"vector_search_mcp.engine.qdrant_engine.AsyncQdrantClient"
|
||||
"searchbox.engine.qdrant_engine.AsyncQdrantClient"
|
||||
) as mock_client_class,
|
||||
):
|
||||
mock_settings_class.return_value = mock_settings
|
||||
|
||||
Reference in New Issue
Block a user