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

cognite/neat/__init__.py

@@ -1,4 +1,5 @@
+from ._session import NeatSession
+from ._utils.auth import get_cognite_client
 from ._version import __version__
-from .utils.auth import get_cognite_client
 
-__all__ = ["__version__", "get_cognite_client"]
+__all__ = ["__version__", "get_cognite_client", "NeatSession"]

cognite/neat/{app → _app}/api/configuration.py

@@ -7,13 +7,13 @@ from cognite.client import CogniteClient
 from fastapi import FastAPI
 
 from cognite import neat
-from cognite.neat.config import Config, configure_logging, copy_examples_to_directory, create_data_dir_structure
-from cognite.neat.constants import PACKAGE_DIRECTORY
-from cognite.neat.workflows.cdf_store import CdfStore
-from cognite.neat.workflows.manager import WorkflowManager
-from cognite.neat.workflows.triggers import TriggerManager
+from cognite.neat._config import Config, configure_logging, copy_examples_to_directory, create_data_dir_structure
+from cognite.neat._constants import PACKAGE_DIRECTORY
+from cognite.neat._workflows.cdf_store import CdfStore
+from cognite.neat._workflows.manager import WorkflowManager
+from cognite.neat._workflows.triggers import TriggerManager
 
-UI_PATH = PACKAGE_DIRECTORY / "app" / "ui" / "neat-app" / "build"
+UI_PATH = PACKAGE_DIRECTORY / "_app" / "ui" / "neat-app" / "build"
 
 
 class NeatApp:
@@ -87,7 +87,9 @@ def create_neat_app() -> NeatApp:
 
     configure_logging(config.log_level, config.log_format)
     logging.info(f" Starting NEAT version {neat.__version__}")
-    logging.debug(f" Config: {config.model_dump(exclude={'cdf_client': {'client_secret': ...}})}")
+    logging.debug(
+        f" Config: {config.model_dump(exclude={'cdf_client': {'client_secret': ...}})}"  # type: ignore[arg-type]
+    )
 
     return NeatApp(config)
 

cognite/neat/_app/api/context_manager/__init__.py

@@ -0,0 +1,3 @@
+from cognite.neat._app.api.context_manager.manager import lifespan
+
+__all__ = ["lifespan"]

cognite/neat/{app → _app}/api/context_manager/manager.py

@@ -3,7 +3,7 @@ from contextlib import asynccontextmanager
 
 from fastapi import FastAPI
 
-from cognite.neat.app.api.configuration import NEAT_APP
+from cognite.neat._app.api.configuration import NEAT_APP
 
 
 @asynccontextmanager

cognite/neat/{app → _app}/api/explorer.py

@@ -7,16 +7,16 @@ from fastapi.responses import RedirectResponse
 from fastapi.staticfiles import StaticFiles
 
 from cognite import neat
-from cognite.neat.app.api.asgi.metrics import prometheus_app
-from cognite.neat.app.api.configuration import NEAT_APP, UI_PATH
-from cognite.neat.app.api.context_manager import lifespan
-from cognite.neat.app.api.routers import (
+from cognite.neat._app.api.asgi.metrics import prometheus_app
+from cognite.neat._app.api.configuration import NEAT_APP, UI_PATH
+from cognite.neat._app.api.context_manager import lifespan
+from cognite.neat._app.api.routers import (
     configuration,
     crud,
     metrics,
     workflows,
 )
-from cognite.neat.app.api.utils.logging import EndpointFilter
+from cognite.neat._app.api.utils.logging import EndpointFilter
 
 app = FastAPI(title="Neat", lifespan=lifespan)
 

cognite/neat/{app → _app}/api/routers/configuration.py

@@ -4,8 +4,8 @@ from pathlib import Path
 
 from fastapi import APIRouter
 
-from cognite.neat.app.api.configuration import NEAT_APP
-from cognite.neat.config import Config
+from cognite.neat._app.api.configuration import NEAT_APP
+from cognite.neat._config import Config
 
 router = APIRouter()
 

cognite/neat/{app → _app}/api/routers/crud.py

@@ -6,10 +6,10 @@ import shutil
 
 from fastapi import APIRouter, UploadFile
 
-from cognite.neat.app.api.configuration import NEAT_APP
-from cognite.neat.config import Config
-from cognite.neat.workflows.model import FlowMessage
-from cognite.neat.workflows.utils import get_file_hash
+from cognite.neat._app.api.configuration import NEAT_APP
+from cognite.neat._config import Config
+from cognite.neat._workflows.model import FlowMessage
+from cognite.neat._workflows.utils import get_file_hash
 
 router = APIRouter()
 

cognite/neat/{app → _app}/api/routers/workflows.py

@@ -5,15 +5,15 @@ from typing import cast
 from fastapi import APIRouter, Depends, HTTPException, Request, UploadFile
 from fastapi.responses import FileResponse, JSONResponse
 
-from cognite.neat.app.api.configuration import NEAT_APP
-from cognite.neat.app.api.data_classes.rest import (
+from cognite.neat._app.api.configuration import NEAT_APP
+from cognite.neat._app.api.data_classes.rest import (
     RunWorkflowRequest,
 )
-from cognite.neat.workflows import WorkflowFullStateReport
-from cognite.neat.workflows.base import WorkflowDefinition
-from cognite.neat.workflows.model import FlowMessage
-from cognite.neat.workflows.steps.data_contracts import NeatGraph
-from cognite.neat.workflows.steps.step_model import DataContract
+from cognite.neat._workflows import WorkflowFullStateReport
+from cognite.neat._workflows.base import WorkflowDefinition
+from cognite.neat._workflows.model import FlowMessage
+from cognite.neat._workflows.steps.data_contracts import NeatGraph
+from cognite.neat._workflows.steps.step_model import DataContract
 
 router = APIRouter()
 

cognite/neat/{app → _app}/main.py

@@ -10,7 +10,7 @@ app = FastAPI()
 def run():
     (Path.cwd() / "data").mkdir(parents=True, exist_ok=True)
     webbrowser.open("http://localhost:8000/")
-    uvicorn.run("cognite.neat.app.api.explorer:app", host="0.0.0.0", port=8000, reload=False)
+    uvicorn.run("cognite.neat._app.api.explorer:app", host="0.0.0.0", port=8000, reload=False)
 
 
 if __name__ == "__main__":