|
86 | 86 | ) |
87 | 87 | from dstack._internal.server.services.resources import set_resources_defaults |
88 | 88 | from dstack._internal.utils import random_names |
89 | | -from dstack._internal.utils.common import EntityID, EntityName, EntityNameOrID |
| 89 | +from dstack._internal.utils.common import ( |
| 90 | + EntityID, |
| 91 | + EntityName, |
| 92 | + EntityNameOrID, |
| 93 | + get_current_datetime, |
| 94 | +) |
90 | 95 | from dstack._internal.utils.logging import get_logger |
91 | 96 | from dstack._internal.utils.ssh import pkey_from_str |
92 | 97 |
|
@@ -999,13 +1004,16 @@ async def _create_fleet( |
999 | 1004 | else: |
1000 | 1005 | spec.configuration.name = await generate_fleet_name(session=session, project=project) |
1001 | 1006 |
|
| 1007 | + now = get_current_datetime() |
1002 | 1008 | fleet_model = FleetModel( |
1003 | 1009 | id=uuid.uuid4(), |
1004 | 1010 | name=spec.configuration.name, |
1005 | 1011 | project=project, |
1006 | 1012 | status=FleetStatus.ACTIVE, |
1007 | 1013 | spec=spec.json(), |
1008 | 1014 | instances=[], |
| 1015 | + created_at=now, |
| 1016 | + last_processed_at=now, |
1009 | 1017 | ) |
1010 | 1018 | session.add(fleet_model) |
1011 | 1019 | events.emit( |
@@ -1057,9 +1065,10 @@ async def _create_fleet( |
1057 | 1065 | targets=[events.Target.from_model(instance_model)], |
1058 | 1066 | ) |
1059 | 1067 | fleet_model.instances.append(instance_model) |
1060 | | - pipeline_hinter.hint_fetch(FleetModel.__name__) |
1061 | | - pipeline_hinter.hint_fetch(InstanceModel.__name__) |
1062 | 1068 | await session.commit() |
| 1069 | + if spec.configuration.ssh_config is None: |
| 1070 | + pipeline_hinter.hint_fetch(FleetModel.__name__) |
| 1071 | + pipeline_hinter.hint_fetch(InstanceModel.__name__) |
1063 | 1072 | return fleet_model_to_fleet(fleet_model) |
1064 | 1073 |
|
1065 | 1074 |
|
|
0 commit comments