Agent Config

"""MS Agent Framework agent backing the Agent Config Object demo.Reads three forwarded properties -- tone, expertise, responseLength -- from theAG-UI run input's ``forwardedProps`` and composes its system prompt dynamicallyper turn.The CopilotKit provider's ``properties`` prop is wired through the runtime as``forwardedProps`` on each AG-UI run. We subclass ``AgentFrameworkAgent`` andprepend a request-local system message built from those properties. Each turnreplaces the prior injected message instead of mutating the shared agent.Invalid or missing values fall back to the corresponding ``DEFAULT_*``constant -- this function never raises so the demo can't deadlock on a badpayload."""from __future__ import annotationsfrom collections.abc import AsyncGeneratorfrom textwrap import dedentfrom typing import Any, Literalfrom uuid import uuid4from ag_ui.core import BaseEventfrom agent_framework import Agent, BaseChatClientfrom agent_framework_ag_ui import AgentFrameworkAgentTone = Literal["professional", "casual", "enthusiastic"]Expertise = Literal["beginner", "intermediate", "expert"]ResponseLength = Literal["concise", "detailed"]DEFAULT_TONE: Tone = "professional"DEFAULT_EXPERTISE: Expertise = "intermediate"DEFAULT_RESPONSE_LENGTH: ResponseLength = "concise"VALID_TONES: set[str] = {"professional", "casual", "enthusiastic"}VALID_EXPERTISE: set[str] = {"beginner", "intermediate", "expert"}VALID_RESPONSE_LENGTHS: set[str] = {"concise", "detailed"}def read_properties(forwarded_props: Any) -> dict[str, str]:    """Read forwarded props with defensive defaults.    Any missing or unrecognized value falls back to the corresponding    ``DEFAULT_*`` constant. Never raises.    """    props = forwarded_props if isinstance(forwarded_props, dict) else {}    tone = props.get("tone", DEFAULT_TONE)    expertise = props.get("expertise", DEFAULT_EXPERTISE)    response_length = props.get("responseLength", DEFAULT_RESPONSE_LENGTH)    if tone not in VALID_TONES:        tone = DEFAULT_TONE    if expertise not in VALID_EXPERTISE:        expertise = DEFAULT_EXPERTISE    if response_length not in VALID_RESPONSE_LENGTHS:        response_length = DEFAULT_RESPONSE_LENGTH    return {        "tone": tone,        "expertise": expertise,        "response_length": response_length,    }def build_system_prompt(tone: str, expertise: str, response_length: str) -> str:    """Compose the system prompt from the three axes."""    tone_rules = {        "professional": ("Use neutral, precise language. No emoji. Short sentences."),        "casual": (            "Use friendly, conversational language. Contractions OK. "            "Light humor welcome."        ),        "enthusiastic": (            "Use upbeat, energetic language. Exclamation points OK. Emoji OK."        ),    }    expertise_rules = {        "beginner": "Assume no prior knowledge. Define jargon. Use analogies.",        "intermediate": (            "Assume common terms are understood; explain specialized terms."        ),        "expert": ("Assume technical fluency. Use precise terminology. Skip basics."),    }    length_rules = {        "concise": "Respond in 1-3 sentences.",        "detailed": ("Respond in multiple paragraphs with examples where relevant."),    }    return (        "You are a helpful assistant.\n\n"        f"Tone: {tone_rules[tone]}\n"        f"Expertise level: {expertise_rules[expertise]}\n"        f"Response length: {length_rules[response_length]}"    )class AgentConfigFrameworkAgent(AgentFrameworkAgent):    """AgentFrameworkAgent that rebuilds its system prompt per request.    Overrides ``run`` to read ``forwardedProps`` from the AG-UI input and add    the resulting instructions to a request-local system message.    """    async def run(  # type: ignore[override]        self,        input_data: dict[str, Any],    ) -> AsyncGenerator[BaseEvent, None]:        props = read_properties(input_data.get("forwardedProps"))        system_prompt = build_system_prompt(            props["tone"], props["expertise"], props["response_length"]        )        messages = input_data.get("messages")        if not isinstance(messages, list) or len(messages) == 0:            async for event in super().run(input_data):                yield event            return        run_id = input_data.get("runId") or str(uuid4())        request_input = dict(input_data)        request_input["runId"] = run_id        request_input["messages"] = [            {                "id": f"{run_id}-agent-config",                "role": "system",                "content": system_prompt,            },            *[                message                for message in messages                if not (                    isinstance(message, dict)                    and isinstance(message.get("id"), str)                    and message["id"].endswith("-agent-config")                )            ],        ]        async for event in super().run(request_input):            yield eventdef create_agent_config_agent(chat_client: BaseChatClient) -> AgentConfigFrameworkAgent:    """Instantiate the Agent Config demo agent.    The base MS Agent Framework ``Agent`` carries only a neutral fallback    instruction. The real behavioural steering happens in the request-local    system message applied by ``AgentConfigFrameworkAgent.run``.    """    base_agent = Agent(        client=chat_client,        name="agent_config",        instructions=dedent(            """            You are a helpful assistant. Follow the tone, expertise level, and            response-length directives provided in the system message for each            turn. If no directive is provided, use professional / intermediate            / concise defaults.            """.strip()        ),        tools=[],    )    return AgentConfigFrameworkAgent(        agent=base_agent,        name="AgentConfigObjectDemo",        description=(            "Reads tone / expertise / responseLength from forwardedProps "            "and builds its system prompt per turn."        ),        require_confirmation=False,    )