Skip to content

Commit a282b3c

Browse files
authored
Merge branch 'develop' into release/4.5.0
2 parents 7264fa1 + 88947ab commit a282b3c

File tree

2 files changed

+21
-1
lines changed

2 files changed

+21
-1
lines changed

openbb_platform/providers/congress_gov/openbb_congress_gov/models/bill_info.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -83,6 +83,8 @@ async def aextract_data(
8383
) -> dict:
8484
"""Extract data from the query."""
8585
# pylint: disable=import-outside-toplevel
86+
from openbb_core.app.model.abstract.error import OpenBBError
87+
from openbb_core.provider.utils.errors import UnauthorizedError
8688
from openbb_core.provider.utils.helpers import amake_request
8789

8890
api_key = credentials.get("congress_gov_api_key", "") if credentials else ""
@@ -96,6 +98,14 @@ async def aextract_data(
9698

9799
url = bill_url + "&api_key=" + api_key
98100
base_info: dict = await amake_request(url) # type: ignore
101+
102+
if isinstance(base_info, dict) and (error := base_info.get("error", {})):
103+
if "API_KEY" in error.get("code", ""):
104+
raise UnauthorizedError(
105+
f"{error.get('code', '' )} -> {error.get('message', '')}"
106+
)
107+
raise OpenBBError(f"{error.get('code', '' )} -> {error.get('message', '')}")
108+
99109
base_info = base_info.get("bill", {})
100110
cosponsors = base_info.get("cosponsors", {})
101111

openbb_platform/providers/congress_gov/openbb_congress_gov/models/congress_bills.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -212,7 +212,8 @@ async def aextract_data(
212212
) -> list:
213213
"""Extract data from the Congress API."""
214214
# pylint: disable=import-outside-toplevel
215-
from openbb_core.provider.utils.helpers import amake_request # noqa
215+
from openbb_core.provider.utils.errors import UnauthorizedError # noqa
216+
from openbb_core.provider.utils.helpers import amake_request
216217
from openbb_congress_gov.utils.helpers import (
217218
get_all_bills_by_type,
218219
)
@@ -296,6 +297,15 @@ async def aextract_data(
296297

297298
try:
298299
response = await amake_request(url=url)
300+
if isinstance(response, dict) and (error := response.get("error", {})):
301+
if "API_KEY" in error.get("code", ""):
302+
raise UnauthorizedError(
303+
f"{error.get('code', '' )} -> {error.get('message', '')}"
304+
)
305+
raise OpenBBError(
306+
f"{error.get('code', '' )} -> {error.get('message', '')}"
307+
)
308+
299309
except Exception as e:
300310
# Handle exceptions
301311
raise OpenBBError(e) from e

0 commit comments

Comments
 (0)