Source code for langchain_community.agent_toolkits.polygon.toolkit

from typing import List

from langchain_core.tools import BaseTool
from langchain_core.tools.base import BaseToolkit

from langchain_community.tools.polygon import (
    PolygonAggregates,
    PolygonFinancials,
    PolygonLastQuote,
    PolygonTickerNews,
)
from langchain_community.utilities.polygon import PolygonAPIWrapper


[docs]class PolygonToolkit(BaseToolkit): """Polygon Toolkit. Parameters: tools: List[BaseTool]. The tools in the toolkit. """ tools: List[BaseTool] = []
[docs] @classmethod def from_polygon_api_wrapper( cls, polygon_api_wrapper: PolygonAPIWrapper ) -> "PolygonToolkit": """Create a Polygon Toolkit from a Polygon API Wrapper. Args: polygon_api_wrapper: PolygonAPIWrapper. The Polygon API Wrapper. Returns: PolygonToolkit. The Polygon Toolkit. """ tools = [ PolygonAggregates( api_wrapper=polygon_api_wrapper, ), PolygonLastQuote( api_wrapper=polygon_api_wrapper, ), PolygonTickerNews( api_wrapper=polygon_api_wrapper, ), PolygonFinancials( api_wrapper=polygon_api_wrapper, ), ] return cls(tools=tools)
[docs] def get_tools(self) -> List[BaseTool]: """Get the tools in the toolkit.""" return self.tools