agent-identity-python-sdk 0.1.1__tar.gz → 0.1.3__tar.gz

{agent_identity_python_sdk-0.1.1 → agent_identity_python_sdk-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: agent_identity_python_sdk
-Version: 0.1.1
+Version: 0.1.3
 Summary: Python SDK for using Agent Identity Service
 Home-page: https://github.com/aliyun/agent-identity-dev-kit
 Author: shaoheng
@@ -110,6 +110,20 @@ def my_function(sts_credential: STSCredential):
 my_function()
 ```
 
+### Using Decorators to Obtain Workload Credentials
+
+```python
+from agent_identity_python_sdk.core.decorators import requires_workload_access_token
+@requires_workload_access_token(inject_param_name="workload_access_token")
+def my_function(workload_access_token: str):
+    # use workload_access_token here
+    print(f"Workload Access Token: {workload_access_token}")
+    # Your business logic
+
+# call the function
+my_function()
+```
+
 ## Core Modules
 
 ### IdentityClient

{agent_identity_python_sdk-0.1.1 → agent_identity_python_sdk-0.1.3}/README.md

@@ -89,6 +89,20 @@ def my_function(sts_credential: STSCredential):
 my_function()
 ```
 
+### Using Decorators to Obtain Workload Credentials
+
+```python
+from agent_identity_python_sdk.core.decorators import requires_workload_access_token
+@requires_workload_access_token(inject_param_name="workload_access_token")
+def my_function(workload_access_token: str):
+    # use workload_access_token here
+    print(f"Workload Access Token: {workload_access_token}")
+    # Your business logic
+
+# call the function
+my_function()
+```
+
 ## Core Modules
 
 ### IdentityClient

{agent_identity_python_sdk-0.1.1 → agent_identity_python_sdk-0.1.3}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", encoding="utf-8") as f:
 
 setup(
     name="agent_identity_python_sdk",
-    version="0.1.1",
+    version="0.1.3",
     description="Python SDK for using Agent Identity Service",
     long_description=long_description,
     long_description_content_type="text/markdown",
@@ -20,9 +20,9 @@ setup(
         "alibabacloud-agentidentity20250901>=1.0.1",
         "alibabacloud-agentidentitydata20251127>=1.0.2",
         "setuptools",
-        "pydantic==2.11.7",
-        "urllib3==2.3.0",
-        "utils==1.0.2",
+        "pydantic>=2.11.7",
+        "urllib3>=2.3.0",
+        "utils>=1.0.2",
     ],
     extras_require={
         "dev": [

{agent_identity_python_sdk-0.1.1 → agent_identity_python_sdk-0.1.3}/src/agent_identity_python_sdk/core/decorators.py

@@ -117,7 +117,6 @@ def requires_access_token(
 
     return decorator
 
-
 def requires_api_key(*, credential_provider_name: str, inject_param_name: str = "api_key") -> Callable:
     """Decorator that fetches an api key before calling the decorated function.
 
@@ -235,6 +234,52 @@ def requires_sts_token(*, inject_param_name: str = "sts_credential",
 
     return decorator
 
+def requries_workload_access_token(*, inject_param_name: str = "workload_access_token") -> Callable:
+    """Decorator that fetches a workload access token before calling the decorated function.
+
+    Args:
+        inject_param_name: Parameter name to inject the workload access token into
+
+    Returns:
+        Decorator function that handles workload access token acquisition and injection
+    """
+
+    def decorator(func: Callable) -> Callable:
+        client = IdentityClient(get_region())
+
+        async def _get_workload_token() -> str:
+            user_id = AgentIdentityContext.get_user_id()
+            id_token = AgentIdentityContext.get_user_token()
+            
+            return await _get_workload_access_token(client, user_id=user_id, id_token=id_token)
+
+        @wraps(func)
+        async def async_wrapper(*args: Any, **kwargs: Any) -> Any:
+            workload_access_token = await _get_workload_token()
+            kwargs[inject_param_name] = workload_access_token
+            return await func(*args, **kwargs)
+
+        @wraps(func)
+        def sync_wrapper(*args: Any, **kwargs: Any) -> Any:
+            if _has_running_loop():
+                ctx = contextvars.copy_context()
+                import concurrent.futures
+
+                with concurrent.futures.ThreadPoolExecutor() as executor:
+                    future = executor.submit(ctx.run, asyncio.run, _get_workload_token())
+                    workload_access_token = future.result()
+            else:
+                workload_access_token = asyncio.run(_get_workload_token())
+
+            kwargs[inject_param_name] = workload_access_token
+            return func(*args, **kwargs)
+
+        if asyncio.iscoroutinefunction(func):
+            return async_wrapper
+        else:
+            return sync_wrapper
+
+    return decorator
 
 async def _get_workload_access_token_local(client: IdentityClient, user_id: Optional[str] = None, id_token: Optional[str] = None) -> str:
     workload_identity_name = os.environ.get("AGENT_IDENTITY_WORKLOAD_IDENTITY_NAME", None)
@@ -251,7 +296,6 @@ async def _get_workload_access_token_local(client: IdentityClient, user_id: Opti
 
     return client.get_workload_access_token(workload_identity_name, user_id=user_id, user_token=id_token)
 
-
 async def _get_workload_access_token(client: IdentityClient,
         user_id: Optional[str] = None,
         id_token: Optional[str] = None) -> str:
@@ -261,10 +305,9 @@ async def _get_workload_access_token(client: IdentityClient,
     else:
         return await _get_workload_access_token_local(client, user_id, id_token)
 
-
 def _has_running_loop() -> bool:
     try:
         asyncio.get_running_loop()
         return True
     except RuntimeError:
-        return False
+        return False

{agent_identity_python_sdk-0.1.1 → agent_identity_python_sdk-0.1.3}/src/agent_identity_python_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: agent-identity-python-sdk
-Version: 0.1.1
+Version: 0.1.3
 Summary: Python SDK for using Agent Identity Service
 Home-page: https://github.com/aliyun/agent-identity-dev-kit
 Author: shaoheng
@@ -110,6 +110,20 @@ def my_function(sts_credential: STSCredential):
 my_function()
 ```
 
+### Using Decorators to Obtain Workload Credentials
+
+```python
+from agent_identity_python_sdk.core.decorators import requires_workload_access_token
+@requires_workload_access_token(inject_param_name="workload_access_token")
+def my_function(workload_access_token: str):
+    # use workload_access_token here
+    print(f"Workload Access Token: {workload_access_token}")
+    # Your business logic
+
+# call the function
+my_function()
+```
+
 ## Core Modules
 
 ### IdentityClient

{agent_identity_python_sdk-0.1.1 → agent_identity_python_sdk-0.1.3}/src/agent_identity_python_sdk.egg-info/requires.txt

@@ -1,9 +1,9 @@
 alibabacloud-agentidentity20250901>=1.0.1
 alibabacloud-agentidentitydata20251127>=1.0.2
-pydantic==2.11.7
+pydantic>=2.11.7
 setuptools
-urllib3==2.3.0
-utils==1.0.2
+urllib3>=2.3.0
+utils>=1.0.2
 
 [dev]
 pytest-asyncio>=0.24.0