Gmail And Connector Tools

from pathlib import Path
import sys

ROOT = (
    Path.cwd().resolve().parent
    if Path.cwd().name == "notebooks"
    else Path.cwd().resolve()
)
if str(ROOT) not in sys.path:
    sys.path.insert(0, str(ROOT))

from examples.run_multi_tool_agent import build_llm_from_env
from shipit_agent import (
    Agent,
    CredentialRecord,
    FileCredentialStore,
    GmailTool,
    GoogleCalendarTool,
    GoogleDriveTool,
    JiraTool,
    LinearTool,
    SlackTool,
)

credential_store = FileCredentialStore(
    ROOT / ".shipit_notebooks" / "connector_creds.json"
)
credential_store.set(
    CredentialRecord(
        key="gmail",
        provider="gmail",
        secrets={
            "access_token": "ACCESS_TOKEN",
            "refresh_token": "REFRESH_TOKEN",
            "client_id": "CLIENT_ID",
            "client_secret": "CLIENT_SECRET",
        },
    )
)

llm = build_llm_from_env("bedrock")

agent = Agent(
    llm=llm,
    credential_store=credential_store,
    tools=[
        GmailTool(),
        GoogleCalendarTool(),
        GoogleDriveTool(),
        SlackTool(),
        LinearTool(),
        JiraTool(),
    ],
)

sorted(tool.name for tool in agent.tools)