Skip to content

Commit d60d5b1

Browse files
CopilotCopilot
authored andcommitted
fix: use contents= for Message construction in agent-framework 1.3.0
- Update InputObserverMiddleware to use Message(contents=) instead of Message(text=) since agent-framework 1.3.0 renamed the parameter - Update corresponding tests to verify contents field - Fix teardown_module signature to accept optional module parameter Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
1 parent d86c63e commit d60d5b1

4 files changed

Lines changed: 9 additions & 9 deletions

File tree

src/processor/src/libs/agent_framework/middlewares.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -159,7 +159,7 @@ async def process(
159159
f"[InputObserverMiddleware] Updated: '{original_text}' -> '{updated_text}'"
160160
)
161161

162-
modified_message = Message(role=message.role, text=updated_text)
162+
modified_message = Message(role=message.role, contents=updated_text)
163163
modified_messages.append(modified_message)
164164
modified_count += 1
165165
else:

src/processor/src/tests/unit/libs/agent_framework/test_groupchat_orchestrator_internals.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ def __init__(self, *, role, text=None, contents=None, author_name=None):
4040
)
4141

4242

43-
def teardown_module():
43+
def teardown_module(module=None):
4444
"""Restore the original Message class to avoid leaking into other tests."""
4545
if _original_message is not None:
4646
groupchat_module.Message = _original_message

src/processor/src/tests/unit/libs/agent_framework/test_input_observer_middleware.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ def __init__(self, *, role, text=None, contents=None, author_name=None):
2626
from libs.agent_framework.middlewares import InputObserverMiddleware # noqa: E402
2727

2828

29-
def teardown_module():
29+
def teardown_module(module=None):
3030
"""Restore the original Message class to avoid leaking into other tests."""
3131
if _original_message is not None:
3232
middlewares_module.Message = _original_message

src/processor/src/tests/unit/libs/agent_framework/test_middlewares_extras.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ def __init__(self, *, role, text=None, contents=None, author_name=None):
3333
)
3434

3535

36-
def teardown_module():
36+
def teardown_module(module=None):
3737
"""Restore the original Message class to avoid leaking into other tests."""
3838
if _original_message is not None:
3939
middlewares_module.Message = _original_message
@@ -110,22 +110,22 @@ class TestInputObserverMiddleware:
110110
def test_replaces_user_messages_when_replacement_set(self):
111111
from libs.agent_framework.middlewares import InputObserverMiddleware
112112

113-
msg_user = Message(role=ROLE_USER, text="orig user")
114-
msg_assistant = Message(role=ROLE_ASSISTANT, text="hi")
113+
msg_user = Message(role=ROLE_USER, text="orig user", contents="orig user")
114+
msg_assistant = Message(role=ROLE_ASSISTANT, text="hi", contents="hi")
115115
ctx = MagicMock()
116116
ctx.messages = [msg_user, msg_assistant]
117117
next_fn = AsyncMock()
118118
mw = InputObserverMiddleware(replacement="REDACTED")
119119
_run(mw.process(ctx, next_fn))
120120
# First message replaced, second untouched
121-
assert ctx.messages[0].text == "REDACTED"
122-
assert ctx.messages[1].text == "hi"
121+
assert ctx.messages[0].contents == "REDACTED"
122+
assert ctx.messages[1].contents == "hi"
123123
next_fn.assert_awaited_once()
124124

125125
def test_no_replacement_keeps_text(self):
126126
from libs.agent_framework.middlewares import InputObserverMiddleware
127127

128-
msg = Message(role=ROLE_USER, text="keep me")
128+
msg = Message(role=ROLE_USER, text="keep me", contents="keep me")
129129
ctx = MagicMock()
130130
ctx.messages = [msg]
131131
mw = InputObserverMiddleware(replacement=None)

0 commit comments

Comments
 (0)