Skip to content

Commit

Permalink
fix linting in load_tools
Browse files Browse the repository at this point in the history
  • Loading branch information
kritgrover authored Nov 25, 2024
1 parent 0cc4f17 commit 76cf86b
Showing 1 changed file with 3 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -392,6 +392,7 @@ def _get_golden_query(**kwargs: Any) -> BaseTool:
def _get_pubmed(**kwargs: Any) -> BaseTool:
return PubmedQueryRun(api_wrapper=PubMedAPIWrapper(**kwargs))


def _get_finance_polygon_all_tickers(**kwargs: Any) -> BaseTool:
return PolygonAllTickers(api_wrapper=FinancePolygonAPIWrapper(**kwargs))

Expand Down Expand Up @@ -511,11 +512,13 @@ def _get_finance_polygon_ref_ticker_news(**kwargs: Any) -> BaseTool:
def _get_finance_polygon_trades(**kwargs: Any) -> BaseTool:
return PolygonTrades(api_wrapper=FinancePolygonAPIWrapper(**kwargs))


def _get_google_books(**kwargs: Any) -> BaseTool:
from langchain_community.tools.google_books import GoogleBooksQueryRun

return GoogleBooksQueryRun(api_wrapper=GoogleBooksAPIWrapper(**kwargs))


def _get_google_jobs(**kwargs: Any) -> BaseTool:
return GoogleJobsQueryRun(api_wrapper=GoogleJobsAPIWrapper(**kwargs))

Expand Down

0 comments on commit 76cf86b

Please sign in to comment.