From 9ed334d0bdc8366a87e8b902a0a2b56496e05a1d Mon Sep 17 00:00:00 2001 From: Rishabh Srivastava Date: Thu, 30 May 2024 20:04:35 +0800 Subject: [PATCH] linting --- utils/dialects.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/utils/dialects.py b/utils/dialects.py index 2b12dda..97c6fbf 100644 --- a/utils/dialects.py +++ b/utils/dialects.py @@ -78,7 +78,9 @@ async def amend_invalid_sql( """ Use LLM to correct a list of invalid SQL queries given a common question, specific error message and the database schema """ - openai = AsyncOpenAI(api_key=os.environ.get("OPENAI_API_KEY", "ADD_YOUR_OPENAI_KEY_HERE")) + openai = AsyncOpenAI( + api_key=os.environ.get("OPENAI_API_KEY", "ADD_YOUR_OPENAI_KEY_HERE") + ) completion_dict_list = [] for i in range(len(sql_list)): sql = sql_list[i]