cognite-neat 0.85.12__py3-none-any.whl → 0.87.0__py3-none-any.whl

cognite/neat/rules/models/information/_rules.py

@@ -1,13 +1,13 @@
 import math
 import sys
 from datetime import datetime
-from typing import TYPE_CHECKING, Any, ClassVar, Literal, cast
+from typing import TYPE_CHECKING, Any, ClassVar, Literal
 
 from pydantic import Field, field_serializer, field_validator, model_validator
 from pydantic.main import IncEx
 from rdflib import Namespace
 
-from cognite.neat.constants import PREFIXES
+from cognite.neat.constants import get_default_prefixes
 from cognite.neat.issues import MultiValueError
 from cognite.neat.rules import exceptions, issues
 from cognite.neat.rules.models._base import (
@@ -38,10 +38,9 @@ from cognite.neat.rules.models.data_types import DataType
 from cognite.neat.rules.models.domain import DomainRules
 from cognite.neat.rules.models.entities import (
     ClassEntity,
+    ClassEntityList,
     EntityTypes,
     MultiValueTypeInfo,
-    ParentClassEntity,
-    ParentEntityList,
     ReferenceEntity,
     Undefined,
     UnknownEntity,
@@ -59,7 +58,7 @@ else:
 
 
 class InformationMetadata(BaseMetadata):
-    role: ClassVar[RoleTypes] = RoleTypes.information_architect
+    role: ClassVar[RoleTypes] = RoleTypes.information
     data_model_type: DataModelType = Field(DataModelType.enterprise, alias="dataModelType")
     schema_: SchemaCompleteness = Field(SchemaCompleteness.partial, alias="schema")
     extension: ExtensionCategoryType | None = ExtensionCategory.addition
@@ -114,6 +113,9 @@ class InformationMetadata(BaseMetadata):
     def as_identifier(self) -> str:
         return f"{self.prefix}:{self.name}"
 
+    def get_prefix(self) -> str:
+        return self.prefix
+
 
 class InformationClass(SheetEntity):
     """
@@ -130,7 +132,7 @@ class InformationClass(SheetEntity):
     class_: ClassEntity = Field(alias="Class")
     name: str | None = Field(alias="Name", default=None)
     description: str | None = Field(alias="Description", default=None)
-    parent: ParentEntityList | None = Field(alias="Parent Class", default=None)
+    parent: ClassEntityList | None = Field(alias="Parent Class", default=None)
     reference: URLEntity | ReferenceEntity | None = Field(alias="Reference", default=None, union_mode="left_to_right")
     match_type: MatchType | None = Field(alias="Match Type", default=None)
     comment: str | None = Field(alias="Comment", default=None)
@@ -259,7 +261,7 @@ class InformationRules(BaseRules):
     metadata: InformationMetadata = Field(alias="Metadata")
     properties: SheetList[InformationProperty] = Field(alias="Properties")
     classes: SheetList[InformationClass] = Field(alias="Classes")
-    prefixes: dict[str, Namespace] = Field(default_factory=lambda: PREFIXES.copy(), alias="Prefixes")
+    prefixes: dict[str, Namespace] = Field(default_factory=get_default_prefixes, alias="Prefixes")
     last: "InformationRules | None" = Field(None, alias="Last")
     reference: "InformationRules | None" = Field(None, alias="Reference")
 
@@ -268,7 +270,7 @@ class InformationRules(BaseRules):
         if isinstance(values, dict):
             return {key: Namespace(value) if isinstance(value, str) else value for key, value in values.items()}
         elif values is None:
-            values = PREFIXES.copy()
+            values = get_default_prefixes()
         return values
 
     @model_validator(mode="after")
@@ -287,7 +289,7 @@ class InformationRules(BaseRules):
         # update parent classes
         for class_ in self.classes:
             if class_.parent:
-                for parent in cast(list[ParentClassEntity], class_.parent):
+                for parent in class_.parent:
                     if not isinstance(parent.prefix, str):
                         parent.prefix = self.metadata.prefix
             if class_.class_.prefix is Undefined:
@@ -348,7 +350,7 @@ class InformationRules(BaseRules):
 
         return _InformationRulesConverter(self).as_asset_architect_rules()
 
-    def as_dms_architect_rules(self) -> "DMSRules":
+    def as_dms_rules(self) -> "DMSRules":
         from ._converter import _InformationRulesConverter
 
-        return _InformationRulesConverter(self).as_dms_architect_rules()
+        return _InformationRulesConverter(self).as_dms_rules()

cognite/neat/rules/models/information/_rules_input.py

@@ -15,7 +15,6 @@ from cognite.neat.rules.models.data_types import DataType
 from cognite.neat.rules.models.entities import (
     ClassEntity,
     MultiValueTypeInfo,
-    ParentClassEntity,
     Unknown,
     UnknownEntity,
 )
@@ -227,7 +226,7 @@ class InformationClassInput:
             "Reference": self.reference,
             "Match Type": self.match_type,
             "Parent Class": (
-                [ParentClassEntity.load(parent, prefix=default_prefix) for parent in self.parent.split(",")]
+                [ClassEntity.load(parent, prefix=default_prefix) for parent in self.parent.split(",")]
                 if self.parent
                 else None
             ),

cognite/neat/rules/models/information/_validation.py

@@ -161,7 +161,7 @@ class InformationPostValidation:
             class_subclass_pairs[class_.class_] = []
             if class_.parent is None:
                 continue
-            class_subclass_pairs[class_.class_].extend([parent.as_class_entity() for parent in class_.parent])
+            class_subclass_pairs[class_.class_].extend(class_.parent)
 
         return class_subclass_pairs
 

cognite/neat/utils/utils.py

@@ -7,20 +7,30 @@ from collections import Counter, OrderedDict
 from collections.abc import Iterable
 from datetime import datetime
 from functools import wraps
-from typing import TypeAlias, cast, overload
+from typing import Literal, TypeAlias, cast, overload
 
 import pandas as pd
 from cognite.client import ClientConfig, CogniteClient
-from cognite.client.credentials import CredentialProvider, OAuthClientCredentials, OAuthInteractive, Token
+from cognite.client.credentials import (
+    CredentialProvider,
+    OAuthClientCredentials,
+    OAuthInteractive,
+    Token,
+)
 from cognite.client.exceptions import CogniteDuplicatedError, CogniteReadTimeout
 from pydantic import HttpUrl, TypeAdapter, ValidationError
 from pydantic_core import ErrorDetails
 from pyparsing import Any
-from rdflib import Literal, Namespace
+from rdflib import Literal as RdfLiteral
+from rdflib import Namespace
 from rdflib.term import URIRef
 
 from cognite.neat import _version
-from cognite.neat.utils.cdf import CogniteClientConfig, InteractiveCogniteClient, ServiceCogniteClient
+from cognite.neat.utils.cdf import (
+    CogniteClientConfig,
+    InteractiveCogniteClient,
+    ServiceCogniteClient,
+)
 
 if sys.version_info >= (3, 11):
     from datetime import UTC
@@ -30,12 +40,15 @@ else:
     UTC = timezone.utc
 
 
-Triple: TypeAlias = tuple[URIRef, URIRef, Literal | URIRef]
+Triple: TypeAlias = tuple[URIRef, URIRef, RdfLiteral | URIRef]
 
 
 def get_cognite_client_from_config(config: ServiceCogniteClient) -> CogniteClient:
     credentials = OAuthClientCredentials(
-        token_url=config.token_url, client_id=config.client_id, client_secret=config.client_secret, scopes=config.scopes
+        token_url=config.token_url,
+        client_id=config.client_id,
+        client_secret=config.client_secret,
+        scopes=config.scopes,
     )
 
     return _get_cognite_client(config, credentials)
@@ -75,15 +88,25 @@ def _get_cognite_client(config: CogniteClientConfig, credentials: CredentialProv
 
 
 @overload
-def remove_namespace_from_uri(*URI: URIRef | str, special_separator: str = "#_") -> str: ...
+def remove_namespace_from_uri(
+    *URI: URIRef | str,
+    special_separator: str = "#_",
+    validation: Literal["full", "prefix"] = "prefix",
+) -> str: ...
 
 
 @overload
-def remove_namespace_from_uri(*URI: tuple[URIRef | str, ...], special_separator: str = "#_") -> tuple[str, ...]: ...
+def remove_namespace_from_uri(
+    *URI: tuple[URIRef | str, ...],
+    special_separator: str = "#_",
+    validation: Literal["full", "prefix"] = "prefix",
+) -> tuple[str, ...]: ...
 
 
 def remove_namespace_from_uri(
-    *URI: URIRef | str | tuple[URIRef | str, ...], special_separator: str = "#_"
+    *URI: URIRef | str | tuple[URIRef | str, ...],
+    special_separator: str = "#_",
+    validation: Literal["full", "prefix"] = "prefix",
 ) -> tuple[str, ...] | str:
     """Removes namespace from URI
 
@@ -93,6 +116,9 @@ def remove_namespace_from_uri(
         special_separator : str
             Special separator to use instead of # or / if present in URI
             Set by default to "#_" which covers special client use case
+        validation: str
+            Validation type to use for URI. If set to "full", URI will be validated using pydantic
+            If set to "prefix", only check if URI starts with http or https will be made
 
     Returns
         Entities id without namespace
@@ -114,11 +140,17 @@ def remove_namespace_from_uri(
 
     output = []
     for u in uris:
-        try:
-            _ = TypeAdapter(HttpUrl).validate_python(u)
-            output.append(u.split(special_separator if special_separator in u else "#" if "#" in u else "/")[-1])
-        except ValidationError:
-            output.append(str(u))
+        if validation == "full":
+            try:
+                _ = TypeAdapter(HttpUrl).validate_python(u)
+                output.append(u.split(special_separator if special_separator in u else "#" if "#" in u else "/")[-1])
+            except ValidationError:
+                output.append(str(u))
+        else:
+            if u.lower().startswith("http"):
+                output.append(u.split(special_separator if special_separator in u else "#" if "#" in u else "/")[-1])
+            else:
+                output.append(str(u))
 
     return tuple(output) if len(output) > 1 else output[0]
 
@@ -154,8 +186,8 @@ def as_neat_compliant_uri(uri: URIRef) -> URIRef:
     return URIRef(f"{namespace}{compliant_uri}")
 
 
-def convert_rdflib_content(content: Literal | URIRef | dict | list) -> Any:
-    if isinstance(content, Literal) or isinstance(content, URIRef):
+def convert_rdflib_content(content: RdfLiteral | URIRef | dict | list) -> Any:
+    if isinstance(content, RdfLiteral) or isinstance(content, URIRef):
         return content.toPython()
     elif isinstance(content, dict):
         return {key: convert_rdflib_content(value) for key, value in content.items()}
@@ -192,7 +224,9 @@ def _traverse(hierarchy: dict, graph: dict, names: list[str]) -> dict:
 
 
 def get_generation_order(
-    class_linkage: pd.DataFrame, parent_col: str = "source_class", child_col: str = "target_class"
+    class_linkage: pd.DataFrame,
+    parent_col: str = "source_class",
+    child_col: str = "target_class",
 ) -> dict:
     parent_child_list = class_linkage[[parent_col, child_col]].values.tolist()
     # Build a directed graph and a list of all names that have no parent
@@ -318,7 +352,9 @@ def generate_exception_report(exceptions: list[dict] | list[ErrorDetails] | None
     return report
 
 
-def _order_expectations_by_type(exceptions: list[dict] | list[ErrorDetails]) -> dict[str, list[str]]:
+def _order_expectations_by_type(
+    exceptions: list[dict] | list[ErrorDetails],
+) -> dict[str, list[str]]:
     exception_dict: dict[str, list[str]] = {}
     for exception in exceptions:
         if not isinstance(exception["loc"], str) and isinstance(exception["loc"], Iterable):

cognite/neat/workflows/steps/lib/current/graph_store.py

@@ -2,11 +2,15 @@ import logging
 from pathlib import Path
 from typing import ClassVar, cast
 
-from cognite.neat.constants import PREFIXES
+from cognite.neat.constants import DEFAULT_NAMESPACE, get_default_prefixes
 from cognite.neat.legacy.graph import stores
 from cognite.neat.workflows._exceptions import StepNotInitialized
 from cognite.neat.workflows.model import FlowMessage
-from cognite.neat.workflows.steps.data_contracts import RulesData, SolutionGraph, SourceGraph
+from cognite.neat.workflows.steps.data_contracts import (
+    RulesData,
+    SolutionGraph,
+    SourceGraph,
+)
 from cognite.neat.workflows.steps.step_model import Configurable, Step
 
 __all__ = ["GraphStoreConfiguration", "GraphStoreReset"]
@@ -40,9 +44,21 @@ class GraphStoreConfiguration(Step):
             value="source-graph-store",
             label="Local directory that is used as local graph store.Only for oxigraph, file store types",
         ),
-        Configurable(name="Query URL", value="", label="Query URL for SPARQL endpoint. Only for SPARQL store type"),
-        Configurable(name="Update URL", value="", label="Update URL for SPARQL endpoint. Only for SPARQL store type"),
-        Configurable(name="GraphDB API root URL", value="", label="Root url for GraphDB. Only for graphdb"),
+        Configurable(
+            name="Query URL",
+            value="",
+            label="Query URL for SPARQL endpoint. Only for SPARQL store type",
+        ),
+        Configurable(
+            name="Update URL",
+            value="",
+            label="Update URL for SPARQL endpoint. Only for SPARQL store type",
+        ),
+        Configurable(
+            name="GraphDB API root URL",
+            value="",
+            label="Root url for GraphDB. Only for graphdb",
+        ),
         Configurable(
             name="Init procedure",
             value="reset",
@@ -72,7 +88,7 @@ class GraphStoreConfiguration(Step):
             graph_store = None
             logging.info("Graph reset complete")
 
-        prefixes = rules_data.rules.prefixes if rules_data else PREFIXES.copy()
+        prefixes = rules_data.rules.prefixes if rules_data else get_default_prefixes()
 
         if store_type == stores.OxiGraphStore.rdf_store_type and graph_store is not None:
             # OXIGRAPH doesn't like to be initialized twice without a good reason
@@ -83,7 +99,7 @@ class GraphStoreConfiguration(Step):
         except KeyError:
             return FlowMessage(output_text="Invalid store type")
 
-        new_graph_store = store_cls(prefixes=prefixes, base_prefix="neat", namespace=PREFIXES["neat"])
+        new_graph_store = store_cls(prefixes=prefixes, base_prefix="neat", namespace=DEFAULT_NAMESPACE)
         new_graph_store.init_graph(
             self.configs["Query URL"],
             self.configs["Update URL"],
@@ -93,7 +109,11 @@ class GraphStoreConfiguration(Step):
 
         return (
             FlowMessage(output_text="Graph store configured successfully"),
-            SourceGraph(graph=new_graph_store) if graph_name == "SourceGraph" else SolutionGraph(graph=new_graph_store),
+            (
+                SourceGraph(graph=new_graph_store)
+                if graph_name == "SourceGraph"
+                else SolutionGraph(graph=new_graph_store)
+            ),
         )
 
 

cognite/neat/workflows/steps/lib/legacy/graph_extractor.py

@@ -8,14 +8,20 @@ from typing import ClassVar, cast
 
 from rdflib import RDF, XSD, Literal, Namespace, URIRef
 
-from cognite.neat.constants import PREFIXES
+from cognite.neat.constants import DEFAULT_NAMESPACE
 from cognite.neat.legacy.graph import extractors
-from cognite.neat.legacy.graph.extractors._mock_graph_generator import generate_triples as generate_mock_triples
+from cognite.neat.legacy.graph.extractors._mock_graph_generator import (
+    generate_triples as generate_mock_triples,
+)
 from cognite.neat.legacy.rules.exporters._rules2triples import get_instances_as_triples
 from cognite.neat.utils.utils import create_sha256_hash
 from cognite.neat.workflows._exceptions import StepNotInitialized
 from cognite.neat.workflows.model import FlowMessage, StepExecutionStatus
-from cognite.neat.workflows.steps.data_contracts import RulesData, SolutionGraph, SourceGraph
+from cognite.neat.workflows.steps.data_contracts import (
+    RulesData,
+    SolutionGraph,
+    SourceGraph,
+)
 from cognite.neat.workflows.steps.step_model import Configurable, Step
 
 __all__ = [
@@ -196,7 +202,10 @@ class ExtractGraphFromMockGraph(Step):
             label="Target number of instances for each class",
         ),
         Configurable(
-            name="graph_name", value="solution", label="The name of target graph.", options=["source", "solution"]
+            name="graph_name",
+            value="solution",
+            label="The name of target graph.",
+            options=["source", "solution"],
         ),
     ]
 
@@ -225,7 +234,10 @@ class ExtractGraphFromMockGraph(Step):
         try:
             triples = generate_mock_triples(transformation_rules=transformation_rules.rules, class_count=class_count)
         except Exception as e:
-            return FlowMessage(error_text=f"Error: {e}", step_execution_status=StepExecutionStatus.ABORT_AND_FAIL)
+            return FlowMessage(
+                error_text=f"Error: {e}",
+                step_execution_status=StepExecutionStatus.ABORT_AND_FAIL,
+            )
 
         logging.info("Adding mock triples to graph")
         graph_store.graph.add_triples(triples, verbose=True)  # type: ignore[arg-type]
@@ -243,7 +255,10 @@ class ExtractGraphFromRulesInstanceSheet(Step):
 
     configurables: ClassVar[list[Configurable]] = [
         Configurable(
-            name="graph_name", value="solution", label="The name of target graph.", options=["source", "solution"]
+            name="graph_name",
+            value="solution",
+            label="The name of target graph.",
+            options=["source", "solution"],
         ),
     ]
 
@@ -264,7 +279,10 @@ class ExtractGraphFromRulesInstanceSheet(Step):
         try:
             graph_store.graph.add_triples(triples, verbose=True)  # type: ignore[arg-type]
         except Exception as e:
-            return FlowMessage(error_text=f"Error: {e}", step_execution_status=StepExecutionStatus.ABORT_AND_FAIL)
+            return FlowMessage(
+                error_text=f"Error: {e}",
+                step_execution_status=StepExecutionStatus.ABORT_AND_FAIL,
+            )
 
         return FlowMessage(output_text=output_text)
 
@@ -281,7 +299,7 @@ class ExtractGraphFromRulesDataModel(Step):
     def run(  # type: ignore[override, syntax]
         self, transformation_rules: RulesData, source_graph: SourceGraph
     ) -> FlowMessage:
-        ns = PREFIXES["neat"]
+        ns = DEFAULT_NAMESPACE
         classes = transformation_rules.rules.classes
         properties = transformation_rules.rules.properties
         counter = 0
@@ -291,16 +309,29 @@ class ExtractGraphFromRulesDataModel(Step):
             source_graph.graph.graph.add((rdf_instance_id, RDF.type, URIRef(ns + class_def.class_id)))
             if class_def.parent_class:
                 source_graph.graph.graph.add(
-                    (rdf_instance_id, URIRef(ns + "hasParent"), URIRef(ns + "_" + cast(str, class_def.parent_class)))
+                    (
+                        rdf_instance_id,
+                        URIRef(ns + "hasParent"),
+                        URIRef(ns + "_" + cast(str, class_def.parent_class)),
+                    )
                 )
             counter += 1
 
         for _property_name, property_def in properties.items():
             rdf_instance_id = URIRef(ns + "_" + property_def.class_id)
             source_graph.graph.graph.add(
-                (rdf_instance_id, URIRef(ns + property_def.property_id), Literal(property_def.expected_value_type))
+                (
+                    rdf_instance_id,
+                    URIRef(ns + property_def.property_id),
+                    Literal(property_def.expected_value_type),
+                )
             )
-            if property_def.expected_value_type.suffix not in ("string", "integer", "float", "boolean"):
+            if property_def.expected_value_type.suffix not in (
+                "string",
+                "integer",
+                "float",
+                "boolean",
+            ):
                 source_graph.graph.graph.add(
                     (
                         rdf_instance_id,
@@ -324,10 +355,15 @@ class ExtractGraphFromJsonFile(Step):
     version = "legacy"
     configurables: ClassVar[list[Configurable]] = [
         Configurable(
-            name="file_name", value="data_dump.json", label="Full path to the file containing data dump in JSON format"
+            name="file_name",
+            value="data_dump.json",
+            label="Full path to the file containing data dump in JSON format",
         ),
         Configurable(
-            name="graph_name", value="solution", label="The name of target graph.", options=["source", "solution"]
+            name="graph_name",
+            value="solution",
+            label="The name of target graph.",
+            options=["source", "solution"],
         ),
         Configurable(
             name="object_id_generation_method",
@@ -340,7 +376,12 @@ class ExtractGraphFromJsonFile(Step):
                 hash_of_json_element - takes a hash of the JSON element.Very generic method but \
                      can be slow working with big objects. \
                 uuid - generates a random UUID, the option produces unstables ids . ",
-            options=["source_object_properties", "source_object_id_mapping", "hash_of_json_element", "uuid"],
+            options=[
+                "source_object_properties",
+                "source_object_id_mapping",
+                "hash_of_json_element",
+                "uuid",
+            ],
         ),
         Configurable(
             name="json_object_id_mapping",
@@ -359,10 +400,21 @@ class ExtractGraphFromJsonFile(Step):
             value="http://purl.org/cognite/neat#",
             label="Namespace to be used for the generated objects.",
         ),
-        Configurable(name="namespace_prefix", value="neat", label="The prefix to be used for the namespace."),
+        Configurable(
+            name="namespace_prefix",
+            value="neat",
+            label="The prefix to be used for the namespace.",
+        ),
     ]
 
-    def get_json_object_id(self, method, object_name: str, json_object: dict, parent_object_id: str, id_mapping: dict):
+    def get_json_object_id(
+        self,
+        method,
+        object_name: str,
+        json_object: dict,
+        parent_object_id: str,
+        id_mapping: dict,
+    ):
         if method == "source_object_properties":
             object_id = ""
             if object_name in id_mapping:
@@ -385,7 +437,11 @@ class ExtractGraphFromJsonFile(Step):
                 # back to hashing
                 logging.debug(f"Object {object_name} doesn't have a valid id.Error : {e}")
                 object_id = self.get_json_object_id(
-                    "hash_of_json_element", object_name, json_object, parent_object_id, id_mapping
+                    "hash_of_json_element",
+                    object_name,
+                    json_object,
+                    parent_object_id,
+                    id_mapping,
                 )
         else:
             raise ValueError(
@@ -436,7 +492,11 @@ class ExtractGraphFromJsonFile(Step):
 
         # Iterate through the JSON data and convert it to triples
         def convert_json_to_triples(
-            data: dict, parent_node: URIRef, parent_object_id: str, parent_node_path: str, property_name=None
+            data: dict,
+            parent_node: URIRef,
+            parent_object_id: str,
+            parent_node_path: str,
+            property_name=None,
         ):
             nonlocal nodes_counter, property_counter
             if isinstance(data, dict):
@@ -456,7 +516,13 @@ class ExtractGraphFromJsonFile(Step):
                     new_node = URIRef(ns + object_id)
                     graph.graph.add((new_node, RDF.type, URIRef(ns + property_name)))
                     if labels_mapping and property_name in labels_mapping:
-                        graph.graph.add((new_node, URIRef(ns + "label"), Literal(data[labels_mapping[property_name]])))
+                        graph.graph.add(
+                            (
+                                new_node,
+                                URIRef(ns + "label"),
+                                Literal(data[labels_mapping[property_name]]),
+                            )
+                        )
                     else:
                         graph.graph.add((new_node, URIRef(ns + "label"), Literal(property_name)))
                     graph.graph.add((new_node, URIRef(ns + "parent"), parent_node))
@@ -470,7 +536,13 @@ class ExtractGraphFromJsonFile(Step):
                         convert_json_to_triples(value, parent_node, parent_object_id, parent_node_path, key)
                 else:
                     for item in data:
-                        convert_json_to_triples(item, parent_node, parent_object_id, parent_node_path, property_name)
+                        convert_json_to_triples(
+                            item,
+                            parent_node,
+                            parent_object_id,
+                            parent_node_path,
+                            property_name,
+                        )
             else:
                 # Convert scalar values to RDF literals
                 if isinstance(data, bool):
@@ -513,7 +585,10 @@ class ExtractGraphFromAvevaPiAssetFramework(Step):
             containing data dump in XML format",
         ),
         Configurable(
-            name="graph_name", value="solution", label="The name of target graph.", options=["source", "solution"]
+            name="graph_name",
+            value="solution",
+            label="The name of target graph.",
+            options=["source", "solution"],
         ),
         Configurable(
             name="root_node_external_id",
@@ -535,7 +610,11 @@ class ExtractGraphFromAvevaPiAssetFramework(Step):
             value="http://purl.org/cognite/neat#",
             label="Namespace to be used for the generated objects.",
         ),
-        Configurable(name="namespace_prefix", value="neat", label="The prefix to be used for the namespace."),
+        Configurable(
+            name="namespace_prefix",
+            value="neat",
+            label="The prefix to be used for the namespace.",
+        ),
     ]
 
     def add_root_asset_to_source_graph(self) -> str:
@@ -556,7 +635,10 @@ class ExtractGraphFromAvevaPiAssetFramework(Step):
         if source_file := self.configs["file_name"]:
             source_pi_dump = Path(self.data_store_path) / source_file
         else:
-            return FlowMessage(output_text="No source file specified", next_step_ids=["step_error_handler"])
+            return FlowMessage(
+                output_text="No source file specified",
+                next_step_ids=["step_error_handler"],
+            )
 
         # self.graph.bind
         if self.configs["graph_name"] == "solution":
@@ -586,7 +668,11 @@ class ExtractGraphFromAvevaPiAssetFramework(Step):
             self.graph_store.graph.add((rdf_instance_id, URIRef(self.ns + "Path"), Literal(new_element_path)))
             if parent_element_id:
                 self.graph_store.graph.add(
-                    (rdf_instance_id, URIRef(self.ns + "hasParent"), URIRef(self.ns + parent_element_id))
+                    (
+                        rdf_instance_id,
+                        URIRef(self.ns + "hasParent"),
+                        URIRef(self.ns + parent_element_id),
+                    )
                 )
             for child in af_element:
                 if child.tag == "AFAttribute":
@@ -595,7 +681,13 @@ class ExtractGraphFromAvevaPiAssetFramework(Step):
                     pass
                 else:
                     try:
-                        self.graph_store.graph.add((rdf_instance_id, URIRef(self.ns + child.tag), Literal(child.text)))
+                        self.graph_store.graph.add(
+                            (
+                                rdf_instance_id,
+                                URIRef(self.ns + child.tag),
+                                Literal(child.text),
+                            )
+                        )
                     except Exception as e:
                         logging.error(f"Error parsing AFAttribute {name} : {e}")
 
@@ -610,7 +702,11 @@ class ExtractGraphFromAvevaPiAssetFramework(Step):
             self.graph_store.graph.add((rdf_instance_id, URIRef(self.ns + "Path"), Literal(new_element_path)))
             if parent_element_id:
                 self.graph_store.graph.add(
-                    (rdf_instance_id, URIRef(self.ns + "hasParent"), URIRef(self.ns + parent_element_id))
+                    (
+                        rdf_instance_id,
+                        URIRef(self.ns + "hasParent"),
+                        URIRef(self.ns + parent_element_id),
+                    )
                 )
 
             for child in af_element:
@@ -624,7 +720,13 @@ class ExtractGraphFromAvevaPiAssetFramework(Step):
                     counter += 1
                     process_af_element(child, new_element_path, element_id)
                 else:
-                    self.graph_store.graph.add((rdf_instance_id, URIRef(self.ns + child.tag), Literal(child.text)))
+                    self.graph_store.graph.add(
+                        (
+                            rdf_instance_id,
+                            URIRef(self.ns + child.tag),
+                            Literal(child.text),
+                        )
+                    )
 
             if template:
                 self.graph_store.graph.add((rdf_instance_id, RDF.type, URIRef(self.ns + template)))