Fixed some bugs:

- Fixed vector search post like
- Fixed ibis query read_ids
- Fixed native datatype

CHANGELOG.md

@@ -32,6 +32,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - Fix the bug where shared artifacts are deleted when removing a component.
 - Fix compatibility issues with the latest version of pymongo.
 - Fix the query parser incompatibility with '.' symbol.
+- Fix the post like in the service vector_search.
 
 #### New Features & Functionality
 

plugins/ibis/superduper_ibis/data_backend.py

@@ -152,7 +152,15 @@ def check_ready_ids(
         conditions = []
         for key in keys:
             conditions.append(query[key].notnull())
-        docs = query.filter(*conditions).select(query.primary_id).execute()
+
+        # TODO: Hotfix, will be removed by the refactor PR
+        try:
+            docs = query.filter(*conditions).select(query.primary_id).execute()
+        except Exception as e:
+            if "Table not found" in str(e) or "Can't find table" in str(e):
+                return []
+            else:
+                raise e
         ready_ids = [doc[query.primary_id] for doc in docs]
         self._log_check_ready_ids_message(ids, ready_ids)
         return ready_ids

plugins/sqlalchemy/superduper_sqlalchemy/metadata.py

@@ -127,7 +127,7 @@ def _init_tables(self):
             'ARTIFACT_RELATIONS',
             metadata,
             Column('uuid', type_string),
-            Column('artifact_id', type_integer),
+            Column('artifact_id', type_string),
             *component_table_args,
         )
 

superduper/components/datatype.py

@@ -418,6 +418,8 @@ def get_hash(data):
             bytes_ = data.encode()
         elif isinstance(data, bytes):
             bytes_ = data
+        elif isinstance(data, Native):
+            bytes_ = str([type(data), data.x]).encode()
         else:
             raise ValueError(f'Unsupported data type: {type(data)}')
         return hashlib.sha1(bytes_).hexdigest()

superduper/vector_search/interface.py

@@ -161,7 +161,11 @@ def find_nearest_from_array(
                 service='vector_search',
                 data=h,
                 endpoint='query/search',
-                args={'vector_index': self.vector_index, 'n': n},
+                args={
+                    'vector_index': self.vector_index,
+                    'n': n,
+                    "within_ids": ",".join(within_ids) if within_ids else "",
+                },
             )
             return response['ids'], response['scores']