📚 에이전트 메모리 관리

📚 에이전트 메모리 관리 (Manage your agent's memory)

결국 에이전트는 단순한 구성 요소로 정의할 수 있어요: 도구와 프롬프트가 있고, 가장 중요하게는 계획·실행·오류의 기록을 그린 과거 단계의 메모리를 가집니다.

출처: 공식문서 - Manage your agent's memory

에이전트 메모리 재생 (Replay)

과거 에이전트 실행을 검사하는 몇 가지 기능을 제공해요. 계측 가이드에서 강조하듯 특정 단계를 확대/축소할 수 있는 멋진 UI에 실행을 표시하도록 계측할 수 있어요. agent.replay()를 쓸 수도 있습니다.

에이전트가 실행된 뒤:

from smolagents import InferenceClientModel, CodeAgent

agent = CodeAgent(tools=[], model=InferenceClientModel(), verbosity_level=0)

result = agent.run("What's the 20th Fibonacci number?")

마지막 실행을 재생하려면:

agent.replay()

에이전트 메모리를 동적으로 변경하기

agent.memory.get_full_steps()로 전체 단계를 딕셔너리로 가져올 수 있어요. 스텝 콜백을 사용해 에이전트 메모리를 동적으로 바꿀 수도 있습니다. 스텝 콜백은 인자에서 agent 자체에 접근할 수 있어, 위에서 강조한 대로 어떤 메모리 스텝이든 접근하고 필요하면 바꿀 수 있어요. 예를 들어 웹 브라우저 에이전트가 수행하는 각 단계의 스크린샷을 관찰하고 있다고 해보죠.

참고: 이 코드는 간결함을 위해 일부 import와 객체 정의가 제거된 불완전한 코드입니다. 전체 동작 코드는 원본 스크립트를 보세요.

CodeAgent(
    tools=[WebSearchTool(), go_back, close_popups, search_item_ctrl_f],
    model=model,
    additional_authorized_imports=["helium"],
    step_callbacks=[update_screenshot],
    max_steps=20,
    verbosity_level=2,
)

전체 동작 예시는 비전 웹 브라우저 코드를 참고하세요.

에이전트를 한 단계씩 실행하기

며칠이 걸리는 도구 호출이 있을 때 유용해요. 에이전트를 단계별로 실행하면 각 단계에서 메모리도 갱신할 수 있습니다.

from smolagents import InferenceClientModel, CodeAgent, ActionStep, TaskStep

agent = CodeAgent(tools=[], model=InferenceClientModel(), verbosity_level=1)
agent.python_executor.send_tools({**agent.tools})
print(agent.memory.system_prompt)

task = "What is the 20th Fibonacci number?"

더 알아보기 (Learn more)

• 좋은 에이전트 만들기
• OpenTelemetry로 실행 검사