エージェントのツール連携用フレームワーク「Composio」を試す

!pip install composio_core composio_openai

!pip freeze | egrep -i "composio|openai"

composio_core==0.3.20
composio_openai==0.3.20
openai==1.35.13

!composio add github

Error: User not logged in, please login using `composio login`

!composio login

Authenticating...
> Redirecting you to the login page
> https://app.composio.dev/?cliKey=XXXXXXXXXXXXXXX
> Enter authentication code: 

> Enter authentication code: XXXXXXXXXXXXXXX
✔ Authenticated successfully!

!composio add github

> Adding integration: Github...

Please authenticate github in the browser and come back here. URL: https://github.com/login/oauth/authorize?client_id=XXXXXXXXXXXXXXXX
⣾ ⚠ Waiting for github authentication...

✔ github added successfully!

import os
from google.colab import userdata
from openai import OpenAI
from composio_openai import ComposioToolSet, App, Action

openai_client = OpenAI(api_key=userdata.get('OPENAI_API_KEY'))
composio_toolset = ComposioToolSet(api_key=userdata.get('COMPOSIO_API_KEY'))

actions = composio_toolset.get_actions(actions=[Action.GITHUB_ACTIVITY_STAR_REPO_FOR_AUTHENTICATED_USER])

MetadataFileNotFound: Metadata file for `GITHUB_ACTIVITY_STAR_REPO_FOR_AUTHENTICATED_USER` not found, Please run `composio apps update` to fix this

!composio apps update

⚠️ Apps does not require update
✔ Tags updated
⚠️ Actions does not require update
⚠️ Triggers does not require update

actions = composio_toolset.get_actions(actions=[Action.GITHUB_ACTIVITY_STAR_REPO_FOR_AUTHENTICATED_USER])

import json

print(json.dumps(actions, ensure_ascii=False, indent=2))

[
  {
    "type": "function",
    "function": {
      "name": "github_activity_star_repo_for_authenticated_user",
      "description": "Note that you'll need to set `Content-Length` to zero when calling out to this endpoint. For more information, see \"[HTTP method](https://docs.github.com/rest/guides/getting-started-with-the-rest-api#http-method).\"",
      "parameters": {
        "properties": {
          "owner": {
            "description": "The account owner of the repository. The name is not case sensitive.",
            "title": "Owner",
            "type": "string"
          },
          "repo": {
            "description": "The name of the repository without the `.git` extension. The name is not case sensitive.",
            "title": "Repo",
            "type": "string"
          }
        },
        "title": "activity_star_repo_for_authenticated_userRequest",
        "type": "object",
        "required": [
          "owner",
          "repo"
        ]
      }
    }
  }
]

# アシスタントの作成
assistant = openai_client.beta.assistants.create(
    name="パーソナルアシスタント",
    instructions="あなたは親切な日本語のアシスタントです。",
    model="gpt-4o",
    tools=actions,
)

# スレッドの作成
thread = openai_client.beta.threads.create()

# スレッドへのメッセージ追加
message = openai_client.beta.threads.messages.create(
    thread_id=thread.id,
    role="user",
    content="GitHubのkun432/kun432レポジトリにスターを付けて。"
)

# runオブジェクトを作成
run = openai_client.beta.threads.runs.create(
    thread_id=thread.id,
    assistant_id=assistant.id
)

response_after_tool_calls = composio_toolset.wait_and_handle_assistant_tool_calls(
    client=openai_client,
    run=run,
    thread=thread,
)

print(response_after_tool_calls)

Run(id='run_BHtwqwd8bKqdQF2Ff9yul1yj', assistant_id='asst_8VndcUHGW5nq6oWUkBt7MMsc', cancelled_at=None, completed_at=1720987583, created_at=1720987524, expires_at=None, failed_at=None, incomplete_details=None, instructions='あなたは親切な日本語のアシスタントです。', last_error=None, max_completion_tokens=None, max_prompt_tokens=None, metadata={}, model='gpt-4o', object='thread.run', parallel_tool_calls=True, required_action=None, response_format='auto', started_at=1720987582, status='completed', thread_id='thread_9AbmPZWUK88fiFJdgMFcK8XF', tool_choice='auto', tools=[FunctionTool(function=FunctionDefinition(name='github_activity_star_repo_for_authenticated_user', description='Note that you\'ll need to set `Content-Length` to zero when calling out to this endpoint. For more information, see "[HTTP method](https://docs.github.com/rest/guides/getting-started-with-the-rest-api#http-method)."', parameters={'properties': {'owner': {'description': 'The account owner of the repository. The name is not case sensitive.', 'title': 'Owner', 'type': 'string'}, 'repo': {'description': 'The name of the repository without the `.git` extension. The name is not case sensitive.', 'title': 'Repo', 'type': 'string'}}, 'title': 'activity_star_repo_for_authenticated_userRequest', 'type': 'object', 'required': ['owner', 'repo']}), type='function')], truncation_strategy=TruncationStrategy(type='auto', last_messages=None), usage=Usage(completion_tokens=45, prompt_tokens=833, total_tokens=878), temperature=1.0, top_p=1.0, tool_resources={})

messages = openai_client.beta.threads.messages.list(thread_id=thread.id)
for message in messages.data:
    print("{}: {}".format(message.role, message.content[0].text.value))

assistant: kun432/kun432レポジトリにスターを付けました。
user: GitHubのkun432/kun432レポジトリにスターを付けて。

import os
from google.colab import userdata
from openai import OpenAI
from composio_openai import ComposioToolSet, App, Action

openai_client = OpenAI(api_key=userdata.get('OPENAI_API_KEY'))
composio_toolset = ComposioToolSet(api_key=userdata.get('COMPOSIO_API_KEY'))

actions = composio_toolset.get_actions(actions=[Action.GITHUB_ACTIVITY_STAR_REPO_FOR_AUTHENTICATED_USER])

messages=[
    {"role": "system", "content": "あなたは親切な日本語のアシスタントです。"},
    {"role": "user", "content": "GitHubのkun432/kun432にスターを付けて。"}
]

response = openai_client.chat.completions.create(
    model="gpt-4o",
    messages=messages,
    tools=actions,
)

print(json.dumps(response.to_dict()["choices"][0]["message"]["tool_calls"], ensure_ascii=False, indent=2))

[
  {
    "id": "call_zW96CkXe6fEbsQDNCRpmzts7",
    "function": {
      "arguments": "{\"owner\":\"kun432\",\"repo\":\"kun432\"}",
      "name": "github_activity_star_repo_for_authenticated_user"
    },
    "type": "function"
  }
]

result = composio_toolset.handle_tool_calls(response=response)
print(result)

[{'execution_details': {'executed': True}, 'response_data': ''}]

actions = composio_toolset.get_actions(actions=[Action.GITHUB_REPO_S_GET_CODE_FREQUENCY_STATS])

messages=[
    {"role": "system", "content": "あなたは親切な日本語のアシスタントです。"},
    {"role": "user", "content": "GitHubのkun432/kun432のコミット頻度を教えて"}
]

response = openai_client.chat.completions.create(
    model="gpt-4o",
    messages=messages,
    tools=actions,
)

result = composio_toolset.handle_tool_calls(response=response)
print(json.dumps(result, indent=2))

[
  {
    "execution_details": {
      "executed": true
    },
    "response_data": [
      [
        1707004800,
        116,
        -18
      ],
      [
        1707609600,
        32,
        -18
      ],
      [
        1708214400,
        3,
        -3
      ],
      [
        1708819200,
        4,
        -4
      ],
      [
        1709424000,
        9,
        -9
      ],
      [
        1710028800,
        10,
        -10
      ],
      [
        1710633600,
        7,
        -7
      ],
      [
        1711238400,
        6,
        -6
      ],
      [
        1711843200,
        5,
        -5
      ],
      [
        1712448000,
        5,
        -5
      ],
      [
        1713052800,
        11,
        -9
      ],
      [
        1713657600,
        11,
        -11
      ],
      [
        1714262400,
        6,
        -6
      ],
      [
        1714867200,
        9,
        -9
      ],
      [
        1715472000,
        6,
        -6
      ],
      [
        1716076800,
        2,
        -2
      ],
      [
        1716681600,
        6,
        -6
      ],
      [
        1717286400,
        6,
        -6
      ],
      [
        1717891200,
        7,
        -7
      ],
      [
        1718496000,
        3,
        -3
      ],
      [
        1719100800,
        6,
        -6
      ],
      [
        1719705600,
        9,
        -9
      ],
      [
        1720310400,
        6,
        -6
      ],
      [
        1720915200,
        0,
        0
      ]
    ]
  }
]

actions = composio_toolset.get_actions(actions=[Action.GITHUB_REPO_S_GET_CODE_FREQUENCY_STATS])

messages=[
    {"role": "system", "content": "あなたは親切な日本語のアシスタントです。"},
    {"role": "user", "content": "GitHubのkun432/kun432のコミット頻度を教えて"}
]

response = openai_client.chat.completions.create(
    model="gpt-4o",
    messages=messages,
    tools=actions,
)

while True:
    if response.choices[0].message.tool_calls:
        messages.append(response.choices[0].message)
        tool_calls = response.choices[0].message.tool_calls
        for tool_call in tool_calls:
            tool_call_id = tool_call.id
            tool_name = tool_call.function.name
            tool_result = composio_toolset.handle_tool_calls(response=response)
            messages.append(
                {
                    "tool_call_id": tool_call.id,
                    "role": "tool",
                    "name": tool_name,
                    "content": str(tool_result),
                }
            )
        response = openai_client.chat.completions.create(
            model="gpt-4o",
            messages=messages,
            tools=actions,
        )
    else:
        break

print(response.choices[0].message.content)

kun432/kun432リポジトリの最近のコミット頻度を以下のように確認しました：

| 週 (Unix time)       | 追加された行数 | 削除された行数 |
|---------------------|----------------|----------------|
| 1707004800          | 116            | -18            |
| 1707609600          | 32             | -18            |
| 1708214400          | 3              | -3             |
| 1708819200          | 4              | -4             |
| 1709424000          | 9              | -9             |
| 1710028800          | 10             | -10            |
| 1710633600          | 7              | -7             |
| 1711238400          | 6              | -6             |
| 1711843200          | 5              | -5             |
| 1712448000          | 5              | -5             |
| 1713052800          | 11             | -9             |
| 1713657600          | 11             | -11            |
| 1714262400          | 6              | -6             |
| 1714867200          | 9              | -9             |
| 1715472000          | 6              | -6             |
| 1716076800          | 2              | -2             |
| 1716681600          | 6              | -6             |
| 1717286400          | 6              | -6             |
| 1717891200          | 7              | -7             |
| 1718496000          | 3              | -3             |
| 1719100800          | 6              | -6             |
| 1719705600          | 9              | -9             |
| 1720310400          | 6              | -6             |
| 1720915200          | 0              | 0              |

このデータは、各週に追加および削除された行数を示しています。

!pip install composio-llamaindex

!pip freeze | egrep -i "llama-|composio"

composio_core==0.3.20
composio_langchain==0.3.20
composio_llamaindex==0.3.20
llama-cloud==0.0.9
llama-index==0.10.55
llama-index-agent-openai==0.2.8
llama-index-cli==0.1.12
llama-index-core==0.10.55
llama-index-embeddings-openai==0.1.10
llama-index-indices-managed-llama-cloud==0.2.5
llama-index-legacy==0.9.48
llama-index-llms-openai==0.1.25
llama-index-multi-modal-llms-openai==0.1.7
llama-index-program-openai==0.1.6
llama-index-question-gen-openai==0.1.3
llama-index-readers-file==0.1.30
llama-index-readers-llama-parse==0.1.6
llama-parse==0.4.7

import os
from google.colab import userdata

os.environ["OPENAI_API_KEY"] = userdata.get('OPENAI_API_KEY')
os.environ["COMPOSIO_API_KEY"] = userdata.get('COMPOSIO_API_KEY')

!composio apps update

from llama_index.llms.openai import OpenAI
from llama_index.core.llms import ChatMessage
from llama_index.core.agent import FunctionCallingAgentWorker

from composio_llamaindex import Action, ComposioToolSet

llm = OpenAI(model="gpt-4o")

composio_toolset = ComposioToolSet()
tools = composio_toolset.get_actions(actions=[Action.GITHUB_REPO_S_GET_CODE_FREQUENCY_STATS]) 

prefix_messages = [
    ChatMessage(
        role="system",
        content=(
            "あなたは、色々なツールが統合された日本語のエージェントです。依頼内容に応じて、様々なツールを活用・実行して、タスクを解決します。"
        ),
    )
]

agent = FunctionCallingAgentWorker(
    tools=tools,
    llm=llm,
    prefix_messages=prefix_messages,
    max_function_calls=10,
    allow_parallel_tool_calls=False,
    verbose=True,
).as_agent()

response = agent.chat("GitHubのkun432/kun432のコミット頻度を教えて")

Added user message to memory: GitHubのkun432/kun432のコミット頻度を教えて
=== Calling Function ===
Calling function: github_repo_s_get_code_frequency_stats with args: {"owner": "kun432", "repo": "kun432"}
=== Function Output ===
{'execution_details': {'executed': True}, 'response_data': [[1707004800, 116, -18], [1707609600, 32, -18], [1708214400, 3, -3], [1708819200, 4, -4], [1709424000, 9, -9], [1710028800, 10, -10], [1710633600, 7, -7], [1711238400, 6, -6], [1711843200, 5, -5], [1712448000, 5, -5], [1713052800, 11, -9], [1713657600, 11, -11], [1714262400, 6, -6], [1714867200, 9, -9], [1715472000, 6, -6], [1716076800, 2, -2], [1716681600, 6, -6], [1717286400, 6, -6], [1717891200, 7, -7], [1718496000, 3, -3], [1719100800, 6, -6], [1719705600, 9, -9], [1720310400, 6, -6], [1720915200, 0, 0]]}
=== LLM Response ===
以下は、GitHubリポジトリ「kun432/kun432」の最近のコミット頻度（週ごとの追加行数と削除行数）です：

| 週の開始日（UNIXタイムスタンプ） | 追加行数 | 削除行数 |
|-------------------------------|--------|--------|
| 1707004800                    | 116    | -18    |
| 1707609600                    | 32     | -18    |
| 1708214400                    | 3      | -3     |
| 1708819200                    | 4      | -4     |
| 1709424000                    | 9      | -9     |
| 1710028800                    | 10     | -10    |
| 1710633600                    | 7      | -7     |
| 1711238400                    | 6      | -6     |
| 1711843200                    | 5      | -5     |
| 1712448000                    | 5      | -5     |
| 1713052800                    | 11     | -9     |
| 1713657600                    | 11     | -11    |
| 1714262400                    | 6      | -6     |
| 1714867200                    | 9      | -9     |
| 1715472000                    | 6      | -6     |
| 1716076800                    | 2      | -2     |
| 1716681600                    | 6      | -6     |
| 1717286400                    | 6      | -6     |
| 1717891200                    | 7      | -7     |
| 1718496000                    | 3      | -3     |
| 1719100800                    | 6      | -6     |
| 1719705600                    | 9      | -9     |
| 1720310400                    | 6      | -6     |
| 1720915200                    | 0      | 0      |

このデータは、各週ごとの追加行数と削除行数を示しています。

from llama_index.llms.openai import OpenAI
from llama_index.agent.openai import OpenAIAgent
from llama_index.llms.openai import OpenAI

from composio_llamaindex import Action, ComposioToolSet

llm = OpenAI(model="gpt-4o")

composio_toolset = ComposioToolSet()
tools = composio_toolset.get_actions(actions=[Action.GITHUB_REPO_S_GET_CODE_FREQUENCY_STATS]) 

agent = OpenAIAgent.from_tools(tools, llm=llm, verbose=True)
response = agent.chat("GitHubのkun432/kun432のコミット頻度を教えて")
print("\n回答: ", response)

Added user message to memory: GitHubのkun432/kun432のコミット頻度を教えて
=== Calling Function ===
Calling function: github_repo_s_get_code_frequency_stats with args: {"owner":"kun432","repo":"kun432"}
Got output: {'execution_details': {'executed': True}, 'response_data': [[1707004800, 116, -18], [1707609600, 32, -18], [1708214400, 3, -3], [1708819200, 4, -4], [1709424000, 9, -9], [1710028800, 10, -10], [1710633600, 7, -7], [1711238400, 6, -6], [1711843200, 5, -5], [1712448000, 5, -5], [1713052800, 11, -9], [1713657600, 11, -11], [1714262400, 6, -6], [1714867200, 9, -9], [1715472000, 6, -6], [1716076800, 2, -2], [1716681600, 6, -6], [1717286400, 6, -6], [1717891200, 7, -7], [1718496000, 3, -3], [1719100800, 6, -6], [1719705600, 9, -9], [1720310400, 6, -6], [1720915200, 0, 0]]}
========================


回答:  kun432/kun432リポジトリの最近のコミット頻度は以下の通りです（週ごと）：

- 1707004800: 追加 116, 削除 18
- 1707609600: 追加 32, 削除 18
- 1708214400: 追加 3, 削除 3
- 1708819200: 追加 4, 削除 4
- 1709424000: 追加 9, 削除 9
- 1710028800: 追加 10, 削除 10
- 1710633600: 追加 7, 削除 7
- 1711238400: 追加 6, 削除 6
- 1711843200: 追加 5, 削除 5
- 1712448000: 追加 5, 削除 5
- 1713052800: 追加 11, 削除 9
- 1713657600: 追加 11, 削除 11
- 1714262400: 追加 6, 削除 6
- 1714867200: 追加 9, 削除 9
- 1715472000: 追加 6, 削除 6
- 1716076800: 追加 2, 削除 2
- 1716681600: 追加 6, 削除 6
- 1717286400: 追加 6, 削除 6
- 1717891200: 追加 7, 削除 7
- 1718496000: 追加 3, 削除 3
- 1719100800: 追加 6, 削除 6
- 1719705600: 追加 9, 削除 9
- 1720310400: 追加 6, 削除 6
- 1720915200: 追加 0, 削除 0

このデータは週ごとの追加と削除の数を示しています。