Function select_database_data

bmtools/tools/database/api.py:104–132

select_database_data(query : str) Read the data stored in database based on the SQL query from the translate_nlp_to_sql API. query : str is a string that represents the SQL query outputted by the translate_nlp_to_sql API. Final answer should be complete. This is

(query : str)

    @tool.get("/select_database_data")
    def select_database_data(query : str):
        """select_database_data(query : str) Read the data stored in database based on the SQL query from the translate_nlp_to_sql API.
            query : str is a string that represents the SQL query outputted by the translate_nlp_to_sql API.
            Final answer should be complete.

            This is an example:
            Thoughts: Now that I have the database schema and SQL query, I will use the \\\'select_database_data\\\' command to retrieve the data from the database based on the SQL query
            Reasoning: I will use the \\\'select_database_data\\\' command to retrieve the data from the database based on the SQL query
            Plan: - Use the \\\'select_database_data\\\' command to retrieve the data from the database based on the SQL query.\\\\n- Use the \\\'finish\\\' command to signal that I have completed all my objectives.
            Command: {"name": "select_database_data", "args": {query: "SELECT s_comment FROM supplier BY s_comment DESC"}}
            Result: Command select_database_data returned: "The number of result rows is: 394"
        """

        if query == "":
            raise RuntimeError("SQL query is empty")

        print("=========== database query:", query)
        res_completion = db.pgsql_results(query) # list format

        if res_completion == "<fail>":
            raise RuntimeError("Database query failed")

        #data = json.loads(str(res_completion).strip())
        if isinstance(res_completion, list):
            text_output = f"The number of result rows is: "+"".join(str(len(res_completion)))
        else:
            text_output = f"The number of result rows is: "+"".join(str(res_completion))

        return text_output

    @tool.get("/rewrite_sql")
    def rewrite_sql(sql: str="select distinct l_orderkey, sum(l_extendedprice + 3 + (1 - l_discount)) as revenue, o_orderkey, o_shippriority from customer, orders, lineitem where c_mktsegment = 'BUILDING' and c_custkey = o_custkey and l_orderkey = o_orderkey and o_orderdate < date '1995-03-15' and l_shipdate > date '1995-03-15' group by l_orderkey, o_orderkey, o_shippriority order by revenue desc, o_orderkey;"):