Skip to content

Commit

Permalink
Dh-5025/reformat with black
Browse files Browse the repository at this point in the history
  • Loading branch information
MohammadrezaPourreza committed Nov 22, 2023
1 parent 7e86d6f commit ffc71c9
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion dataherald/sql_generator/dataherald_finetuning_agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -123,6 +123,7 @@ async def _arun(
) -> str:
raise NotImplementedError("SystemTime tool does not support async")


class TablesSQLDatabaseTool(BaseSQLDatabaseTool, BaseTool):
"""Tool which takes in the given question and returns a list of tables with their relevance score to the question"""

Expand All @@ -139,7 +140,7 @@ def _run(
run_manager: CallbackManagerForToolRun | None = None, # noqa: ARG002
) -> str:
"""Use the concatenation of table name, columns names, and the description of the table as the table representation"""
tables= []
tables = []
for table in self.db_scan:
tables.append(table.table_name)
return f"Tables in the database: {','.join(tables)}"
Expand Down

0 comments on commit ffc71c9

Please sign in to comment.