feat: SQL Prompt class and update prompt from file function

Author: nonegom

llm_utils/prompts_class.py

@@ -0,0 +1,33 @@
+from langchain.chains.sql_database.prompt import SQL_PROMPTS
+import os
+
+from langchain_core.prompts import load_prompt
+
+
+class SQLPrompt():
+    def __init__(self):
+        # os library를 확인해서 SQL_PROMPTS key에 해당하는ㅁ prompt가 있으면, 이를 교체
+        self.sql_prompts = SQL_PROMPTS
+        self.target_db_list = list(SQL_PROMPTS.keys())
+        self.prompt_path = '../prompt'
+
+    def update_prompt_from_path(self):
+        if os.path.exists(self.prompt_path):
+            path_list = os.listdir(self.prompt_path)
+            # yaml 파일만 가져옴
+            file_list = [file for file in path_list if file.endswith('.yaml')]
+            key_path_dict = {key.split('.')[0]: os.path.join(self.prompt_path, key) for key in file_list if key.split('.')[0] in self.target_db_list}
+            # file_list에서 sql_prompts의 key에 해당하는 파일이 있는 것만 가져옴
+            for key, path in key_path_dict.items():
+                self.sql_prompts[key] = load_prompt(path, encoding='utf-8')
+        else:
+            raise FileNotFoundError(f"Prompt file not found in {self.prompt_path}")
+        return False
+
+if __name__ == '__main__':
+    sql_prompts_class = SQLPrompt()
+    print(sql_prompts_class.sql_prompts['mysql'])
+    print(sql_prompts_class.update_prompt_from_path())
+
+    print(sql_prompts_class.sql_prompts['mysql'])
+    print(sql_prompts_class.sql_prompts)

prompt/mysql.yaml

@@ -0,0 +1,21 @@
+_type: prompt
+template: |
+  커스텀 MySQL 프롬프트입니다.
+  You are a MySQL expert. Given an input question, first create a syntactically correct MySQL query to run, then look at the results of the query and return the answer to the input question.
+  Unless the user specifies in the question a specific number of examples to obtain, query for at most {top_k} results using the LIMIT clause as per MySQL. You can order the results to return the most informative data in the database.
+  Never query for all columns from a table. You must query only the columns that are needed to answer the question. Wrap each column name in backticks (`) to denote them as delimited identifiers.
+  Pay attention to use only the column names you can see in the tables below. Be careful to not query for columns that do not exist. Also, pay attention to which column is in which table.
+  Pay attention to use CURDATE() function to get the current date, if the question involves "today".
+  
+  Use the following format:
+
+  Question: Question here
+  SQLQuery: SQL Query to run
+  SQLResult: Result of the SQLQuery
+  Answer: Final answer here
+  
+  Only use the following tables:
+  {table_info}
+
+  Question: {input}
+input_variables: ["input", "table_info", "top_k"]