updated function

franciscojavierarceo · franciscojavierarceo · commit 8e9dcb245633 · 2024-05-15T14:18:47.000-04:00
Signed-off-by: Francisco Javier Arceo &lt;farceo@redhat.com&gt;
diff --git a/sdk/python/feast/infra/online_stores/sqlite.py b/sdk/python/feast/infra/online_stores/sqlite.py
@@ -111,7 +111,7 @@ def online_write_batch(
                 for feature_name, val in values.items():
                     vector_val = None
                     if config.online_store.vss_enabled:
-                        vector_val = get_list_val_str(val)
+                        vector_val = self._get_list_val_str(val)
                         conn.execute(
                             f"""
                                 UPDATE {_table_id(project, table)}
@@ -281,6 +281,19 @@ def teardown(
         except FileNotFoundError:
             pass
 
+    def _get_list_val_str(self, val: ValueProto) -> str:
+        if val.HasField("string_list_val"):
+            return ",".join(val.string_list_val.val)
+        elif val.HasField("bytes_list_val"):
+            return ",".join(map(str, val.bytes_list_val.val))
+        elif val.HasField("int64_list_val"):
+            return ",".join(map(str, val.int64_list_val.val))
+        elif val.HasField("float_list_val"):
+            return ",".join(map(str, val.float_list_val.val))
+        elif val.HasField("double_list_val"):
+            return ",".join(map(str, val.double_list_val.val))
+        else:
+            raise ValueError("Unsupported list value type")
 
     def retrieve_online_documents(
         self,
@@ -360,7 +373,6 @@ def retrieve_online_documents(
         return result
 
 
-
 def _initialize_conn(db_path: str):
     Path(db_path).parent.mkdir(exist_ok=True)
     return sqlite3.connect(
