Skip to content

Commit 1d328e1

Browse files
authored
test: fix role enum value in TCK SUT agent (#725)
1 parent 72a1007 commit 1d328e1

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

tck/sut_agent.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@
2323
AgentSkill,
2424
Message,
2525
Part,
26+
Role,
2627
TaskState,
2728
TaskStatus,
2829
TaskStatusUpdateEvent,
@@ -87,7 +88,7 @@ async def execute(
8788
status=TaskStatus(
8889
state=TaskState.TASK_STATE_WORKING,
8990
message=Message(
90-
role='agent',
91+
role=Role.ROLE_AGENT,
9192
message_id=str(uuid.uuid4()),
9293
parts=[Part(text='Processing your question')],
9394
task_id=task_id,
@@ -108,7 +109,7 @@ async def execute(
108109
logger.info('[SUTAgentExecutor] Response: %s', agent_reply_text)
109110

110111
agent_message = Message(
111-
role='agent',
112+
role=Role.ROLE_AGENT,
112113
message_id=str(uuid.uuid4()),
113114
parts=[Part(text=agent_reply_text)],
114115
task_id=task_id,

0 commit comments

Comments
 (0)