graphiti-core 0.7.3__py3-none-any.whl → 0.7.4__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of graphiti-core might be problematic. Click here for more details.

Files changed (6) hide show
  1. graphiti_core/search/search_utils.py +5 -4
  2. graphiti_core/utils/maintenance/node_operations.py +3 -1
  3. {graphiti_core-0.7.3.dist-info → graphiti_core-0.7.4.dist-info}/METADATA +1 -1
  4. {graphiti_core-0.7.3.dist-info → graphiti_core-0.7.4.dist-info}/RECORD +6 -6
  5. {graphiti_core-0.7.3.dist-info → graphiti_core-0.7.4.dist-info}/LICENSE +0 -0
  6. {graphiti_core-0.7.3.dist-info → graphiti_core-0.7.4.dist-info}/WHEEL +0 -0
graphiti_core/search/search_utils.py CHANGED
@@ -229,8 +229,8 @@ async def edge_similarity_search(
229
229
 
230
230
  query: LiteralString = (
231
231
  """
232
- MATCH (n:Entity)-[r:RELATES_TO]->(m:Entity)
233
- """
232
+ MATCH (n:Entity)-[r:RELATES_TO]->(m:Entity)
233
+ """
234
234
  + group_filter_query
235
235
  + filter_query
236
236
  + """\nWITH DISTINCT r, vector.similarity.cosine(r.fact_embedding, $search_vector) AS score
@@ -765,8 +765,9 @@ async def node_distance_reranker(
765
765
  # rerank on shortest distance
766
766
  filtered_uuids.sort(key=lambda cur_uuid: scores[cur_uuid])
767
767
 
768
- # add back in filtered center uuid
769
- filtered_uuids = [center_node_uuid] + filtered_uuids
768
+ # add back in filtered center uuid if it was filtered out
769
+ if center_node_uuid in node_uuids:
770
+ filtered_uuids = [center_node_uuid] + filtered_uuids
770
771
 
771
772
  return filtered_uuids
772
773
 
graphiti_core/utils/maintenance/node_operations.py CHANGED
@@ -172,7 +172,9 @@ async def extract_nodes(
172
172
  new_nodes = []
173
173
  for name in extracted_node_names:
174
174
  entity_type = node_classifications.get(name)
175
- labels = ['Entity'] if entity_type is None else ['Entity', entity_type]
175
+ labels = (
176
+ ['Entity'] if entity_type is None or entity_type == 'None' else ['Entity', entity_type]
177
+ )
176
178
 
177
179
  new_node = EntityNode(
178
180
  name=name,
{graphiti_core-0.7.3.dist-info → graphiti_core-0.7.4.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: graphiti-core
3
- Version: 0.7.3
3
+ Version: 0.7.4
4
4
  Summary: A temporal graph building library
5
5
  License: Apache-2.0
6
6
  Author: Paul Paliychuk
{graphiti_core-0.7.3.dist-info → graphiti_core-0.7.4.dist-info}/RECORD RENAMED
@@ -44,7 +44,7 @@ graphiti_core/search/search.py,sha256=DX-tcIa0SiKI2HX-b_WdjGE74A8RLWQor4p90dJluU
44
44
  graphiti_core/search/search_config.py,sha256=UZN8jFA4pBlw2O5N1cuhVRBdTwMLR9N3Oyo6sQ4MDVw,3117
45
45
  graphiti_core/search/search_config_recipes.py,sha256=yUqiLnn9vFg39M8eVwjVKfBCL_ptGrfDMQ47m_Blb0g,6885
46
46
  graphiti_core/search/search_filters.py,sha256=4MJmCXD-blMc71xB4F9K4a72qidDCigADQ_ztdG15kc,5884
47
- graphiti_core/search/search_utils.py,sha256=mZqq97XEjTga3pz50l__b8qVFVPRes1dNGB8tZktJ3A,25665
47
+ graphiti_core/search/search_utils.py,sha256=Bywp7trqP_gEwqH8I-JHuj3Mljw9P2K6_XooXI75jHI,25739
48
48
  graphiti_core/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
49
49
  graphiti_core/utils/bulk_utils.py,sha256=P4LKO46Yle4tBdNcQ3hDHcSQFaR8UBLfoL-z1M2Wua0,14690
50
50
  graphiti_core/utils/datetime_utils.py,sha256=Ti-2tnrDFRzBsbfblzsHybsM3jaDLP4-VT2t0VhpIzU,1357
@@ -52,10 +52,10 @@ graphiti_core/utils/maintenance/__init__.py,sha256=TRY3wWWu5kn3Oahk_KKhltrWnh0NA
52
52
  graphiti_core/utils/maintenance/community_operations.py,sha256=gIw1M5HGgc2c3TXag5ygPPpAv5WsG-yoC8Lhmfr6FMs,10011
53
53
  graphiti_core/utils/maintenance/edge_operations.py,sha256=tNw56vN586JYZMgie6RLRTiHZ680-kWzDIxW8ucL6SU,12780
54
54
  graphiti_core/utils/maintenance/graph_data_operations.py,sha256=qds9ALk9PhpQs1CNZTZGpi70mqJ93Y2KhIh9X2r8MUI,6533
55
- graphiti_core/utils/maintenance/node_operations.py,sha256=gihbPEBH6StLQCSd9wSu582d4Owaw3l5JLR1IBDrnVs,14137
55
+ graphiti_core/utils/maintenance/node_operations.py,sha256=OAw8IHNGLLQ32lODffsZGZykQ0qOgidpLHrTAHVyUUM,14186
56
56
  graphiti_core/utils/maintenance/temporal_operations.py,sha256=RdNtubCyYhOVrvcOIq2WppHls1Q-BEjtsN8r38l-Rtc,3691
57
57
  graphiti_core/utils/maintenance/utils.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
58
- graphiti_core-0.7.3.dist-info/LICENSE,sha256=KCUwCyDXuVEgmDWkozHyniRyWjnWUWjkuDHfU6o3JlA,11325
59
- graphiti_core-0.7.3.dist-info/METADATA,sha256=CA2-V0qWnuIo_sQecfWTQQJ-9GLb8lMvMd5FqbUVFPQ,10541
60
- graphiti_core-0.7.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
61
- graphiti_core-0.7.3.dist-info/RECORD,,
58
+ graphiti_core-0.7.4.dist-info/LICENSE,sha256=KCUwCyDXuVEgmDWkozHyniRyWjnWUWjkuDHfU6o3JlA,11325
59
+ graphiti_core-0.7.4.dist-info/METADATA,sha256=5FWWCZo1fKxmMuDrLF3_qxRhz18MBwOy2lMFrhVIFzk,10541
60
+ graphiti_core-0.7.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
61
+ graphiti_core-0.7.4.dist-info/RECORD,,