U h'@sldZddlmZmZddlmZddlmZmZddl m Z ddl m Z GdddeZ Gd d d e Zd S) z"Tool for the Dataherald Hosted API)OptionalType)CallbackManagerForToolRun) BaseModelField)BaseTool)DataheraldAPIWrapperc@s eZdZUeddZeed<dS)DataheraldTextToSQLInputz7Natural language query to be translated to a SQL query.) descriptionpromptN)__name__ __module__ __qualname__rr str__annotations__rrM/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/tools/dataherald/tool.pyr s r c@sZeZdZUdZdZeed<dZeed<eed<e Z e e ed<d ee eed d d ZdS) DataheraldTextToSQLz+Tool that queries using the Dataherald SDK.Z dataheraldnamezcA wrapper around Dataherald. Text to SQL. Input should be a prompt and an existing db_connection_idr api_wrapper args_schemaN)r run_managerreturncCs |j|S)zUse the Dataherald tool.)rrun)selfr rrrr_runszDataheraldTextToSQL._run)N)r r r__doc__rrrr rr rrrrrrrrrrrs   rN)rtypingrrZlangchain_core.callbacksrZlangchain_core.pydantic_v1rrZlangchain_core.toolsrZ(langchain_community.utilities.dataheraldrr rrrrrs