diff --git a/openbb_platform/providers/fmp/openbb_fmp/models/company_news.py b/openbb_platform/providers/fmp/openbb_fmp/models/company_news.py index 358cd85c83e..dbb1b825704 100644 --- a/openbb_platform/providers/fmp/openbb_fmp/models/company_news.py +++ b/openbb_platform/providers/fmp/openbb_fmp/models/company_news.py @@ -7,6 +7,7 @@ from openbb_core.provider.standard_models.company_news import ( CompanyNewsData, CompanyNewsQueryParams, ) +from openbb_core.provider.utils.helpers import filter_by_dates from openbb_fmp.utils.helpers import get_data_many from pydantic import Field @@ -75,4 +76,7 @@ class FMPCompanyNewsFetcher( query: FMPCompanyNewsQueryParams, data: List[Dict], **kwargs: Any ) -> List[FMPCompanyNewsData]: """Return the transformed data.""" - return [FMPCompanyNewsData.model_validate(d) for d in data] + modeled_data = [FMPCompanyNewsData.model_validate(d) for d in data] + return filter_by_dates( + modeled_data, query.start_date, query.end_date, query.date + ) diff --git a/openbb_platform/providers/intrinio/openbb_intrinio/models/company_news.py b/openbb_platform/providers/intrinio/openbb_intrinio/models/company_news.py index c54cbb766b3..1b5f9a0740b 100644 --- a/openbb_platform/providers/intrinio/openbb_intrinio/models/company_news.py +++ b/openbb_platform/providers/intrinio/openbb_intrinio/models/company_news.py @@ -12,6 +12,7 @@ from openbb_core.provider.standard_models.company_news import ( from openbb_core.provider.utils.helpers import ( ClientResponse, amake_requests, + filter_by_dates, get_querystring, ) from pydantic import Field, field_validator @@ -94,4 +95,7 @@ class IntrinioCompanyNewsFetcher( query: IntrinioCompanyNewsQueryParams, data: List[Dict], **kwargs: Any ) -> List[IntrinioCompanyNewsData]: """Return the transformed data.""" - return [IntrinioCompanyNewsData.model_validate(d) for d in data] + modeled_data = [IntrinioCompanyNewsData.model_validate(d) for d in data] + return filter_by_dates( + modeled_data, query.start_date, query.end_date, query.date + ) diff --git a/openbb_platform/providers/polygon/openbb_polygon/models/company_news.py b/openbb_platform/providers/polygon/openbb_polygon/models/company_news.py index 259f19a5c06..0d452af5243 100644 --- a/openbb_platform/providers/polygon/openbb_polygon/models/company_news.py +++ b/openbb_platform/providers/polygon/openbb_polygon/models/company_news.py @@ -8,7 +8,7 @@ from openbb_core.provider.standard_models.company_news import ( CompanyNewsData, CompanyNewsQueryParams, ) -from openbb_core.provider.utils.helpers import get_querystring +from openbb_core.provider.utils.helpers import filter_by_dates, get_querystring from openbb_polygon.utils.helpers import get_data_many, get_date_condition from pydantic import BaseModel, Field, field_validator @@ -122,4 +122,7 @@ class PolygonCompanyNewsFetcher( **kwargs: Any, ) -> List[PolygonCompanyNewsData]: """Transform data.""" - return [PolygonCompanyNewsData.model_validate(d) for d in data] + modeled_data = [PolygonCompanyNewsData.model_validate(d) for d in data] + return filter_by_dates( + modeled_data, query.start_date, query.end_date, query.date + ) diff --git a/openbb_platform/providers/tiingo/openbb_tiingo/models/company_news.py b/openbb_platform/providers/tiingo/openbb_tiingo/models/company_news.py index fa521554ec4..70ec4d0eeef 100644 --- a/openbb_platform/providers/tiingo/openbb_tiingo/models/company_news.py +++ b/openbb_platform/providers/tiingo/openbb_tiingo/models/company_news.py @@ -9,7 +9,7 @@ from openbb_core.provider.standard_models.company_news import ( CompanyNewsData, CompanyNewsQueryParams, ) -from openbb_core.provider.utils.helpers import get_querystring +from openbb_core.provider.utils.helpers import filter_by_dates, get_querystring from openbb_tiingo.utils.helpers import get_data_many from pydantic import Field, field_validator @@ -93,4 +93,7 @@ class TiingoCompanyNewsFetcher( query: TiingoCompanyNewsQueryParams, data: List[Dict], **kwargs: Any ) -> List[TiingoCompanyNewsData]: """Return the transformed data.""" - return [TiingoCompanyNewsData.model_validate(d) for d in data] + modeled_data = [TiingoCompanyNewsData.model_validate(d) for d in data] + return filter_by_dates( + modeled_data, query.start_date, query.end_date, query.date + ) diff --git a/openbb_platform/providers/ultima/openbb_ultima/models/company_news.py b/openbb_platform/providers/ultima/openbb_ultima/models/company_news.py index 880be92796d..b5f16022340 100644 --- a/openbb_platform/providers/ultima/openbb_ultima/models/company_news.py +++ b/openbb_platform/providers/ultima/openbb_ultima/models/company_news.py @@ -9,6 +9,7 @@ from openbb_core.provider.standard_models.company_news import ( CompanyNewsData, CompanyNewsQueryParams, ) +from openbb_core.provider.utils.helpers import filter_by_dates from openbb_ultima.utils.helpers import get_data from pydantic import Field @@ -93,4 +94,4 @@ class UltimaCompanyNewsFetcher( item["publisher"] = item["publisher"] item["risk_category"] = item["riskCategory"] results.append(UltimaCompanyNewsData.model_validate(item)) - return results + return filter_by_dates(results, query.start_date, query.end_date, query.date)