community[minor]: Add Datahareld tool (#19680)

**Description:** Integrate [dataherald](https://www.dataherald.com)
tool, It is a natural language-to-SQL tool.
**Dependencies:** Install dataherald sdk to use it,
```
pip install dataherald
```

---------

Co-authored-by: Bagatur <22008038+baskaryan@users.noreply.github.com>
Co-authored-by: Bagatur <baskaryan@gmail.com>
Co-authored-by: Christophe Bornet <cbornet@hotmail.com>
This commit is contained in:
Juan Carlos José Camacho
2024-04-13 17:27:16 -06:00
committed by GitHub
parent ece008f117
commit 450c458f8f
11 changed files with 306 additions and 0 deletions

View File

@@ -0,0 +1,9 @@
"""Integration test for Dataherald API Wrapper."""
from langchain_community.utilities.dataherald import DataheraldAPIWrapper
def test_call() -> None:
"""Test that call gives the correct answer."""
search = DataheraldAPIWrapper(db_connection_id="65fb766367dd22c99ce1a12d")
output = search.run("How many employees are in the company?")
assert "Answer: SELECT \n COUNT(*) FROM \n employees" in output