USKit

简介

在真实的对话系统中，通常会要求同时支持多种对话技能，比如育儿对话机器人中要求同时支持音乐、故事、古诗、百科等技能。这涉及到多个技能的管理、召回、排序和选择等策略，在具体的实现中，通常由一个统一调度对话中控 US (Unified Scheduler) 来提供这些能力。

USKit 作为 UNIT 的开源对话中控模块，主要负责多个对话技能的接入、调度、排序、选择、回复生成以及多轮对话 session 的管理，无缝对接 UNIT 平台能力，可以方便开发者通过配置快速搭建和定制符合自己业务场景的对话中控，将多种技能整合为统一的对话机器人为用户提供服务。

主要功能

核心概念

在开始之前，我们先定义以下两个概念：

  1. 对话技能 (bot skill)：指某个特定场景下的对话能力，一个对话技能对应一个具体的对话场景，例如天气技能负责提供天气场景下的对话能力。
  2. 对话机器人 (bot service)：由多个对话技能整合而成的统一对话服务，可以同时支持多个对话场景的人机对话，一个对话机器人对应一个具体业务的整体解决方案，比如车载对话机器人、育儿对话机器人等。USKit 提供的正是快速搭建对话机器人的能力。

功能介绍

USKit 作为 UNIT 的开源中控模块，通过配置驱动和内置表达式运算支持，提供了以下能力：

架构

USKit 针对对话中控的能力做了抽象，抽取出不同对话场景中下中控的通用能力，避免不必要的重复开发。同时按照配置化驱动和高扩展性的思想进行整体架构的设计，方便开发者通过配置快速构建和定制适用于特定业务场景的对话中控。USKit的整体架构如下图：

USKit整体架构

系统主要由以下4个核心部分组成：

开发者在使用 USKit 的时候，只需配置上述3个配置文件，即可完成对话中控的搭建，后续策略的变动和升级也通过修改配置文件即可完成，可以让开发者关注对话机器人中控本身的策略逻辑，不需要重复开发框架代码

3个配置文件的详细说明可以参考详细配置说明

安装

从 GitHub 获取 USKit 的源代码:

git clone https://github.com/baidu/unit-uskit.git
cd unit-uskit

USKit 提供了两种安装方式：使用 Docker 构建镜像和手动安装。推荐使用 Docker 进行安装。

Docker 安装

docker build -t unit-uskit .

我们也提供成功编译的 Docker 镜像，可使用如下命令拉取镜像：

docker pull docker.io/shanetian/unit-uskit:v3.0.0

手动安装

1. 安装依赖

USKit 基于以下工具包进行开发：

目前支持 Ubuntu，CentOS 等系统环境，Ubuntu 推荐使用 16.04 或以上版本，CentOS 推荐使用 7 以上版本，编译 USKit 前请确保对应版本的依赖已经正确安装：

Ubuntu
sh deps.sh ubuntu
CentOS
sh deps.sh centos

2. 编译USKit

mkdir _build && cd _build && cmake .. && make

注：编译前请检查 gcc 版本，若版本较低不支持 C++14，则使用 source /opt/rh/devtoolset-7/enable 切换到新安装的 gcc 再编译即可。

如果希望打印更多 Debug 信息，可以在生成执行 cmake 时指定为 Debug，在运行 USKit 的过程中将会打出更详细的 log 信息，方便问题定位：

mkdir _build && cd _build && cmake -DCMAKE_BUILD_TYPE=Debug .. && make

注：打开该选项会影响性能，建议只在开发环境中使用。

USKit 代码目录结构

.
├── CMakeLists.txt
├── README.md
├── _build              # 编译产出，在该目录下修改配置文件以及运行 ./uskit 启动服务
│   ├── conf            # 配置目录
│   │   ├── us          # 对话机器人的配置，每个子目录对应一个具体业务场景下的对话机器人
│   │   │   └── demo    # USKit 教程示例的配置及配置生成工具
│   │   │       ├── options.py           # demo 场景里需要开发者自行配置的选项，如技能 ID，API Key 等
│   │   │       ├── conf_generator.py    # 根据 options.py 里声明的配置，自动生成所需配置文件
│   │   │       └── conf_templates       # 配置模板目录
│   │   ├── gflags.conf                  # USKit 启动参数默认参数配置
│   │   └── us.conf                      # USKit 本身相关的配置，用于指定加载的对话机器人
│   └── uskit           # USKit 主程序
├── conf                # 配置目录，编译成功后会被复制到 _build/conf
├── docs                # 详细文档
├── proto               # protobuf 文件
├── src                 # 源代码
└── third_party         # 第三方依赖目录

使用

USKit 主要通过配置来驱动中控的搭建和定制，使用方只需要配置对应目录下的 backend.conf，rank.conf 以及 flow.conf，然后重启服务即可生效。

新建对话中控

在 USKit 中，新建一个对话场景下的机器人中控，只需在 _build/conf/us 目录下新增对应的目录，例如，新增一个对话机器人 demo_service：

mkdir -p _build/conf/us/demo_service

在该目录下，编写对应的配置文件：

注册对话中控

同时修改 us.conf，注册需要加载的对话机器人，新添如下一行：

load : "demo_service"

启动服务

进入 _build 目录运行 USKit：

./uskit

命令参数说明：

成功启动 USKit 服务后，可以通过 <HOST>:8888/us 发起 HTTP POST 请求，请求体使用 json 格式，请求参数如下：

样例：

{"usid": "demo_service", "logid": "123456", "query": "北京今天天气怎么样", "uuid": "123"}

返回结果同样采用 json 格式，包含以下三个字段：

样例：

{"error_code": 0, "error_msg": "OK", "result": "好的"}

更多文档

FAQ

  1. 执行 deps.sh 的过程中 clone boost 失败导致后续依赖安装失败

    由于 boost 库较大，GitHub 访问较慢导致下载失败。可以在 boost 官网手动下载包至 third_party/ 并编译：

    tar -zxf boost_1_76_0.tar.gz && mv boost_1_76_0 boost
cd boost
mkdir -p _build/output
./bootstrap.sh --prefix=./_build/output
./b2 install

如何贡献

讨论

Copyright and License

USKit is licensed under the Apache License, Version 2.0