Source code for langchain_community.agent_toolkits.cassandra_database.toolkit

"""Apache Cassandra 工具包。"""
from typing import List

from langchain_core.pydantic_v1 import Field

from langchain_community.agent_toolkits.base import BaseToolkit
from langchain_community.tools import BaseTool
from langchain_community.tools.cassandra_database.tool import (
    GetSchemaCassandraDatabaseTool,
    GetTableDataCassandraDatabaseTool,
    QueryCassandraDatabaseTool,
)
from langchain_community.utilities.cassandra_database import CassandraDatabase



[docs]class CassandraDatabaseToolkit(BaseToolkit):
    """与Apache Cassandra数据库交互的工具包。"""

    db: CassandraDatabase = Field(exclude=True)

    class Config:
        """此pydantic对象的配置。"""

        arbitrary_types_allowed = True


[docs]    def get_tools(self) -> List[BaseTool]:
        """获取工具包中的工具。"""
        return [
            GetSchemaCassandraDatabaseTool(db=self.db),
            QueryCassandraDatabaseTool(db=self.db),
            GetTableDataCassandraDatabaseTool(db=self.db),
        ]