added lint
This commit is contained in:
parent
982f0831a4
commit
19c6aaa53d
1 changed files with 41 additions and 41 deletions
|
|
@ -341,47 +341,47 @@ class FalkorDBAdapter(VectorDBInterface, GraphDBInterface):
|
|||
|
||||
return collection_name in collections
|
||||
|
||||
async def create_data_points(self, data_points: list[DataPoint]):
|
||||
"""
|
||||
Add a list of data points to the graph database via batching.
|
||||
|
||||
Can raise exceptions if there are issues during the database operations.
|
||||
|
||||
Parameters:
|
||||
-----------
|
||||
|
||||
- data_points (list[DataPoint]): A list of DataPoint instances to be inserted into
|
||||
the database.
|
||||
"""
|
||||
embeddable_values = []
|
||||
vector_map = {}
|
||||
|
||||
for data_point in data_points:
|
||||
property_names = DataPoint.get_embeddable_property_names(data_point)
|
||||
key = str(data_point.id)
|
||||
vector_map[key] = {}
|
||||
|
||||
for property_name in property_names:
|
||||
property_value = getattr(data_point, property_name, None)
|
||||
|
||||
if property_value is not None:
|
||||
vector_map[key][property_name] = len(embeddable_values)
|
||||
embeddable_values.append(property_value)
|
||||
else:
|
||||
vector_map[key][property_name] = None
|
||||
|
||||
vectorized_values = await self.embed_data(embeddable_values)
|
||||
|
||||
for data_point in data_points:
|
||||
vectorized_data = [
|
||||
vectorized_values[vector_map[str(data_point.id)][property_name]]
|
||||
if vector_map[str(data_point.id)][property_name] is not None
|
||||
else None
|
||||
for property_name in DataPoint.get_embeddable_property_names(data_point)
|
||||
]
|
||||
|
||||
query, params = await self.create_data_point_query(data_point, vectorized_data)
|
||||
self.query(query, params)
|
||||
# async def create_data_points(self, data_points: list[DataPoint]):
|
||||
# """
|
||||
# Add a list of data points to the graph database via batching.
|
||||
#
|
||||
# Can raise exceptions if there are issues during the database operations.
|
||||
#
|
||||
# Parameters:
|
||||
# -----------
|
||||
#
|
||||
# - data_points (list[DataPoint]): A list of DataPoint instances to be inserted into
|
||||
# the database.
|
||||
# """
|
||||
# embeddable_values = []
|
||||
# vector_map = {}
|
||||
#
|
||||
# for data_point in data_points:
|
||||
# property_names = DataPoint.get_embeddable_property_names(data_point)
|
||||
# key = str(data_point.id)
|
||||
# vector_map[key] = {}
|
||||
#
|
||||
# for property_name in property_names:
|
||||
# property_value = getattr(data_point, property_name, None)
|
||||
#
|
||||
# if property_value is not None:
|
||||
# vector_map[key][property_name] = len(embeddable_values)
|
||||
# embeddable_values.append(property_value)
|
||||
# else:
|
||||
# vector_map[key][property_name] = None
|
||||
#
|
||||
# vectorized_values = await self.embed_data(embeddable_values)
|
||||
#
|
||||
# for data_point in data_points:
|
||||
# vectorized_data = [
|
||||
# vectorized_values[vector_map[str(data_point.id)][property_name]]
|
||||
# if vector_map[str(data_point.id)][property_name] is not None
|
||||
# else None
|
||||
# for property_name in DataPoint.get_embeddable_property_names(data_point)
|
||||
# ]
|
||||
#
|
||||
# query, params = await self.create_data_point_query(data_point, vectorized_data)
|
||||
# self.query(query, params)
|
||||
|
||||
async def create_vector_index(self, index_name: str, index_property_name: str):
|
||||
"""
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue