Agent/examples/application_reference/providers/context.py

from cyber_agent.application import ContextMaterial, ContextResourceDescriptor
from cyber_agent.application.models import stable_hash


class ReferenceContextProvider:
    def __init__(self) -> None:
        self.list_calls = 0
        self.resolve_calls = 0
        self.content = {
            "source": "reference-fixture",
            "topic": "agent framework responsibilities",
        }

    async def list_context(self, request):
        self.list_calls += 1
        return [ContextResourceDescriptor(
            application_ref=request.application_ref,
            root_trace_id=request.root_trace_id,
            trace_id=request.trace_id,
            uid=request.uid,
            role_id=request.role_id,
            source_id="reference-brief",
            source_version="1",
            content_hash=stable_hash(self.content),
            kind="creative_brief",
            summary="Reference creative brief",
            inheritance="inheritable",
            estimated_chars=len(str(self.content)),
        )]

    async def resolve_context(self, request, descriptor):
        self.resolve_calls += 1
        del request
        return ContextMaterial(descriptor=descriptor, content=self.content)