Skip to content

Commit f772177

Browse files
committed
Update Library code to use snake_case
1 parent bff0cdf commit f772177

File tree

6 files changed

+22
-22
lines changed

6 files changed

+22
-22
lines changed

src/a2a/client/helpers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,5 +18,5 @@ def create_text_message_object(
1818
A `Message` object with a new UUID messageId.
1919
"""
2020
return Message(
21-
role=role, parts=[Part(TextPart(text=content))], messageId=str(uuid4())
21+
role=role, parts=[Part(TextPart(text=content))], message_id=str(uuid4())
2222
)

src/a2a/server/agent_execution/context.py

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -53,14 +53,14 @@ def __init__( # noqa: PLR0913
5353
# match the request. Otherwise, create them
5454
if self._params:
5555
if task_id:
56-
self._params.message.taskId = task_id
56+
self._params.message.task_id = task_id
5757
if task and task.id != task_id:
5858
raise ServerError(InvalidParamsError(message='bad task id'))
5959
else:
6060
self._check_or_generate_task_id()
6161
if context_id:
62-
self._params.message.contextId = context_id
63-
if task and task.contextId != context_id:
62+
self._params.message.context_id = context_id
63+
if task and task.context_id != context_id:
6464
raise ServerError(
6565
InvalidParamsError(message='bad context id')
6666
)
@@ -148,17 +148,17 @@ def _check_or_generate_task_id(self) -> None:
148148
if not self._params:
149149
return
150150

151-
if not self._task_id and not self._params.message.taskId:
152-
self._params.message.taskId = str(uuid.uuid4())
153-
if self._params.message.taskId:
154-
self._task_id = self._params.message.taskId
151+
if not self._task_id and not self._params.message.task_id:
152+
self._params.message.task_id = str(uuid.uuid4())
153+
if self._params.message.task_id:
154+
self._task_id = self._params.message.task_id
155155

156156
def _check_or_generate_context_id(self) -> None:
157157
"""Ensures a context ID is present, generating one if necessary."""
158158
if not self._params:
159159
return
160160

161-
if not self._context_id and not self._params.message.contextId:
162-
self._params.message.contextId = str(uuid.uuid4())
163-
if self._params.message.contextId:
164-
self._context_id = self._params.message.contextId
161+
if not self._context_id and not self._params.message.context_id:
162+
self._params.message.context_id = str(uuid.uuid4())
163+
if self._params.message.context_id:
164+
self._context_id = self._params.message.context_id

src/a2a/server/agent_execution/simple_request_context_builder.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ def __init__(
1818
1919
Args:
2020
should_populate_referred_tasks: If True, the builder will fetch tasks
21-
referenced in `params.message.referenceTaskIds` and populate the
21+
referenced in `params.message.reference_task_ids` and populate the
2222
`related_tasks` field in the RequestContext. Defaults to False.
2323
task_store: The TaskStore instance to use for fetching referred tasks.
2424
Required if `should_populate_referred_tasks` is True.
@@ -38,7 +38,7 @@ async def build(
3838
3939
This method assembles the RequestContext object. If the builder was
4040
initialized with `should_populate_referred_tasks=True`, it fetches all tasks
41-
referenced in `params.message.referenceTaskIds` from the `task_store`.
41+
referenced in `params.message.reference_task_ids` from the `task_store`.
4242
4343
Args:
4444
params: The parameters of the incoming message send request.
@@ -57,12 +57,12 @@ async def build(
5757
self._task_store
5858
and self._should_populate_referred_tasks
5959
and params
60-
and params.message.referenceTaskIds
60+
and params.message.reference_task_ids
6161
):
6262
tasks = await asyncio.gather(
6363
*[
6464
self._task_store.get(task_id)
65-
for task_id in params.message.referenceTaskIds
65+
for task_id in params.message.reference_task_ids
6666
]
6767
)
6868
related_tasks = [x for x in tasks if x is not None]

src/a2a/server/apps/jsonrpc/fastapi_app.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,7 @@ def add_routes_to_app(
8989
)(self._handle_requests)
9090
app.get(agent_card_url)(self._handle_get_agent_card)
9191

92-
if self.agent_card.supportsAuthenticatedExtendedCard:
92+
if self.agent_card.supports_authenticated_extended_card:
9393
app.get(extended_agent_card_url)(
9494
self._handle_get_authenticated_extended_agent_card
9595
)

src/a2a/server/apps/jsonrpc/jsonrpc_app.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -135,11 +135,11 @@ def __init__(
135135
agent_card=agent_card, request_handler=http_handler
136136
)
137137
if (
138-
self.agent_card.supportsAuthenticatedExtendedCard
138+
self.agent_card.supports_authenticated_extended_card
139139
and self.extended_agent_card is None
140140
):
141141
logger.error(
142-
'AgentCard.supportsAuthenticatedExtendedCard is True, but no extended_agent_card was provided. The /agent/authenticatedExtendedCard endpoint will return 404.'
142+
'AgentCard.supports_authenticated_extended_card is True, but no extended_agent_card was provided. The /agent/authenticatedExtendedCard endpoint will return 404.'
143143
)
144144
self._context_builder = context_builder or DefaultCallContextBuilder()
145145

@@ -421,7 +421,7 @@ async def _handle_get_authenticated_extended_agent_card(
421421
self, request: Request
422422
) -> JSONResponse:
423423
"""Handles GET requests for the authenticated extended agent card."""
424-
if not self.agent_card.supportsAuthenticatedExtendedCard:
424+
if not self.agent_card.supports_authenticated_extended_card:
425425
return JSONResponse(
426426
{'error': 'Extended agent card not supported or not enabled.'},
427427
status_code=404,
@@ -435,7 +435,7 @@ async def _handle_get_authenticated_extended_agent_card(
435435
by_alias=True,
436436
)
437437
)
438-
# If supportsAuthenticatedExtendedCard is true, but no specific
438+
# If supports_authenticated_extended_card is true, but no specific
439439
# extended_agent_card was provided during server initialization,
440440
# return a 404
441441
return JSONResponse(

src/a2a/server/apps/jsonrpc/starlette_app.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,7 @@ def routes(
8686
),
8787
]
8888

89-
if self.agent_card.supportsAuthenticatedExtendedCard:
89+
if self.agent_card.supports_authenticated_extended_card:
9090
app_routes.append(
9191
Route(
9292
extended_agent_card_url,

0 commit comments

Comments
 (0)