Skip to content

Commit 9314c09

Browse files
committed
chore(ve_identity): Remove redundant logging from identity and token modules
Eliminated unnecessary logger.info statements from IdentityClient and WorkloadTokenManager to reduce log verbosity and improve clarity. No changes to core logic or functionality.
1 parent 9bed908 commit 9314c09

File tree

6 files changed

+10
-23
lines changed

6 files changed

+10
-23
lines changed

docs/content/91.auth/1.agent-identity-intro.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ A: Agent Identity 会自动处理:
8282
## 相关资源
8383

8484
### Agent Identity 官方文档
85-
- [Agent Identity API 参考](https://www.volcengine.com/docs/6758/1261038)
85+
- [Agent Identity API 参考](https://www.volcengine.com/docs/86848/1918752)
8686

8787
### 外部资源
8888
- [OAuth2 规范](https://tools.ietf.org/html/rfc6749)
@@ -92,5 +92,5 @@ A: Agent Identity 会自动处理:
9292

9393
如有问题,请:
9494
1. 查看相应文档的常见问题部分
95-
2. 查看 [Agent Identity 官方文档](https://www.volcengine.com/docs/6758/1261038)
95+
2. 查看 [Agent Identity 官方文档](https://www.volcengine.com/docs/86848/1913964)
9696
3. 联系火山引擎技术支持

docs/content/91.auth/2.api-key-outbound.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -94,5 +94,5 @@ A: 实现错误处理和重试逻辑。
9494
- [Agent Identity 产品介绍](./1.agent-identity-intro.md)
9595
- [OAuth2 M2M 认证](./3.oauth2-m2m-outbound.md)
9696
- [OAuth2 USER_FEDERATION 认证](./4.oauth2-user-federation-outbound.md)
97-
- [Agent Identity API 参考](https://www.volcengine.com/docs/6758/1261038)
97+
- [Agent Identity API 参考](https://www.volcengine.com/docs/86848/1918752)
9898

docs/content/91.auth/3.oauth2-m2m-outbound.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -112,5 +112,5 @@ auth_config = oauth2_auth(
112112
- [Agent Identity 产品介绍](./1.agent-identity-intro.md)
113113
- [API Key 认证](./2.api-key-outbound.md)
114114
- [OAuth2 USER_FEDERATION 认证](./4.oauth2-user-federation-outbound.md)
115-
- [Agent Identity API 参考](https://www.volcengine.com/docs/6758/1261038)
115+
- [Agent Identity API 参考](https://www.volcengine.com/docs/86848/1918752)
116116

docs/content/91.auth/4.oauth2-user-federation-outbound.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -179,4 +179,4 @@ A: 如需自定义回调地址,需要将授权码(code)、状态(state
179179
- [Agent Identity 产品介绍](./1.agent-identity-intro.md)
180180
- [API Key 认证](./2.api-key-outbound.md)
181181
- [OAuth2 M2M 认证](./3.oauth2-m2m-outbound.md)
182-
- [Agent Identity API 参考](https://www.volcengine.com/docs/6758/1261038)
182+
- [Agent Identity API 参考](https://www.volcengine.com/docs/86848/1918752)

veadk/integrations/ve_identity/identity_client.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -234,7 +234,6 @@ def convert_response(
234234
if user_token:
235235
if user_id is not None:
236236
logger.warning("Both user_token and user_id provided, using user_token")
237-
logger.info("Retrieving workload access token with JWT authentication...")
238237
resp: volcenginesdkid.GetWorkloadAccessTokenForJWTResponse = (
239238
self._api_client.get_workload_access_token_for_jwt(
240239
volcenginesdkid.GetWorkloadAccessTokenForJWTRequest(
@@ -244,9 +243,6 @@ def convert_response(
244243
)
245244

246245
elif user_id:
247-
logger.info(
248-
"Retrieving workload access token with user ID authentication..."
249-
)
250246
resp: volcenginesdkid.GetWorkloadAccessTokenForUserIdResponse = (
251247
self._api_client.get_workload_access_token_for_user_id(
252248
volcenginesdkid.GetWorkloadAccessTokenForUserIdRequest(
@@ -255,14 +251,12 @@ def convert_response(
255251
)
256252
)
257253
else:
258-
logger.info("Retrieving workload access token...")
259254
resp: volcenginesdkid.GetWorkloadAccessTokenResponse = (
260255
self._api_client.get_workload_access_token(
261256
volcenginesdkid.GetWorkloadAccessTokenRequest(name=workload_name),
262257
)
263258
)
264259

265-
logger.info("Successfully retrieved workload access token")
266260
return convert_response(resp)
267261

268262
@refresh_credentials

veadk/integrations/ve_identity/token_manager.py

Lines changed: 5 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,9 @@ def __init__(
5858
"""
5959
self._identity_client = identity_client or IdentityClient(region=region)
6060

61-
def _build_cache_key(self, tool_context: Union[ToolContext|ReadonlyContext]) -> str:
61+
def _build_cache_key(
62+
self, tool_context: Union[ToolContext | ReadonlyContext]
63+
) -> str:
6264
"""Build a unique cache key for storing the workload token.
6365
6466
The cache key is composed of the agent name and user ID to ensure
@@ -87,7 +89,7 @@ def _is_token_expired(self, expires_at: int) -> bool:
8789

8890
async def get_workload_token(
8991
self,
90-
tool_context: Union[ToolContext|ReadonlyContext],
92+
tool_context: Union[ToolContext | ReadonlyContext],
9193
workload_name: Optional[str] = None,
9294
user_token: Optional[str] = None,
9395
) -> str:
@@ -126,11 +128,6 @@ async def get_workload_token(
126128
f"Cached workload token expired for agent '{tool_context.agent_name}', refreshing..."
127129
)
128130

129-
# Token not cached or expired - fetch a new one
130-
logger.info(
131-
f"Fetching new workload token for agent '{tool_context.agent_name}'..."
132-
)
133-
134131
# Default to agent_name if workload_name not specified
135132
if not workload_name:
136133
workload_name = tool_context.agent_name
@@ -147,15 +144,11 @@ async def get_workload_token(
147144

148145
tool_context._invocation_context.session.state[cache_key] = workload_token
149146

150-
logger.info(
151-
f"Successfully cached new workload token for agent '{tool_context.agent_name}'",
152-
)
153-
154147
return workload_token.workload_access_token
155148

156149

157150
async def get_workload_token(
158-
tool_context: Union[ToolContext|ReadonlyContext],
151+
tool_context: Union[ToolContext | ReadonlyContext],
159152
identity_client: Optional[IdentityClient] = None,
160153
workload_name: Optional[str] = None,
161154
user_token: Optional[str] = None,

0 commit comments

Comments
 (0)