Skip to content

Commit b9206c5

Browse files
committed
fix(a2a): message id
1 parent 4fb9f16 commit b9206c5

File tree

2 files changed

+11
-1
lines changed

2 files changed

+11
-1
lines changed

veadk/cli/services/vefaas/template/deploy.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818

1919
from veadk.cloud.cloud_agent_engine import CloudAgentEngine
2020
from fastmcp.client import Client
21+
from veadk.cloud.cloud_app import get_message_id
2122

2223
SESSION_ID = "cloud_app_test_session"
2324
USER_ID = "cloud_app_test_user"
@@ -48,7 +49,7 @@ async def main():
4849
query_example, SESSION_ID, USER_ID
4950
)
5051
print(f"VeFaaS application ID: {cloud_app.vefaas_application_id}")
51-
print(f"Message ID: {response_message.message_id}")
52+
print(f"Message ID: {get_message_id(response_message)}")
5253
print(
5354
f"Response from {cloud_app.vefaas_endpoint}: {response_message.parts[0].root.text}"
5455
)

veadk/cloud/cloud_app.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -189,3 +189,12 @@ async def message_send(
189189
# TODO(floritange): show error log on VeFaaS function
190190
print(e)
191191
return None
192+
193+
194+
def get_message_id(message: Message):
195+
"""Get the messageId of the a2a message"""
196+
if getattr(message, "messageId", None):
197+
# Compatible with the messageId of the old version
198+
return message.messageId
199+
else:
200+
return message.message_id

0 commit comments

Comments
 (0)