from composio import Composio
from composio_langgraph import LanggraphProvider

TOOL_SLUGS = [
    # Slack
    "SLACK_SENDS_A_MESSAGE_TO_A_SLACK_CHANNEL",
    "SLACK_FETCH_CONVERSATION_HISTORY",
    "SLACK_FIND_CHANNELS",
    "SLACK_FIND_USERS",
    "SLACK_ADD_REACTION_TO_AN_ITEM",
    "SLACK_FETCH_MESSAGE_THREAD_FROM_A_CONVERSATION",
    # Notion
    "NOTION_SEARCH_NOTION_PAGE",
    "NOTION_RETRIEVE_PAGE",
    "NOTION_UPDATE_PAGE",
    "NOTION_CREATE_NOTION_PAGE",
    "NOTION_ADD_PAGE_CONTENT",
    # GitHub
    "GITHUB_CREATE_AN_ISSUE",
    "GITHUB_LIST_REPO_ISSUES",
    "GITHUB_GET_A_REPOSITORY",
    # Google Drive
    "GOOGLEDRIVE_FIND_FILE",
    "GOOGLEDRIVE_GET_FILE_METADATA",
]


def get_composio_tools():
    client = Composio(provider=LanggraphProvider())
    tools = client.tools.get(user_id="default", tools=TOOL_SLUGS)
    return tools