style: apply Black formatter to Python files

Author: ehddnr301

interface/streamlit_app.py

@@ -8,4 +8,3 @@
 )
 
 pg.run()
-

llm_utils/chains.py

@@ -1,5 +1,9 @@
 import os
-from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder, SystemMessagePromptTemplate
+from langchain_core.prompts import (
+    ChatPromptTemplate,
+    MessagesPlaceholder,
+    SystemMessagePromptTemplate,
+)
 
 from .llm_factory import get_llm
 
@@ -15,8 +19,9 @@
 
 llm = get_llm()
 
+
 def create_query_refiner_chain(llm):
-    prompt = get_prompt_template('query_refiner_prompt')
+    prompt = get_prompt_template("query_refiner_prompt")
     tool_choice_prompt = ChatPromptTemplate.from_messages(
         [
             SystemMessagePromptTemplate.from_template(prompt),
@@ -38,7 +43,7 @@ def create_query_refiner_chain(llm):
 # QueryMakerChain
 def create_query_maker_chain(llm):
     # SystemPrompt만 yaml 파일로 불러와서 사용
-    prompt = get_prompt_template('query_maker_prompt')
+    prompt = get_prompt_template("query_maker_prompt")
     query_maker_prompt = ChatPromptTemplate.from_messages(
         [
             SystemMessagePromptTemplate.from_template(prompt),
@@ -67,4 +72,4 @@ def create_query_maker_chain(llm):
 query_maker_chain = create_query_maker_chain(llm)
 
 if __name__ == "__main__":
-    query_refiner_chain.invoke()
+    query_refiner_chain.invoke()

llm_utils/graph.py

@@ -35,7 +35,6 @@ class QueryMakerState(TypedDict):
 
 # 노드 함수: QUERY_REFINER 노드
 def query_refiner_node(state: QueryMakerState):
-    print('query_refiner_node 진입 [md]')
     res = query_refiner_chain.invoke(
         input={
             "user_input": [state["messages"][0].content],
@@ -61,7 +60,6 @@ def get_table_info_node(state: QueryMakerState):
         )
     except:
         documents = get_info_from_db()
-        print("db_embedding 진입")
         db = FAISS.from_documents(documents, embeddings)
         db.save_local(os.getcwd() + "/table_info_db")
     doc_res = db.similarity_search(state["messages"][-1].content)

llm_utils/prompts_class.py

@@ -4,30 +4,26 @@
 from langchain_core.prompts import load_prompt
 
 
-class SQLPrompt():
+class SQLPrompt:
     def __init__(self):
-        # os library를 확인해서 SQL_PROMPTS key에 해당하는ㅁ prompt가 있으면, 이를 교체
+        # os library를 확인해서 SQL_PROMPTS key에 해당하는 prompt가 있으면, 이를 교체
         self.sql_prompts = SQL_PROMPTS
         self.target_db_list = list(SQL_PROMPTS.keys())
-        self.prompt_path = '../prompt'
+        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 = [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')
+                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/template_loader.py

@@ -1,15 +1,14 @@
 import os
 
 
-
 def get_prompt_template(prompt_name: str) -> str:
     try:
-        with open(os.path.join(os.path.dirname(__file__), f"{prompt_name}.md"), "r", encoding="utf-8") as f:
+        with open(
+            os.path.join(os.path.dirname(__file__), f"{prompt_name}.md"),
+            "r",
+            encoding="utf-8",
+        ) as f:
             template = f.read()
     except FileNotFoundError:
         raise FileNotFoundError(f"경고: '{prompt_name}.md' 파일을 찾을 수 없습니다.")
     return template
-
-if __name__ == "__main__":
-    print(get_prompt_template("system_prompt"))
-    # print(apply_prompt_template("prompt_md_sample", {"messages": []}))