Merge pull request #52 from AdamWozniewski/feature/aidx-198-rar-support-auth-details

pmalouin · web-flow · commit 1a42df8cd906 · 2025-10-03T09:11:18.000-04:00
Add Authorization_details
diff --git a/examples/async-user-confirmation/langchain-examples/.env.example b/examples/async-user-confirmation/langchain-examples/.env.example
@@ -6,6 +6,7 @@ AUTH0_CLIENT_SECRET="<auth0-client-secret>"
 # API
 API_URL=http://localhost:8081/
 AUDIENCE=sample-api
+REQUESTED_EXPIRY=300
 
 # OpenAI
 OPENAI_API_KEY="<openai-api-key>"
diff --git a/examples/async-user-confirmation/langchain-examples/README.md b/examples/async-user-confirmation/langchain-examples/README.md
@@ -26,6 +26,7 @@ AUTH0_CLIENT_SECRET="<auth0-client-secret>"
 # API
 API_URL=http://localhost:8081/
 AUDIENCE=sample-api
+REQUESTED_EXPIRY=300
 
 # OpenAI
 OPENAI_API_KEY="<openai-api-key>"
diff --git a/examples/async-user-confirmation/langchain-examples/src/agents/conditional_trade.py b/examples/async-user-confirmation/langchain-examples/src/agents/conditional_trade.py
@@ -103,6 +103,7 @@ def check_trade_status(state: State):
 auth0_ai = Auth0AI()
 protect_tool = auth0_ai.with_async_user_confirmation(
     audience=os.getenv("AUDIENCE"),
+    requested_expiry=os.getenv("REQUESTED_EXPIRY"),
     scopes=["stock:trade"],
     binding_message=lambda ticker, qty: f"Authorize the purchase of {qty} {ticker}",
     user_id=lambda *_, **__: ensure_config().get("configurable", {}).get("user_id")
diff --git a/examples/async-user-confirmation/llama-index-examples/.env.example b/examples/async-user-confirmation/llama-index-examples/.env.example
@@ -6,6 +6,7 @@ AUTH0_CLIENT_SECRET="<auth0-client-secret>"
 # API
 API_URL=http://localhost:8081/
 AUDIENCE=sample-api
+REQUESTED_EXPIRY=300
 
 # OpenAI
 OPENAI_API_KEY="<openai-api-key>"
diff --git a/examples/async-user-confirmation/llama-index-examples/README.md b/examples/async-user-confirmation/llama-index-examples/README.md
@@ -25,6 +25,7 @@ AUTH0_CLIENT_SECRET="<auth0-client-secret>"
 # API
 API_URL=http://localhost:8081/
 AUDIENCE=sample-api
+REQUESTED_EXPIRY=300
 
 # OpenAI
 OPENAI_API_KEY="<openai-api-key>"
diff --git a/examples/async-user-confirmation/llama-index-examples/src/auth0/auth0_ai.py b/examples/async-user-confirmation/llama-index-examples/src/auth0/auth0_ai.py
@@ -17,6 +17,7 @@ async def user_id(**_kwargs):
 with_async_user_confirmation = auth0_ai.with_async_user_confirmation(
     scopes=["stock:trade"],
     audience=os.getenv("AUDIENCE"),
+    requested_expiry=os.getenv("REQUESTED_EXPIRY"),
     binding_message=lambda ticker, qty: f"Authorize the purchase of {qty} {ticker}",
     user_id=user_id,
     # When this flag is set to `"block"`, the execution of the tool awaits until the user approves or rejects the request.
diff --git a/packages/auth0-ai-langchain/README.md b/packages/auth0-ai-langchain/README.md
@@ -32,6 +32,7 @@ auth0_ai = Auth0AI()
 with_async_user_confirmation = auth0_ai.with_async_user_confirmation(
     scopes=["stock:trade"],
     audience=os.getenv("AUDIENCE"),
+    requested_expiry=os.getenv("REQUESTED_EXPIRY"),
     binding_message=lambda ticker, qty: f"Authorize the purchase of {qty} {ticker}",
     user_id=lambda *_, **__: ensure_config().get("configurable", {}).get("user_id"),
     # Optional:
@@ -67,6 +68,7 @@ When defining the tool authorizer, you can specify the `authorization_details` p
 with_async_user_confirmation = auth0_ai.with_async_user_confirmation(
     scopes=["stock:trade"],
     audience=os.getenv("AUDIENCE"),
+    requested_expiry=os.getenv("REQUESTED_EXPIRY"),
     binding_message=lambda ticker, qty: f"Authorize the purchase of {qty} {ticker}",
     authorization_details=lambda ticker, qty: [
         {
diff --git a/packages/auth0-ai-langchain/auth0_ai_langchain/auth0_ai.py b/packages/auth0-ai-langchain/auth0_ai_langchain/auth0_ai.py
@@ -46,6 +46,7 @@ def with_async_user_confirmation(self, **params: CIBAAuthorizerParams) -> Callab
             with_async_user_confirmation = auth0_ai.with_async_user_confirmation(
                 scopes=["stock:trade"],
                 audience=os.getenv("AUDIENCE"),
+                requested_expiry=os.getenv("REQUESTED_EXPIRY"),
                 binding_message=lambda ticker, qty: f"Authorize the purchase of {qty} {ticker}",
                 user_id=lambda *_, **__: ensure_config().get("configurable", {}).get("user_id")
             )
diff --git a/packages/auth0-ai-llamaindex/README.md b/packages/auth0-ai-llamaindex/README.md
@@ -31,6 +31,7 @@ auth0_ai = Auth0AI()
 with_async_user_confirmation = auth0_ai.with_async_user_confirmation(
     scopes=["stock:trade"],
     audience=os.getenv("AUDIENCE"),
+    requested_expiry=os.getenv("REQUESTED_EXPIRY"),
     binding_message=lambda ticker, qty: f"Authorize the purchase of {qty} {ticker}",
     user_id=lambda *_, **__: session["user"]["userinfo"]["sub"],
     # Optional:
@@ -67,6 +68,7 @@ When defining the tool authorizer, you can specify the `authorization_details` p
 with_async_user_confirmation = auth0_ai.with_async_user_confirmation(
     scopes=["stock:trade"],
     audience=os.getenv("AUDIENCE"),
+    requested_expiry=os.getenv("REQUESTED_EXPIRY"),
     binding_message=lambda ticker, qty: f"Authorize the purchase of {qty} {ticker}",
     authorization_details=lambda ticker, qty: [
         {
diff --git a/packages/auth0-ai-llamaindex/auth0_ai_llamaindex/auth0_ai.py b/packages/auth0-ai-llamaindex/auth0_ai_llamaindex/auth0_ai.py
@@ -89,6 +89,7 @@ def with_async_user_confirmation(self, **params: CIBAAuthorizerParams) -> Callab
             with_async_user_confirmation = auth0_ai.with_async_user_confirmation(
                 scopes=["stock:trade"],
                 audience=os.getenv("AUDIENCE"),
+                requested_expiry=os.getenv("REQUESTED_EXPIRY"),
                 binding_message=lambda ticker, qty: f"Authorize the purchase of {qty} {ticker}",
                 user_id=lambda *_, **__: session["user"]["userinfo"]["sub"]
             )
diff --git a/packages/auth0-ai/__init__.py b/packages/auth0-ai/__init__.py
diff --git a/packages/auth0-ai/auth0_ai/authorizers/ciba/ciba_authorizer_params.py b/packages/auth0-ai/auth0_ai/authorizers/ciba/ciba_authorizer_params.py
@@ -1,6 +1,7 @@
 from typing import Awaitable, Callable, Generic, Literal, Optional, TypedDict, Union
 from auth0_ai.authorizers.context import AuthContext
 from auth0_ai.authorizers.types import ToolInput
+from auth0_ai.credentials import AuthorizationDetails
 from auth0_ai.stores import Store
 
 
@@ -37,12 +38,12 @@ class CIBAAuthorizerParams(TypedDict, Generic[ToolInput]):
         Callable[ToolInput, Awaitable[str]]
     ]
     authorization_details: Optional[Union[
-        list[dict],
-        Callable[ToolInput, list[dict]],
-        Callable[ToolInput, Awaitable[list[dict]]]
+        list[AuthorizationDetails],
+        Callable[ToolInput, list[AuthorizationDetails]],
+        Callable[ToolInput, Awaitable[list[AuthorizationDetails]]],
     ]]
     store: Optional[Store]
     audience: Optional[str]
     requested_expiry: Optional[int]
     on_authorization_request: Optional[Literal["block", "interrupt"]]
-    credentials_context: Optional[AuthContext]
+    credentials_context: Optional[AuthContext]
diff --git a/packages/auth0-ai/auth0_ai/credentials.py b/packages/auth0-ai/auth0_ai/credentials.py
@@ -1,9 +1,13 @@
 from typing import TypedDict, Optional
 
+class AuthorizationDetails(TypedDict):
+    type: str
+
 class TokenResponse(TypedDict):
     access_token: str
     expires_in: int
     scope: list[str]
     token_type: Optional[str]
     id_token: Optional[str]
     refresh_token: Optional[str]
+    authorization_details: Optional[list[AuthorizationDetails]]

Original file line number	Diff line number	Diff line change
`@@ -46,6 +46,7 @@ def with_async_user_confirmation(self, **params: CIBAAuthorizerParams) -> Callab`
`46`	`46`	`with_async_user_confirmation = auth0_ai.with_async_user_confirmation(`
`47`	`47`	`scopes=["stock:trade"],`
`48`	`48`	`audience=os.getenv("AUDIENCE"),`
	`49`	`+ requested_expiry=os.getenv("REQUESTED_EXPIRY"),`
`49`	`50`	`binding_message=lambda ticker, qty: f"Authorize the purchase of {qty} {ticker}",`
`50`	`51`	`user_id=lambda _, *__: ensure_config().get("configurable", {}).get("user_id")`
`51`	`52`	`)`
Original file line number	Diff line number	Diff line change
`@@ -89,6 +89,7 @@ def with_async_user_confirmation(self, **params: CIBAAuthorizerParams) -> Callab`
`89`	`89`	`with_async_user_confirmation = auth0_ai.with_async_user_confirmation(`
`90`	`90`	`scopes=["stock:trade"],`
`91`	`91`	`audience=os.getenv("AUDIENCE"),`
	`92`	`+ requested_expiry=os.getenv("REQUESTED_EXPIRY"),`
`92`	`93`	`binding_message=lambda ticker, qty: f"Authorize the purchase of {qty} {ticker}",`
`93`	`94`	`user_id=lambda _, *__: session["user"]["userinfo"]["sub"]`
`94`	`95`	`)`