Skip to content

Commit 9d74d4c

Browse files
committed
feat: pgvector 스크립트 추가 및 README 업데이트
- pgvector를 실행하기 위한 Docker 스크립트 추가 - README에 pgvector 지원 및 사용 방법에 대한 설명 추가
1 parent 839d7b5 commit 9d74d4c

File tree

2 files changed

+27
-0
lines changed

2 files changed

+27
-0
lines changed

README.md

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@ Lang2SQL은 자연어 쿼리를 최적화된 SQL 문으로 변환하는 오픈
3434
- **🔍 스키마 인식**: DataHub 메타데이터를 활용한 정확한 컬럼 매핑
3535
- **🛠️ 웹 인터페이스**: 대화형 Streamlit 앱을 통한 사용
3636
- **📈 시각화**: 생성된 SQL 쿼리 결과를 다양한 차트와 그래프로 시각화하여 데이터 인사이트를 직관적으로 파악
37+
- **🗄️ 유연한 VectorDB**: FAISS(로컬)와 pgvector(PostgreSQL) 중 선택 가능한 벡터 데이터베이스 지원
3738

3839
### 🤔 해결하는 문제
3940

@@ -85,6 +86,27 @@ lang2sql run-streamlit
8586
lang2sql --datahub_server http://your-datahub-server:8080 run-streamlit -p 8888
8687
```
8788

89+
### VectorDB 선택
90+
91+
FAISS(로컬) 또는 pgvector(PostgreSQL) 중 선택:
92+
93+
```bash
94+
# FAISS 사용 (기본값)
95+
lang2sql --vectordb-type faiss run-streamlit
96+
97+
# pgvector 사용
98+
lang2sql --vectordb-type pgvector run-streamlit
99+
```
100+
101+
### 자연어 쿼리 실행
102+
103+
```bash
104+
# 기본 FAISS 사용
105+
lang2sql query "고객 데이터를 기반으로 유니크한 유저 수를 카운트하는 쿼리"
106+
107+
# pgvector 사용
108+
lang2sql query "고객 데이터를 기반으로 유니크한 유저 수를 카운트하는 쿼리" --vectordb-type pgvector --vectordb-location "postgresql://postgres:postgres@localhost:5432/postgres"
109+
88110
### 환경 설정
89111

90112
- 현재는 pip 패키지 설치로 프로젝트 시작이 어려운 상황입니다.

pgvector.sh

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
docker run -d \
2+
--name pgvector \
3+
-e POSTGRES_PASSWORD=postgres \
4+
-p 5431:5432 \
5+
pgvector/pgvector:pg17

0 commit comments

Comments
 (0)