nucliadb 6.3.4.post3780__py3-none-any.whl → 6.3.4.post3796__py3-none-any.whl

nucliadb/search/search/query_parser/parsers/graph.py

@@ -148,6 +148,9 @@ def _parse_path_query(expr: graph_requests.GraphPathQuery) -> nodereader_pb2.Gra
     elif isinstance(expr, graph_requests.Relation):
         _set_relation_to_pb(expr, pb.path.relation)
 
+    elif isinstance(expr, graph_requests.Generated):
+        _set_generated_to_pb(expr, pb)
+
     else:  # pragma: nocover
         # This is a trick so mypy generates an error if this branch can be reached,
         # that is, if we are missing some ifs
@@ -174,6 +177,9 @@ def _parse_node_query(expr: graph_requests.GraphNodesQuery) -> nodereader_pb2.Gr
         _set_node_to_pb(expr, pb.path.source)
         pb.path.undirected = True
 
+    elif isinstance(expr, graph_requests.Generated):
+        _set_generated_to_pb(expr, pb)
+
     else:  # pragma: nocover
         # This is a trick so mypy generates an error if this branch can be reached,
         # that is, if we are missing some ifs
@@ -201,6 +207,9 @@ def _parse_relation_query(
     elif isinstance(expr, graph_requests.Relation):
         _set_relation_to_pb(expr, pb.path.relation)
 
+    elif isinstance(expr, graph_requests.Generated):
+        _set_generated_to_pb(expr, pb)
+
     else:  # pragma: nocover
         # This is a trick so mypy generates an error if this branch can be reached,
         # that is, if we are missing some ifs
@@ -213,10 +222,10 @@ def _set_node_to_pb(node: graph_requests.GraphNode, pb: nodereader_pb2.GraphQuer
     if node.value is not None:
         pb.value = node.value
         if node.match == graph_requests.NodeMatchKind.EXACT:
-            pb.match_kind = nodereader_pb2.GraphQuery.Node.MatchKind.EXACT
+            pb.match_kind = nodereader_pb2.GraphQuery.Node.MatchKind.DEPRECATED_EXACT
 
         elif node.match == graph_requests.NodeMatchKind.FUZZY:
-            pb.match_kind = nodereader_pb2.GraphQuery.Node.MatchKind.FUZZY
+            pb.match_kind = nodereader_pb2.GraphQuery.Node.MatchKind.DEPRECATED_FUZZY
 
         else:  # pragma: nocover
             # This is a trick so mypy generates an error if this branch can be reached,
@@ -235,3 +244,23 @@ def _set_relation_to_pb(relation: graph_requests.GraphRelation, pb: nodereader_p
         pb.value = relation.label
     if relation.type is not None:
         pb.relation_type = RelationTypeMap[relation.type]
+
+
+def _set_generated_to_pb(generated: graph_requests.Generated, pb: nodereader_pb2.GraphQuery.PathQuery):
+    if generated.by == graph_requests.Generator.USER:
+        pb.facet.facet = "/g/u"
+
+    elif generated.by == graph_requests.Generator.PROCESSOR:
+        pb.bool_not.facet.facet = "/g"
+
+    elif generated.by == graph_requests.Generator.DATA_AUGMENTATION:
+        facet = "/g/da"
+        if generated.da_task is not None:
+            facet += f"/{generated.da_task}"
+
+        pb.facet.facet = facet
+
+    else:  # pragma: nocover
+        # This is a trick so mypy generates an error if this branch can be reached,
+        # that is, if we are missing some ifs
+        _a: int = "a"

{nucliadb-6.3.4.post3780.dist-info → nucliadb-6.3.4.post3796.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nucliadb
-Version: 6.3.4.post3780
+Version: 6.3.4.post3796
 Summary: NucliaDB
 Author-email: Nuclia <nucliadb@nuclia.com>
 License: AGPL
@@ -20,11 +20,11 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3 :: Only
 Requires-Python: <4,>=3.9
 Description-Content-Type: text/markdown
-Requires-Dist: nucliadb-telemetry[all]>=6.3.4.post3780
-Requires-Dist: nucliadb-utils[cache,fastapi,storages]>=6.3.4.post3780
-Requires-Dist: nucliadb-protos>=6.3.4.post3780
-Requires-Dist: nucliadb-models>=6.3.4.post3780
-Requires-Dist: nidx-protos>=6.3.4.post3780
+Requires-Dist: nucliadb-telemetry[all]>=6.3.4.post3796
+Requires-Dist: nucliadb-utils[cache,fastapi,storages]>=6.3.4.post3796
+Requires-Dist: nucliadb-protos>=6.3.4.post3796
+Requires-Dist: nucliadb-models>=6.3.4.post3796
+Requires-Dist: nidx-protos>=6.3.4.post3796
 Requires-Dist: nucliadb-admin-assets>=1.0.0.post1224
 Requires-Dist: nuclia-models>=0.24.2
 Requires-Dist: uvicorn

{nucliadb-6.3.4.post3780.dist-info → nucliadb-6.3.4.post3796.dist-info}/RECORD

@@ -258,7 +258,7 @@ nucliadb/search/search/query_parser/old_filters.py,sha256=-zbfN-RsXoj_DRjh3Lfp-w
 nucliadb/search/search/query_parser/parsers/__init__.py,sha256=ySCNSdbesLXGZyR88919njulA6UE10_3PhqMG_Yj1o4,1034
 nucliadb/search/search/query_parser/parsers/catalog.py,sha256=XdBiTweGTQkj8m_V_i2xbwp7P5pPO8K1Tud692XKhMw,7149
 nucliadb/search/search/query_parser/parsers/find.py,sha256=q3wH_i0DGceeKckYEH3c5MqM5EvRiMCL7r-6nCAId9Q,4666
-nucliadb/search/search/query_parser/parsers/graph.py,sha256=SfbMV-H-oCqZraQIwD8n5Hqi9dwOlvqzh_I3aZAKatg,8369
+nucliadb/search/search/query_parser/parsers/graph.py,sha256=43S8iSg4j9I_XD8zpW1VggDspQD-NMyS26B5Mw6Dktw,9358
 nucliadb/standalone/__init__.py,sha256=cp15ZcFnHvpcu_5-aK2A4uUyvuZVV_MJn4bIXMa20ks,835
 nucliadb/standalone/api_router.py,sha256=hgq9FXpihzgjHkwcVGfGCSwyXy67fqXTfLFHuINzIi0,5567
 nucliadb/standalone/app.py,sha256=mAApNK_iVsQgJyd-mtwCeZq5csSimwnXmlQGH9a70pE,5586
@@ -354,8 +354,8 @@ nucliadb/writer/tus/local.py,sha256=7jYa_w9b-N90jWgN2sQKkNcomqn6JMVBOVeDOVYJHto,
 nucliadb/writer/tus/s3.py,sha256=vF0NkFTXiXhXq3bCVXXVV-ED38ECVoUeeYViP8uMqcU,8357
 nucliadb/writer/tus/storage.py,sha256=ToqwjoYnjI4oIcwzkhha_MPxi-k4Jk3Lt55zRwaC1SM,2903
 nucliadb/writer/tus/utils.py,sha256=MSdVbRsRSZVdkaum69_0wku7X3p5wlZf4nr6E0GMKbw,2556
-nucliadb-6.3.4.post3780.dist-info/METADATA,sha256=fWU0d62hr5LmRJRO4TGC9prZ4nZ261zU12xkYtDo2ng,4291
-nucliadb-6.3.4.post3780.dist-info/WHEEL,sha256=DK49LOLCYiurdXXOXwGJm6U4DkHkg4lcxjhqwRa0CP4,91
-nucliadb-6.3.4.post3780.dist-info/entry_points.txt,sha256=XqGfgFDuY3zXQc8ewXM2TRVjTModIq851zOsgrmaXx4,1268
-nucliadb-6.3.4.post3780.dist-info/top_level.txt,sha256=hwYhTVnX7jkQ9gJCkVrbqEG1M4lT2F_iPQND1fCzF80,20
-nucliadb-6.3.4.post3780.dist-info/RECORD,,
+nucliadb-6.3.4.post3796.dist-info/METADATA,sha256=CAxQ5xeSrVmN4NWzZwAdBUv2q5O5cP1INsDQ14RJYdk,4291
+nucliadb-6.3.4.post3796.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+nucliadb-6.3.4.post3796.dist-info/entry_points.txt,sha256=XqGfgFDuY3zXQc8ewXM2TRVjTModIq851zOsgrmaXx4,1268
+nucliadb-6.3.4.post3796.dist-info/top_level.txt,sha256=hwYhTVnX7jkQ9gJCkVrbqEG1M4lT2F_iPQND1fCzF80,20
+nucliadb-6.3.4.post3796.dist-info/RECORD,,

{nucliadb-6.3.4.post3780.dist-info → nucliadb-6.3.4.post3796.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (78.0.2)
+Generator: setuptools (78.1.0)
 Root-Is-Purelib: true
 Tag: py3-none-any