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,
|
||||
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!")
|
||||
|
|
|
|||
|
|
@ -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)
|
||||
|
||||
|
|
|
|||
|
|
@ -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)
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue