Skip to content

Commit 47d7966

Browse files
committed
fix format
1 parent e83136b commit 47d7966

4 files changed

Lines changed: 23 additions & 12 deletions

File tree

src/sap_cloud_sdk/agentgateway/_customer.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -496,6 +496,7 @@ async def get_mcp_tools_customer(
496496
logger.info("Discovering tools from %d MCP server(s)", len(dependencies))
497497

498498
system_token = None
499+
499500
# Define a helper closure to refetch the system token on demand, since it may need to be
500501
# refreshed during the discovery loop if any server returns a 401
501502
async def refetch_system_token() -> str:

src/sap_cloud_sdk/agentgateway/_lob.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,10 @@
2424

2525
from sap_cloud_sdk.agentgateway._models import MCPTool
2626
from sap_cloud_sdk.agentgateway._token_cache import _TokenCache
27-
from sap_cloud_sdk.agentgateway.exceptions import AgentGatewaySDKError, MCPServerNotFoundError
27+
from sap_cloud_sdk.agentgateway.exceptions import (
28+
AgentGatewaySDKError,
29+
MCPServerNotFoundError,
30+
)
2831

2932
logger = logging.getLogger(__name__)
3033

src/sap_cloud_sdk/agentgateway/_token_cache.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -117,15 +117,12 @@ def get_system_token(self, client_id: str) -> str | None:
117117
del self._system_tokens[key]
118118
return None
119119

120-
def set_system_token(self, token: str, expires_at: float,
121-
client_id: str) -> None:
120+
def set_system_token(self, token: str, expires_at: float, client_id: str) -> None:
122121
"""Cache a system token under `client_id`; evict LRU once size exceeds limit."""
123122
key = client_id
124-
self._system_tokens[key] = _CachedToken(token=token,
125-
expires_at=expires_at)
123+
self._system_tokens[key] = _CachedToken(token=token, expires_at=expires_at)
126124
self._system_tokens.move_to_end(key)
127-
while len(self._system_tokens
128-
) > self._config.max_system_token_cache_size:
125+
while len(self._system_tokens) > self._config.max_system_token_cache_size:
129126
evicted, _ = self._system_tokens.popitem(last=False)
130127
logger.debug("System token cache full — evicted '%s'", evicted)
131128

@@ -157,8 +154,7 @@ def set_user_token(
157154
) -> None:
158155
"""Cache an exchanged user token; evict LRU once size exceeds limit."""
159156
key = self._hash_key(user_jwt, client_id)
160-
self._user_tokens[key] = _CachedToken(token=token,
161-
expires_at=expires_at)
157+
self._user_tokens[key] = _CachedToken(token=token, expires_at=expires_at)
162158
self._user_tokens.move_to_end(key)
163159
while len(self._user_tokens) > self._config.max_user_token_cache_size:
164160
evicted, _ = self._user_tokens.popitem(last=False)
@@ -185,7 +181,11 @@ def compute_expires_at_from_bearer(self, auth_header: str) -> float:
185181
now_mono = time.monotonic()
186182
buffer = self._config.token_expiry_buffer_seconds
187183

188-
jwt = auth_header[7:] if auth_header.lower().startswith("bearer ") else auth_header
184+
jwt = (
185+
auth_header[7:]
186+
if auth_header.lower().startswith("bearer ")
187+
else auth_header
188+
)
189189
exp = _parse_jwt_exp(jwt)
190190
if exp is not None:
191191
remaining = exp - time.time()

src/sap_cloud_sdk/agentgateway/agw_client.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -172,7 +172,9 @@ async def list_mcp_tools(
172172
logger.warning("app_tid parameter ignored for LoB agent flow")
173173

174174
tenant = self._resolve_tenant_subdomain()
175-
return await get_mcp_tools_lob(tenant, self._config.timeout, self._token_cache)
175+
return await get_mcp_tools_lob(
176+
tenant, self._config.timeout, self._token_cache
177+
)
176178

177179
except AgentGatewaySDKError:
178180
# Re-raise SDK errors as-is
@@ -274,7 +276,12 @@ async def call_mcp_tool(
274276

275277
tenant = self._resolve_tenant_subdomain()
276278
return await call_mcp_tool_lob(
277-
tool, resolved_user_token, tenant, self._config.timeout, self._token_cache, **kwargs
279+
tool,
280+
resolved_user_token,
281+
tenant,
282+
self._config.timeout,
283+
self._token_cache,
284+
**kwargs,
278285
)
279286

280287
except AgentGatewaySDKError:

0 commit comments

Comments
 (0)