77)
88from dstack ._internal .core .backends .digitalocean_base .backend import BaseDigitalOceanBackend
99from dstack ._internal .core .backends .digitalocean_base .models import (
10- AnyBaseDigitalOceanBackendConfig ,
1110 AnyBaseDigitalOceanCreds ,
1211 BaseDigitalOceanBackendConfig ,
1312 BaseDigitalOceanBackendConfigWithCreds ,
@@ -33,16 +32,20 @@ def create_backend(
3332 auth = BaseDigitalOceanCreds .parse_obj (config .creds ).json (),
3433 )
3534
36- def get_backend_config (
37- self , record : BackendRecord , include_creds : bool
38- ) -> AnyBaseDigitalOceanBackendConfig :
35+ def get_backend_config_with_creds (
36+ self , record : BackendRecord
37+ ) -> BaseDigitalOceanBackendConfigWithCreds :
38+ config = self ._get_config (record )
39+ return BaseDigitalOceanBackendConfigWithCreds .__response__ .parse_obj (config )
40+
41+ def get_backend_config_without_creds (
42+ self , record : BackendRecord
43+ ) -> BaseDigitalOceanBackendConfig :
3944 config = self ._get_config (record )
40- if include_creds :
41- return BaseDigitalOceanBackendConfigWithCreds .__response__ .parse_obj (config )
4245 return BaseDigitalOceanBackendConfig .__response__ .parse_obj (config )
4346
4447 def get_backend (self , record : BackendRecord ) -> BaseDigitalOceanBackend :
45- pass
48+ raise NotImplementedError ( "Subclasses must implement get_backend" )
4649
4750 def _get_config (self , record : BackendRecord ) -> BaseDigitalOceanConfig :
4851 return BaseDigitalOceanConfig .__response__ (
0 commit comments