|
31 | 31 | from dagster_cloud_cli.core.errors import DagsterCloudHTTPError, raise_http_error |
32 | 32 | from dagster_cloud_cli.core.workspace import CodeLocationDeployData |
33 | 33 |
|
| 34 | +from dagster_cloud.agent.instrumentation.constants import DAGSTER_CLOUD_AGENT_METRIC_PREFIX |
| 35 | +from dagster_cloud.agent.instrumentation.run_launch import extract_run_attributes |
| 36 | +from dagster_cloud.agent.instrumentation.schedule import inspect_schedule_result |
| 37 | +from dagster_cloud.agent.instrumentation.sensor import inspect_sensor_result |
| 38 | +from dagster_cloud.agent.queries import ( |
| 39 | + ADD_AGENT_HEARTBEATS_MUTATION, |
| 40 | + DEPLOYMENTS_QUERY, |
| 41 | + GET_USER_CLOUD_REQUESTS_QUERY, |
| 42 | + WORKSPACE_ENTRIES_QUERY, |
| 43 | +) |
34 | 44 | from dagster_cloud.api.dagster_cloud_api import ( |
35 | 45 | AgentHeartbeat, |
36 | 46 | AgentUtilizationMetrics, |
|
48 | 58 | ) |
49 | 59 | from dagster_cloud.batching import Batcher |
50 | 60 | from dagster_cloud.instance import DagsterCloudAgentInstance |
| 61 | +from dagster_cloud.opentelemetry.observers.execution_observer import observe_execution |
| 62 | +from dagster_cloud.util import SERVER_HANDLE_TAG, compressed_namedtuple_upload_file, is_isolated_run |
| 63 | +from dagster_cloud.version import __version__ |
51 | 64 | from dagster_cloud.workspace.user_code_launcher import ( |
52 | 65 | DagsterCloudUserCodeLauncher, |
53 | 66 | UserCodeLauncherEntry, |
54 | 67 | ) |
55 | 68 | from dagster_cloud.workspace.user_code_launcher.utils import get_instance_ref_for_user_code |
56 | 69 |
|
57 | | -from ..opentelemetry.observers.execution_observer import observe_execution |
58 | | -from ..util import SERVER_HANDLE_TAG, compressed_namedtuple_upload_file, is_isolated_run |
59 | | -from ..version import __version__ |
60 | | -from .instrumentation.constants import DAGSTER_CLOUD_AGENT_METRIC_PREFIX |
61 | | -from .instrumentation.run_launch import extract_run_attributes |
62 | | -from .instrumentation.schedule import inspect_schedule_result |
63 | | -from .instrumentation.sensor import inspect_sensor_result |
64 | | -from .queries import ( |
65 | | - ADD_AGENT_HEARTBEATS_MUTATION, |
66 | | - DEPLOYMENTS_QUERY, |
67 | | - GET_USER_CLOUD_REQUESTS_QUERY, |
68 | | - WORKSPACE_ENTRIES_QUERY, |
69 | | -) |
70 | | - |
71 | 70 | if TYPE_CHECKING: |
72 | 71 | import datetime |
73 | 72 |
|
|
0 commit comments