langchain.agents.agent_toolkits.spark_sql.toolkit.SparkSQLToolkit

class langchain.agents.agent_toolkits.spark_sql.toolkit.SparkSQLToolkit(*, db: SparkSQL, llm: BaseLanguageModel)

Bases: BaseToolkit

Toolkit for interacting with Spark SQL.

Create a new model by parsing and validating input data from keyword arguments.

Raises ValidationError if the input data cannot be parsed to form a valid model.

param db: langchain.utilities.spark_sql.SparkSQL [Required]

param llm: langchain.schema.language_model.BaseLanguageModel [Required]

get_tools() → List[BaseTool]

Get the tools in the toolkit.

model Config

Bases: object

Configuration for this pydantic object.

arbitrary_types_allowed = True

Examples using SparkSQLToolkit

• Spark SQL Agent