Import Solver + neighbors via sparql query

backend/app/graph_export.py

@@ -5,16 +5,25 @@ from typing import Any
 
 def edge_retrieval_query(*, edge_limit: int, include_bnodes: bool) -> str:
     bnode_filter = "" if include_bnodes else "FILTER(!isBlank(?s) && !isBlank(?o))"
+    
     return f"""
+PREFIX rdf: <http://www.w3.org/1999/02/22-rdf-syntax-ns#>
+PREFIX rdfs: <http://www.w3.org/2000/01/rdf-schema#>
+PREFIX owl: <http://www.w3.org/2002/07/owl#>
+
 SELECT ?s ?p ?o
 WHERE {{
-  ?s ?p ?o .
+  {{
+    VALUES ?p {{ rdf:type }}
+    ?s ?p ?o .
+    ?o rdf:type owl:Class .
+  }}
+  UNION
+  {{
+    VALUES ?p {{ rdfs:subClassOf }}
+    ?s ?p ?o .
+  }}
   FILTER(!isLiteral(?o))
-  FILTER(?p NOT IN (
-    <http://www.w3.org/2000/01/rdf-schema#label>,
-    <http://www.w3.org/2004/02/skos/core#prefLabel>,
-    <http://www.w3.org/2004/02/skos/core#altLabel>
-  ))
   {bnode_filter}
 }}
 LIMIT {edge_limit}
@@ -91,4 +100,3 @@ def graph_from_sparql_bindings(
     ]
 
     return out_nodes, out_edges
-