@nine-lab/nine-connector 0.1.2 → 0.1.4

package/package.json

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

package/src/database/core/DatabaseManager.js

@@ -9,7 +9,7 @@ class DatabaseManager {
         this.dialect = Dialects[this.type];
 
         if (!this.dialect) {
-            throw new Error(`❌ Unsupported Dialect for: ${this.type}`);
+            throw new Error(`Unsupported Dialect for: ${this.type}`);
         }
     }
 
@@ -23,9 +23,9 @@ class DatabaseManager {
 
             // 단순 연결 확인 쿼리
             await this.query('SELECT 1');
-            console.log(`✅ [${this.type}] Database connection successful.`);
+            console.log(`[${this.type}] Database connection successful.`);
         } catch (error) {
-            console.error(`❌ [${this.type}] Connection failed:`, error.message);
+            console.error(`[${this.type}] Connection failed:`, error.message);
             throw error;
         }
     }

package/src/index.js

@@ -1,19 +1,24 @@
 #!/usr/bin/env node
 import 'dotenv/config';
-import express from 'express';
-import cors from 'cors';
-import fs from 'fs';
-import path from 'path'; // path 모듈 추가
-import DatabaseManager from './database/core/DatabaseManager.js';
 import { runInit } from './core/init.js';
-import { AIProcessor } from './ai/AIProcessor.js';
-import { AIService } from './ai/AIService.js';
 
 /**
  * 서버 실행 로직
  */
 async function bootstrap() {
-    // 1. 필수 환경 변수 체크
+    // 1. 서버 실행 시점에만 필요한 모듈들을 다이나믹 임포트
+    // 상단에 static import가 있으면 이 로직이 무의미해지므로,
+    // 관련 모듈들은 여기서만 로드합니다.
+    const express = (await import('express')).default;
+    const cors = (await import('cors')).default;
+    const fs = (await import('fs')).default;
+    const path = (await import('path')).default;
+
+    const { default: DatabaseManager } = await import('./database/core/DatabaseManager.js');
+    const { AIProcessor } = await import('./ai/AIProcessor.js');
+    const { AIService } = await import('./ai/AIService.js');
+
+    // 2. 필수 환경 변수 체크
     if (!process.env.DB_HOST || !process.env.DB_NAME) {
         console.error("설정 정보가 부족합니다. 'nine-connector init'을 먼저 실행해주세요.");
         process.exit(1);
@@ -22,7 +27,7 @@ async function bootstrap() {
     const app = express();
     const SERVER_PORT = process.env.SERVER_PORT || 3000;
 
-    // 2. DB 매니저 초기화
+    // 3. DB 매니저 초기화
     const db = new DatabaseManager({
         type: process.env.DB_TYPE,
         host: process.env.DB_HOST,
@@ -38,11 +43,10 @@ async function bootstrap() {
     app.use(cors());
     app.use(express.json());
 
-    // 3. AI 프로세서 초기화 및 체인 등록
+    // 4. AI 프로세서 초기화 및 체인 등록
     const ai = new AIProcessor(process.env.GEMINI_API_KEY, process.env.GEMINI_MODEL);
 
     try {
-        // init.js에서 생성한 경로와 일치시킵니다. (보통 루트의 prompts 폴더)
         const promptPath = path.join(process.cwd(), 'prompts');
 
         const tableFilterPrompt = fs.readFileSync(
@@ -61,25 +65,19 @@ async function bootstrap() {
         process.exit(1);
     }
 
-    // 서비스 인스턴스 생성
     const aiService = new AIService(db, ai);
 
     // [API] 자연어 질의 엔드포인트
     app.post('/api/ask', async (req, res) => {
         const { question } = req.body;
-        if (!question) {
-            return res.status(400).json({ success: false, error: "질문을 입력해주세요." });
-        }
+        if (!question) return res.status(400).json({ success: false, error: "질문을 입력해주세요." });
 
         try {
             const result = await aiService.processNaturalLanguageQuery(question);
             res.json({ success: true, ...result });
         } catch (err) {
             console.error("AI 질의 처리 중 오류:", err);
-            res.status(500).json({
-                success: false,
-                error: err.message || "쿼리 처리 중 서버 오류가 발생했습니다."
-            });
+            res.status(500).json({ success: false, error: err.message || "서버 오류" });
         }
     });
 
@@ -87,29 +85,18 @@ async function bootstrap() {
         await db.connect();
         console.log("데이터베이스 연결 성공");
 
-        // 스키마 조회 API
         app.get('/api/schema', async (req, res) => {
-            try {
-                const schema = await db.getTableSchema();
-                res.json({ success: true, data: schema });
-            } catch (err) {
-                res.status(500).json({ success: false, error: "Failed to fetch schema" });
-            }
+            const schema = await db.getTableSchema();
+            res.json({ success: true, data: schema });
         });
 
-        // 쿼리 실행 API
         app.post('/api/query', async (req, res) => {
             const { sql } = req.body;
-            if (!sql || !sql.trim().toLowerCase().startsWith("select")) {
+            if (!sql?.trim().toLowerCase().startsWith("select")) {
                 return res.status(400).json({ success: false, error: "Only SELECT queries are allowed" });
             }
-
-            try {
-                const rows = await db.query(sql);
-                res.json({ success: true, data: rows });
-            } catch (err) {
-                res.status(500).json({ success: false, error: "Database query error" });
-            }
+            const rows = await db.query(sql);
+            res.json({ success: true, data: rows });
         });
 
         app.listen(SERVER_PORT, () => {