File tree Expand file tree Collapse file tree 2 files changed +21
-1
lines changed
openbb_platform/providers/congress_gov/openbb_congress_gov/models Expand file tree Collapse file tree 2 files changed +21
-1
lines changed Original file line number Diff line number Diff 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
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments