diff --git a/number_management/src/vonage_numbers/number_management.py b/number_management/src/vonage_numbers/number_management.py index fdef03da..2a3c2fee 100644 --- a/number_management/src/vonage_numbers/number_management.py +++ b/number_management/src/vonage_numbers/number_management.py @@ -3,10 +3,11 @@ from pydantic import validate_call from vonage_http_client.http_client import HttpClient from vonage_numbers.errors import NumbersError + from .requests import ( - SearchAvailableNumbersFilter, ListOwnedNumbersFilter, NumberParams, + SearchAvailableNumbersFilter, UpdateNumberParams, ) from .responses import AvailableNumber, NumbersStatus, OwnedNumber diff --git a/number_management/src/vonage_numbers/requests.py b/number_management/src/vonage_numbers/requests.py index 9c756e0f..1bc1462b 100644 --- a/number_management/src/vonage_numbers/requests.py +++ b/number_management/src/vonage_numbers/requests.py @@ -1,11 +1,10 @@ from typing import Optional from pydantic import BaseModel, Field, model_validator - from vonage_numbers.enums import NumberFeatures, NumberType, VoiceCallbackType +from vonage_utils.types import PhoneNumber from .errors import NumbersError -from vonage_utils.types import PhoneNumber class ListNumbersFilter(BaseModel): @@ -38,9 +37,9 @@ class SearchAvailableNumbersFilter(ListNumbersFilter): class NumberParams(BaseModel): """Specify the two-letter country code and the number you are referring to. - If you'd like to perform an action on a subaccount, provide the api_key of - that account in the `target_api_key` field. If you'd like to perform an action - on your own account, you do not need to provide this field. + If you'd like to perform an action on a subaccount, provide the api_key of that account in the + `target_api_key` field. If you'd like to perform an action on your own account, you do not need + to provide this field. """ country: str = Field(..., min_length=2, max_length=2) diff --git a/number_management/tests/test_numbers.py b/number_management/tests/test_numbers.py index 3b1a126f..1b31ea0b 100644 --- a/number_management/tests/test_numbers.py +++ b/number_management/tests/test_numbers.py @@ -2,12 +2,9 @@ import responses from pytest import raises - +from vonage_http_client.http_client import HttpClient from vonage_numbers.errors import NumbersError from vonage_numbers.number_management import Numbers -from vonage_http_client.http_client import HttpClient - -from testutils import build_response, get_mock_api_key_auth from vonage_numbers.requests import ( ListOwnedNumbersFilter, NumberParams, @@ -15,6 +12,8 @@ UpdateNumberParams, ) +from testutils import build_response, get_mock_api_key_auth + path = abspath(__file__) numbers = Numbers(HttpClient(get_mock_api_key_auth()))