auth0-ai-langchain 0.1.1__py3-none-any.whl → 0.1.2__py3-none-any.whl

auth0_ai_langchain/ciba/ciba_poller_graph.py

@@ -1,25 +1,33 @@
 import os
-from langgraph.graph import StateGraph, END, START
-from typing import Awaitable, Callable, Union, Optional, TypedDict
-from langgraph_sdk import get_client
-from langgraph_sdk.schema import Command
-from auth0_ai.authorizers.ciba_authorizer import CIBAAuthorizer, CibaAuthorizerCheckResponse, AuthorizeResponse
+from typing import Awaitable, Callable, Optional, TypedDict, Union
+
+from auth0_ai.authorizers.ciba_authorizer import (
+    AuthorizeResponse,
+    CIBAAuthorizer,
+    CibaAuthorizerCheckResponse,
+)
 from auth0_ai.credentials import Credentials
 from auth0_ai.token_response import TokenResponse
-from langchain_auth0_ai.ciba.types import Auth0Graphs
+from langgraph.graph import END, START, StateGraph
+from langgraph_sdk import get_client
+from langgraph_sdk.schema import Command
+
+from auth0_ai_langchain.ciba.types import Auth0Graphs
+
 
 class State(TypedDict):
     ciba_response: AuthorizeResponse
     on_resume_invoke: str
     thread_id: str
     user_id: str
-    
+
     # Internal
     task_id: str
     tool_id: str
     status: CibaAuthorizerCheckResponse
     token_response: Optional[TokenResponse]
 
+
 def ciba_poller_graph(on_stop_scheduler: Union[str, Callable[[State], Awaitable[None]]]):
     """
     A LangGraph graph to monitor the status of a CIBA transaction.
@@ -35,22 +43,24 @@ def ciba_poller_graph(on_stop_scheduler: Union[str, Callable[[State], Awaitable[
         except Exception as e:
             print(f"Error in check_status: {e}")
         return state
-    
+
     async def stop_scheduler(state: State):
         try:
             if isinstance(on_stop_scheduler, str):
-                langgraph = get_client(url=os.getenv("LANGGRAPH_API_URL", "http://localhost:54367"))
+                langgraph = get_client(url=os.getenv(
+                    "LANGGRAPH_API_URL", "http://localhost:54367"))
                 await langgraph.crons.create_for_thread(state.thread_id, Auth0Graphs.CIBA_POLLER.value)
             elif callable(on_stop_scheduler):
                 await on_stop_scheduler(state)
         except Exception as e:
             print(f"Error in stop_scheduler: {e}")
         return state
-    
+
     async def resume_agent(state: State):
-        langgraph = get_client(url=os.getenv("LANGGRAPH_API_URL", "http://localhost:54367"))
+        langgraph = get_client(url=os.getenv(
+            "LANGGRAPH_API_URL", "http://localhost:54367"))
         _credentials: Credentials = None
-        
+
         try:
             if state["status"] == CibaAuthorizerCheckResponse.APPROVED:
                 _credentials = {
@@ -64,15 +74,16 @@ def ciba_poller_graph(on_stop_scheduler: Union[str, Callable[[State], Awaitable[
                 state["thread_id"],
                 state["on_resume_invoke"],
                 config={
-                    "configurable": {"_credentials": _credentials} # this is only for this run / thread_id
+                    # this is only for this run / thread_id
+                    "configurable": {"_credentials": _credentials}
                 },
                 command=Command(resume={"status": state["status"]})
             )
         except Exception as e:
             print(f"Error in resume_agent: {e}")
-        
+
         return state
-    
+
     async def should_continue(state: State):
         status = state.get("status")
         if status == CibaAuthorizerCheckResponse.PENDING:
@@ -82,7 +93,7 @@ def ciba_poller_graph(on_stop_scheduler: Union[str, Callable[[State], Awaitable[
         elif status in [CibaAuthorizerCheckResponse.APPROVED, CibaAuthorizerCheckResponse.REJECTED]:
             return "resume_agent"
         return END
-    
+
     state_graph = StateGraph(State)
     state_graph.add_node("check_status", check_status)
     state_graph.add_node("stop_scheduler", stop_scheduler)
@@ -90,5 +101,5 @@ def ciba_poller_graph(on_stop_scheduler: Union[str, Callable[[State], Awaitable[
     state_graph.add_edge(START, "check_status")
     state_graph.add_edge("resume_agent", "stop_scheduler")
     state_graph.add_conditional_edges("check_status", should_continue)
-    
-    return state_graph
+
+    return state_graph

{auth0_ai_langchain-0.1.1.dist-info → auth0_ai_langchain-0.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: auth0-ai-langchain
-Version: 0.1.1
+Version: 0.1.2
 Summary: This package is an SDK for building secure AI-powered applications using Auth0, Okta FGA and LangChain.
 License: Apache-2.0
 Author: Auth0
@@ -11,7 +11,7 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
-Requires-Dist: auth0-ai (>=0.1.0,<0.2.0)
+Requires-Dist: auth0-ai (>=0.1.1,<0.2.0)
 Requires-Dist: langchain (>=0.3.20,<0.4.0)
 Requires-Dist: langchain-core (>=0.3.43,<0.4.0)
 Requires-Dist: langgraph (>=0.3.25,<0.4.0)
@@ -28,7 +28,8 @@ Description-Content-Type: text/markdown
 
 ## Installation
 
-> [!WARNING] > `auth0-ai-langchain` is currently under development and it is not intended to be used in production, and therefore has no official support.
+> [!WARNING]
+> `auth0-ai-langchain` is currently under development and it is not intended to be used in production, and therefore has no official support.
 
 ```bash
 pip install auth0-ai-langchain
@@ -38,12 +39,12 @@ pip install auth0-ai-langchain
 
 The `FGAAuthorizer` can leverage Okta FGA to authorize tools executions. The `FGAAuthorizer.create` function can be used to create an authorizer that checks permissions before executing the tool.
 
-Full example of [Authorization for Tools](../../examples/authorization-for-tools/langchain-examples/).
+Full example of [Authorization for Tools](https://github.com/auth0-lab/auth0-ai-python/tree/main/examples/authorization-for-tools/langchain-examples).
 
 1. Create an instance of FGA Authorizer:
 
 ```python
-from langchain_auth0_ai.fga.fga_authorizer import FGAAuthorizer, FGAAuthorizerOptions
+from auth0_ai_langchain.fga.fga_authorizer import FGAAuthorizer, FGAAuthorizerOptions
 
 fga = FGAAuthorizer.create()
 ```
@@ -104,13 +105,13 @@ buy_tool = StructuredTool(
 
 The `Auth0AI.with_federated_connection` function exchanges user's refresh token taken from the runnable configuration (`config.configurable._credentials.refresh_token`) for a Federated Connection API token.
 
-Full Example of [Calling APIs On User's Behalf](../../examples/calling-apis/langchain-examples/).
+Full Example of [Calling APIs On User's Behalf](https://github.com/auth0-lab/auth0-ai-python/tree/main/examples/calling-apis/langchain-examples).
 
 1. Define a tool with the proper authorizer:
 
 ```python
-from langchain_auth0_ai.auth0_ai import Auth0AI
-from langchain_auth0_ai.federated_connections import get_access_token_for_connection
+from auth0_ai_langchain.auth0_ai import Auth0AI
+from auth0_ai_langchain.federated_connections import get_access_token_for_connection
 from langchain_core.tools import StructuredTool
 
 auth0_ai = Auth0AI()
@@ -160,14 +161,14 @@ workflow = (
 
 The `FGARetriever` can be used to filter documents based on access control checks defined in Okta FGA. This retriever performs batch checks on retrieved documents, returning only the ones that pass the specified access criteria.
 
-Full Example of [RAG Application](../../examples/authorization-for-rag/langchain-examples/).
+Full Example of [RAG Application](https://github.com/auth0-lab/auth0-ai-python/tree/main/examples/authorization-for-rag/langchain-examples).
 
 Create a retriever instance using the `FGARetriever` class.
 
 ```python
 from langchain.vectorstores import VectorStoreIndex
 from langchain.schema import Document
-from langchain_auth0_ai import FGARetriever
+from auth0_ai_langchain import FGARetriever
 from openfga_sdk.client.models import ClientCheckRequest
 from openfga_sdk import ClientConfiguration
 from openfga_sdk.credentials import CredentialConfiguration, Credentials
@@ -217,5 +218,5 @@ print(response)
 </p>
 <p align="center">Auth0 is an easy to implement, adaptable authentication and authorization platform. To learn more checkout <a href="https://auth0.com/why-auth0">Why Auth0?</a></p>
 <p align="center">
-This project is licensed under the Apache 2.0 license. See the <a href="/LICENSE"> LICENSE</a> file for more info.</p>
+This project is licensed under the Apache 2.0 license. See the <a href="https://github.com/auth0-lab/auth0-ai-python/blob/main/LICENSE"> LICENSE</a> file for more info.</p>
 

{auth0_ai_langchain-0.1.1.dist-info → auth0_ai_langchain-0.1.2.dist-info}/RECORD

@@ -7,13 +7,13 @@ auth0_ai_langchain/ciba/ciba_graph/initialize_ciba.py,sha256=a41KedBzxfLqG2AhvkF
 auth0_ai_langchain/ciba/ciba_graph/initialize_hitl.py,sha256=CR3jMolZYYOBHx1AXb6yERBaZThMg677qGGo_vRy6I8,1901
 auth0_ai_langchain/ciba/ciba_graph/types.py,sha256=NZS99vPOgJRc2O7mO5MWj6nAa4RSG1R5oSvzYhkz0RA,4234
 auth0_ai_langchain/ciba/ciba_graph/utils.py,sha256=ZPAh0Gs7Hj59_xngg8M7yx1v52dTn2pNpMNRpFKCSII,560
-auth0_ai_langchain/ciba/ciba_poller_graph.py,sha256=aKTOsnlZ-Uy6jAvEh2JBOYP0UGgZiDJ5DzDv7fXMRvM,3792
+auth0_ai_langchain/ciba/ciba_poller_graph.py,sha256=rjlxsTheJrN2J6E5BCE9aVyDO8V7UFhdKyX4KT_hB8k,3828
 auth0_ai_langchain/ciba/types.py,sha256=gybqYEprklZwcMBgaWFooBsJ1GcNUK8ZWRvAX5PZWdE,177
 auth0_ai_langchain/federated_connections/__init__.py,sha256=OJCWTnxYuaWjn8FyFGjCAF7m5Y4Eigkzx7a59atFFFg,356
 auth0_ai_langchain/federated_connections/federated_connection_authorizer.py,sha256=ZLF4p7fPTrODOeHWIShfBTsReSy27u73rIp0L_Umhjg,2218
 auth0_ai_langchain/fga/fga_authorizer.py,sha256=uDaGDSXaxQd1X-2w2zTvnfizMB-DtQ-1G6SIaDNBrho,137
 auth0_ai_langchain/utils/interrupt.py,sha256=JoYJkigDEAPRHZtjo6gw6k3439E4i1O7F4_0ExkL_RE,405
-auth0_ai_langchain-0.1.1.dist-info/LICENSE,sha256=Lu_2YH0oK8b_VVisAhNQ2WIdtwY8pSU2PLbll-y6Cj8,9792
-auth0_ai_langchain-0.1.1.dist-info/METADATA,sha256=6Ymkyi-C9a8vrVomX2vvFZavWSQ6sMAZeMgMRyVfyqY,7588
-auth0_ai_langchain-0.1.1.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-auth0_ai_langchain-0.1.1.dist-info/RECORD,,
+auth0_ai_langchain-0.1.2.dist-info/LICENSE,sha256=Lu_2YH0oK8b_VVisAhNQ2WIdtwY8pSU2PLbll-y6Cj8,9792
+auth0_ai_langchain-0.1.2.dist-info/METADATA,sha256=synK995x8fRtUFH2yR0oyh5vwGzZTEdC2zUJygtzAwQ,7786
+auth0_ai_langchain-0.1.2.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+auth0_ai_langchain-0.1.2.dist-info/RECORD,,