Skip to content

Commit a2a10fe

Browse files
committed
기존 모듈 경로 수정 #134
1 parent 7530ac0 commit a2a10fe

File tree

5 files changed

+6
-6
lines changed

5 files changed

+6
-6
lines changed
Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
"""
2-
token_utils.py
2+
token_usage.py
33
44
LLM 응답 메시지에서 토큰 사용량을 집계하기 위한 유틸리티 모듈입니다.
55
@@ -29,7 +29,7 @@ class TokenUtils:
2929
활용할 수 있습니다.
3030
3131
사용 예:
32-
>>> from token_utils import TokenUtils
32+
>>> from infra.observability.token_usage import TokenUtils
3333
>>> summary = TokenUtils.get_token_usage_summary(messages)
3434
>>> print(summary["total_tokens"])
3535

interface/lang2sql.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
from viz.display_chart import DisplayChart
1818
from engine.query_executor import execute_query as execute_query_common
1919
from llm_utils.llm_response_parser import LLMResponseParser
20-
from llm_utils.token_utils import TokenUtils
20+
from infra.observability.token_usage import TokenUtils
2121
from llm_utils.graph_utils.enriched_graph import builder as enriched_builder
2222
from llm_utils.graph_utils.basic_graph import builder
2323
from llm_utils.graph_utils.simplified_graph import builder as simplified_builder

llm_utils/README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,10 +8,10 @@ Lang2SQL 파이프라인에서 LLM, 검색(RAG), 그래프 워크플로우, DB
88
- (Moved) `viz/display_chart.py`: LLM 활용 Plotly 시각화 유틸.
99
- (Moved) `infra/monitoring/check_server.py`: GMS 헬스체크.
1010
- (Moved) `infra/db/connect_db.py`: ClickHouse 연결/실행.
11-
- **`token_utils.py`**: LLM 메시지의 `usage_metadata` 합산 토큰 집계.
11+
- (Moved) `infra/observability/token_usage.py`: LLM 메시지의 `usage_metadata` 합산 토큰 집계.
1212
- **`llm_response_parser.py`**: LLM 응답에서 `<SQL>`, `<해석>` 블록 추출.
1313
- **`prompts_class.py`**: LangChain SQL 프롬프트를 로컬 YAML로 오버라이드.
14-
- **`utils.py`**: `profile_to_text(profile)`소형 유틸.
14+
- (Moved) `graph_utils/profile_utils.py`: `profile_to_text(profile)`그래프 관련 포맷 유틸.
1515

1616
### Depth 1: LLM/임베딩/검색
1717

llm_utils/graph_utils/base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818

1919
from llm_utils.tools import get_info_from_db
2020
from llm_utils.retrieval import search_tables
21-
from llm_utils.utils import profile_to_text
21+
from llm_utils.graph_utils.profile_utils import profile_to_text
2222

2323
# 노드 식별자 정의
2424
QUERY_REFINER = "query_refiner"

0 commit comments

Comments
 (0)