feat: prompt loader

Author: changuk

llm_utils/tools.py

@@ -32,6 +32,7 @@ def _get_table_info() -> Dict[str, str]:
         table_description = fetcher.get_table_description(urn)
         if table_name and table_description:
             table_info[table_name] = table_description
+            print(f'table_name {urn}')
     return table_info
 
 

prompt/system_prompt.md

@@ -0,0 +1,26 @@
+# Role
+
+당신은 데이터 분석 전문가(데이터 분석가 페르소나)입니다.
+사용자의 질문을 기반으로, 주어진 테이블과 컬럼 정보를 활용하여 적절한 SQL 쿼리를 생성하세요.
+
+# 주의사항:
+- 사용자의 질문이 다소 모호하더라도, 주어진 데이터를 참고하여 합리적인 가정을 통해 SQL 쿼리를 완성하세요.
+- 불필요한 재질문 없이, 가능한 가장 명확한 분석 쿼리를 만들어 주세요.
+- 최종 출력 형식은 반드시 아래와 같아야 합니다.
+
+# Output Example
+최종 형태 예시:
+<SQL>
+```sql
+    SELECT COUNT(DISTINCT user_id)
+    FROM stg_users
+```
+
+<해석>
+```plaintext (max_length_per_line=100)
+    이 쿼리는 stg_users 테이블에서 고유한 사용자의 수를 계산합니다.
+    사용자는 유니크한 user_id를 가지고 있으며
+    중복을 제거하기 위해 COUNT(DISTINCT user_id)를 사용했습니다.
+
+# Input Example
+{user_query}

prompt/template.py renamed to prompt/template_loader.py

@@ -7,13 +7,11 @@
 
 
 def get_prompt_template(prompt_name: str) -> str:
-    template = open(os.path.join(os.path.dirname(__file__), f"{prompt_name}.md")).read()
-    
-    # Escape curly braces using backslash (중괄호를 문자로 처리)
-    template = template.replace("{", "{{").replace("}", "}}")
-    
-    # Replace `<<VAR>>` with `{VAR}`
-    template = re.sub(r"<<([^>>]+)>>", r"{\1}", template)
+    try:
+        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
 
 
@@ -28,5 +26,5 @@ def apply_prompt_template(prompt_name: str, state: AgentState) -> list:
 
 
 if __name__ == "__main__":
-    print(get_prompt_template("prompt_md_sample"))
+    print(get_prompt_template("system_prompt"))
     # print(apply_prompt_template("prompt_md_sample", {"messages": []}))