@nine-lab/nine-connector 0.1.5 → 0.1.7

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@nine-lab/nine-connector",
-  "version": "0.1.5",
+  "version": "0.1.7",
   "description": "NineQuery AI Connector for Database",
   "main": "src/index.js",
   "type": "module",

package/src/core/init.js

@@ -1,6 +1,7 @@
 import fs from 'fs';
 import path from 'path';
 import readline from 'readline';
+import { DEFAULT_PROMPTS } from './prompts.js'; // 분리한 프롬프트 가져오기
 
 /**
  * 1. .env 파일 생성 담당 함수
@@ -40,7 +41,7 @@ async function createEnvFile(rl) {
 
     const envContent = Object.entries(answers)
         .map(([key, val]) => `${key}=${val}`)
-        .join('\n');
+        .join('\n') + '\n';
 
     fs.writeFileSync(envPath, envContent);
     console.log('.env 파일 생성이 완료되었습니다.');
@@ -57,61 +58,14 @@ function createPromptFiles() {
         fs.mkdirSync(promptDir, { recursive: true });
     }
 
-    const prompts = {
-        'table-filter.md': `## TABLE_FILTER_PROMPT
-당신은 SQL 전문가입니다. 사용자의 질문을 분석하여 데이터베이스에서 조회해야 할 필수 테이블들만 선정하세요.
-
-[전체 테이블 목록]
-{schema_summary}
-
-[사용자 질문]
-"{question}"
-
-[응답 규칙]
-1. 반드시 JSON 형식으로만 응답하세요.
-2. 질문을 해결하기 위해 반드시 참조해야 하는 테이블 이름만 'selected_tables' 배열에 넣으세요.
-3. 만약 질문이 데이터베이스 조회와 관련이 없거나, 제공된 테이블 정보로 답할 수 없다면 'is_executable'을 false로 설정하세요.
-4. 'reasoning' 필드에는 왜 해당 테이블들을 선택했는지 간략하게 설명하세요.
-
-[응답 포맷]
-{{
-"reasoning": "선정 이유를 여기에 작성",
-"selected_tables": ["table1", "table2"],
-"is_executable": true
-}}`,
-
-        'query-generator.md': `## SQL_GENERATE_PROMPT
-당신은 SQL 생성기입니다. 제공된 테이블의 컬럼 정보를 바탕으로 사용자의 질문에 답하는 최적의 SQL 쿼리를 작성하세요.
-
-[선택된 테이블 정보]
-{detailed_schema}
-
-[사용자 질문]
-"{question}"
-
-[데이터베이스 타입]
-{db_type}
-
-[응답 규칙]
-1. 반드시 JSON 형식으로만 응답하세요.
-2. SQL은 {db_type} 문법에 완벽하게 맞아야 하며, 즉시 실행 가능해야 합니다.
-3. 쿼리문 안에 마크다운 코드 블록(\`\`\`sql)을 포함하지 마세요. 오직 순수 문자열로만 작성하세요.
-4. 사용자가 이해하기 쉽도록 쿼리에 대한 설명(explanation)을 친절하게 작성하세요.
-5. 존재하지 않는 컬럼이나 테이블은 절대 사용하지 마세요.
-
-[응답 포맷]
-{{
-"sql": "SELECT ... FROM ... WHERE ...",
-"explanation": "이 쿼리는 어떤 데이터를 어떻게 조회하는지에 대한 설명입니다."
-}}`
-    };
-
-    Object.entries(prompts).forEach(([filename, content]) => {
+    // 외부에서 가져온 DEFAULT_PROMPTS 사용
+    Object.entries(DEFAULT_PROMPTS).forEach(([filename, content]) => {
         const filePath = path.join(promptDir, filename);
-        fs.writeFileSync(filePath, content.trim());
+        // trim() 후 개행문자 추가하여 깔끔하게 저장
+        fs.writeFileSync(filePath, content.trim() + '\n', 'utf-8');
     });
 
-    console.log('프롬프트 파일 생성이 완료되었습니다.');
+    console.log('프롬프트 파일(prompts/*.md) 생성이 완료되었습니다.');
 }
 
 /**

package/src/core/prompts.js

@@ -0,0 +1,48 @@
+export const DEFAULT_PROMPTS = {
+    'table-filter.md': `### TABLE_FILTER_PROMPT
+당신은 SQL 전문가입니다. 사용자의 질문을 분석하여 데이터베이스에서 조회해야 할 필수 테이블들만 선정하세요.
+
+[전체 테이블 목록]
+{schema_summary}
+
+[사용자 질문]
+"{question}"
+
+[응답 규칙]
+1. 반드시 JSON 형식으로만 응답하세요.
+2. 질문을 해결하기 위해 반드시 참조해야 하는 테이블 이름만 'selected_tables' 배열에 넣으세요.
+3. 만약 질문이 데이터베이스 조회와 관련이 없거나, 제공된 테이블 정보로 답할 수 없다면 'is_executable'을 false로 설정하세요.
+4. 'reasoning' 필드에는 왜 해당 테이블들을 선택했는지 간략하게 설명하세요.
+
+[응답 포맷]
+{{
+"reasoning": "선정 이유를 여기에 작성",
+"selected_tables": ["table1", "table2"],
+"is_executable": true
+}}`,
+
+    'query-generator.md': `### SQL_GENERATE_PROMPT
+당신은 SQL 생성기입니다. 제공된 테이블의 컬럼 정보를 바탕으로 사용자의 질문에 답하는 최적의 SQL 쿼리를 작성하세요.
+
+[선택된 테이블 정보]
+{detailed_schema}
+
+[사용자 질문]
+"{question}"
+
+[데이터베이스 타입]
+{db_type}
+
+[응답 규칙]
+1. 반드시 JSON 형식으로만 응답하세요.
+2. SQL은 {db_type} 문법에 완벽하게 맞아야 하며, 즉시 실행 가능해야 합니다.
+3. 쿼리문 안에 마크다운 코드 블록(\`\`\`sql)을 포함하지 마세요. 오직 순수 문자열로만 작성하세요.
+4. 사용자가 이해하기 쉽도록 쿼리에 대한 설명(explanation)을 친절하게 작성하세요.
+5. 존재하지 않는 컬럼이나 테이블은 절대 사용하지 마세요.
+
+[응답 포맷]
+{{
+"sql": "SELECT ... FROM ... WHERE ...",
+"explanation": "이 쿼리는 어떤 데이터를 어떻게 조회하는지에 대한 설명입니다."
+}}`
+};

package/prompts/query-generator.md

@@ -1,24 +0,0 @@
-### SQL_GENERATE_PROMPT
-당신은 SQL 생성기입니다. 제공된 테이블의 컬럼 정보를 바탕으로 사용자의 질문에 답하는 최적의 SQL 쿼리를 작성하세요.
-
-[선택된 테이블 정보]
-{detailed_schema}
-
-[사용자 질문]
-"{question}"
-
-[데이터베이스 타입]
-{db_type}
-
-[응답 규칙]
-1. 반드시 JSON 형식으로만 응답하세요.
-2. SQL은 {db_type} 문법에 완벽하게 맞아야 하며, 즉시 실행 가능해야 합니다.
-3. 쿼리문 안에 마크다운 코드 블록(```sql)을 포함하지 마세요. 오직 순수 문자열로만 작성하세요.
-4. 사용자가 이해하기 쉽도록 쿼리에 대한 설명(explanation)을 친절하게 작성하세요.
-5. 존재하지 않는 컬럼이나 테이블은 절대 사용하지 마세요.
-
-[응답 포맷]
-{{
-"sql": "SELECT ... FROM ... WHERE ...",
-"explanation": "이 쿼리는 어떤 데이터를 어떻게 조회하는지에 대한 설명입니다."
-}}

package/prompts/table-filter.md

@@ -1,21 +0,0 @@
-### TABLE_FILTER_PROMPT
-당신은 SQL 전문가입니다. 사용자의 질문을 분석하여 데이터베이스에서 조회해야 할 필수 테이블들만 선정하세요.
-
-[전체 테이블 목록]
-{schema_summary}
-
-[사용자 질문]
-"{question}"
-
-[응답 규칙]
-1. 반드시 JSON 형식으로만 응답하세요.
-2. 질문을 해결하기 위해 반드시 참조해야 하는 테이블 이름만 'selected_tables' 배열에 넣으세요.
-3. 만약 질문이 데이터베이스 조회와 관련이 없거나, 제공된 테이블 정보로 답할 수 없다면 'is_executable'을 false로 설정하세요.
-4. 'reasoning' 필드에는 왜 해당 테이블들을 선택했는지 간략하게 설명하세요.
-
-[응답 포맷]
-{{
-"reasoning": "선정 이유를 여기에 작성",
-"selected_tables": ["table1", "table2"],
-"is_executable": true
-}}