chore: fixes PR issues regarding vector normalization and cognee graph
This commit is contained in:
parent
a59517409c
commit
163bdc527c
3 changed files with 2 additions and 13 deletions
|
|
@ -146,8 +146,7 @@ class LanceDBAdapter(VectorDBInterface):
|
||||||
self,
|
self,
|
||||||
collection_name: str,
|
collection_name: str,
|
||||||
query_text: str = None,
|
query_text: str = None,
|
||||||
query_vector: List[float] = None,
|
query_vector: List[float] = None
|
||||||
with_vector: bool = False
|
|
||||||
):
|
):
|
||||||
if query_text is None and query_vector is None:
|
if query_text is None and query_vector is None:
|
||||||
raise ValueError("One of query_text or query_vector must be provided!")
|
raise ValueError("One of query_text or query_vector must be provided!")
|
||||||
|
|
|
||||||
|
|
@ -2,17 +2,7 @@ from typing import List
|
||||||
|
|
||||||
|
|
||||||
def normalize_distances(result_values: List[dict]) -> List[float]:
|
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)
|
min_value = min(result["_distance"] for result in result_values)
|
||||||
max_value = max(result["_distance"] for result in result_values)
|
max_value = max(result["_distance"] for result in result_values)
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -158,7 +158,7 @@ class CogneeGraph(CogneeAbstractGraph):
|
||||||
print(f"Error mapping vector distances to edges: {ex}")
|
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 = []
|
min_heap = []
|
||||||
for i, edge in enumerate(self.edges):
|
for i, edge in enumerate(self.edges):
|
||||||
source_node = self.get_node(edge.node1.id)
|
source_node = self.get_node(edge.node1.id)
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue