Skip to content

Commit 8fe7c52

Browse files
fix
1 parent ddf542e commit 8fe7c52

File tree

2 files changed

+5
-9
lines changed

2 files changed

+5
-9
lines changed

packages/models-library/src/models_library/api_schemas_directorv2/dynamic_services.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from typing import Annotated, TypeAlias
22

3-
from pydantic import AnyHttpUrl, BaseModel, ByteSize, ConfigDict, Field
3+
from pydantic import AnyHttpUrl, BaseModel, BeforeValidator, ByteSize, ConfigDict, Field
44
from pydantic.config import JsonDict
55

66
from ..resource_tracker import HardwareInfo, PricingInfo
@@ -41,11 +41,9 @@ class DynamicServiceCreate(ServiceDetails):
4141

4242
product_name: Annotated[str, Field(..., description="Current product name")]
4343
product_api_base_url: Annotated[
44-
AnyHttpUrl,
45-
Field(
46-
...,
47-
description="Base url of the product",
48-
),
44+
str,
45+
BeforeValidator(lambda v: f"{AnyHttpUrl(v)}"),
46+
Field(..., description="Current product API base URL"),
4947
]
5048
can_save: Annotated[
5149
bool, Field(..., description="the service data must be saved when closing")

services/dynamic-scheduler/tests/unit/conftest.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,9 +13,7 @@
1313
@pytest.fixture
1414
def get_dynamic_service_start() -> Callable[[NodeID], DynamicServiceStart]:
1515
def _(node_id: NodeID) -> DynamicServiceStart:
16-
dict_data = deepcopy(
17-
DynamicServiceStart.model_config["json_schema_extra"]["example"]
18-
)
16+
dict_data = deepcopy(DynamicServiceStart.model_json_schema()["example"])
1917
dict_data["service_uuid"] = f"{node_id}"
2018
return TypeAdapter(DynamicServiceStart).validate_python(dict_data)
2119

0 commit comments

Comments
 (0)