GamesAI

• GamesAI
  • 安装
  • 使用
  • 配置
    • 1.prefix
    • 2.permission
    • 3.max_history
    • 4.all_ai
    • 5.default_ai
  • 工具与自定义工具
    • 工具
    • 自定义工具
  • 本次更新
    • 1.允许你将提示词文件化
    • 2.加入了错误码自动识别的功能
  • 鸣谢与声明
  • 许可证

想要知道如何实现搜索baidu.com，参考GamesAI源代码中的工具search_minecraft_wiki无疑是最好的选择：

@register_tool(description="搜索Minecraft Wiki以获取相关信息, 请不要使用此方法搜索与Minecraft无关的东西。如果返回了Search results页面, 你可以通过先浏览此页面, 再进行一次精确查询", tr_key="searching_minecraft_wiki", parameters={
    "type": "object",
    "properties": {
        "query": {
            "type": "string",
            "description": "要搜索的内容，例如某个物品、怪物、机制等的名称。"
        }
    },
    "required": ["query"]
})
def search_minecraft_wiki(source: CommandSource, ai_prefix: str, query: str):
    source.reply(f'{ai_prefix}{source.get_server().rtr("games_ai.tools.searching_minecraft_wiki", query=query)}')
    lang = source.get_server().get_mcdr_language()
    if lang == "en_us":
        search_url = f"https://minecraft.wiki/?search={query}"
    else:
        search_url = f"https://zh.minecraft.wiki/?search={query}"
    response = requests.get(search_url)
    if response.status_code == 200:
        return f"一下是搜索内容 {query} 的结果:\n{response.content.decode('utf-8')}"
    else:
        return "无法访问Minecraft Wiki进行搜索"

首先来看工具的注册，其中的description类似于提示词，是必填项，是给AI提供的；其中的tr_key是内部方法，外部编写tools.py时不需要包含此项；而parameters就是AI需要传入的参数，可根据实际情况决定是否填写，其中的properties用于填写所有的传入参数，其中的required项则表示有哪些参数是必填的，注意properties与函数定义中传入参数的位置关系，应该需要一一对应。

例如：

@register_tool(description="搜索百度")

但是，搜索百度肯定需要传入搜索内容，所以：

@register_tool(description="搜索百度", parameters={
    "type": "object",
    "properties": {
        "query": {
            "type": "string",
            "description": "要搜索的内容"
        }
    },
    "required": ["query"]
})

好了，你已经学会如何注册工具了，接下来就可以写函数定义了：

def search_baidu(source: CommandSource, ai_prefix: str, query: str):

上面的代码中，source和ai_prefix项必须编写，因为它们始终会被传入

接下来写函数体：

def search_baidu(source: CommandSource, ai_prefix: str, query: str):
    ...
    return ...

函数体中可以是任何你需要的操作，最后记得一定要return，否则AI将无法正确处理结果

tools.py的完整示例：

import requests
from games_ai.games_ai_tool import register_tool

@register_tool(
    description="使用百度搜索互联网上的信息。当你需要查找实时信息、新闻、百科知识等时使用此工具。",
    parameters={
        "type": "object",
        "properties": {
            "query": {
                "type": "string",
                "description": "要在百度搜索的关键词或问题"
            }
        },
        "required": ["query"]
    }
)
def search_baidu(source, ai_prefix: str, query: str):
    source.reply(f'{ai_prefix}正在搜索百度：{query}...')

    try:
        url = "https://www.baidu.com/s"
        headers = {
            "User-Agent": (
                "Mozilla/5.0 (Windows NT 10.0; Win64; x64) "
                "AppleWebKit/537.36 (KHTML, like Gecko) "
                "Chrome/120.0.0.0 Safari/537.36"
            )
        }
        response = requests.get(url, params={"wd": query}, headers=headers, timeout=10)

        if response.status_code != 200:
            return f"百度搜索失败，HTTP 状态码：{response.status_code}"

        # 简单提取页面文本（去除 HTML 标签）
        import re
        text = response.text
        # 移除 script 和 style 标签内容
        text = re.sub(r'<script[^>]*>.*?</script>', '', text, flags=re.DOTALL | re.IGNORECASE)
        text = re.sub(r'<style[^>]*>.*?</style>', '', text, flags=re.DOTALL | re.IGNORECASE)
        # 移除 HTML 标签
        text = re.sub(r'<[^>]+>', '', text)
        # 合并多余空白
        text = re.sub(r'\s+', ' ', text).strip()

        # 截断过长内容（保留前 3000 字符，适合 AI 上下文）
        max_len = 3000
        if len(text) > max_len:
            text = text[:max_len] + "\n...(内容已截断)"

        return f"百度搜索「{query}」的结果：\n{text}"

    except requests.Timeout:
        return "百度搜索请求超时，请稍后重试"
    except Exception as e:
        return f"百度搜索出错：{str(e)}"