Using Composio With Google AI

Composio enables your Google AI models to connect with many tools!

Install Packages & Connect a Tool

Goal: Enable Google AI models to perform tasks like starring a repository on GitHub via natural language commands

These steps prepare your environment to enable interactions between Google AI and GitHub through Composio.

Run Command

$ pip install composio-google
> 
> # Connect your GitHub so models can interact with it
> 
> composio add github
> 
> # Check all supported apps
> 
> composio apps

Import Base Packages & Initialize Google AI Model

Replace {google_api_key} with your actual API key.

Default Imports & Configuration

1 import dotenv
2 from composio_google import App, ComposioToolset
3 from vertexai.generative_models import GenerativeModel
4 
5 # Load environment variables from .env
6 dotenv.load_dotenv()
7 
8 # Initialize the Composio Toolset
9 composio_toolset = ComposioToolset()
10 
11 # Get GitHub tools that are pre-configured
12 tool = composio_toolset.get_tool(apps=[App.GITHUB])
13 
14 # Initialize the Google AI Gemini model
15 model = GenerativeModel("gemini-1.5-pro", tools=[tool])

Start a Chat Session with the Model

Start Chat Session

1 # Start a chat session
2 chat = model.start_chat()

Execute the Task via Google AI Model

Execute Task

1 # Define task
2 task = "Star a repo composiohq/composio on GitHub"
3 
4 # Send a message to the model
5 response = chat.send_message(task)
6 
7 print("Model response:")
8 print(response)

Handle the Tool Calls

Handle Tool Calls

1 result = composio_toolset.handle_response(response)
2 print("Function call result:")
3 print(result)

Use Specific Actions

Filter Specific Action

1 # To restrict models from executing any actions, filter specific actions 
2 actions = composio_toolset.get_tool(actions=[Action.GITHUB_CREATE_ISSUE])

Use Specific Apps

Filter Specific App

1 # To restrict models from using all tools, filter specific tools 
2 actions = composio_toolset.get_tool(apps=[App.ASANA, App.GITHUB])

Filter apps actions by tags

Filter Actions by Tags

1 actions = composio_toolset.get_tool(apps=[App.ASANA], tags=[Tag.ASANA_TASKS])