WIP

habedi · Mar 1, 2025 · 745f4ec · 745f4ec
1 parent cc4a6a3
commit 745f4ec
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -14,6 +14,7 @@ repository = "https://github.com/habedi/easy-letters"
 python = "^3.10"
 openai = "^1.16.1"
 qdrant-client = "^1.8.2"
+numpy = "^2.2.3"
 
 [tool.poetry.group.dev.dependencies]
 poetry-dynamic-versioning = "^1.4.0"

diff --git a/tests/test_connectors.py b/tests/test_connectors.py
@@ -20,6 +20,7 @@ def test_find_similar():
     response = ranker.find_similar(embedding_to_search, collection_name, 2)
 
     # Assert
+    print(type(response))
     assert response[1].id == search_response[1]["id"]
     assert np.isclose(response[1].score, search_response[1]["score"], atol=1e-4)
     assert response[1].payload == search_response[1]["payload"]