🔥 fix(perf): bug with is_dupe

Prismadic · Jan 10, 2024 · d264e6d · d264e6d
1 parent 491bbbc
commit d264e6d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/magnet/ic/field.py b/magnet/ic/field.py
@@ -35,7 +35,7 @@ async def on(self, category: str = 'no_category', stream: str = 'documents', cre
         self.category = category
         self.stream = stream
         try:
-            nc = await nats.connect(f'{self.server}')
+            nc = await nats.connect(f'nats://{self.server}:4222')
             self.nc = nc
             self.js = self.nc.jetstream()
             self.js.purge_stream

diff --git a/magnet/ize/memory.py b/magnet/ize/memory.py
@@ -195,7 +195,7 @@ def is_dupe(self, q):
         Returns:
             bool: True if the query embedding is a duplicate in the Milvus database, False otherwise.
         """
-        match = self.collection.search(
+        match = self.db.collection.search(
             data=[q]
             , anns_field = "embeddings"
             , param=self.config['INDEX_PARAMS']