cognite-neat 0.92.2__py3-none-any.whl → 0.93.0__py3-none-any.whl

cognite/neat/{store → _store}/_base.py

@@ -3,22 +3,22 @@ import warnings
 from collections.abc import Iterable
 from datetime import datetime, timezone
 from pathlib import Path
-from typing import cast
+from typing import Any, cast
 
 import pandas as pd
+from pandas import Index
 from rdflib import Graph, Namespace, URIRef
 from rdflib.plugins.stores.sparqlstore import SPARQLUpdateStore
 
-from cognite.neat.constants import DEFAULT_NAMESPACE
-from cognite.neat.graph._shared import MIMETypes
-from cognite.neat.graph.extractors import RdfFileExtractor, TripleExtractors
-from cognite.neat.graph.models import InstanceType, Triple
-from cognite.neat.graph.queries import Queries
-from cognite.neat.graph.transformers import Transformers
-from cognite.neat.rules.analysis import InformationAnalysis
-from cognite.neat.rules.models import InformationRules
-from cognite.neat.rules.models.entities import ClassEntity
-from cognite.neat.utils.auxiliary import local_import
+from cognite.neat._constants import DEFAULT_NAMESPACE
+from cognite.neat._graph.extractors import RdfFileExtractor, TripleExtractors
+from cognite.neat._graph.models import InstanceType, Triple
+from cognite.neat._graph.queries import Queries
+from cognite.neat._graph.transformers import Transformers
+from cognite.neat._rules.analysis import InformationAnalysis
+from cognite.neat._rules.models import InformationRules
+from cognite.neat._rules.models.entities import ClassEntity
+from cognite.neat._utils.auxiliary import local_import
 
 from ._provenance import Change, Provenance
 
@@ -158,20 +158,29 @@ class NeatGraphStore:
 
     def write(self, extractor: TripleExtractors) -> None:
         _start = datetime.now(timezone.utc)
+        success = True
 
-        if isinstance(extractor, RdfFileExtractor):
-            self._parse_file(extractor.filepath, extractor.mime_type, extractor.base_uri)
+        if isinstance(extractor, RdfFileExtractor) and not extractor.issue_list.has_errors:
+            self._parse_file(extractor.filepath, cast(str, extractor.mime_type), extractor.base_uri)
+        elif isinstance(extractor, RdfFileExtractor):
+            success = False
+            issue_text = "\n".join([issue.as_message() for issue in extractor.issue_list])
+            warnings.warn(
+                f"Cannot write to graph store with {type(extractor).__name__}, errors found in file:\n{issue_text}",
+                stacklevel=2,
+            )
         else:
             self._add_triples(extractor.extract())
 
-        self.provenance.append(
-            Change.record(
-                activity=f"{type(extractor).__name__}",
-                start=_start,
-                end=datetime.now(timezone.utc),
-                description=f"Extracted triples to graph store using {type(extractor).__name__}",
+        if success:
+            self.provenance.append(
+                Change.record(
+                    activity=f"{type(extractor).__name__}",
+                    start=_start,
+                    end=datetime.now(timezone.utc),
+                    description=f"Extracted triples to graph store using {type(extractor).__name__}",
+                )
             )
-        )
 
     def read(self, class_: str) -> Iterable[tuple[str, dict[str | InstanceType, list[str]]]]:
         """Read instances for given view from the graph store."""
@@ -234,7 +243,7 @@ class NeatGraphStore:
     def _parse_file(
         self,
         filepath: Path,
-        mime_type: MIMETypes = "application/rdf+xml",
+        mime_type: str = "application/rdf+xml",
         base_uri: URIRef | None = None,
     ) -> None:
         """Imports graph data from file.
@@ -342,16 +351,38 @@ class NeatGraphStore:
     def _repr_html_(self) -> str:
         provenance = self.provenance._repr_html_()
         summary: pd.DataFrame = self.summary
-        summary_text = (
-            "<br /><strong>Graph is empty</strong><br />"
-            if summary.empty
-            else f"<br /><strong>Graph content</strong><br />{cast(pd.DataFrame, summary)._repr_html_()}"  # type: ignore[operator]
-        )
 
-        return (
-            f"<strong>{type(self).__name__}</strong> A graph store is a container for storing triples. "
-            "It can be queried and transformed to extract information.<br />"
-            "<strong>Provenance</strong> Provenance is a record of changes that have occurred in the graph store.<br />"
-            f"{provenance}"
-            f"{summary_text}"
+        if summary.empty:
+            summary_text = "<br /><strong>Graph is empty</strong><br />"
+        else:
+            summary_text = (
+                "<br /><strong>Overview</strong>:"  # type: ignore
+                f"<ul><li>{len(summary)} types</strong></li>"
+                f"<li>{sum(summary['Occurrence'])} instances</strong></li></ul>"
+                f"{cast(pd.DataFrame, self._shorten_summary(summary))._repr_html_()}"  # type: ignore[operator]
+            )
+
+        return f"{summary_text}" f"{provenance}"
+
+    def _shorten_summary(self, summary: pd.DataFrame) -> pd.DataFrame:
+        """Shorten summary to top 5 types by occurrence."""
+        top_5_rows = summary.head(5)
+        last_row = summary.tail(1)
+
+        indexes = [
+            *top_5_rows.index.tolist(),
+            "...",
+            *last_row.index.tolist(),
+        ]
+
+        shorter_summary = pd.concat(
+            [
+                top_5_rows,
+                pd.DataFrame([["..."] * summary.shape[1]], columns=summary.columns),
+                last_row,
+            ],
+            ignore_index=True,
         )
+        shorter_summary.index = cast(Index[Any], indexes)
+
+        return shorter_summary

cognite/neat/{store → _store}/_provenance.py

@@ -15,8 +15,8 @@ from typing import TypeVar
 
 from rdflib import PROV, RDF, Literal, URIRef
 
+from cognite.neat._constants import DEFAULT_NAMESPACE
 from cognite.neat._shared import FrozenNeatObject, NeatList
-from cognite.neat.constants import DEFAULT_NAMESPACE
 
 
 @dataclass(frozen=True)
@@ -114,3 +114,13 @@ class Provenance(NeatList[Change]):
 
     def __setitem__(self, *args, **kwargs):
         raise TypeError("Cannot modify change from provenance")
+
+    def _repr_html_(self) -> str:
+        text = "<br /><strong>Provenance</strong>:<ul>"
+
+        for change in self:
+            text += f"<li>{change.description}</li>"
+
+        text += "</ul>"
+
+        return text

cognite/neat/{utils → _utils}/auth.py

@@ -9,7 +9,7 @@ from cognite.client import ClientConfig, CogniteClient
 from cognite.client.credentials import CredentialProvider, OAuthClientCredentials, OAuthInteractive, Token
 
 from cognite.neat import _version
-from cognite.neat.utils.auxiliary import local_import
+from cognite.neat._utils.auxiliary import local_import
 
 __all__ = ["get_cognite_client", "EnvironmentVariables"]
 
@@ -18,7 +18,18 @@ _VALID_LOGIN_FLOWS = get_args(_LOGIN_FLOW)
 _CLIENT_NAME = f"CogniteNeat:{_version.__version__}"
 
 
-def get_cognite_client(env_file_name: str = ".env") -> CogniteClient:
+def get_cognite_client(env_file_name: str) -> CogniteClient:
+    """Instantiate a CogniteClient using environment variables. If the environment variables are not set, the user will
+    be prompted to enter them.
+
+    Args:
+        env_file_name: The name of the .env file to look for in the repository root. If the file is found, the variables
+            will be loaded from the file. If the file is not found, the user will be prompted to enter the variables.
+
+    Returns:
+        CogniteClient: A CogniteClient instance.
+
+    """
     if not env_file_name.endswith(".env"):
         raise ValueError("env_file_name must end with '.env'")
     with suppress(KeyError):
@@ -320,5 +331,5 @@ def _env_in_gitignore(repo_root: Path, env_file_name: str) -> bool:
 
 
 if __name__ == "__main__":
-    c = get_cognite_client()
+    c = get_cognite_client(".env")
     print(c.iam.token.inspect())

cognite/neat/{utils → _utils}/auxiliary.py

@@ -10,7 +10,7 @@ from types import ModuleType
 
 from cognite.client.exceptions import CogniteDuplicatedError, CogniteReadTimeout
 
-from cognite.neat.issues.errors import NeatImportError
+from cognite.neat._issues.errors import NeatImportError
 
 
 def local_import(module: str, extra: str) -> ModuleType:

cognite/neat/{utils → _utils}/cdf/loaders/_data_modeling.py

@@ -31,6 +31,7 @@ from cognite.client.data_classes.data_modeling import (
 from cognite.client.data_classes.data_modeling.ids import (
     ContainerId,
     DataModelId,
+    NodeId,
     ViewId,
 )
 from cognite.client.exceptions import CogniteAPIError
@@ -115,9 +116,14 @@ class SpaceLoader(DataModelingLoader[str, SpaceApply, Space, SpaceApplyList, Spa
         nodes = self.client.data_modeling.instances.list(
             "node", limit=-1, filter=filters.Equals(["node", "space"], space)
         )
-        if nodes:
-            instances = self.client.data_modeling.instances.delete(nodes=nodes.as_ids())
+        node_types = {NodeId(node.type.space, node.type.external_id) for node in nodes if node.type}
+        node_data = set(nodes.as_ids()) - node_types
+        if node_data:
+            instances = self.client.data_modeling.instances.delete(nodes=list(node_data))
             print(f"Deleted {len(instances.nodes)} nodes")
+        if node_types:
+            instances = self.client.data_modeling.instances.delete(nodes=list(node_types))
+            print(f"Deleted {len(instances.nodes)} node types")
         views = self.client.data_modeling.views.list(limit=-1, space=space)
         if views:
             deleted_views = self.client.data_modeling.views.delete(views.as_ids())

cognite/neat/{utils → _utils}/cdf/loaders/_ingestion.py

@@ -15,7 +15,7 @@ from cognite.client.data_classes import (
 from cognite.client.exceptions import CogniteAPIError
 from cognite.client.utils.useful_types import SequenceNotStr
 
-from cognite.neat.utils.cdf.data_classes import RawTable, RawTableID, RawTableList, RawTableWrite, RawTableWriteList
+from cognite.neat._utils.cdf.data_classes import RawTable, RawTableID, RawTableList, RawTableWrite, RawTableWriteList
 
 from ._base import ResourceLoader
 

cognite/neat/{utils → _utils}/upload.py

@@ -3,8 +3,8 @@ from dataclasses import dataclass, field
 from functools import total_ordering
 from typing import Any, Generic
 
+from cognite.neat._issues import NeatIssueList
 from cognite.neat._shared import T_ID, NeatList, NeatObject
-from cognite.neat.issues import NeatIssueList
 
 
 @total_ordering

cognite/neat/_version.py

@@ -1 +1 @@
-__version__ = "0.92.2"
+__version__ = "0.93.0"

cognite/neat/_workflows/__init__.py

@@ -0,0 +1,17 @@
+from cognite.neat._workflows.base import BaseWorkflow
+from cognite.neat._workflows.manager import WorkflowManager
+from cognite.neat._workflows.model import (
+    FlowMessage,
+    WorkflowFullStateReport,
+    WorkflowStepDefinition,
+    WorkflowStepEvent,
+)
+
+__all__ = [
+    "BaseWorkflow",
+    "WorkflowStepDefinition",
+    "WorkflowManager",
+    "WorkflowStepEvent",
+    "WorkflowFullStateReport",
+    "FlowMessage",
+]

cognite/neat/{workflows → _workflows}/base.py

@@ -11,13 +11,13 @@ import yaml
 from cognite.client import ClientConfig, CogniteClient
 from prometheus_client import Gauge
 
-from cognite.neat.app.monitoring.metrics import NeatMetricsCollector
-from cognite.neat.config import Config
-from cognite.neat.issues.errors import WorkflowConfigurationNotSetError, WorkflowStepOutputError
-from cognite.neat.utils.auxiliary import retry_decorator
-from cognite.neat.workflows import cdf_store, utils
-from cognite.neat.workflows.cdf_store import CdfStore
-from cognite.neat.workflows.model import (
+from cognite.neat._app.monitoring.metrics import NeatMetricsCollector
+from cognite.neat._config import Config
+from cognite.neat._issues.errors import WorkflowConfigurationNotSetError, WorkflowStepOutputError
+from cognite.neat._utils.auxiliary import retry_decorator
+from cognite.neat._workflows import cdf_store, utils
+from cognite.neat._workflows.cdf_store import CdfStore
+from cognite.neat._workflows.model import (
     FlowMessage,
     StepExecutionStatus,
     StepType,
@@ -31,9 +31,9 @@ from cognite.neat.workflows.model import (
     WorkflowStepEvent,
     WorkflowSystemComponent,
 )
-from cognite.neat.workflows.steps.step_model import DataContract
-from cognite.neat.workflows.steps_registry import StepsRegistry
-from cognite.neat.workflows.tasks import WorkflowTaskBuilder
+from cognite.neat._workflows.steps.step_model import DataContract
+from cognite.neat._workflows.steps_registry import StepsRegistry
+from cognite.neat._workflows.tasks import WorkflowTaskBuilder
 
 summary_metrics = Gauge("neat_workflow_summary_metrics", "Workflow execution summary metrics", ["wf_name", "name"])
 steps_metrics = Gauge("neat_workflow_steps_metrics", "Workflow step level metrics", ["wf_name", "step_name", "name"])

cognite/neat/{workflows → _workflows}/cdf_store.py

@@ -11,9 +11,9 @@ from cognite.client.data_classes import Event, FileMetadataUpdate
 from fastapi.encoders import jsonable_encoder
 from pydantic import BaseModel
 
-from cognite.neat.issues.errors import WorkflowConfigurationNotSetError
-from cognite.neat.workflows.model import WorkflowFullStateReport, WorkflowState, WorkflowStepEvent
-from cognite.neat.workflows.utils import get_file_hash
+from cognite.neat._issues.errors import WorkflowConfigurationNotSetError
+from cognite.neat._workflows.model import WorkflowFullStateReport, WorkflowState, WorkflowStepEvent
+from cognite.neat._workflows.utils import get_file_hash
 
 
 class NeatCdfResource(BaseModel):

cognite/neat/{workflows → _workflows}/examples/Export_DMS/workflow.yaml

@@ -1,89 +1,89 @@
-configs: []
-description: null
-implementation_module: null
-name: Export DMS
-steps:
--   complex_configs: {}
-    configs: {}
-    description: null
-    enabled: true
-    id: step_861205
-    label: Upload Rules Spreadsheet
-    max_retries: 0
-    method: null
-    params:
-        file_type: rules
-    retry_delay: 3
-    stype: file_uploader
-    system_component_id: null
-    transition_to:
-    - step_295479
-    trigger: true
-    ui_config:
-        pos_x: 629
-        pos_y: 57
--   complex_configs: {}
-    configs:
-        File name: ''
-        Report formatter: BasicHTML
-        Role: infer
-    description: null
-    enabled: true
-    id: step_295479
-    label: Validate
-    max_retries: 0
-    method: ExcelToRules
-    params: {}
-    retry_delay: 3
-    stype: stdstep
-    system_component_id: null
-    transition_to:
-    - step_50885
-    trigger: false
-    ui_config:
-        pos_x: 629
-        pos_y: 161
--   complex_configs:
-        Components:
-            containers: true
-            data_models: true
-            spaces: true
-            views: true
-    configs:
-        Dry run: 'False'
-        Existing component handling: update
-        Multi-space components create: 'True'
-    description: null
-    enabled: true
-    id: step_50885
-    label: Export Data Model to CDF
-    max_retries: 0
-    method: RulesToDMS
-    params: {}
-    retry_delay: 3
-    stype: stdstep
-    system_component_id: null
-    transition_to:
-    - step_171560
-    trigger: false
-    ui_config:
-        pos_x: 629
-        pos_y: 243
--   complex_configs: {}
-    configs: {}
-    description: null
-    enabled: true
-    id: step_171560
-    label: Export Transformations
-    max_retries: 0
-    method: RulesToCDFTransformations
-    params: {}
-    retry_delay: 3
-    stype: stdstep
-    system_component_id: null
-    transition_to: []
-    trigger: false
-    ui_config:
-        pos_x: 629
-        pos_y: 342
-system_components: []
+configs: []
+description: null
+implementation_module: null
+name: Export DMS
+steps:
+-   complex_configs: {}
+    configs: {}
+    description: null
+    enabled: true
+    id: step_861205
+    label: Upload Rules Spreadsheet
+    max_retries: 0
+    method: null
+    params:
+        file_type: rules
+    retry_delay: 3
+    stype: file_uploader
+    system_component_id: null
+    transition_to:
+    - step_295479
+    trigger: true
+    ui_config:
+        pos_x: 629
+        pos_y: 57
+-   complex_configs: {}
+    configs:
+        File name: ''
+        Report formatter: BasicHTML
+        Role: infer
+    description: null
+    enabled: true
+    id: step_295479
+    label: Validate
+    max_retries: 0
+    method: ExcelToRules
+    params: {}
+    retry_delay: 3
+    stype: stdstep
+    system_component_id: null
+    transition_to:
+    - step_50885
+    trigger: false
+    ui_config:
+        pos_x: 629
+        pos_y: 161
+-   complex_configs:
+        Components:
+            containers: true
+            data_models: true
+            spaces: true
+            views: true
+    configs:
+        Dry run: 'False'
+        Existing component handling: update
+        Multi-space components create: 'True'
+    description: null
+    enabled: true
+    id: step_50885
+    label: Export Data Model to CDF
+    max_retries: 0
+    method: RulesToDMS
+    params: {}
+    retry_delay: 3
+    stype: stdstep
+    system_component_id: null
+    transition_to:
+    - step_171560
+    trigger: false
+    ui_config:
+        pos_x: 629
+        pos_y: 243
+-   complex_configs: {}
+    configs: {}
+    description: null
+    enabled: true
+    id: step_171560
+    label: Export Transformations
+    max_retries: 0
+    method: RulesToCDFTransformations
+    params: {}
+    retry_delay: 3
+    stype: stdstep
+    system_component_id: null
+    transition_to: []
+    trigger: false
+    ui_config:
+        pos_x: 629
+        pos_y: 342
+system_components: []

cognite/neat/{workflows → _workflows}/manager.py

@@ -9,12 +9,12 @@ from cognite.client import CogniteClient
 from prometheus_client import Gauge
 from pydantic import BaseModel
 
-from cognite.neat.config import Config
-from cognite.neat.workflows import BaseWorkflow
-from cognite.neat.workflows.base import WorkflowDefinition
-from cognite.neat.workflows.model import FlowMessage, InstanceStartMethod, WorkflowState, WorkflowStepDefinition
-from cognite.neat.workflows.steps_registry import StepsRegistry
-from cognite.neat.workflows.tasks import WorkflowTaskBuilder
+from cognite.neat._config import Config
+from cognite.neat._workflows import BaseWorkflow
+from cognite.neat._workflows.base import WorkflowDefinition
+from cognite.neat._workflows.model import FlowMessage, InstanceStartMethod, WorkflowState, WorkflowStepDefinition
+from cognite.neat._workflows.steps_registry import StepsRegistry
+from cognite.neat._workflows.tasks import WorkflowTaskBuilder
 
 live_workflow_instances = Gauge("neat_workflow_live_instances", "Count of live workflow instances", ["itype"])
 

cognite/neat/{workflows → _workflows}/steps/data_contracts.py

@@ -9,14 +9,14 @@ from cognite.client.data_classes import (
 )
 from cognite.client.data_classes.data_modeling import EdgeApply, NodeApply
 
-from cognite.neat.rules.models import (
+from cognite.neat._rules.models import (
     AssetRules,
     DMSRules,
     DomainRules,
     InformationRules,
 )
-from cognite.neat.store import NeatGraphStore
-from cognite.neat.workflows.steps.step_model import DataContract
+from cognite.neat._store import NeatGraphStore
+from cognite.neat._workflows.steps.step_model import DataContract
 
 
 class MultiRuleData(DataContract):

cognite/neat/{workflows → _workflows}/steps/lib/current/graph_extractor.py

@@ -2,16 +2,13 @@ import json
 from pathlib import Path
 from typing import ClassVar, cast
 
-from rdflib import URIRef
-
-from cognite.neat.constants import DEFAULT_NAMESPACE
-from cognite.neat.graph.extractors import RdfFileExtractor
-from cognite.neat.graph.extractors._mock_graph_generator import MockGraphGenerator
-from cognite.neat.issues.errors import WorkflowStepNotInitializedError
-from cognite.neat.rules._shared import DMSRules, InformationRules
-from cognite.neat.workflows.model import FlowMessage, StepExecutionStatus
-from cognite.neat.workflows.steps.data_contracts import MultiRuleData, NeatGraph
-from cognite.neat.workflows.steps.step_model import Configurable, Step
+from cognite.neat._graph.extractors import RdfFileExtractor
+from cognite.neat._graph.extractors._mock_graph_generator import MockGraphGenerator
+from cognite.neat._issues.errors import WorkflowStepNotInitializedError
+from cognite.neat._rules._shared import DMSRules, InformationRules
+from cognite.neat._workflows.model import FlowMessage, StepExecutionStatus
+from cognite.neat._workflows.steps.data_contracts import MultiRuleData, NeatGraph
+from cognite.neat._workflows.steps.step_model import Configurable, Step
 
 __all__ = ["GraphFromRdfFile", "GraphFromMockData"]
 
@@ -83,25 +80,7 @@ class GraphFromRdfFile(Step):
             name="File path",
             value="source-graphs/source-graph-dump.xml",
             label="File name of source graph data dump in RDF format",
-        ),
-        Configurable(
-            name="MIME type",
-            value="application/rdf+xml",
-            label="MIME type of file containing RDF graph",
-            options=[
-                "application/rdf+xml",
-                "text/turtle",
-                "application/n-triples",
-                "application/n-quads",
-                "application/trig",
-            ],
-        ),
-        Configurable(
-            name="Add base URI",
-            value="True",
-            label="Whether to add base URI to graph in case if entity ids are relative",
-            options=["True", "False"],
-        ),
+        )
     ]
 
     def run(self, store: NeatGraph) -> FlowMessage:  # type: ignore[override, syntax]
@@ -112,8 +91,6 @@ class GraphFromRdfFile(Step):
             store.graph.write(
                 RdfFileExtractor(  # type: ignore[abstract]
                     filepath=self.data_store_path / Path(source_file),
-                    mime_type=self.configs["MIME type"],  # type: ignore[arg-type]
-                    base_uri=(URIRef(DEFAULT_NAMESPACE) if self.configs["Add base URI"] == "True" else None),
                 )
             )
 

cognite/neat/{workflows → _workflows}/steps/lib/current/graph_loader.py

@@ -2,10 +2,10 @@ import time
 from pathlib import Path
 from typing import ClassVar
 
-from cognite.neat.issues.errors import WorkflowStepNotInitializedError
-from cognite.neat.workflows.model import FlowMessage
-from cognite.neat.workflows.steps.data_contracts import NeatGraph
-from cognite.neat.workflows.steps.step_model import Configurable, Step
+from cognite.neat._issues.errors import WorkflowStepNotInitializedError
+from cognite.neat._workflows.model import FlowMessage
+from cognite.neat._workflows.steps.data_contracts import NeatGraph
+from cognite.neat._workflows.steps.step_model import Configurable, Step
 
 __all__ = [
     "GraphToRdfFile",

cognite/neat/{workflows → _workflows}/steps/lib/current/graph_store.py

@@ -1,12 +1,12 @@
 from typing import ClassVar
 
-from cognite.neat.store import NeatGraphStore
-from cognite.neat.workflows.model import FlowMessage
-from cognite.neat.workflows.steps.data_contracts import (
+from cognite.neat._store import NeatGraphStore
+from cognite.neat._workflows.model import FlowMessage
+from cognite.neat._workflows.steps.data_contracts import (
     MultiRuleData,
     NeatGraph,
 )
-from cognite.neat.workflows.steps.step_model import Configurable, Step
+from cognite.neat._workflows.steps.step_model import Configurable, Step
 
 __all__ = ["GraphStoreConfiguration"]
 

cognite/neat/{workflows → _workflows}/steps/lib/current/rules_exporter.py

@@ -2,20 +2,20 @@ import time
 from pathlib import Path
 from typing import ClassVar, Literal, cast
 
-from cognite.neat.issues.errors import WorkflowStepNotInitializedError
-from cognite.neat.rules import exporters
-from cognite.neat.rules._shared import DMSRules, InformationRules, VerifiedRules
-from cognite.neat.rules.models import AssetRules, RoleTypes
-from cognite.neat.rules.transformers import (
+from cognite.neat._issues.errors import WorkflowStepNotInitializedError
+from cognite.neat._rules import exporters
+from cognite.neat._rules._shared import DMSRules, InformationRules, VerifiedRules
+from cognite.neat._rules.models import AssetRules, RoleTypes
+from cognite.neat._rules.transformers import (
     AssetToInformation,
     DMSToInformation,
     InformationToAsset,
     InformationToDMS,
     RulesPipeline,
 )
-from cognite.neat.workflows.model import FlowMessage, StepExecutionStatus
-from cognite.neat.workflows.steps.data_contracts import CogniteClient, MultiRuleData
-from cognite.neat.workflows.steps.step_model import Configurable, Step
+from cognite.neat._workflows.model import FlowMessage, StepExecutionStatus
+from cognite.neat._workflows.steps.data_contracts import CogniteClient, MultiRuleData
+from cognite.neat._workflows.steps.step_model import Configurable, Step
 
 __all__ = [
     "RulesToDMS",