chore: fixes PR issues regarding vector normalization and cognee graph

This commit is contained in:
hajdul88 2024-11-26 15:37:34 +01:00
parent a59517409c
commit 163bdc527c
3 changed files with 2 additions and 13 deletions

View file

@ -146,8 +146,7 @@ class LanceDBAdapter(VectorDBInterface):
self,
collection_name: str,
query_text: str = None,
query_vector: List[float] = None,
with_vector: bool = False
query_vector: List[float] = None
):
if query_text is None and query_vector is None:
raise ValueError("One of query_text or query_vector must be provided!")

View file

@ -2,17 +2,7 @@ from typing import List
def normalize_distances(result_values: List[dict]) -> List[float]:
min_value = 100
max_value = 0
for result in result_values:
value = float(result["_distance"])
if value > max_value:
max_value = value
if value < min_value:
min_value = value
normalized_values = []
min_value = min(result["_distance"] for result in result_values)
max_value = max(result["_distance"] for result in result_values)

View file

@ -158,7 +158,7 @@ class CogneeGraph(CogneeAbstractGraph):
print(f"Error mapping vector distances to edges: {ex}")
async def calculate_top_triplet_importances(self, k = int) -> List:
async def calculate_top_triplet_importances(self, k: int) -> List:
min_heap = []
for i, edge in enumerate(self.edges):
source_node = self.get_node(edge.node1.id)