diff --git a/openbb_platform/core/openbb_core/provider/standard_models/balance_sheet.py b/openbb_platform/core/openbb_core/provider/standard_models/balance_sheet.py index 22e6fcf84b6..0b8832bcca9 100644 --- a/openbb_platform/core/openbb_core/provider/standard_models/balance_sheet.py +++ b/openbb_platform/core/openbb_core/provider/standard_models/balance_sheet.py @@ -29,25 +29,11 @@ class BalanceSheetQueryParams(QueryParams): default=5, description=QUERY_DESCRIPTIONS.get("limit", "") ) - # pylint: disable=inconsistent-return-statements @field_validator("symbol", mode="before", check_fields=False) @classmethod def upper_symbol(cls, v: Union[str, List[str], Set[str]]): """Convert symbol to uppercase.""" - if isinstance(v, str): - if v.isdigit() or v.isalpha(): - return v.upper() - raise ValueError(f"Invalid symbol: {v}. Must be either a ticker or CIK.") - if isinstance(v, List): - symbols = [] - for symbol in v: - if symbol.isdigit() or symbol.isalpha(): - symbols.append(symbol.upper()) - else: - raise ValueError( - f"Invalid symbol: {symbol}. Must be either a ticker or CIK." - ) - return ",".join(symbols) + return v.split(",")[0].upper() if "," in v else v.upper() class BalanceSheetData(Data): diff --git a/openbb_platform/core/openbb_core/provider/standard_models/cash_flow.py b/openbb_platform/core/openbb_core/provider/standard_models/cash_flow.py index 066098497f2..3a2f4bc1ae4 100644 --- a/openbb_platform/core/openbb_core/provider/standard_models/cash_flow.py +++ b/openbb_platform/core/openbb_core/provider/standard_models/cash_flow.py @@ -26,25 +26,11 @@ class CashFlowStatementQueryParams(QueryParams): default=5, description=QUERY_DESCRIPTIONS.get("limit", "") ) - # pylint: disable=inconsistent-return-statements @field_validator("symbol", mode="before", check_fields=False) @classmethod def upper_symbol(cls, v: Union[str, List[str], Set[str]]): """Convert symbol to uppercase.""" - if isinstance(v, str): - if v.isdigit() or v.isalpha(): - return v.upper() - raise ValueError(f"Invalid symbol: {v}. Must be either a ticker or CIK.") - if isinstance(v, List): - symbols = [] - for symbol in v: - if symbol.isdigit() or symbol.isalpha(): - symbols.append(symbol.upper()) - else: - raise ValueError( - f"Invalid symbol: {symbol}. Must be either a ticker or CIK." - ) - return ",".join(symbols) + return v.split(",")[0].upper() if "," in v else v.upper() class CashFlowStatementData(Data): diff --git a/openbb_platform/core/openbb_core/provider/standard_models/income_statement.py b/openbb_platform/core/openbb_core/provider/standard_models/income_statement.py index b77ef8c4dc4..9eb5f84f882 100644 --- a/openbb_platform/core/openbb_core/provider/standard_models/income_statement.py +++ b/openbb_platform/core/openbb_core/provider/standard_models/income_statement.py @@ -26,25 +26,11 @@ class IncomeStatementQueryParams(QueryParams): default=5, description=QUERY_DESCRIPTIONS.get("limit", "") ) - # pylint: disable=inconsistent-return-statements @field_validator("symbol", mode="before", check_fields=False) @classmethod def upper_symbol(cls, v: Union[str, List[str], Set[str]]): """Convert symbol to uppercase.""" - if isinstance(v, str): - if v.isdigit() or v.isalpha(): - return v.upper() - raise ValueError(f"Invalid symbol: {v}. Must be either a ticker or CIK.") - if isinstance(v, List): - symbols = [] - for symbol in v: - if symbol.isdigit() or symbol.isalpha(): - symbols.append(symbol.upper()) - else: - raise ValueError( - f"Invalid symbol: {symbol}. Must be either a ticker or CIK." - ) - return ",".join(symbols) + return v.split(",")[0].upper() if "," in v else v.upper() class IncomeStatementData(Data):