JavaでAIMLのプログラムを実行するための備忘録

AIチャット

JavaでAIMLのプログラムを実行して、対話ができるようにするための備忘録です。

ライブラリはGoldRenardのProgram-JBを利用しています。

※ これはAIML2.0リファレンス実装です。AIMLは、チャットボットや、ALICE、Mitsuku、English Tutor、The Professor、SUPERなどのモバイル仮想アシスタントを作成するために広く採用されている標準です。

使い方

Mavenの利用ができます。

<repositories>
    <repository>
        <id>jitpack.io</id>
        <url>https://jitpack.io</url>
    </repository>
</repositories>
<dependencies>
    <dependency>
        <groupId>com.github.blue-islands</groupId>
        <artifactId>Program-JB</artifactId>
        <version>1.1.0</version>
    </dependency>
    <dependency>
        <groupId>org.apache.commons</groupId>
        <artifactId>commons-lang3</artifactId>
        <version>3.12.0</version>
    </dependency>
    <dependency>
        <groupId>org.slf4j</groupId>
        <artifactId>slf4j-api</artifactId>
        <version>1.7.31</version>
    </dependency>
    <dependency>
        <groupId>org.slf4j</groupId>
        <artifactId>slf4j-simple</artifactId>
        <version>1.7.31</version>
    </dependency>
</dependencies>

2024/06/13 更新

Program-JBについては、JCenterのリポジトリで公開されていましたが、2022年2月1日で廃止になります。
その為、Program-JBをフォークしたプロジェクトを利用しています。

日本語での実装方法

ベースは英語を対象に作られているため、日本語で利用するためには設定が必要になります。

import org.alicebot.ab.Bot;
import org.alicebot.ab.BotConfiguration;
import org.alicebot.ab.Chat;
import org.alicebot.ab.configuration.LanguageConfiguration;

public class Chatbot {
    public static void main(String[] args) {
        final var defaultResponse = "私にはその答えがありません。";
        final var errorResponse = "私の脳に何か問題があります。";
        final var scheduleError = "そのイベントをスケジュールできません。";
        final var systemFailed = "システムコマンドの実行に失敗しました。";
        final var templateFailed = "テンプレートが失敗しました。";
        final var tooMuchRecursion = "AIMLの再帰が多すぎます。";
        final var tooMuchLooping = "AIMLのループが多すぎます。";

        final var language = LanguageConfiguration.builder()
                .defaultResponse(defaultResponse)
                .errorResponse(errorResponse)
                .scheduleError(scheduleError)
                .systemFailed(systemFailed)
                .templateFailed(templateFailed)
                .tooMuchRecursion(tooMuchRecursion)
                .tooMuchLooping(tooMuchLooping).build();

        final var bot = new Bot(BotConfiguration.builder()
                .name("alice")
                .path("src/main/resources")
                .jpTokenize(true)
                .language(language)
                .build());

        final var chatSession = new Chat(bot);

        final var text = "こんにちは、元気ですか？";
        final var answer = chatSession.multisentenceRespond(text);
        System.out.println(answer);
    }
}

Servletなどで利用する場合は以下のような実装になるかと思います。

import javax.servlet.ServletContext;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.alicebot.ab.Bot;
import org.alicebot.ab.BotConfiguration;
import org.alicebot.ab.Chat;
import org.alicebot.ab.configuration.LanguageConfiguration;
import java.io.IOException;

public class ChatbotServlet extends HttpServlet {
    private Chat chatSession;

    @Override
    public void init() {
        ServletContext context = getServletContext();
        String rootPath = context.getRealPath("/WEB-INF/classes");

        final var defaultResponse = "私にはその答えがありません。";
        final var errorResponse = "私の脳に何か問題があります。";
        final var scheduleError = "そのイベントをスケジュールできません。";
        final var systemFailed = "システムコマンドの実行に失敗しました。";
        final var templateFailed = "テンプレートが失敗しました。";
        final var tooMuchRecursion = "AIMLの再帰が多すぎます。";
        final var tooMuchLooping = "AIMLのループが多すぎます。";

        final var language = LanguageConfiguration.builder()
                .defaultResponse(defaultResponse)
                .errorResponse(errorResponse)
                .scheduleError(scheduleError)
                .systemFailed(systemFailed)
                .templateFailed(templateFailed)
                .tooMuchRecursion(tooMuchRecursion)
                .tooMuchLooping(tooMuchLooping).build();

        final var bot = new Bot(BotConfiguration.builder()
                .name("alice")
                .path(rootPath)
                .jpTokenize(true)
                .language(language)
                .build());

        chatSession = new Chat(bot);
    }

    @Override
    protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws IOException {
        String text = req.getParameter("text");
        String answer = chatSession.multisentenceRespond(text);
        resp.getWriter().write(answer);
    }
}

AIML2.0リファレンス

AIML2.0リファレンス