Refactor

Author: tibor-reiss

weaviate/collections/batch/grpc_batch_objects.py

@@ -49,15 +49,14 @@ def __init__(self, connection: ConnectionV4, consistency_level: Optional[Consist
 
     def __grpc_objects(self, objects: List[_BatchObject]) -> List[batch_pb2.BatchObject]:
         def pack_vector(vector: Any) -> bytes:
-            vector_list = _get_vector_v4(vector)
-            return struct.pack("{}f".format(len(vector_list)), *vector_list)
+            return struct.pack("{}f".format(len(vector)), *vector)
 
         return [
             batch_pb2.BatchObject(
                 collection=obj.collection,
                 vector_bytes=(
                     pack_vector(obj.vector)
-                    if obj.vector is not None and not isinstance(obj.vector, dict)
+                    if obj.vector is not None and isinstance(obj.vector, list)
                     else None
                 ),
                 uuid=str(obj.uuid) if obj.uuid is not None else str(uuid_package.uuid4()),

weaviate/collections/data/data.py

@@ -363,7 +363,12 @@ async def insert_many(
             (
                 _BatchObject(
                     collection=self.name,
-                    vector=obj.vector,
+                    vector=(
+                        _get_vector_v4(obj.vector)
+                        if obj.vector is not None
+                        and not isinstance(obj.vector, dict)
+                        else obj.vector
+                    ),
                     uuid=str(obj.uuid if obj.uuid is not None else uuid_package.uuid4()),
                     properties=cast(dict, obj.properties),
                     tenant=self._tenant,