fix(identity): refactor imports to use absolute paths in ve_identity

- Fix Non-standard docstrings
- Remove veadk prefix in logger

Author: loveyana

veadk/integrations/ve_identity/__init__.py

@@ -22,20 +22,28 @@
 - WorkloadTokenManager: Manages workload access tokens with caching
 - AuthRequestProcessor: Handles OAuth2 flows in agent conversations
 
-Example usage:
-    from veadk.integrations.ve_identity import VeIdentityFunctionTool
-
-    @VeIdentityFunctionTool(
-        provider_name="github",
-        scopes=["repo", "user"],
-        auth_flow="USER_FEDERATION",
+Examples:
+    from veadk.integrations.ve_identity import (
+        VeIdentityFunctionTool,
+        oauth2_auth,
     )
+
     async def get_github_repos(access_token: str):
         # Tool implementation
         pass
+
+    tool = VeIdentityFunctionTool(
+        func=get_github_repos,
+        auth_config=oauth2_auth(
+            provider_name="github",
+            scopes=["repo", "user"],
+            auth_flow="USER_FEDERATION",
+        ),
+        into="access_token",
+    )
 """
 
-from .auth_processor import (
+from veadk.integrations.ve_identity.auth_processor import (
     AuthRequestConfig,
     AuthRequestProcessor,
     _NoOpAuthProcessor,
@@ -46,7 +54,7 @@ async def get_github_repos(access_token: str):
 )
 
 # New unified tools
-from .auth_config import (
+from veadk.integrations.ve_identity.auth_config import (
     api_key_auth,
     oauth2_auth,
     workload_auth,
@@ -55,16 +63,19 @@ async def get_github_repos(access_token: str):
     WorkloadAuthConfig,
     VeIdentityAuthConfig,
 )
-from .function_tool import VeIdentityFunctionTool
-from .mcp_tool import VeIdentityMcpTool
-from .mcp_toolset import VeIdentityMcpToolset
-from .identity_client import IdentityClient
-from .models import (
+from veadk.integrations.ve_identity.function_tool import VeIdentityFunctionTool
+from veadk.integrations.ve_identity.mcp_tool import VeIdentityMcpTool
+from veadk.integrations.ve_identity.mcp_toolset import VeIdentityMcpToolset
+from veadk.integrations.ve_identity.identity_client import IdentityClient
+from veadk.integrations.ve_identity.models import (
     OAuth2TokenResponse,
     OAuth2AuthPoller,
     WorkloadToken,
 )
-from .token_manager import WorkloadTokenManager, get_workload_token
+from veadk.integrations.ve_identity.token_manager import (
+    WorkloadTokenManager,
+    get_workload_token,
+)
 
 __all__ = [
     # Client

veadk/integrations/ve_identity/auth_config.py

@@ -23,8 +23,8 @@
 
 from pydantic import BaseModel, model_validator, field_validator
 
-from .models import OAuth2AuthPoller
-from .identity_client import IdentityClient
+from veadk.integrations.ve_identity.models import OAuth2AuthPoller
+from veadk.integrations.ve_identity.identity_client import IdentityClient
 
 
 class AuthConfig(BaseModel, ABC):
@@ -121,6 +121,7 @@ def _validate_required_fields(self):
     def auth_type(self) -> str:
         return "oauth2"
 
+
 class WorkloadAuthConfig(AuthConfig):
     """Workload Access Token authentication configuration."""
 
@@ -129,7 +130,6 @@ def auth_type(self) -> str:
         return "workload"
 
 
-
 # Type alias for all auth configs
 VeIdentityAuthConfig = Union[ApiKeyAuthConfig, OAuth2AuthConfig, WorkloadAuthConfig]
 
@@ -145,6 +145,7 @@ def api_key_auth(
         provider_name=provider_name, identity_client=identity_client, region=region
     )
 
+
 def workload_auth(
     provider_name: str,
     identity_client: Optional[IdentityClient] = None,
@@ -155,6 +156,7 @@ def workload_auth(
         provider_name=provider_name, identity_client=identity_client, region=region
     )
 
+
 def oauth2_auth(
     provider_name: str,
     scopes: List[str],

veadk/integrations/ve_identity/auth_mixins.py

@@ -37,19 +37,19 @@
 from google.adk.tools.openapi_tool.auth.auth_helpers import dict_to_auth_scheme
 from google.adk.agents.readonly_context import ReadonlyContext
 
-from .models import OAuth2AuthPoller, OAuth2TokenResponse
-from .identity_client import IdentityClient
-from .auth_config import (
+from veadk.integrations.ve_identity.models import OAuth2AuthPoller, OAuth2TokenResponse
+from veadk.integrations.ve_identity.identity_client import IdentityClient
+from veadk.integrations.ve_identity.auth_config import (
     VeIdentityAuthConfig,
     ApiKeyAuthConfig,
     OAuth2AuthConfig,
     WorkloadAuthConfig,
 )
-from .token_manager import get_workload_token
+from veadk.integrations.ve_identity.token_manager import get_workload_token
 
 from veadk.utils.logger import get_logger
 
-logger = get_logger("veadk." + __name__)
+logger = get_logger(__name__)
 
 
 # OAuth2 scheme definition (shared across all OAuth2 tools)
@@ -509,7 +509,7 @@ async def _handle_oauth2_flow_for_readonly_context(
 
     def _create_default_oauth2_poller(self, auth_uri: str, request_dict: dict):
         """Create a default OAuth2 poller for ReadonlyContext scenarios."""
-        from .auth_processor import _DefaultOauth2AuthPoller
+        from veadk.integrations.ve_identity.auth_processor import _DefaultOauth2AuthPoller
 
         async def async_token_fetcher():
             response = await self._identity_client.get_oauth2_token_or_auth_url(

veadk/integrations/ve_identity/auth_processor.py

@@ -25,9 +25,9 @@
 from google.genai import types
 from google.adk.auth.auth_credential import OAuth2Auth
 
-from .identity_client import IdentityClient
-from .models import AuthRequestConfig, OAuth2AuthPoller
-from .utils import (
+from veadk.integrations.ve_identity.identity_client import IdentityClient
+from veadk.integrations.ve_identity.models import AuthRequestConfig, OAuth2AuthPoller
+from veadk.integrations.ve_identity.utils import (
     get_function_call_auth_config,
     get_function_call_id,
     is_pending_auth_event,
@@ -43,7 +43,7 @@
     from veadk.runner import Runner
 
 
-logger = get_logger("veadk." + __name__)
+logger = get_logger(__name__)
 
 # Default configuration for token polling
 DEFAULT_POLLING_INTERVAL_SECONDS = 5
@@ -102,7 +102,10 @@ class _DefaultOauth2AuthPoller(OAuth2AuthPoller):
     def __init__(
         self,
         auth_url: str,
-        polling_func: Callable[[], Optional[OAuth2Auth]] | Callable[[], Awaitable[Optional[OAuth2Auth]]],
+        polling_func: (
+            Callable[[], Optional[OAuth2Auth]]
+            | Callable[[], Awaitable[Optional[OAuth2Auth]]]
+        ),
     ):
         """Initialize the OAuth2 auth poller.
 
@@ -134,6 +137,7 @@ async def poll_for_auth(self) -> OAuth2Auth:
 
             # Check if polling_func is async or sync
             import inspect
+
             if inspect.iscoroutinefunction(self.polling_func):
                 oauth2auth = await self.polling_func()
             else:
@@ -226,7 +230,9 @@ async def process_auth_request(
         # Use custom poller or default poller
         # Create async token fetcher for default poller
         async def async_token_fetcher():
-            response = await self._identity_client.get_oauth2_token_or_auth_url(**request_dict)
+            response = await self._identity_client.get_oauth2_token_or_auth_url(
+                **request_dict
+            )
             return (
                 OAuth2Auth(access_token=response.access_token)
                 if response.access_token and response.access_token.strip()

veadk/integrations/ve_identity/function_tool.py

@@ -23,17 +23,20 @@
 from google.adk.tools.function_tool import FunctionTool
 from google.adk.tools.tool_context import ToolContext
 
-from .auth_config import (
+from veadk.integrations.ve_identity.auth_config import (
     VeIdentityAuthConfig,
     ApiKeyAuthConfig,
     OAuth2AuthConfig,
     WorkloadAuthConfig,
 )
-from .auth_mixins import VeIdentityAuthMixin, AuthRequiredException
+from veadk.integrations.ve_identity.auth_mixins import (
+    VeIdentityAuthMixin,
+    AuthRequiredException,
+)
 
 from veadk.utils.logger import get_logger
 
-logger = get_logger("veadk." + __name__)
+logger = get_logger(__name__)
 
 
 class VeIdentityFunctionTool(VeIdentityAuthMixin, FunctionTool):

veadk/integrations/ve_identity/identity_client.py

@@ -27,7 +27,7 @@
 import volcenginesdkid
 import volcenginesdkcore
 
-from .models import (
+from veadk.integrations.ve_identity.models import (
     DCRRegistrationRequest,
     DCRRegistrationResponse,
     OAuth2TokenResponse,
@@ -36,7 +36,7 @@
 
 from veadk.utils.logger import get_logger
 
-logger = get_logger("veadk." + __name__)
+logger = get_logger(__name__)
 
 
 def refresh_credentials(func):
@@ -123,9 +123,7 @@ async def create_oauth2_credential_provider(
 
         # Use the SDK's built-in async support
         return await self._api_client.create_oauth2_credential_provider(
-            volcenginesdkid.CreateOauth2CredentialProviderRequest(
-                **request_params
-            ),
+            volcenginesdkid.CreateOauth2CredentialProviderRequest(**request_params),
             async_req=True,
         )
 
@@ -145,9 +143,7 @@ async def create_api_key_credential_provider(
 
         # Use the SDK's built-in async support
         return await self._api_client.create_api_key_credential_provider(
-            volcenginesdkid.CreateApiKeyCredentialProviderRequest(
-                **request_params
-            ),
+            volcenginesdkid.CreateApiKeyCredentialProviderRequest(**request_params),
             async_req=True,
         )
 
@@ -229,9 +225,7 @@ def convert_response(
             logger.info("Retrieving workload access token...")
             resp: volcenginesdkid.GetWorkloadAccessTokenForUserIdResponse = (
                 await self._api_client.get_workload_access_token(
-                    volcenginesdkid.GetWorkloadAccessTokenRequest(
-                        name=workload_name
-                    ),
+                    volcenginesdkid.GetWorkloadAccessTokenRequest(name=workload_name),
                     async_req=True,
                 )
             )

veadk/integrations/ve_identity/mcp_tool.py

@@ -29,6 +29,7 @@
     from mcp.types import Tool as McpBaseTool
 except ImportError as e:
     import sys
+
     if sys.version_info < (3, 10):
         raise ImportError(
             "MCP Tool requires Python 3.10 or above. Please upgrade your Python"
@@ -40,13 +41,16 @@
 from google.adk.tools.mcp_tool.mcp_session_manager import MCPSessionManager
 from google.adk.tools.mcp_tool.mcp_session_manager import retry_on_closed_resource
 
-from .auth_config import VeIdentityAuthConfig
-from .auth_mixins import VeIdentityAuthMixin, AuthRequiredException
-from .utils import generate_headers
+from veadk.integrations.ve_identity.auth_config import VeIdentityAuthConfig
+from veadk.integrations.ve_identity.auth_mixins import (
+    VeIdentityAuthMixin,
+    AuthRequiredException,
+)
+from veadk.integrations.ve_identity.utils import generate_headers
 
 from veadk.utils.logger import get_logger
 
-logger = get_logger("veadk." + __name__)
+logger = get_logger(__name__)
 
 
 class VeIdentityMcpTool(VeIdentityAuthMixin, BaseTool):
@@ -149,11 +153,11 @@ async def run_async(
             return e.message
 
     async def _execute_with_credential(
-        self, 
-        *, 
-        args: dict[str, Any], 
-        tool_context: ToolContext, 
-        credential: AuthCredential
+        self,
+        *,
+        args: dict[str, Any],
+        tool_context: ToolContext,
+        credential: AuthCredential,
     ) -> Any:
         """Execute the MCP tool with the provided credential.
 
@@ -190,4 +194,4 @@ async def _run_async_impl(
         session = await self._mcp_session_manager.create_session(headers=headers)
 
         response = await session.call_tool(self._mcp_tool.name, arguments=args)
-        return response
+        return response