Processing Tools

1
def gmail_schema_processor(schema: dict) -> dict:
2
    del schema["recipient_email"]
3
    del schema["attachment"]
4
5
processed_send_email_tool = toolset.get_tools(
6
    actions=[Action.GMAIL_SEND_EMAIL],
7
    processors={
8
        "schema": {Action.GMAIL_SEND_EMAIL: gmail_schema_processor},
9
    },
10
)

1
def gmail_preprocessor(inputs: dict) -> dict:
2
    inputs["recipient_email"] = "sam@openai.com"
3
    return inputs
4
5
processed_send_email_tool = toolset.get_tools(
6
    actions=[Action.GMAIL_SEND_EMAIL],
7
    processors={
8
        "schema": {Action.GMAIL_SEND_EMAIL: gmail_schema_processor},
9
        "pre": {Action.GMAIL_SEND_EMAIL: gmail_preprocessor},
10
    },
11
)

1
import tiktoken
2
from composio_openai import ComposioToolset, Action
3
4
def num_tokens_from_string(string: str, encoding_name: str) -> int:
5
    encoding = tiktoken.get_encoding(encoding_name)
6
    num_tokens = len(encoding.encode(string))
7
    return num_tokens
8
9
encoding = tiktoken.encoding_for_model("gpt-4o-mini")
10
toolset = ComposioToolset()
11
12
emails = toolset.execute_action(Action.GMAIL_FETCH_EMAILS, {})
13
print(num_tokens_from_string(str(emails), encoding.name))

>>291344

1
def gmail_postprocessor(result: dict) -> dict:
2
    processed_result = result.copy()
3
    processed_response = []
4
    for email in result["data"]["messages"]:
5
        processed_response.append(
6
            {
7
                "subject": email["subject"],
8
                "sender": email["sender"],
9
            }
10
        )
11
    processed_result["data"] = processed_response
12
    return processed_result
13
14
processed_emails = gmail_postprocessor(emails)
15
print(num_tokens_from_string(str(processed_emails), encoding.name))

> 356

1
[{'data': [{'subject': 'Update on AGI Development',
2
    'sender': 'Greg Brockman <gdb@composio.dev>'},
3
   {'subject': 'Security alert',
4
    'sender': 'Google <no-reply@accounts.google.com>'},
5
   {'subject': 'Hiddens Flaw in Most RAG Systems from ChromaDB, LlamaIndex, and Glean',
6
    'sender': 'Jason Liu <work@jxnl.co>'},
7
   {'subject': 'AGI, Inc. ❇️',
8
    'sender': 'Cerebral Valley <cerebralvalley@mail.beehiiv.com>'}],
9
  'error': None,
10
  'successfull': True,
11
  'successful': True}]

1
from openai import OpenAI
2
from composio_openai import ComposioToolSet
3
4
client = OpenAI()
5
toolset = ComposioToolSet()
6
7
def gmail_postprocessor(result: dict) -> dict:
8
    processed_result = result.copy()
9
    processed_response = []
10
    for email in result["data"]["messages"]:
11
        processed_response.append(
12
            {
13
                "subject": email["subject"],
14
                "sender": email["sender"],
15
            }
16
        )
17
    processed_result["data"] = processed_response
18
    return processed_result
19
20
processed_fetch_emails_tool = toolset.get_tools(
21
    actions=[Action.GMAIL_FETCH_EMAILS],
22
    processors={
23
        "post": {Action.GMAIL_FETCH_EMAILS: gmail_postprocessor},
24
    },
25
)
26
27
messages = [
28
    {"role": "system", "content": "You are an Email Agent"},
29
    {
30
        "role": "user",
31
        "content": "get me my emails and then answer who has sent me an email about agi?",
32
    },
33
]
34
while True:
35
    response = client.chat.completions.create(
36
        model="gpt-4o-mini",
37
        tools=processed_fetch_emails_tool,
38
        messages=messages,
39
    )
40
    result = toolset.handle_tool_calls(response)
41
    if response.choices[0].finish_reason != "tool_calls":
42
        print(response.choices[0].message.content)
43
        break
44
    messages.append(
45
        {
46
            "role": "assistant",
47
            "content": "",
48
            "tool_calls": response.choices[0].message.tool_calls,
49
        }
50
    )
51
    for tool_call in response.choices[0].message.tool_calls:
52
        messages.append(
53
            {
54
                "role": "tool",
55
                "content": str(result),
56
                "tool_call_id": tool_call.id,
57
            }
58
        )

You have received two emails related to AGI:
1. **Email from gdb@composio.dev**  
   - Subject: Update on AGI Development  
2. **Email from Cerebral Valley**  
   - Subject: AGI, Inc. - from MultiOn's Div Garg ❇️  
Would you like to take any specific actions regarding these emails?