a bg9@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__rrw/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/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.toolsrZpydanticrrZ(langchain_community.utilities.dataheraldrr rrrrrs