o Zh9@sldZddlmZmZddlmZddlmZddlm Z m Z ddl m Z Gddde Z Gd d d eZd S) z"Tool for the Dataherald Hosted API)OptionalType)CallbackManagerForToolRun)BaseTool) BaseModelField)DataheraldAPIWrapperc@s eZdZUeddZeed<dS)DataheraldTextToSQLInputz7Natural language query to be translated to a SQL query.) descriptionpromptN)__name__ __module__ __qualname__rr str__annotations__rr`/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/tools/dataherald/tool.pyr s r c@s`eZdZUdZdZeed<dZeed<eed<e Z e e ed< dd ed e ed efd 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_schemaNr run_managerreturncCs |j|S)zUse the Dataherald tool.)rrun)selfr rrrr_runs zDataheraldTextToSQL._run)N)r r r__doc__rrrr rr rrrrrrrrrrrs   rN)rtypingrrZlangchain_core.callbacksrZlangchain_core.toolsrZpydanticrrZ(langchain_community.utilities.dataheraldrr rrrrrs