consolidate run functions (#126)

consolidating logic for when a chain is able to run with single input
text, single output text

open to feedback on naming, logic, usefulness
This commit is contained in:
Harrison Chase
2022-11-13 18:14:35 -08:00
committed by GitHub
parent 1fe3a4f724
commit f23b3ceb49
20 changed files with 39 additions and 143 deletions
@@ -25,6 +25,6 @@ def test_sql_database_run() -> None:
conn.execute(stmt)
db = SQLDatabase(engine)
db_chain = SQLDatabaseChain(llm=OpenAI(temperature=0), database=db)
output = db_chain.query("What company does Harrison work at?")
output = db_chain.run("What company does Harrison work at?")
expected_output = " Harrison works at Foo."
assert output == expected_output