cognite-toolkit 0.7.56__py3-none-any.whl → 0.7.58__py3-none-any.whl

cognite_toolkit/_cdf_tk/apps/_auth_app.py

@@ -54,6 +54,7 @@ class AuthApp(typer.Typer):
         "projectsAcl": ["LIST", "READ"],
         "groupsAcl": ["LIST", "READ", "CREATE", "UPDATE", "DELETE"]
         """
+        # We do not pass in a client here as this is typically used to create the .env file needed for authentication.
         cmd = AuthCommand()
         cmd.run(
             lambda: cmd.init(
@@ -92,9 +93,8 @@ class AuthApp(typer.Typer):
         ] = False,
     ) -> None:
         """Verify that the current user or service principal has the required capabilities to run the CDF Toolkit commands."""
-        cmd = AuthCommand()
         client = EnvironmentVariables.create_from_environment().get_client()
-
+        cmd = AuthCommand(client=client)
         cmd.run(
             lambda: cmd.verify(
                 client,

cognite_toolkit/_cdf_tk/apps/_core_app.py

@@ -210,9 +210,9 @@ class CoreApp(typer.Typer):
             print_warning = False
 
         cmd = (
-            BuildCommandV2(print_warning=print_warning)
+            BuildCommandV2(print_warning=print_warning, client=client)
             if Flags.v08.is_enabled()
-            else BuildCommand(print_warning=print_warning)
+            else BuildCommand(print_warning=print_warning, client=client)
         )
         cmd.run(
             lambda: cmd.execute(
@@ -302,8 +302,8 @@ class CoreApp(typer.Typer):
         ] = False,
     ) -> None:
         """Deploys the configuration files in the build directory to the CDF project."""
-        cmd = DeployCommand(print_warning=True)
         env_vars = EnvironmentVariables.create_from_environment()
+        cmd = DeployCommand(print_warning=True, client=env_vars.get_client())
         cmd.run(
             lambda: cmd.deploy_build_directory(
                 env_vars=env_vars,
@@ -371,8 +371,8 @@ class CoreApp(typer.Typer):
     ) -> None:
         """Cleans the resources in the build directory from the CDF project."""
         # Override cluster and project from the options/env variables
-        cmd = CleanCommand(print_warning=True)
         env = EnvironmentVariables.create_from_environment()
+        cmd = CleanCommand(print_warning=True, client=env.get_client())
         cmd.run(
             lambda: cmd.execute(
                 env,

cognite_toolkit/_cdf_tk/apps/_dev_app.py

@@ -1,3 +1,4 @@
+import contextlib
 from pathlib import Path
 from typing import Annotated, Any
 
@@ -5,8 +6,10 @@ import typer
 from rich import print
 
 from cognite_toolkit._cdf_tk.cdf_toml import CDFToml
+from cognite_toolkit._cdf_tk.client import ToolkitClient
 from cognite_toolkit._cdf_tk.commands import ResourcesCommand
 from cognite_toolkit._cdf_tk.feature_flags import FeatureFlag, Flags
+from cognite_toolkit._cdf_tk.utils.auth import EnvironmentVariables
 
 from ._run import RunApp
 
@@ -74,7 +77,13 @@ class DevApp(typer.Typer):
         ] = CDF_TOML.cdf.default_organization_dir,
     ) -> None:
         """create resource YAMLs."""
-        cmd = ResourcesCommand()
+        client: ToolkitClient | None = None
+        with contextlib.redirect_stdout(None), contextlib.suppress(Exception):
+            # Try to load client if possible, but ignore errors.
+            # This is only used for logging purposes in the command.
+            client = EnvironmentVariables.create_from_environment().get_client()
+
+        cmd = ResourcesCommand(client=client)
         cmd.run(
             lambda: cmd.create(
                 organization_dir=organization_dir,

cognite_toolkit/_cdf_tk/apps/_download_app.py

@@ -193,9 +193,9 @@ class DownloadApp(typer.Typer):
         ] = False,
     ) -> None:
         """This command will download RAW tables from CDF into a temporary directory."""
-        cmd = DownloadCommand()
-
         client = EnvironmentVariables.create_from_environment().get_client()
+        cmd = DownloadCommand(client=client)
+
         if tables and database:
             selectors = [RawTable(db_name=database, table_name=table) for table in tables]
         elif tables and not database:
@@ -289,7 +289,7 @@ class DownloadApp(typer.Typer):
             )
 
         selectors = [DataSetSelector(kind="Assets", data_set_external_id=data_set) for data_set in data_sets]
-        cmd = DownloadCommand()
+        cmd = DownloadCommand(client=client)
         cmd.run(
             lambda: cmd.download(
                 selectors=selectors,
@@ -407,7 +407,7 @@ class DownloadApp(typer.Typer):
             )
 
         selectors = [DataSetSelector(kind="TimeSeries", data_set_external_id=data_set) for data_set in data_sets]
-        cmd = DownloadCommand()
+        cmd = DownloadCommand(client=client)
         cmd.run(
             lambda: cmd.download(
                 selectors=selectors,
@@ -486,7 +486,7 @@ class DownloadApp(typer.Typer):
             )
 
         selectors = [DataSetSelector(kind="Events", data_set_external_id=data_set) for data_set in data_sets]
-        cmd = DownloadCommand()
+        cmd = DownloadCommand(client=client)
 
         cmd.run(
             lambda: cmd.download(
@@ -613,7 +613,7 @@ class DownloadApp(typer.Typer):
             selectors = [DataSetSelector(kind="FileMetadata", data_set_external_id=data_set) for data_set in data_sets]
             io = FileMetadataIO(client)
 
-        cmd = DownloadCommand()
+        cmd = DownloadCommand(client=client)
         cmd.run(
             lambda: cmd.download(
                 selectors=selectors,  # type: ignore[misc]
@@ -678,9 +678,9 @@ class DownloadApp(typer.Typer):
         ] = False,
     ) -> None:
         """This command will download an asset hierarchy from CDF into a temporary directory."""
-        cmd = DownloadCommand()
-
         client = EnvironmentVariables.create_from_environment().get_client()
+        cmd = DownloadCommand(client=client)
+
         if hierarchy is None:
             selector = AssetInteractiveSelect(client, "download")
             hierarchy = selector.select_hierarchy(allow_empty=False)
@@ -782,7 +782,8 @@ class DownloadApp(typer.Typer):
         ] = False,
     ) -> None:
         """This command will download Instances from CDF into a temporary directory."""
-        cmd = DownloadCommand()
+        client = EnvironmentVariables.create_from_environment().get_client()
+        cmd = DownloadCommand(client=client)
 
         client = EnvironmentVariables.create_from_environment().get_client()
         if instance_space is None:
@@ -957,7 +958,7 @@ class DownloadApp(typer.Typer):
                 ).unsafe_ask()
             )
 
-        cmd = DownloadCommand()
+        cmd = DownloadCommand(client=client)
         selector = DataPointsDataSetSelector(
             data_set_external_id=dataset,
             start=start_time,
@@ -1028,8 +1029,8 @@ class DownloadApp(typer.Typer):
         ] = False,
     ) -> None:
         """This command will download Charts from CDF into a temporary directory."""
-        cmd = DownloadCommand()
         client = EnvironmentVariables.create_from_environment().get_client()
+        cmd = DownloadCommand(client=client)
         selector: ChartSelector
         if external_ids is None:
             selected_external_ids = InteractiveChartSelect(client).select_external_ids()
@@ -1101,8 +1102,8 @@ class DownloadApp(typer.Typer):
         ] = False,
     ) -> None:
         """This command will download Canvas from CDF into a temporary directory."""
-        cmd = DownloadCommand()
         client = EnvironmentVariables.create_from_environment().get_client()
+        cmd = DownloadCommand(client=client)
         selector: CanvasSelector
         if external_ids is None:
             selected_external_ids = InteractiveCanvasSelect(client).select_external_ids()

cognite_toolkit/_cdf_tk/apps/_dump_app.py

@@ -143,7 +143,7 @@ class DumpConfigApp(typer.Typer):
             selected_data_model = DataModelId(*data_model_id)
         client = EnvironmentVariables.create_from_environment().get_client()
 
-        cmd = DumpResourceCommand()
+        cmd = DumpResourceCommand(client=client)
         cmd.run(
             lambda: cmd.dump_to_yamls(
                 DataModelFinder(client, selected_data_model, include_global=include_global),
@@ -199,7 +199,7 @@ class DumpConfigApp(typer.Typer):
             selected_workflow = WorkflowVersionId(*workflow_id)
         client = EnvironmentVariables.create_from_environment().get_client()
 
-        cmd = DumpResourceCommand()
+        cmd = DumpResourceCommand(client=client)
         cmd.run(
             lambda: cmd.dump_to_yamls(
                 WorkflowFinder(client, selected_workflow),
@@ -248,7 +248,7 @@ class DumpConfigApp(typer.Typer):
         """This command will dump the selected transformation as yaml to the folder specified, defaults to /tmp."""
         client = EnvironmentVariables.create_from_environment().get_client()
 
-        cmd = DumpResourceCommand()
+        cmd = DumpResourceCommand(client=client)
         cmd.run(
             lambda: cmd.dump_to_yamls(
                 TransformationFinder(client, tuple(transformation_id) if transformation_id else None),
@@ -297,7 +297,7 @@ class DumpConfigApp(typer.Typer):
         """This command will dump the selected group as yaml to the folder specified, defaults to /tmp."""
         client = EnvironmentVariables.create_from_environment().get_client()
 
-        cmd = DumpResourceCommand()
+        cmd = DumpResourceCommand(client=client)
         cmd.run(
             lambda: cmd.dump_to_yamls(
                 GroupFinder(client, tuple(group_name) if group_name else None),
@@ -346,7 +346,7 @@ class DumpConfigApp(typer.Typer):
         """Dump on or more agents as yaml to the specified folder, defaults to /tmp."""
 
         client = EnvironmentVariables.create_from_environment().get_client()
-        cmd = DumpResourceCommand()
+        cmd = DumpResourceCommand(client=client)
         cmd.run(
             lambda: cmd.dump_to_yamls(
                 AgentFinder(client, tuple(external_id) if external_id else None),
@@ -405,7 +405,7 @@ class DumpConfigApp(typer.Typer):
                 )
             selected_view_id = ViewId(*view_id)
 
-        cmd = DumpResourceCommand()
+        cmd = DumpResourceCommand(client=client)
         cmd.run(
             lambda: cmd.dump_to_yamls(
                 NodeFinder(client, selected_view_id),
@@ -453,7 +453,7 @@ class DumpConfigApp(typer.Typer):
     ) -> None:
         """This command will dump the selected location filters as yaml to the folder specified, defaults to /tmp."""
         client = EnvironmentVariables.create_from_environment().get_client()
-        cmd = DumpResourceCommand()
+        cmd = DumpResourceCommand(client=client)
         cmd.run(
             lambda: cmd.dump_to_yamls(
                 LocationFilterFinder(client, tuple(external_id) if external_id else None),
@@ -501,7 +501,7 @@ class DumpConfigApp(typer.Typer):
     ) -> None:
         """This command will dump the selected extraction pipeline as yaml to the folder specified, defaults to /tmp."""
         client = EnvironmentVariables.create_from_environment().get_client()
-        cmd = DumpResourceCommand()
+        cmd = DumpResourceCommand(client=client)
         cmd.run(
             lambda: cmd.dump_to_yamls(
                 ExtractionPipelineFinder(client, tuple(external_id) if external_id else None),
@@ -549,7 +549,7 @@ class DumpConfigApp(typer.Typer):
     ) -> None:
         """This command will dump the selected datasets as yaml to the folder specified, defaults to /tmp."""
         client = EnvironmentVariables.create_from_environment().get_client()
-        cmd = DumpResourceCommand()
+        cmd = DumpResourceCommand(client=client)
         cmd.run(
             lambda: cmd.dump_to_yamls(
                 DataSetFinder(client, tuple(external_id) if external_id else None),
@@ -597,7 +597,7 @@ class DumpConfigApp(typer.Typer):
     ) -> None:
         """This command will dump the selected functions as yaml to the folder specified, defaults to /tmp."""
         client = EnvironmentVariables.create_from_environment().get_client()
-        cmd = DumpResourceCommand()
+        cmd = DumpResourceCommand(client=client)
         cmd.run(
             lambda: cmd.dump_to_yamls(
                 FunctionFinder(client, tuple(external_id) if external_id else None),
@@ -645,7 +645,7 @@ class DumpConfigApp(typer.Typer):
     ) -> None:
         """This command will dump the selected Streamlit apps as yaml to the folder specified, defaults to /tmp."""
         client = EnvironmentVariables.create_from_environment().get_client()
-        cmd = DumpResourceCommand()
+        cmd = DumpResourceCommand(client=client)
         cmd.run(
             lambda: cmd.dump_to_yamls(
                 StreamlitFinder(client, tuple(external_id) if external_id else None),
@@ -693,7 +693,7 @@ class DumpConfigApp(typer.Typer):
     ) -> None:
         """This command will dump the selected instance spaces as yaml to the folder specified, defaults to /tmp."""
         client = EnvironmentVariables.create_from_environment().get_client()
-        cmd = DumpResourceCommand()
+        cmd = DumpResourceCommand(client=client)
         cmd.run(
             lambda: cmd.dump_to_yamls(
                 SpaceFinder(client, tuple(spaces) if spaces else None),
@@ -748,7 +748,7 @@ class DumpConfigApp(typer.Typer):
                     "View ID must be provided as exactly two arguments: externalId and space."
                 )
             selected_view_id = SearchConfigViewId(*view_id)
-        cmd = DumpResourceCommand()
+        cmd = DumpResourceCommand(client=client)
         cmd.run(
             lambda: cmd.dump_to_yamls(
                 SearchConfigFinder(client, tuple([selected_view_id]) if selected_view_id else None),

cognite_toolkit/_cdf_tk/apps/_landing_app.py

@@ -1,8 +1,11 @@
+import contextlib
 from typing import Annotated
 
 import typer
 
+from cognite_toolkit._cdf_tk.client import ToolkitClient
 from cognite_toolkit._cdf_tk.commands import InitCommand
+from cognite_toolkit._cdf_tk.utils.auth import EnvironmentVariables
 
 
 class LandingApp(typer.Typer):
@@ -21,7 +24,13 @@ class LandingApp(typer.Typer):
         ] = False,
     ) -> None:
         """Getting started checklist"""
-        cmd = InitCommand()
+        client: ToolkitClient | None = None
+        with contextlib.redirect_stdout(None), contextlib.suppress(Exception):
+            # Remove the Error message from failing to load the config
+            # This is verified in check_auth
+            client = EnvironmentVariables.create_from_environment().get_client()
+
+        cmd = InitCommand(client=client)
         # Tracking  command with the usual lambda run construct
         # is intentionally left out because we don't want to expose the user to the warning
         # before they've had the chance to opt in (which is something they'll do later using this command).

cognite_toolkit/_cdf_tk/apps/_migrate_app.py

@@ -108,7 +108,7 @@ class MigrateApp(typer.Typer):
         depend on the primary resources 3D and annotations.
         """
         client = EnvironmentVariables.create_from_environment().get_client(enable_set_pending_ids=True)
-        cmd = MigrationPrepareCommand()
+        cmd = MigrationPrepareCommand(client=client)
         cmd.run(
             lambda: cmd.deploy_cognite_migration(
                 client,
@@ -167,7 +167,7 @@ class MigrateApp(typer.Typer):
             )
             verbose = questionary.confirm("Do you want verbose output?", default=verbose).unsafe_ask()
 
-        cmd = MigrationCommand()
+        cmd = MigrationCommand(client=client)
         cmd.run(
             lambda: cmd.create(
                 client,
@@ -244,7 +244,7 @@ class MigrateApp(typer.Typer):
         elif data_set is None or instance_space is None:
             raise typer.BadParameter("Both data_set and instance_space must be provided together.")
 
-        cmd = MigrationCommand()
+        cmd = MigrationCommand(client=client)
         cmd.run(
             lambda: cmd.create(
                 client,
@@ -337,7 +337,7 @@ class MigrateApp(typer.Typer):
             container_id=ContainerId("cdf_cdm", "CogniteAsset"),
         )
 
-        cmd = MigrationCommand()
+        cmd = MigrationCommand(client=client)
         cmd.run(
             lambda: cmd.migrate(
                 selected=selected,
@@ -484,7 +484,7 @@ class MigrateApp(typer.Typer):
             container_id=ContainerId("cdf_cdm", "CogniteActivity"),
         )
 
-        cmd = MigrationCommand()
+        cmd = MigrationCommand(client=client)
 
         cmd.run(
             lambda: cmd.migrate(
@@ -592,7 +592,7 @@ class MigrateApp(typer.Typer):
                 "Do you want to link old and new TimeSeries?", default=not skip_linking
             ).unsafe_ask()
 
-        cmd = MigrationCommand()
+        cmd = MigrationCommand(client=client)
         cmd.run(
             lambda: cmd.migrate(
                 selected=selected,
@@ -694,7 +694,7 @@ class MigrateApp(typer.Typer):
             resource_type="file",
             container_id=ContainerId("cdf_cdm", "CogniteFile"),
         )
-        cmd = MigrationCommand()
+        cmd = MigrationCommand(client=client)
 
         if data_set_id is None:
             skip_linking = not questionary.confirm(
@@ -841,7 +841,7 @@ class MigrateApp(typer.Typer):
             dry_run = questionary.confirm("Do you want to perform a dry run?", default=dry_run).unsafe_ask()
             verbose = questionary.confirm("Do you want verbose output?", default=verbose).unsafe_ask()
 
-        cmd = MigrationCommand()
+        cmd = MigrationCommand(client=client)
         cmd.run(
             lambda: cmd.migrate(
                 selected=selected,
@@ -914,7 +914,7 @@ class MigrateApp(typer.Typer):
             dry_run = questionary.confirm("Do you want to perform a dry run?", default=dry_run).unsafe_ask()
             verbose = questionary.confirm("Do you want verbose output?", default=verbose).unsafe_ask()
 
-        cmd = MigrationCommand()
+        cmd = MigrationCommand(client=client)
         selector = CanvasExternalIdSelector(external_ids=tuple(external_id))
         cmd.run(
             lambda: cmd.migrate(
@@ -977,7 +977,7 @@ class MigrateApp(typer.Typer):
         else:
             selected_external_ids = InteractiveChartSelect(client).select_external_ids()
 
-        cmd = MigrationCommand()
+        cmd = MigrationCommand(client=client)
         cmd.run(
             lambda: cmd.migrate(
                 selected=ChartExternalIdSelector(external_ids=tuple(selected_external_ids)),
@@ -1039,7 +1039,7 @@ class MigrateApp(typer.Typer):
             selected_models = ThreeDInteractiveSelect(client, "migrate").select_three_d_models("classic")
             selected_ids = [model.id for model in selected_models]
 
-        cmd = MigrationCommand()
+        cmd = MigrationCommand(client=client)
         cmd.run(
             lambda: cmd.migrate(
                 selected=ThreeDModelIdSelector(ids=tuple(selected_ids)),
@@ -1118,12 +1118,12 @@ class MigrateApp(typer.Typer):
             object_3D_space = space_selector.select_instance_space(
                 multiselect=False,
                 message="In which instance space do you want to create the 3D Object nodes?",
-                include_empty=False,
+                include_empty=True,
             )
             cad_node_space = space_selector.select_instance_space(
                 multiselect=False,
                 message="In which instance space do you want to create the CAD Node nodes?",
-                include_empty=False,
+                include_empty=True,
             )
             dry_run = questionary.confirm("Do you want to perform a dry run?", default=dry_run).unsafe_ask()
             verbose = questionary.confirm("Do you want verbose output?", default=verbose).unsafe_ask()
@@ -1133,7 +1133,7 @@ class MigrateApp(typer.Typer):
                 "--object-3d-space and --cad-node-space are required when specifying IDs directly."
             )
 
-        cmd = MigrationCommand()
+        cmd = MigrationCommand(client=client)
         cmd.run(
             lambda: cmd.migrate(
                 selected=ThreeDModelIdSelector(ids=tuple(selected_ids)),
@@ -1179,7 +1179,7 @@ class MigrateApp(typer.Typer):
         """Creates Infield V2 configurations from existing APM Configurations in CDF."""
         client = EnvironmentVariables.create_from_environment().get_client()
 
-        cmd = MigrationCommand()
+        cmd = MigrationCommand(client=client)
         cmd.run(
             lambda: cmd.create(
                 client,

cognite_toolkit/_cdf_tk/apps/_modules_app.py

@@ -1,3 +1,4 @@
+import contextlib
 from pathlib import Path
 from typing import Annotated
 
@@ -5,6 +6,7 @@ import typer
 from rich import print
 
 from cognite_toolkit._cdf_tk.cdf_toml import CDFToml
+from cognite_toolkit._cdf_tk.client import ToolkitClient
 from cognite_toolkit._cdf_tk.commands import ModulesCommand, PullCommand
 from cognite_toolkit._cdf_tk.utils.auth import EnvironmentVariables
 from cognite_toolkit._version import __version__
@@ -77,6 +79,11 @@ class ModulesApp(typer.Typer):
         ] = False,
     ) -> None:
         """Initialize or upgrade a new CDF project with templates interactively."""
+        client: ToolkitClient | None = None
+        with contextlib.redirect_stdout(None), contextlib.suppress(Exception):
+            # Try to load client if possible, but ignore errors.
+            # This is only used for logging purposes in the command.
+            client = EnvironmentVariables.create_from_environment().get_client()
 
         if library_url and not library_checksum:
             raise typer.BadParameter(
@@ -84,7 +91,7 @@ class ModulesApp(typer.Typer):
                 param_hint="--library-checksum",
             )
 
-        with ModulesCommand() as cmd:
+        with ModulesCommand(client=client) as cmd:
             cmd.run(
                 lambda: cmd.init(
                     organization_dir=organization_dir,
@@ -114,7 +121,13 @@ class ModulesApp(typer.Typer):
             ),
         ] = False,
     ) -> None:
-        cmd = ModulesCommand()
+        client: ToolkitClient | None = None
+        with contextlib.redirect_stdout(None), contextlib.suppress(Exception):
+            # Try to load client if possible, but ignore errors.
+            # This is only used for logging purposes in the command.
+            client = EnvironmentVariables.create_from_environment().get_client()
+
+        cmd = ModulesCommand(client=client)
         cmd.run(lambda: cmd.upgrade(organization_dir=organization_dir, verbose=verbose))
 
     # This is a trick to use an f-string for the docstring
@@ -140,7 +153,12 @@ class ModulesApp(typer.Typer):
         ] = False,
     ) -> None:
         """Add one or more new module(s) to the project."""
-        with ModulesCommand() as cmd:
+        client: ToolkitClient | None = None
+        with contextlib.redirect_stdout(None), contextlib.suppress(Exception):
+            # Try to load client if possible, but ignore errors.
+            # This is only used for logging purposes in the command.
+            client = EnvironmentVariables.create_from_environment().get_client()
+        with ModulesCommand(client=client) as cmd:
             cmd.run(lambda: cmd.add(organization_dir=organization_dir))
 
     def pull(
@@ -187,8 +205,8 @@ class ModulesApp(typer.Typer):
         ] = False,
     ) -> None:
         """Pull a module from CDF. This will overwrite the local files with the latest version from CDF."""
-        cmd = PullCommand()
         env_vars = EnvironmentVariables.create_from_environment()
+        cmd = PullCommand(client=env_vars.get_client())
         cmd.run(
             lambda: cmd.pull_module(
                 module_name_or_path=module_name_or_path,
@@ -227,5 +245,11 @@ class ModulesApp(typer.Typer):
         ] = False,
     ) -> None:
         """List all available modules in the project."""
-        cmd = ModulesCommand()
+        client: ToolkitClient | None = None
+        with contextlib.redirect_stdout(None), contextlib.suppress(Exception):
+            # Try to load client if possible, but ignore errors.
+            # This is only used for logging purposes in the command.
+            client = EnvironmentVariables.create_from_environment().get_client()
+
+        cmd = ModulesCommand(client=client)
         cmd.run(lambda: cmd.list(organization_dir=organization_dir, build_env_name=build_env))

cognite_toolkit/_cdf_tk/apps/_profile_app.py

@@ -55,7 +55,7 @@ class ProfileApp(typer.Typer):
         these data sets, and the RAW tables that is used in these transformations.
         """
         client = EnvironmentVariables.create_from_environment().get_client()
-        cmd = ProfileAssetCommand(output_spreadsheet)
+        cmd = ProfileAssetCommand(output_spreadsheet, client=client)
         cmd.run(
             lambda: cmd.assets(
                 client=client,
@@ -90,7 +90,7 @@ class ProfileApp(typer.Typer):
         This shows an approximation of unstructured data count. This can, for example, be used to estimate the
         effort to model this data in data modeling."""
         client = EnvironmentVariables.create_from_environment().get_client()
-        cmd = ProfileAssetCentricCommand(output_spreadsheet)
+        cmd = ProfileAssetCentricCommand(output_spreadsheet, client=client)
         cmd.run(
             lambda: cmd.asset_centric(
                 client,
@@ -127,7 +127,7 @@ class ProfileApp(typer.Typer):
         source of the data in a specific CDF resource.
         """
         client = EnvironmentVariables.create_from_environment().get_client()
-        cmd = ProfileTransformationCommand(output_spreadsheet)
+        cmd = ProfileTransformationCommand(output_spreadsheet, client=client)
         cmd.run(
             lambda: cmd.transformation(
                 client,
@@ -165,7 +165,7 @@ class ProfileApp(typer.Typer):
         source of the data in a specific CDF resource.
         """
         client = EnvironmentVariables.create_from_environment().get_client()
-        cmd = ProfileRawCommand(output_spreadsheet)
+        cmd = ProfileRawCommand(output_spreadsheet, client=client)
         cmd.run(
             lambda: cmd.raw(
                 client,

cognite_toolkit/_cdf_tk/apps/_purge.py

@@ -106,8 +106,8 @@ class PurgeApp(typer.Typer):
         ] = False,
     ) -> None:
         """This command will delete the contents of the specified dataset"""
-        cmd = PurgeCommand()
         client = EnvironmentVariables.create_from_environment().get_client()
+        cmd = PurgeCommand(client=client)
 
         if external_id is None:
             # Is Interactive
@@ -203,9 +203,8 @@ class PurgeApp(typer.Typer):
         ] = False,
     ) -> None:
         """This command will delete the contents of the specified space."""
-
-        cmd = PurgeCommand()
         client = EnvironmentVariables.create_from_environment().get_client()
+        cmd = PurgeCommand(client=client)
 
         if space is None:
             # Is Interactive
@@ -319,9 +318,9 @@ class PurgeApp(typer.Typer):
         ] = False,
     ) -> None:
         """This command will delete the contents of the specified instances."""
-
-        cmd = PurgeCommand()
         client = EnvironmentVariables.create_from_environment().get_client(enable_set_pending_ids=True)
+        cmd = PurgeCommand(client=client)
+
         is_interactive = view is None and instance_list is None
         selector: InstanceSelector
         if is_interactive:

cognite_toolkit/_cdf_tk/apps/_repo_app.py

@@ -1,12 +1,15 @@
+import contextlib
 from pathlib import Path
 from typing import Annotated
 
 import typer
 from rich import print
 
+from cognite_toolkit._cdf_tk.client import ToolkitClient
 from cognite_toolkit._cdf_tk.commands import RepoCommand
 from cognite_toolkit._cdf_tk.commands.repo import REPOSITORY_HOSTING
 from cognite_toolkit._cdf_tk.utils import humanize_collection
+from cognite_toolkit._cdf_tk.utils.auth import EnvironmentVariables
 
 
 class RepoApp(typer.Typer):
@@ -37,6 +40,10 @@ class RepoApp(typer.Typer):
         verbose: bool = typer.Option(False, "-v", "--verbose", help="Verbose output"),
     ) -> None:
         """Initialize a new git repository with files like .gitignore, cdf.toml, and so on."""
-
-        cmd = RepoCommand()
+        client: ToolkitClient | None = None
+        with contextlib.redirect_stdout(None), contextlib.suppress(Exception):
+            # Try to load client if possible, but ignore errors.
+            # This is only used for logging purposes in the command.
+            client = EnvironmentVariables.create_from_environment().get_client()
+        cmd = RepoCommand(client=client)
         cmd.run(lambda: cmd.init(cwd=cwd, host=host, verbose=verbose))

cognite_toolkit/_cdf_tk/apps/_run.py

@@ -58,8 +58,9 @@ class RunApp(typer.Typer):
     ) -> None:
         """This command will run the specified transformation using a one-time session."""
         RunApp._print_deprecation_warning()
-        cmd = RunTransformationCommand()
         client = EnvironmentVariables.create_from_environment().get_client()
+        cmd = RunTransformationCommand(client=client)
+
         cmd.run(lambda: cmd.run_transformation(client, external_id))
 
     @staticmethod
@@ -116,8 +117,8 @@ class RunApp(typer.Typer):
     ) -> None:
         """This command will run the specified workflow."""
         RunApp._print_deprecation_warning()
-        cmd = RunWorkflowCommand()
         env_vars = EnvironmentVariables.create_from_environment()
+        cmd = RunWorkflowCommand(client=env_vars.get_client())
         cmd.run(lambda: cmd.run_workflow(env_vars, organization_dir, env_name, external_id, version, wait))
 
 
@@ -188,8 +189,8 @@ class RunFunctionApp(typer.Typer):
     ) -> None:
         """This command will run the specified function locally."""
         RunApp._print_deprecation_warning()
-        cmd = RunFunctionCommand()
         env_vars = EnvironmentVariables.create_from_environment()
+        cmd = RunFunctionCommand(client=env_vars.get_client())
         cmd.run(
             lambda: cmd.run_local(
                 env_vars,
@@ -254,6 +255,6 @@ class RunFunctionApp(typer.Typer):
     ) -> None:
         """This command will run the specified function (assuming it is deployed) in CDF."""
         RunApp._print_deprecation_warning()
-        cmd = RunFunctionCommand()
         env_vars = EnvironmentVariables.create_from_environment()
+        cmd = RunFunctionCommand(client=env_vars.get_client())
         cmd.run(lambda: cmd.run_cdf(env_vars, organization_dir, env_name, external_id, schedule, wait))