mirror of
https://github.com/Mintplex-Labs/langchain-python.git
synced 2026-08-27 19:29:56 -04:00
Ankush Gola
82baecc892
This PR adds * `ZeroShotAgent.as_sql_agent`, which returns an agent for interacting with a sql database. This builds off of `SQLDatabaseChain`. The main advantages are 1) answering general questions about the db, 2) access to a tool for double checking queries, and 3) recovering from errors * `ZeroShotAgent.as_json_agent` which returns an agent for interacting with json blobs. * Several examples in notebooks --------- Co-authored-by: Harrison Chase <hw.chase.17@gmail.com>
16 lines
364 B
Python
16 lines
364 B
Python
"""Toolkits for agents."""
|
|
from abc import abstractmethod
|
|
from typing import List
|
|
|
|
from pydantic import BaseModel
|
|
|
|
from langchain.tools import BaseTool
|
|
|
|
|
|
class BaseToolkit(BaseModel):
|
|
"""Class responsible for defining a collection of related tools."""
|
|
|
|
@abstractmethod
|
|
def get_tools(self) -> List[BaseTool]:
|
|
"""Get the tools in the toolkit."""
|