小美 - 企业微信预约机器人

使用PaddleNLP,Wechaty和Rasa制作基于企业微信的智能预约机器人

项目GitHub地址

设想初衷

我经常去一家日式预约制理发店,和老板交谈后发现,他们不希望能够让散客立刻能够预约店长,并且需要能够快速管理所有的会员,并且能够分开不同的会员梯度。我们认为,通过有赞和大众点评并不能控制准入的客人,如果使用企业微信,并对联系人进行标签管理,既可以达到管理会员的需要,也不丢失预约制理发店的特质。另外我发现,理发店的日程管理非常原始,采用纸质管理,并且因为理发店工作人员不多,不一定能够及时提醒顾客到店,会产生丢单的情况。所以我们和理发店老板商量希望帮他们在企业微信上线一个预约机器人,把预约日程记录到企业微信的日历里,并且能够利用微信功能自动提醒。

对话流程设计

BotSociety设计

机器人架构

机器人的逻辑核心采用Rasa框架。不过Rasa框架对中文支持有限,这里我编利用PaddleNLP的Transformer API生成embedding,作为每个句子的特征,整合到Rasa里面,提供中文的意图识别和基于CRF的命名实体识别支持。利用bert-wwm-chinese预训练模型,效果还不错。对接微信方面,我们使用了自研的chat-operator中间件进行对接。我们用的是由句子互动提供的wxwork Wechaty puppet,直接对接企业微信。

时间段识别我们使用了Facebook的duckling进行识别。机器人会根据识别出来的时间段,匹配企业微信日历里面相关时间的日程,找出可以预约的时间,给用户选择。当选择成功之后,会生成一个日程在企业微信日历中,帮店家管理每天的日程。

飞桨使用体验

我们使用了PaddleNLP的Transformer,我觉得整体和HuggingFace的Transformers包差距不是特别大。需要注意把数据转换成Paddle的Tensor才能调用模型(用 paddle.to_tensor() 方法)。如果时间充裕一些的话,我会考虑使用Transformer来做命名实体识别,因为我手头上针对这个场景的数据不够多,所以只能用Rasa提供的CRF算法来做。后面希望给理发店部署使用之后,可以多收集一些数据。

当前遇到的问题和限制

  1. PaddleNLP的Bert模型forward时候返回embedding会出现不一致的情况,导致意图识别的准确率不稳定。可能和我的整合方法有关,希望能和Paddle的工程师沟通,看看能否改进
  2. 当前Wechaty和Rasa之间结合不够紧凑,由于时间关系暂时没有办法编写基于Python Wechaty的Rasa Connector组件。希望比赛后可以加上,配合添加企业微信的消息入口,可以制作用户添加机器人的欢迎语流程,更重要的是可以想办法把企业微信这边的external_id和Wechaty的contact id对上。
  3. 对于美业项目和项目的一些询问细节,需要和理发店的老板进一步沟通,当前只有比较简单的项目和耗时mapping,后续会需要根据实际情况多添加几轮问题,判断整个项目的总耗时(详细设计请见上方对话流程设计)
  4. 预约时间还有一些边界条件需要处理,比如说营业时间限制等。预约时间判断也写得比较仓促,可能会存在一些bug

截图



Logo

学大模型,用大模型上飞桨星河社区!每天8点V100G算力免费领!免费领取ERNIE 4.0 100w Token >>>

更多推荐