Qdrant 설정 추가

Author: ehddnr301

cli/utils/env_loader.py

@@ -71,14 +71,14 @@ def set_vectordb(
     """VectorDB 타입과 위치를 설정합니다.
 
     Args:
-        vectordb_type (str): VectorDB 타입 ("faiss" 또는 "pgvector").
+        vectordb_type (str): VectorDB 타입 ("faiss" 또는 "pgvector" 또는 "qdrant").
         vectordb_location (Optional[str]): 경로 또는 연결 URL.
 
     Raises:
         ValueError: 잘못된 타입이나 경로/URL일 경우.
     """
 
-    if vectordb_type not in ("faiss", "pgvector"):
+    if vectordb_type not in ("faiss", "pgvector", "qdrant"):
         raise ValueError(f"지원하지 않는 VectorDB 타입: {vectordb_type}")
 
     os.environ["VECTORDB_TYPE"] = vectordb_type

interface/app_pages/settings_sections/data_source_section.py

@@ -103,10 +103,36 @@ def render_data_source_section(config: Config | None = None) -> None:
                     new_url = st.text_input(
                         "URL", value=existing.url, key="dh_edit_url"
                     )
-                    new_faiss = st.text_input(
-                        "FAISS 저장 경로(선택)",
-                        value=existing.faiss_path or "",
-                        key="dh_edit_faiss",
+                    new_vdb_type = st.selectbox(
+                        "VectorDB 타입",
+                        options=["faiss", "pgvector", "qdrant"],
+                        index=(
+                            0
+                            if existing.vectordb_type == "faiss"
+                            else (1 if existing.vectordb_type == "pgvector" else 2)
+                        ),
+                        key="dh_edit_vdb_type",
+                    )
+                    new_vdb_loc_placeholder = (
+                        "FAISS 디렉토리 경로 (예: ./dev/table_info_db)"
+                        if new_vdb_type == "faiss"
+                        else (
+                            "pgvector 연결 문자열 (postgresql://...)"
+                            if new_vdb_type == "pgvector"
+                            else "Qdrant URL (예: http://localhost:6333)"
+                        )
+                    )
+                    new_vdb_location = st.text_input(
+                        "VectorDB 위치",
+                        value=existing.vectordb_location or existing.faiss_path or "",
+                        key="dh_edit_vdb_loc",
+                        placeholder=new_vdb_loc_placeholder,
+                    )
+                    new_vdb_api_key = st.text_input(
+                        "VectorDB API Key (선택)",
+                        value=existing.vectordb_api_key or "",
+                        type="password",
+                        key="dh_edit_vdb_key",
                     )
                     new_note = st.text_input(
                         "메모", value=existing.note or "", key="dh_edit_note"
@@ -128,7 +154,14 @@ def render_data_source_section(config: Config | None = None) -> None:
                                 update_datahub_source(
                                     name=edit_dh,
                                     url=new_url,
-                                    faiss_path=(new_faiss or None),
+                                    faiss_path=(
+                                        new_vdb_location
+                                        if new_vdb_type == "faiss"
+                                        else None
+                                    ),
+                                    vectordb_type=new_vdb_type,
+                                    vectordb_location=(new_vdb_location or None),
+                                    vectordb_api_key=(new_vdb_api_key or None),
                                     note=(new_note or None),
                                 )
                                 st.success("저장되었습니다.")
@@ -147,10 +180,29 @@ def render_data_source_section(config: Config | None = None) -> None:
             dh_url = st.text_input(
                 "URL", key="dh_url", placeholder="http://localhost:8080"
             )
-            dh_faiss = st.text_input(
-                "FAISS 저장 경로(선택)",
-                key="dh_faiss",
-                placeholder="예: ./dev/table_info_db",
+            dh_vdb_type = st.selectbox(
+                "VectorDB 타입",
+                options=["faiss", "pgvector", "qdrant"],
+                key="dh_new_vdb_type",
+            )
+            dh_vdb_loc_placeholder = (
+                "FAISS 디렉토리 경로 (예: ./dev/table_info_db)"
+                if dh_vdb_type == "faiss"
+                else (
+                    "pgvector 연결 문자열 (postgresql://...)"
+                    if dh_vdb_type == "pgvector"
+                    else "Qdrant URL (예: http://localhost:6333)"
+                )
+            )
+            dh_vdb_location = st.text_input(
+                "VectorDB 위치",
+                key="dh_new_vdb_loc",
+                placeholder=dh_vdb_loc_placeholder,
+            )
+            dh_vdb_api_key = st.text_input(
+                "VectorDB API Key (선택)",
+                type="password",
+                key="dh_new_vdb_key",
             )
             dh_note = st.text_input("메모", key="dh_note", placeholder="선택")
 
@@ -174,7 +226,12 @@ def render_data_source_section(config: Config | None = None) -> None:
                             add_datahub_source(
                                 name=dh_name,
                                 url=dh_url,
-                                faiss_path=(dh_faiss or None),
+                                faiss_path=(
+                                    dh_vdb_location if dh_vdb_type == "faiss" else None
+                                ),
+                                vectordb_type=dh_vdb_type,
+                                vectordb_location=(dh_vdb_location or None),
+                                vectordb_api_key=(dh_vdb_api_key or None),
                                 note=dh_note or None,
                             )
                             st.success("추가되었습니다.")
@@ -216,21 +273,35 @@ def render_data_source_section(config: Config | None = None) -> None:
                 if existing:
                     new_type = st.selectbox(
                         "타입",
-                        options=["faiss", "pgvector"],
-                        index=(0 if existing.type == "faiss" else 1),
+                        options=["faiss", "pgvector", "qdrant"],
+                        index=(
+                            0
+                            if existing.type == "faiss"
+                            else (1 if existing.type == "pgvector" else 2)
+                        ),
                         key="vdb_edit_type",
                     )
                     new_loc_placeholder = (
                         "FAISS 디렉토리 경로 (예: ./dev/table_info_db)"
                         if new_type == "faiss"
-                        else "pgvector 연결 문자열 (postgresql://user:pass@host:port/db)"
+                        else (
+                            "pgvector 연결 문자열 (postgresql://user:pass@host:port/db)"
+                            if new_type == "pgvector"
+                            else "Qdrant URL (예: http://localhost:6333)"
+                        )
                     )
                     new_location = st.text_input(
                         "위치",
                         value=existing.location,
                         key="vdb_edit_location",
                         placeholder=new_loc_placeholder,
                     )
+                    new_api_key = st.text_input(
+                        "API Key (선택)",
+                        value=existing.api_key or "",
+                        type="password",
+                        key="vdb_edit_key",
+                    )
                     new_prefix = st.text_input(
                         "컬렉션 접두사(선택)",
                         value=existing.collection_prefix or "",
@@ -258,6 +329,7 @@ def render_data_source_section(config: Config | None = None) -> None:
                                     name=edit_vdb,
                                     vtype=new_type,
                                     location=new_location,
+                                    api_key=(new_api_key or None),
                                     collection_prefix=(new_prefix or None),
                                     note=(new_note or None),
                                 )
@@ -275,16 +347,23 @@ def render_data_source_section(config: Config | None = None) -> None:
             st.write("VectorDB 추가")
             vdb_name = st.text_input("이름", key="vdb_name")
             vdb_type = st.selectbox(
-                "타입", options=["faiss", "pgvector"], key="vdb_type"
+                "타입", options=["faiss", "pgvector", "qdrant"], key="vdb_type"
             )
             vdb_loc_placeholder = (
                 "FAISS 디렉토리 경로 (예: ./dev/table_info_db)"
                 if vdb_type == "faiss"
-                else "pgvector 연결 문자열 (postgresql://user:pass@host:port/db)"
+                else (
+                    "pgvector 연결 문자열 (postgresql://user:pass@host:port/db)"
+                    if vdb_type == "pgvector"
+                    else "Qdrant URL (예: http://localhost:6333)"
+                )
             )
             vdb_location = st.text_input(
                 "위치", key="vdb_location", placeholder=vdb_loc_placeholder
             )
+            vdb_api_key = st.text_input(
+                "API Key (선택)", type="password", key="vdb_new_key"
+            )
             vdb_prefix = st.text_input(
                 "컬렉션 접두사(선택)", key="vdb_prefix", placeholder="예: app1_"
             )
@@ -312,6 +391,7 @@ def render_data_source_section(config: Config | None = None) -> None:
                                 name=vdb_name,
                                 vtype=vdb_type,
                                 location=vdb_location,
+                                api_key=(vdb_api_key or None),
                                 collection_prefix=(vdb_prefix or None),
                                 note=(vdb_note or None),
                             )

interface/app_pages/sidebar_components/data_source_selector.py

@@ -39,8 +39,18 @@ def render_sidebar_data_source_selector(config=None) -> None:
                 return
             try:
                 update_datahub_server(config, selected.url)
-                # DataHub 선택 시, FAISS 경로가 정의되어 있으면 기본 VectorDB 로케이션으로도 반영
-                if selected.faiss_path:
+                # DataHub 선택 시, VectorDB 설정이 정의되어 있으면 기본 VectorDB 로케이션으로도 반영
+                if selected.vectordb_location:
+                    try:
+                        update_vectordb_settings(
+                            config,
+                            vectordb_type=selected.vectordb_type or "faiss",
+                            vectordb_location=selected.vectordb_location,
+                        )
+                    except Exception as e:
+                        st.sidebar.warning(f"VectorDB 설정 적용 경고: {e}")
+                elif selected.faiss_path:
+                    # Backward compatibility
                     try:
                         update_vectordb_settings(
                             config,

interface/core/config/models.py

@@ -17,14 +17,18 @@ class DataHubSource:
     name: str
     url: str
     faiss_path: Optional[str] = None
+    vectordb_type: str = "faiss"
+    vectordb_location: Optional[str] = None
+    vectordb_api_key: Optional[str] = None
     note: Optional[str] = None
 
 
 @dataclass
 class VectorDBSource:
     name: str
-    type: str  # 'faiss' | 'pgvector'
+    type: str  # 'faiss' | 'pgvector' | 'qdrant'
     location: str
+    api_key: Optional[str] = None
     collection_prefix: Optional[str] = None
     note: Optional[str] = None
 

interface/core/config/persist.py

@@ -63,11 +63,22 @@ def _parse_datahub_list(items: List[Dict[str, Any]]) -> List[DataHubSource]:
         name = str(item.get("name", "")).strip()
         url = str(item.get("url", "")).strip()
         faiss_path = item.get("faiss_path")
+        vectordb_type = item.get("vectordb_type", "faiss")
+        vectordb_location = item.get("vectordb_location")
+        vectordb_api_key = item.get("vectordb_api_key")
         note = item.get("note")
         if not name or not url:
             continue
         parsed.append(
-            DataHubSource(name=name, url=url, faiss_path=faiss_path, note=note)
+            DataHubSource(
+                name=name,
+                url=url,
+                faiss_path=faiss_path,
+                vectordb_type=vectordb_type,
+                vectordb_location=vectordb_location,
+                vectordb_api_key=vectordb_api_key,
+                note=note,
+            )
         )
     return parsed
 
@@ -81,12 +92,14 @@ def _parse_vectordb_list(items: List[Dict[str, Any]]) -> List[VectorDBSource]:
         if not name or not vtype or not location:
             continue
         collection_prefix = item.get("collection_prefix")
+        api_key = item.get("api_key")
         note = item.get("note")
         parsed.append(
             VectorDBSource(
                 name=name,
                 type=vtype,
                 location=location,
+                api_key=api_key,
                 collection_prefix=collection_prefix,
                 note=note,
             )

interface/core/config/registry_data_sources.py

@@ -41,25 +41,53 @@ def _save_registry(registry: DataSourcesRegistry) -> None:
 
 
 def add_datahub_source(
-    *, name: str, url: str, faiss_path: Optional[str] = None, note: Optional[str] = None
+    *,
+    name: str,
+    url: str,
+    faiss_path: Optional[str] = None,
+    vectordb_type: str = "faiss",
+    vectordb_location: Optional[str] = None,
+    vectordb_api_key: Optional[str] = None,
+    note: Optional[str] = None,
 ) -> None:
     registry = get_data_sources_registry()
     if any(s.name == name for s in registry.datahub):
         raise ValueError(f"이미 존재하는 DataHub 이름입니다: {name}")
     registry.datahub.append(
-        DataHubSource(name=name, url=url, faiss_path=faiss_path, note=note)
+        DataHubSource(
+            name=name,
+            url=url,
+            faiss_path=faiss_path,
+            vectordb_type=vectordb_type,
+            vectordb_location=vectordb_location,
+            vectordb_api_key=vectordb_api_key,
+            note=note,
+        )
     )
     _save_registry(registry)
 
 
 def update_datahub_source(
-    *, name: str, url: str, faiss_path: Optional[str], note: Optional[str]
+    *,
+    name: str,
+    url: str,
+    faiss_path: Optional[str],
+    vectordb_type: str = "faiss",
+    vectordb_location: Optional[str] = None,
+    vectordb_api_key: Optional[str] = None,
+    note: Optional[str],
 ) -> None:
     registry = get_data_sources_registry()
     for idx, s in enumerate(registry.datahub):
         if s.name == name:
             registry.datahub[idx] = DataHubSource(
-                name=name, url=url, faiss_path=faiss_path, note=note
+                name=name,
+                url=url,
+                faiss_path=faiss_path,
+                vectordb_type=vectordb_type,
+                vectordb_location=vectordb_location,
+                vectordb_api_key=vectordb_api_key,
+                note=note,
             )
             _save_registry(registry)
             return
@@ -77,12 +105,15 @@ def add_vectordb_source(
     name: str,
     vtype: str,
     location: str,
+    api_key: Optional[str] = None,
     collection_prefix: Optional[str] = None,
     note: Optional[str] = None,
 ) -> None:
     vtype = (vtype or "").lower()
-    if vtype not in ("faiss", "pgvector"):
-        raise ValueError("VectorDB 타입은 'faiss' 또는 'pgvector'여야 합니다")
+    if vtype not in ("faiss", "pgvector", "qdrant"):
+        raise ValueError(
+            "VectorDB 타입은 'faiss', 'pgvector', 'qdrant' 중 하나여야 합니다"
+        )
     registry = get_data_sources_registry()
     if any(s.name == name for s in registry.vectordb):
         raise ValueError(f"이미 존재하는 VectorDB 이름입니다: {name}")
@@ -91,6 +122,7 @@ def add_vectordb_source(
             name=name,
             type=vtype,
             location=location,
+            api_key=api_key,
             collection_prefix=collection_prefix,
             note=note,
         )
@@ -103,19 +135,23 @@ def update_vectordb_source(
     name: str,
     vtype: str,
     location: str,
+    api_key: Optional[str] = None,
     collection_prefix: Optional[str],
     note: Optional[str],
 ) -> None:
     vtype = (vtype or "").lower()
-    if vtype not in ("faiss", "pgvector"):
-        raise ValueError("VectorDB 타입은 'faiss' 또는 'pgvector'여야 합니다")
+    if vtype not in ("faiss", "pgvector", "qdrant"):
+        raise ValueError(
+            "VectorDB 타입은 'faiss', 'pgvector', 'qdrant' 중 하나여야 합니다"
+        )
     registry = get_data_sources_registry()
     for idx, s in enumerate(registry.vectordb):
         if s.name == name:
             registry.vectordb[idx] = VectorDBSource(
                 name=name,
                 type=vtype,
                 location=location,
+                api_key=api_key,
                 collection_prefix=collection_prefix,
                 note=note,
             )