|
1 | 1 | # setup.py |
2 | 2 | from setuptools import setup, find_packages |
| 3 | +import os |
| 4 | +import glob |
3 | 5 |
|
4 | | -with open("docs/README.md", "r", encoding="utf-8") as fh: |
| 6 | +with open("README.md", "r", encoding="utf-8") as fh: |
5 | 7 | long_description = fh.read() |
6 | 8 |
|
| 9 | +# 프롬프트 파일들을 찾습니다 |
| 10 | +prompt_files = glob.glob('prompt/*.md') |
| 11 | +prompt_files = [os.path.basename(f) for f in prompt_files] |
| 12 | + |
7 | 13 | setup( |
8 | | - name="lang2sql", # 패키지 이름 |
| 14 | + name="lang2SQL", # 패키지 이름 |
9 | 15 | version="0.1.9", # 버전 |
10 | 16 | author="ehddnr301", |
11 | 17 | |
12 | 18 | url="https://github.com/CausalInferenceLab/Lang2SQL", |
13 | 19 | description="Lang2SQL - Query Generator for Data Warehouse", |
14 | 20 | long_description=long_description, |
15 | 21 | long_description_content_type="text/markdown", |
16 | | - packages=find_packages(), # my_package를 자동으로 찾음 |
| 22 | + packages=find_packages() + ['prompt'], # prompt 패키지 직접 추가 |
| 23 | + package_data={ |
| 24 | + 'prompt': ['*.md', '*.py'], # prompt 디렉토리의 모든 .md 파일 포함 |
| 25 | + }, |
| 26 | + data_files=[ |
| 27 | + ('prompt', [os.path.join('prompt', f) for f in prompt_files]) |
| 28 | + ], |
| 29 | + include_package_data=True, |
17 | 30 | install_requires=[ |
18 | 31 | "langgraph==0.2.62", |
19 | 32 | "datahub==0.999.1", |
|
29 | 42 | "langchain-ollama>=0.3.2,<0.4.0", |
30 | 43 | "langchain-huggingface>=0.1.2,<0.2.0", |
31 | 44 | "transformers==4.51.2", |
| 45 | + "clickhouse-driver==0.2.9", |
32 | 46 | ], |
33 | 47 | entry_points={ |
34 | 48 | "console_scripts": [ |
|
0 commit comments