admin-valentin/arcade-mcp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update CrewAI example: Custom Auth Flow (#260)

Browse source
This commit is contained in:
Eric Gustin 2025-02-21 16:25:43 -08:00 committed by GitHub
parent 62173da343
commit ad66667bdb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 7 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
examples/crewai/crewai_with_arcade_tool.py
View file

@ -30,16 +30,15 @@ def custom_auth_flow(
Authorization is handled before executing the tool. Authorization is handled before executing the tool.
This function overrides the ArcadeToolManager's default auth flow performed by ArcadeToolManager.authorize_tool This function overrides the ArcadeToolManager's default auth flow performed by ArcadeToolManager.authorize_tool
""" """
print(f"Authorization required for tool: '{tool_name}' with inputs:")
for input_name, input_value in tool_input.items():
print(f" {input_name}: {input_value}")
# Get authorization status # Get authorization status
auth_response = manager.authorize(tool_name, USER_ID) auth_response = manager.authorize(tool_name, USER_ID)
# If the user is not authorized for the tool, # If the user is not authorized for the tool,
# then we need to handle the authorization before executing the tool # then we need to handle the authorization before executing the tool
if not manager.is_authorized(auth_response.id): if not manager.is_authorized(auth_response.id):
print(f"Authorization required for tool: '{tool_name}' with inputs:")
for input_name, input_value in tool_input.items():
print(f" {input_name}: {input_value}")
# Handle authorization # Handle authorization
print(f"\nTo authorize, visit: {auth_response.url}") print(f"\nTo authorize, visit: {auth_response.url}")
# Block until the user has completed the authorization # Block until the user has completed the authorization
@ -48,6 +47,10 @@ def custom_auth_flow(
# Ensure authorization completed successfully # Ensure authorization completed successfully
if not manager.is_authorized(auth_response.id): if not manager.is_authorized(auth_response.id):
raise ValueError(f"Authorization failed for {tool_name}. URL: {auth_response.url}") raise ValueError(f"Authorization failed for {tool_name}. URL: {auth_response.url}")
else:
print(f"Authorization already granted for tool: '{tool_name}' with inputs:")
for input_name, input_value in tool_input.items():
print(f" {input_name}: {input_value}")
def custom_execute_flow( def custom_execute_flow(