From 6271faa49f121eb2ed5b6e5a12860cff8e14209f Mon Sep 17 00:00:00 2001 From: tehcoderer Date: Tue, 28 Nov 2023 14:57:19 -0500 Subject: [PATCH] update merged fred models --- .../providers/fred/openbb_fred/models/search.py | 9 ++++++--- .../providers/fred/openbb_fred/models/series.py | 8 +++----- 2 files changed, 9 insertions(+), 8 deletions(-) diff --git a/openbb_platform/providers/fred/openbb_fred/models/search.py b/openbb_platform/providers/fred/openbb_fred/models/search.py index aed7e051afb..a7d53b545d0 100644 --- a/openbb_platform/providers/fred/openbb_fred/models/search.py +++ b/openbb_platform/providers/fred/openbb_fred/models/search.py @@ -9,7 +9,10 @@ from openbb_core.provider.standard_models.fred_search import ( SearchQueryParams, ) from openbb_core.provider.utils.descriptions import QUERY_DESCRIPTIONS -from openbb_core.provider.utils.helpers import async_make_request, get_querystring +from openbb_core.provider.utils.helpers import ( + async_request, + get_querystring, +) from pydantic import Field, NonNegativeInt @@ -100,7 +103,7 @@ class FredSearchFetcher( if query.is_release is True: url = f"https://api.stlouisfed.org/fred/releases?api_key={api_key}&file_type=json" - response = await async_make_request(url, timeout=5, **kwargs) + response = await async_request(url, timeout=5, **kwargs) return response.get("releases") # type: ignore[return-value, union-attr] @@ -119,7 +122,7 @@ class FredSearchFetcher( querystring = get_querystring(query.model_dump(), exclude).replace(" ", "+") url = url + querystring + f"&file_type=json&api_key={api_key}" - response = await async_make_request(url, timeout=5, **kwargs) + response = await async_request(url, timeout=5, **kwargs) return response.get("seriess") # type: ignore[return-value, union-attr] diff --git a/openbb_platform/providers/fred/openbb_fred/models/series.py b/openbb_platform/providers/fred/openbb_fred/models/series.py index 4a4e614706d..dafff70afdd 100644 --- a/openbb_platform/providers/fred/openbb_fred/models/series.py +++ b/openbb_platform/providers/fred/openbb_fred/models/series.py @@ -11,7 +11,7 @@ from openbb_core.provider.standard_models.fred_series import ( SeriesQueryParams, ) from openbb_core.provider.utils.descriptions import QUERY_DESCRIPTIONS -from openbb_core.provider.utils.helpers import async_make_request, get_querystring +from openbb_core.provider.utils.helpers import async_request, get_querystring from pydantic import Field _warn = warnings.warn @@ -135,10 +135,8 @@ class FredSeriesFetcher( ] async def async_get_fred_data(url, metadata_url, series_id, **kwargs): - response = await async_make_request(url, timeout=5, **kwargs) - metadata_response = await async_make_request( - metadata_url, timeout=5, **kwargs - ) + response = await async_request(url, timeout=5, **kwargs) + metadata_response = await async_request(metadata_url, timeout=5, **kwargs) _metadata = metadata_response.get("seriess")[0] data = response.get("observations") try: