Skip to content

Commit cdd29b9

Browse files
committed
formatting and linting
1 parent 506d782 commit cdd29b9

2 files changed

Lines changed: 6 additions & 6 deletions

File tree

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

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -501,8 +501,8 @@ def _modify_rpc_url(self, agent_card: AgentCard, request: Request) -> None:
501501
if 'X-Forwarded-Host' in request.headers:
502502
host = request.headers['X-Forwarded-Host']
503503
else:
504-
host = request.url.hostname
505-
port = request.url.port
504+
host = request.url.hostname or rpc_url.hostname or 'localhost'
505+
port = request.url.port or rpc_url.port
506506

507507
if 'X-Forwarded-Proto' in request.headers:
508508
scheme = request.headers['X-Forwarded-Proto']
@@ -514,7 +514,7 @@ def _modify_rpc_url(self, agent_card: AgentCard, request: Request) -> None:
514514
if ':' in host: # type: ignore
515515
comps = host.rsplit(':', 1) # type: ignore
516516
host = comps[0]
517-
port = comps[1]
517+
port = int(comps[1]) if comps[1] else port
518518

519519
# Handle URL maps,
520520
# e.g. "agents/my-agent/.well-known/agent-card.json"

src/a2a/server/apps/rest/rest_adapter.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -205,8 +205,8 @@ def _modify_rpc_url(self, agent_card: AgentCard, request: Request) -> None:
205205
if 'X-Forwarded-Host' in request.headers:
206206
host = request.headers['X-Forwarded-Host']
207207
else:
208-
host = request.url.hostname
209-
port = request.url.port
208+
host = request.url.hostname or rpc_url.hostname or 'localhost'
209+
port = request.url.port or rpc_url.port
210210

211211
if 'X-Forwarded-Proto' in request.headers:
212212
scheme = request.headers['X-Forwarded-Proto']
@@ -218,7 +218,7 @@ def _modify_rpc_url(self, agent_card: AgentCard, request: Request) -> None:
218218
if ':' in host: # type: ignore
219219
comps = host.rsplit(':', 1) # type: ignore
220220
host = comps[0]
221-
port = comps[1]
221+
port = int(comps[1]) if comps[1] else port
222222

223223
# Handle URL maps,
224224
# e.g. "agents/my-agent/.well-known/agent-card.json"

0 commit comments

Comments
 (0)