Skip to content

feat: 토큰 비용 추정 (--cost) 및 출력 토큰 예산 제한 (--max-tokens) #261

@TransparentDeveloper

Description

@TransparentDeveloper

개요

AI API 사용 비용 예측과 토큰 예산 제한 기능을 추가합니다.

배경

  • PO: --cost 플래그로 모델별 비용 추정 (code2prompt의 --price 참조)
  • Planner: --max-tokens 플래그로 토큰 예산 제한
  • 교차 검증: 토큰 관련 2개 상호 보완적 기능

--cost 기능

Estimated cost (input tokens):
  Claude 3.5 Sonnet  $0.0037  (vs full codebase: ~$0.0312, saved 88%)
  GPT-4o             $0.0037
  Gemini 1.5 Pro     $0.0009

--max-tokens 기능

brfit . --max-tokens 8000  # 토큰 8000 이내로 제한
  • 파일 크기(Size) 역순 정렬하여 큰 파일부터 제외
  • 예산 초과 파일 수 stderr 경고

구현 방향

  • 모델별 단가를 내장 상수로 관리
  • Packager.Package() 내 토큰 예산 필터 삽입
  • 기존 tiktoken 기반 Tokenizer 인터페이스 활용

제안: PO, Planner (교차 검증)

Metadata

Metadata

Assignees

Labels

enhancementNew feature or requestversion:minorNew feature - minor version bump

Projects

No projects

Relationships

None yet

Development

No branches or pull requests

Issue actions