Skip to content

Commit

Permalink
fix: qdrant properties!
Browse files Browse the repository at this point in the history
qdrant engine had different property from pg engine we had earlier!
  • Loading branch information
amindadgar committed May 20, 2024
1 parent 83aa683 commit 04581b9
Showing 1 changed file with 5 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
from utils.query_engine.base_qdrant_engine import BaseQdrantEngine


class TestBaseEngine(TestCase):
class TestBaseQdrantEngine(TestCase):
def test_setup_vector_store_index(self):
"""
Tests that _setup_vector_store_index creates a PGVectorAccess object
Expand All @@ -19,6 +19,7 @@ def test_setup_vector_store_index(self):
testing=True,
)

expected_dbname = f"community_{community_id}"
self.assertIn(expected_dbname, base_engine.vector_store.connection_string)
self.assertEqual(base_engine.vector_store.table_name, platform_table_name)
expected_collection_name = f"{community_id}_{platform_table_name}"
self.assertEqual(
base_engine.vector_store.collection_name, expected_collection_name
)

0 comments on commit 04581b9

Please sign in to comment.