mirror of
https://github.com/hwchase17/langchain
synced 2024-11-10 01:10:59 +00:00
450c458f8f
**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>
10 lines
422 B
Python
10 lines
422 B
Python
"""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
|