diff options
7 files changed, 51 insertions, 43 deletions
diff --git a/openbb_platform/core/openbb_core/provider/standard_models/equity_search.py b/openbb_platform/core/openbb_core/provider/standard_models/equity_search.py index 8ec02ceeb80..195e99dbeef 100644 --- a/openbb_platform/core/openbb_core/provider/standard_models/equity_search.py +++ b/openbb_platform/core/openbb_core/provider/standard_models/equity_search.py @@ -16,6 +16,10 @@ class EquitySearchQueryParams(QueryParams): is_symbol: bool = Field( description="Whether to search by ticker symbol.", default=False ) + use_cache: Optional[bool] = Field( + default=True, + description="Whether to use the cache or not.", + ) class EquitySearchData(Data): diff --git a/openbb_platform/extensions/equity/integration/test_equity_api.py b/openbb_platform/extensions/equity/integration/test_equity_api.py index 9ec9e781119..c34badc62a5 100644 --- a/openbb_platform/extensions/equity/integration/test_equity_api.py +++ b/openbb_platform/extensions/equity/integration/test_equity_api.py @@ -1212,7 +1212,15 @@ def test_equity_fundamental_latest_attributes(params, headers): ({"query": "AAPl", "is_symbol": True, "provider": "cboe", "use_cache": False}), ({"query": "Apple", "provider": "sec", "use_cache": False, "is_fund": False}), ({"query": "", "provider": "nasdaq", "use_cache": False, "is_etf": True}), - ({"query": "gold", "provider": "intrinio", "active": True, "limit": 100}), + ( + { + "query": "gold", + "provider": "intrinio", + "active": True, + "limit": 100, + "use_cache": None, + } + ), ], ) @pytest.mark.integration diff --git a/openbb_platform/extensions/equity/integration/test_equity_python.py b/openbb_platform/extensions/equity/integration/test_equity_python.py index 2e866217ef2..7c97baaea18 100644 --- a/openbb_platform/extensions/equity/integration/test_equity_python.py +++ b/openbb_platform/extensions/equity/integration/test_equity_python.py @@ -1150,7 +1150,15 @@ def test_equity_fundamental_latest_attributes(params, obb): ({"query": "AAPL", "is_symbol": True, "provider": "cboe", "use_cache": False}), ({"query": "Apple", "provider": "sec", "use_cache": False, "is_fund": False}), ({"query": "", "provider": "nasdaq", "use_cache": False, "is_etf": True}), - ({"query": "gold", "provider": "intrinio", "active": True, "limit": 100}), + ( + { + "query": "gold", + "provider": "intrinio", + "active": True, + "limit": 100, + "use_cache": None, + } + ), ], ) @pytest.mark.integration diff --git a/openbb_platform/providers/cboe/openbb_cboe/models/equity_search.py b/openbb_platform/providers/cboe/openbb_cboe/models/equity_search.py index 428a7d92023..a67f1453bd2 100644 --- a/openbb_platform/providers/cboe/openbb_cboe/models/equity_search.py +++ b/openbb_platform/providers/cboe/openbb_cboe/models/equity_search.py @@ -17,12 +17,6 @@ class CboeEquitySearchQueryParams(EquitySearchQueryParams): Source: https://www.cboe.com/ """ - use_cache: bool = Field( - default=True, - description="When True, the company directory of optionable tickers will be cached for 24 hours." - + " Set as False to bypass.", - ) - class CboeEquitySearchData(EquitySearchData): """CBOE Equity Search Data.""" diff --git a/openbb_platform/providers/intrinio/tests/record/http/test_intrinio_fetchers/test_intrinio_equity_search_fetcher.yaml b/openbb_platform/providers/intrinio/tests/record/http/test_intrinio_fetchers/test_intrinio_equity_search_fetcher.yaml index 292cee77eb4..5dcb8c6f8d5 100644 --- a/openbb_platform/providers/intrinio/tests/record/http/test_intrinio_fetchers/test_intrinio_equity_search_fetcher.yaml +++ b/openbb_platform/providers/intrinio/tests/record/http/test_intrinio_fetchers/test_intrinio_equity_search_fetcher.yaml @@ -3,36 +3,37 @@ interactions: body: null headers: Accept: - - '*/*' + - application/json Accept-Encoding: - - gzip, deflate, br + - gzip, deflate Connection: - keep-alive method: GET - uri: https://api-v2.intrinio.com/companies/search?active=True&api_key=MOCK_API_KEY&page_size=100&query=gold + uri: https://api-v2.intrinio.com/companies/search?active=True&api_key=MOCK_API_KEY&page_size=100&query=gold&use_cache=True response: body: string: !!binary | - H4sIAHb8b2UAA4yWW3PiOBCF/wrF8+yWkGxZnjdfQAFsY2wuhq2tKQMujyfEJlwSYGr++4rNZlCL - YnHewOFT6/ic7v7ZXFYvm7Qssl3z618/m8Wq+fXy1bfF+bW/yJtfmvti+Zxtxbd84Lnic5m+ZOKT - bUVR1+k3Lt82nEEUikfrrBBP0EDr20OnN08YHve5Y08MIh4ui+fLQ/Fn6JSZWvPXF+m4c3vS3pzk - 47rWOLkeV/zBUINX61XDqbabPz8PKw/rtYRuMZ2YFCvoV1tfAHQw4V7nyg4mV/I9MEUGQ4YC9rNX - dgZgLikUZO8f3G65/C2OrpkEIZ0jHjhPIXqinkmBOOxyGIUH5RkbTMENxrE1vp40jh9eACFsIJMo - 3NkPGiQytz/irqRMHz9WRme3kifT43oGlBlFUSKBR0F06xyVrBG9ZZpKyS7N3gA59i0ukePqJa1h - FEPAEYLs08LdWkCOiTV2JPakeKvFxgRpqld6Ln8HcfKTmF/RfnY87D7Y4/gemOg6NRTwibylQyhI - xOeSHvt0W6NmTWSH6Gpwtu4rsF00a8dXdFTssjporDGqxP20Cn60gdRekMiv0SvKw/Gx+RBjwoCQ - vc2RHwF2xOXOFVWndH2TzJu8MF1XBcmPxqYPm1TgWFLVgpc2BmUdVXRDw0Q199GfPYPKff4kG9Av - ymVVQxWMGFZqT7jHpsAnPaGKxO6lYgzUKJwyitTk5JrnUuBueySZ28b5o5I1bJpEsXYeDtfQJBPZ - 2ZNit09vxfivxeKp0xl5/YndZa7G4PxhhGDVkC36boAbjGweyh2r7duDa8+6Kw+59EPlIgkNCQW+ - iXkgT6C4yEvJk//TWTBuKZXnk0UJtbd494q21ulLtbs3iEbuIDHIjM+StqmbskqtliGypRpUa//A - MFoTT45tlL1ldS5ChY2Ymi7khS7cOKaxnC5evWe7/Qfc26/uRwszTKg6q8vFM1wCOtyBw7pTHcrV - 43QxDYny1S7sDUHtHz/6jG36vqvK34XfFQVjis2bSbqlILbeIJBbgleJlrB9rLhmYKq+zrM/9dtA - ccfmcwnupIttrdeJTGYoy8WZBXYFfWnJS8sF+iI6Tpwuv+8a4ffTrlh+ntUede5GAIlhoqq/iRYa - sCXvdOFJjU6RrVc7Wf6mLnoOQqSfdO3ecObYtmtgJQLYUBvFucLTEkbAHXfkCFhu97KQXTqFGA93 - NWu1CKZInbhROAQudSNpcokP/2JvLzFHIeFRFGDic6I14TAQa5S6EvvTlQnezZi7XekS47LIHwxI - U6MaUTeotauvQf3WnMviWPNR27Eer35ipSS6GuFeSG1oqJgPZPguLfe1Zq9Y0FTpczzL2qB0zwe7 - sHd4Kco66x81CSaK4PmQtxcwxmEf7DuHTfpch65jsVuqsvdDLwXKxP6gA5bistpWdeiEtZjoEn+L - n2bH/bdNmmcf//XrHwAAAP//AwB3pkhmKg4AAA== + H4sIAN0HwWUAA4yX23LiOBCGX4XienZLlnyQ586nCPABkDkYtramDHg9zoDNKQkwNe8+YkkGSRTB + uTMOn1p/99/d/GzOq9U6LYts1/z6z89msWh+PX/0bXba+LO8+aW5L+Y/si37lHQDlz2X6SpjT7ZF + advxG+dPG06X9tirZVawN6Cr+nbf6UwTDIc+ceyRgdjLefHj/JL9GZqOTbX56wt33Mkbeesjf1zb + GibX44q/MGiQarloONV2/ffHYeXLcsmhFawhU4cSemNrMwEdjUjwdGVHoyv5HlgHBgaGCM4z3B0L + 4GFsDa/cYfyQCwA0gIkk7uRZjxKe6w+IywXsw8cBa/hWiWR8WE5OPHhAacKBBxG9TahMVpGmmKZI + Pi7dStQ4JtThyHFavsf8GRmoWFFuyFORTD3H4sjBMHIvQQcD9x6Z1RyrOklmK/UXuagGV3Hs4V2L + T1RWDKSrUvqOij41BG47ph0u4l5EauQPGoYi5S939exVyF8cWoRXuVqlNVxisBQCIEU9c7eWUHQj + a8hncFS81mJDBFTJKKcw22Ah7ohwrSTK3i7gdjn/00U01UQAaASQyGn1QEsPTF3oIvjsSimlp45L + 3gTpwyQm15PC7PCyu5w1jO/dAGmabsg32NhArMLWpMVb0rGm1tU6d9VRVUNRpKCP6DXti1mlZMol + dZ9uawivsu6HNDnsrbsRzTPx4iuaFrusDhqqWJcb9mpblgLaIS7fSxIraNdRxDBVLNViYnTGmZDG + uCU0wLjVpd4F3o6ce2iIkWLIYi+iZ08o8yBKeAsFRflyeGxPgDFrsZI9ATWp2F4nLd747NHzHjdY + xcAqlNmHcPJDiDskLd6eYVHOqxpxQ4ChVCW5HYJQSKUXUO+KnhTlfy91ygRhVZfad0ICPBY06ZCA + T2UnXf8ZDZ/BdawDuWflauDqQp3YA87tNswfyaFC00SS17c5CKkgNSX85kOrY7q8aVg3gx1rmmzH + /GCsfXHJiYRZxnhpo1vW8aRmqBBJUzjRe0iXpnDEbzpxkZdc7J80cQgVyfD5aFaKYlukfUVby3RV + 7e718YHbTQw0IZPEMzWzKVU7Vkw5sd4zFFMwCniX0uw1q3MRndUN1uR+G/Rd/iKXL334KH3bVe8F + GewXd8EQ6lCWf3zY6kKtB92I92hQMY9uH0etGlCXJTmF49AT5HdsMuUnUDrb1pIEmNiQVpYTjuxK + zK3Fb7Jn6IrZNE7n33eN3vfjrph/nOUNnu6WEWBjQx7RG9rrCzXqUpcz7fnp/2Xuqn1TYy4FYAp6 + iFAaQRQSpDbFrsZ2UnntD8cLU7jTkLhtTrBhWeQPTGyqbLuTm06vvxQnyIif1qNit09v+/C7D+DY + eRoE/shuY1fF4q8ijBCUh2ynp9tiXmLS5e5g7dJyX6tdsLVMl2/SX2/EdmRFvkCPiN+lNYa4AQFQ + ZTqcZJ5AD0JhiAcvq6Kss1LqJoJIyu7R7wWpuCKE3SdhFS6rbVVHGYQVLPefbLE0RWVaEeWEGbS8 + yHo8ydkqrALG/pd9Mzvsv63TPLv816/fAAAA//8DANBYgPZ5DwAA headers: Connection: - keep-alive @@ -41,7 +42,7 @@ interactions: Content-Type: - application/json Date: - - Wed, 06 Dec 2023 04:45:42 GMT + - Mon, 05 Feb 2024 16:07:58 GMT Transfer-Encoding: - chunked Vary: diff --git a/openbb_platform/providers/nasdaq/openbb_nasdaq/models/equity_search.py b/openbb_platform/providers/nasdaq/openbb_nasdaq/models/equity_search.py index ce5d6d74c66..77ada033b51 100644 --- a/openbb_platform/providers/nasdaq/openbb_nasdaq/models/equity_search.py +++ b/openbb_platform/providers/nasdaq/openbb_nasdaq/models/equity_search.py @@ -33,10 +33,6 @@ class NasdaqEquitySearchQueryParams(EquitySearchQueryParams): default=None, description="If True, returns ETFs.", ) - use_cache: Optional[bool] = Field( - default=True, - description="If True, caches the symbol directory for one day.", - ) class NasdaqEquitySearchData(EquitySearchData): @@ -108,16 +104,17 @@ class NasdaqEquitySearchFetcher( @staticmethod def transform_query(params: Dict[str, Any]) -> NasdaqEquitySearchQueryParams: + """Transform the query parameters.""" return NasdaqEquitySearchQueryParams(**params) + # pylint: disable=unused-argument @staticmethod - async def extract_data( + async def aextract_data( query: NasdaqEquitySearchQueryParams, credentials: Optional[Dict[str, str]], **kwargs: Any, ) -> str: """Extract data from Nasdaq.""" - url = "https://www.nasdaqtrader.com/dynamic/SymDir/nasdaqtraded.txt" def fetch_data(): @@ -133,6 +130,7 @@ class NasdaqEquitySearchFetcher( loop = asyncio.get_running_loop() return await loop.run_in_executor(None, fetch_data) + # pylint: disable=unused-argument @staticmethod def transform_data( query: NasdaqEquitySearchQueryParams, @@ -140,7 +138,6 @@ class NasdaqEquitySearchFetcher( **kwargs: Any, ) -> List[NasdaqEquitySearchData]: """Transform the data and filter the results.""" - directory = read_csv(StringIO(data), sep="|").iloc[:-1] if query.is_etf is True: diff --git a/openbb_platform/providers/sec/openbb_sec/models/equity_search.py b/openbb_platform/providers/sec/openbb_sec/models/equity_search.py index 0438bc694f1..9a3b49409af 100644 --- a/openbb_platform/providers/sec/openbb_sec/models/equity_search.py +++ b/openbb_platform/providers/sec/openbb_sec/models/equity_search.py @@ -25,10 +25,6 @@ class SecEquitySearchQueryParams(EquitySearchQueryParams): default=False, description="Whether to direct the search to the list of mutual funds and ETFs.", ) - use_cache: bool = Field( - default=True, - description="Whether to use the cache or not. Company names, tickers, and CIKs are cached for seven days.", - ) class SecEquitySearchData(EquitySearchData): |