Agentic Context Engineering: 자가 개선 언어 모델을 위한 진화하는 Context 설계

# ACE Curator가 playbook에 새 bullet을 추가하는 핵심 로직 예시
import json
from sentence_transformers import SentenceTransformer, util

class ACEPlaybook:
    def __init__(self):
        self.bullets = []  # {id, section, content, helpful, harmful}
        self.model = SentenceTransformer('all-MiniLM-L6-v2')
        self._counter = 0

    def add_bullet(self, section: str, content: str):
        self._counter += 1
        bullet_id = f"ctx-{self._counter:05d}"
        self.bullets.append({
            "id": bullet_id,
            "section": section,
            "content": content,
            "helpful": 0,
            "harmful": 0
        })
        return bullet_id

    def update_feedback(self, bullet_ids: list[str], tag: str):
        """Reflector가 분류한 helpful/harmful 피드백을 반영"""
        for b in self.bullets:
            if b["id"] in bullet_ids:
                if tag == "helpful":
                    b["helpful"] += 1
                elif tag == "harmful":
                    b["harmful"] += 1

    def deduplicate(self, threshold=0.92):
        """Grow-and-Refine: 유사 bullet 병합으로 context collapse 방지"""
        contents = [b["content"] for b in self.bullets]
        embeddings = self.model.encode(contents, convert_to_tensor=True)
        to_remove = set()
        for i in range(len(self.bullets)):
            for j in range(i + 1, len(self.bullets)):
                if j in to_remove:
                    continue
                sim = util.cos_sim(embeddings[i], embeddings[j]).item()
                if sim > threshold:
                    # helpful 카운터 낮은 쪽 제거
                    loser = j if self.bullets[i]["helpful"] >= self.bullets[j]["helpful"] else i
                    to_remove.add(loser)
        self.bullets = [b for i, b in enumerate(self.bullets) if i not in to_remove]

    def render(self) -> str:
        """Generator에게 주입할 playbook 텍스트 생성"""
        lines = []
        for b in self.bullets:
            lines.append(f"[{b['id']}] helpful={b['helpful']} harmful={b['harmful']} :: {b['content']}")
        return "\n".join(lines)

# --- 사용 예시 ---
playbook = ACEPlaybook()
playbook.add_bullet("strategies", "항상 Phone 앱 contacts에서 관계(roommate 등)를 조회할 것. 거래 내역 파싱으로 추측 금지.")
playbook.add_bullet("apis", "페이지네이션은 while True + break 패턴 사용. for i in range(10) 절대 금지.")

# Reflector 피드백 반영
playbook.update_feedback(["ctx-00001"], "helpful")
playbook.update_feedback(["ctx-00002"], "helpful")

# 새 insight 추가 후 dedup
playbook.add_bullet("strategies", "항상 contacts API로 관계를 확인하고 heuristic 사용 금지.")
playbook.deduplicate()  # 유사 bullet 제거

print(playbook.render())