Skip to content

Commit 96df974

Browse files
fix: replace Role.USER with 'user' string in test_input_observer_middleware
Role is a NewType(str) in agent-framework 1.1.1, not an enum. Role.USER raises AttributeError at runtime. Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
1 parent 13fe76a commit 96df974

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

src/tests/ContentProcessorWorkflow/libs/agent_framework/test_input_observer_middleware.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
import asyncio
88
from types import SimpleNamespace
99

10-
from agent_framework import Message, Role
10+
from agent_framework import Message
1111

1212
from libs.agent_framework.middlewares import InputObserverMiddleware
1313

@@ -16,7 +16,7 @@ def test_input_observer_middleware_replaces_user_text_when_configured() -> None:
1616
async def _run() -> None:
1717
ctx = SimpleNamespace(
1818
messages=[
19-
Message(role=Role.USER, contents=["original"]),
19+
Message(role="user", contents=["original"]),
2020
]
2121
)
2222

@@ -27,7 +27,7 @@ async def _next(_context):
2727

2828
await mw.process(ctx, _next)
2929

30-
assert ctx.messages[0].role == Role.USER
30+
assert ctx.messages[0].role == "user"
3131
assert ctx.messages[0].text == "replacement"
3232

3333
asyncio.run(_run())

0 commit comments

Comments
 (0)