nonebot-plugin-marshoai/README_TOOLS_EN.md
2024-11-26 21:07:05 +08:00

4.1 KiB
Raw Blame History

🛠️Marsho

Marsho is a simple module loader. It allows to load kits and its function from tools in plugin directory, for AI to use. More information for Function Call, please refr to OpenAI Offical Documentation

✍️ Development Kits

📁 Directory Structure

tools in plugin directory is caller Toolset. It contains many Toolkit, Toolkit is similar with Packages in Python in structure. It need __init__.py document and tools.json definition document in it. They are used to store and define functions.

A directory structure of Toolkit:

tools/ # Toolset Directory
└── marshoai-example/ # Toolkit Directory, Named as Packages' name
    └── __init__.py # Tool Module
    └── tools.json # Function Definition Document

In this directory tree:

  • Toolset Directory is named as marshoat-xxxxx, its name is the name of Toolset. Please follow this naming standard.
  • **Tool Module could contain many callable Sync function, it could have parameters or be parameterless and the return value should be supported by AI model. Generally speaking, the str type is accepted to most model.
  • Function Definition Document is for AI model to know how to call these function.

Function Writing

Let's write a function for getting the weather and one for getting time.

__init__.py
from datetime import datetime

async def get_weather(location: str):
    return f"The temperature of {location} is 114514℃。" #To simulate the return value of weather.

async def get_current_time():
    current_time = datetime.now().strftime("%Y.%m.%d %H:%M:%S")
    time_prompt = f"Now is {current_time}。"
    return time_prompt

In this example, we define two functions, get_weather and get_current_time. The former accepts a str typed parameter. Let AI to know the existence of these funxtions, you shuold write Function Definition Document

tools.json
[
    {
        "type": "function",
        "function": {
            "name": "marshoai-example__get_weather", # Name of this Function Call
            "description": "Get the weather of a specified locatin.", # Description, it need to descripte the usage of this Functin
            "parameters": {   # Define the parameters
                "type": "object",
                "properties": {
                    "location": { # 'location' is the name that _init__.py had defined.
                        "type": "string", # the Type of patameters
                        "description": "City or district. Such as Beijing, Hangzhou, Yuhang District" # Descriptionit need to descripte the type or example of Actual Parameter
                    }
                }
            },
            "required": [ # Define the Required Parameters
                "location"
            ]
        }
    },
    {
        "type": "function",
        "function": {
            "name": "marshoai-example__get_current_time",
            "description": "Get time",
            "parameters": {} # No parameter requried, so it is blanked
        }
    }
]

In this document, we defined tow function. This Function Definition Document will be typed into AI model, for letting AI to know when and how to call these function. The name of Name of this Function Call is specific required. Using weather-getting as an example, the Name of this Function Call, marshoai-example__get_weather, contain these information.

  • marshoai-example is the name of its Toolkit.
  • get_weather is the name of function.
  • Two underscores are used as a sseparator.

Using this Naming Standard, it could be compatible with more APIs in the standard format of OpenAI. So don't use two underscores as the name of Toolkit or Function.

Function Testing

After developing the Tools, start the Bot. There loading information of Toolkit in Nonebot Logs. This is the test ecample:

> marsho What's the weather like in Shenzhen?
> marsho Please tell me the weather in Shimokitazawa, Hangzhou, and Suzhou separately.
> marsho What time is it now?