byeongwook.choiandClaude Fable 5.1
11ae3629b2
Co-Authored-By: Claude Fable 5.1 <noreply@anthropic.com>
79 lines
3.3 KiB
Python
79 lines
3.3 KiB
Python
"""Stage 3 — LLM 출력 스키마 (docs/logic-chunk-design.md). pydantic 검증.
|
|
|
|
- ProgramSummary : 프로그램 1건 요약 (unit 추출의 문맥으로도 쓰임)
|
|
- UnitExtraction : unit 하나에서 LLM 이 골라낸 로직 조각 목록 + unit 한 줄 요약
|
|
- LogicChunk : 로직 조각 하나 — 인덱스의 1차 단위
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
CHUNK_KINDS = (
|
|
"sql_select", # DB 조회 (SELECT / OPEN CURSOR)
|
|
"db_write", # DB 갱신 (INSERT / UPDATE / MODIFY / DELETE / COMMIT)
|
|
"fm_call", # BAPI · 펑션모듈 · RFC · 메서드 호출로 업무 처리
|
|
"aggregation", # 내부테이블 집계 · 가공 · 병합 (LOOP / COLLECT / SORT …)
|
|
"validation", # 입력 검증 · 권한 체크 · 존재 확인
|
|
"calculation", # 금액 · 수량 · 환율 · 날짜 계산
|
|
"output", # ALV · 리스트 · 화면 · 파일 · 메일 출력
|
|
"interface", # 외부 시스템 송수신 (파일 · IDoc · HTTP · RFC destination)
|
|
"control_flow", # 처리 흐름 분기 · 하위 로직 호출 순서
|
|
"other",
|
|
)
|
|
|
|
|
|
class LogicChunk(BaseModel):
|
|
"""LLM 이 반환하는 조각. line_* 는 include 기준 줄 번호(코드에 붙여 준 번호 그대로)."""
|
|
line_start: int
|
|
line_end: int
|
|
first_line: str = "" # line_start 줄의 코드 원문 — 줄 번호 검증용 앵커
|
|
kind: str = "other"
|
|
purpose_ko: str
|
|
purpose_en: str = ""
|
|
keywords_ko: list[str] = Field(default_factory=list)
|
|
keywords_en: list[str] = Field(default_factory=list)
|
|
sap_objects: list[str] = Field(default_factory=list) # 테이블 · FM · BAPI · 클래스 · T-Code
|
|
confidence: float = 0.5
|
|
|
|
|
|
class UnitExtraction(BaseModel):
|
|
unit_purpose_ko: str = "" # unit 한 줄 요약 — 얇은 색인용 (조각이 없어도 채운다)
|
|
chunks: list[LogicChunk] = Field(default_factory=list)
|
|
unclear: list[str] = Field(default_factory=list)
|
|
|
|
|
|
class SelectionParam(BaseModel):
|
|
name: str
|
|
desc_ko: str = ""
|
|
|
|
|
|
class KeyInternalTable(BaseModel):
|
|
name: str
|
|
filled_by: list[str] = Field(default_factory=list)
|
|
consumed_by: list[str] = Field(default_factory=list)
|
|
desc_ko: str = ""
|
|
|
|
|
|
class ProgramSummary(BaseModel):
|
|
program: str
|
|
title_ko: str = ""
|
|
business_purpose_ko: str = ""
|
|
business_purpose_en: str = ""
|
|
main_flow: list[str] = Field(default_factory=list)
|
|
selection_screen: list[SelectionParam] = Field(default_factory=list)
|
|
key_internal_tables: list[KeyInternalTable] = Field(default_factory=list)
|
|
tables_read: list[str] = Field(default_factory=list) # 파서 값으로 덮어씀
|
|
tables_write: list[str] = Field(default_factory=list) # 파서 값으로 덮어씀
|
|
external_calls: list[str] = Field(default_factory=list) # 파서 값으로 덮어씀
|
|
output_type: list[str] = Field(default_factory=list)
|
|
business_tags: list[str] = Field(default_factory=list)
|
|
proposed_tags: list[str] = Field(default_factory=list)
|
|
sap_module: str = ""
|
|
keywords_ko: list[str] = Field(default_factory=list)
|
|
keywords_en: list[str] = Field(default_factory=list)
|
|
related_tcodes: list[str] = Field(default_factory=list)
|
|
notes: list[str] = Field(default_factory=list)
|
|
confidence: float = 0.5
|
|
unclear: list[str] = Field(default_factory=list)
|
|
prompt_version: int = 0
|