d12542db55
Note - This Engine PR must go first: https://github.com/ArcadeAI/Engine/pull/65 In this PR: - Add `client.tool.authorize` to authorize a tool by name by @Spartee - Refactored client.auth methods to always pass around scopes (as needed by the above Engine PR) by @nbarbettini - Reduced the scopes needed in the Slack toolkit, which was blocked by this until now! @nbarbettini --------- Co-authored-by: Nate Barbettini <nate@arcade-ai.com>
68 lines
2.3 KiB
Python
68 lines
2.3 KiB
Python
import os
|
|
|
|
from google.oauth2.credentials import Credentials
|
|
from langchain_google_community import GmailToolkit
|
|
from langchain_google_community.gmail.utils import (
|
|
build_resource_service,
|
|
)
|
|
from langchain_openai import ChatOpenAI
|
|
from langgraph.prebuilt import create_react_agent
|
|
|
|
# Step 1: Install required packages
|
|
# Run the following in your terminal:
|
|
# %pip install -qU langchain-google-community[gmail]
|
|
# %pip install -qU langchain-openai
|
|
# %pip install -qU langgraph
|
|
#
|
|
# Step 2: Set environment variables for LangChain and OpenAI API keys
|
|
# Uncomment the following lines if you have the LangSmith API key
|
|
# os.environ["LANGCHAIN_TRACING_V2"] = "true"
|
|
# os.environ["LANGCHAIN_API_KEY"] = getpass.getpass("Enter your LangSmith API key: ")
|
|
#
|
|
# Step 3 (Option 1) Manually authenticate with Gmail by creating your own google app, credentials, and handling tokens and Oauth
|
|
# credentials = get_gmail_credentials(
|
|
# token_file="token.json",
|
|
# scopes=["https://mail.google.com/"],
|
|
# client_secrets_file="credentials.json",
|
|
# )
|
|
#
|
|
# ----------------- OR -----------------
|
|
# Step 3 (Option 2) Use the Arcade SDK to authenticate with Gmail
|
|
from arcade.client import Arcade, AuthProvider
|
|
|
|
client = Arcade(api_key=os.environ["ARCADE_API_KEY"])
|
|
|
|
challenge = client.auth.authorize(
|
|
provider=AuthProvider.google,
|
|
scopes=["https://www.googleapis.com/auth/gmail.readonly"],
|
|
user_id="example_user_id",
|
|
)
|
|
|
|
if challenge.status != "completed":
|
|
print(f"Please visit this URL to authorize: {challenge.auth_url}")
|
|
input("Press Enter after you've completed the authorization...")
|
|
challenge = client.auth.poll_authorization(challenge)
|
|
if challenge.status != "completed":
|
|
print("Authorization not completed. Please try again.")
|
|
exit(1)
|
|
|
|
|
|
creds = Credentials(challenge.context.token)
|
|
api_resource = build_resource_service(credentials=creds)
|
|
toolkit = GmailToolkit(api_resource=api_resource)
|
|
|
|
# Step 4: Get available tools
|
|
tools = toolkit.get_tools()
|
|
|
|
# Step 5: Initialize the LLM and create an agent
|
|
llm = ChatOpenAI(model="gpt-4o")
|
|
agent_executor = create_react_agent(llm, tools)
|
|
|
|
# Step 6: Draft an email using the agent
|
|
example_query = "Read my latest emails to me and summarize them."
|
|
events = agent_executor.stream(
|
|
{"messages": [("user", example_query)]},
|
|
stream_mode="values",
|
|
)
|
|
for event in events:
|
|
event["messages"][-1].pretty_print()
|