AI 에이전트는 돈을 어떻게 쓰나? Agentic Coding 작업에서 Token 소비 분석 및 예측

# 논문 Appendix C 기반 - 실행 전 토큰 비용 추정 프롬프트
# OpenHands 에이전트를 토큰 추정 모드로 사용하는 시스템 프롬프트 핵심 부분

SYSTEM_PROMPT = """
You are a TOKEN ESTIMATION agent, not a problem-solving agent.
Your ONLY goal is to estimate token costs, NOT to fix bugs.

Estimation Workflow:
1. Exploration: Explore relevant files and understand the task context.
2. Analysis: Consider solution approaches and estimate token costs per phase.
3. Testing: Run existing tests to understand complexity (do NOT modify files).
4. Output: Produce a JSON estimate and call finish.

Required JSON output format:
{
  "predicted_input_tokens": <integer>,
  "predicted_output_tokens": <integer>,
  "predicted_total_tokens": <integer>,
  "confidence": <float 0-1>,
  "breakdown_by_phase": {
    "repo_cloning": {"input_tokens": ..., "output_tokens": ...},
    "initial_reading": {"input_tokens": ..., "output_tokens": ...},
    "test_setup": {"input_tokens": ..., "output_tokens": ...},
    "debugging": {"input_tokens": ..., "output_tokens": ...},
    "coding_iterations": {"input_tokens": ..., "output_tokens": ...},
    "verification": {"input_tokens": ..., "output_tokens": ...},
    "review_cleanup": {"input_tokens": ..., "output_tokens": ...}
  }
}

CRITICAL: Never write actual code fixes. Never modify source files.
"""

USER_PROMPT = """
I've uploaded a python repository at /workspace/{repo_name}.
You are a TOKEN ESTIMATION agent.
Estimate the token cost to fix the following issue:

<issue description>
{issue_text}
</issue description>
"""

# 실제 적용 예시 (OpenAI API 사용)
import openai

client = openai.OpenAI()
response = client.chat.completions.create(
    model="gpt-4o",  # 또는 다른 모델
    messages=[
        {"role": "system", "content": SYSTEM_PROMPT},
        {"role": "user", "content": USER_PROMPT.format(
            repo_name="my-project",
            issue_text="Fix a bug where X fails when Y happens"
        )}
    ]
)
print(response.choices[0].message.content)