Toggle Menu
Prev Up Next

# Source code for langchain_community.agent_toolkits.polygon.toolkit

    from typing import List

    from langchain_core.tools import BaseToolkit

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

    class PolygonToolkit(BaseToolkit):
        """Polygon Toolkit."""

        tools: List[BaseTool] = []

        @classmethod
        def from_polygon_api_wrapper(
            cls, polygon_api_wrapper: PolygonAPIWrapper
        ) -> "PolygonToolkit":
            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)

        def get_tools(self) -> List[BaseTool]:
            """Get the tools in the toolkit."""
            return self.tools