Rasa Core实践 报时机器人
创始人
2024-02-25 02:56:05

文章目录

    • 1. 领域 domain
    • 2. 故事 story
      • 用户消息
      • 机器人动作与事件
      • 辅助符号
    • 3. 动作 action
      • 回复动作
      • 表单
      • 默认动作
      • 自定义动作
    • 4. 词槽 slot
      • 词槽和对话行为
      • 词槽类型
      • 词槽映射
    • 5. 策略 policy
    • 6. 端点 endpoints.yml
    • 7. rasa SDK、自定义动作
      • 自定义动作
      • 运行自定义动作
    • 8. rasa 支持的客户端
    • 9. 实战:报时机器人
      • nlu.yml
      • stories.yml
      • domain.yml
      • config.yml
      • endpoints.yml
      • actions.py
      • 测试

learn from https://github.com/Chinese-NLP-book/rasa_chinese_book_code

rasa core 对话记录 和 选择下一个动作

1. 领域 domain

定义了所有信息:

  • 意图、实体、词槽、动作、表单、回复

意图、实体 应该 跟 rasa nlu 中的保持一致

utter_ 开头的回复 表示 渲染同名模板发送给用户

responses:utter_greet:- "你好 {name}!" # {name} 是模板变量

回复 还支持 富文本,指定通道

会话配置:会话过期时间,是否继承历史词槽

2. 故事 story

version: "3.0"
stories:- story: happy pathsteps:- intent: greet- action: utter_greet- story: query timesteps:- intent: query_time- action: action_query_time

必须要有的 key 是 story、steps
steps 表示用户和机器人之间的交互

用户消息

	- intent: inform  # 用户意图entities:- location: "上海" # 实体信息- price: "实惠"

机器人动作与事件

动作: action
返回事件:词槽事件(对词槽的值进行变更)、active_loop 事件(激活or取消激活表单)

辅助符号

  • 检查点符号,checkpoint 减少故事中重复的部分,名字相同的检查点可以互相跳转
    不同的故事之间,可以通过一个尾部,一个首部 相同的 checkpoint 连接成一个新的故事

  • or 语句

stories:
- story:steps:# 上一个step...- action: utter_ask_confirm- or:- intent: affirm- intent: thankyou- action: action_handle_affirmation

大部分相同,仅有其中一个步骤用户的意图不同

3. 动作 action

接受用户输入、对话状态信息,按照业务逻辑处理,并输出改变对话状态的事件和回复消息

回复动作

与 domain 里的 回复 关联在一起

当调用这类动作时,会自动查找回复中的同名的模板并渲染

表单

收集任务所需的所有要素

默认动作

rasa内置的一些默认动作

自定义动作

满足后端交互计算需求,如查数据库、第三方api请求

4. 词槽 slot

词槽必须要有 名字类型

slots:slot_name: # 名字type: text  # 类型influence_conversation: falseinitial_value: "hello"  # 初始值mappings: # 映射- type: from_entityentity: entity_name

词槽和对话行为

设置 influence_conversation bool 选项: 词槽是否影响对话行为

词槽类型

text、bool、category(枚举)、float(需要设置取值范围)、list、any(不影响系统动作预测)

词槽映射

如上mappings 字段,from_entity 表示将读取某个实体(entity指定)的值来赋值词槽

5. 策略 policy

策略负责学习故事,从而预测动作

有一些内置的策略,他们有优先级,除非是专家,不要随意修改优先级

数据增强: 使用 Rasa 命令时,添加 -- augmentation 来设定数据增强的数量

6. 端点 endpoints.yml

定义了 rasa core 和 其他服务进行连接的配置信息

7. rasa SDK、自定义动作

安装 rasa时,默认安装
单独安装 pip install rasa-sdk

自定义动作

class ActionQueryTime(Action):def name(self) -> Text:return "action_query_time"def run(self,dispatcher: CollectingDispatcher,tracker: Tracker,domain: Dict[Text, Any],) -> List[Dict[Text, Any]]:current_time = datetime.now().strftime("%H:%M:%S")dispatcher.utter_message(text=current_time)return []
  • 重写 name() 向服务器申明动作名字
  • 重写 run() 获取当前对话信息
    tracker 对象(对话状态追踪,获取历史实体、词槽等)
    domain 对象
    用户消息对象 dispatcher
    根据这些信息完成业务动作,如想改变对话状态,需要返回事件发送给 rasa服务器,没有的话,返回 []

运行自定义动作

跟rasa一起安装的sdk,rasa run actions
单独安装的 python -m rasa_sdk --actions actions

8. rasa 支持的客户端

支持 Facebook、Rasa Webchat、Chatroom

跟 IM 连接的组件 称为 connector 其负责实现通信协议

rasa支持自定义 连接器,支持同时使用多个连接器连接IM,需要在 credentials.yml 文件中配置如何连接客户端

9. 实战:报时机器人

tree

.
├── actions.py
├── config.yml
├── credentials.yml
├── data
│   ├── nlu.yml
│   └── stories.yml
├── domain.yml
├── endpoints.yml
├── __init__.py
├── media
│   └── demo.png
├── README.md
└── tests└── conversation_tests.md

nlu.yml

version: "3.0"
nlu:- intent: greetexamples: |- 你好- 您好- hello- hi- 喂- 在么- intent: goodbyeexamples: |- 拜拜- 再见- 拜- 退出- 结束- intent: query_timeexamples: |- 现在几点了- 什么时候了- 几点了- 现在什么时候了- 现在的时间- intent: query_dateexamples: |- [今天](date)几号- [今天](date)是几号- [昨天](date)几号- [明天](date)几号- [今天](date)的日期- [今天](date)几号了- [明天](date)的日期- 几号- intent: query_weekdayexamples: |- [今天](date)星期几- [明天](date)星期几- [昨天](date)星期几- [今天](date)是星期几- 星期几

stories.yml

version: "3.0"
stories:- story: happy pathsteps:- intent: greet- action: utter_greet- story: query timesteps:- intent: query_time- action: action_query_time- story: query datesteps:- intent: query_date- action: action_query_date- story: query weekdaysteps:- intent: query_weekday- action: action_query_weekday- story: say goodbyesteps:- intent: goodbye- action: utter_goodbye

domain.yml

version: "3.0"
session_config:session_expiration_time: 60carry_over_slots_to_new_session: true
intents:- greet- goodbye- query_time- query_date- query_weekday
entities:- date
slots:date:type: textinfluence_conversation: falsemappings:- type: from_entityentity: date
responses:utter_greet:- text: 你好,我是 Silly,我可以帮你查询时间、日期和星期几。你可以对我说「现在几点了?」、「今天几号?」或者「明天星期几?」。utter_goodbye:- text: 再见!
actions:- action_query_time- action_query_date- action_query_weekday- utter_goodbye- utter_greet

config.yml

recipe: default.v1
language: zh
pipeline:- name: JiebaTokenizer- name: LanguageModelFeaturizermodel_name: "bert"model_weights: "bert-base-chinese"- name: DIETClassifierepochs: 100tensorboard_log_directory: ./loglearning_rate: 0.001
policies:- name: MemoizationPolicy- name: TEDPolicymax_history: 5epochs: 100- name: RulePolicy

endpoints.yml

action_endpoint:url: "http://localhost:5055/webhook"

actions.py

from typing import Any, Text, Dict, List
from datetime import datetime, timedeltafrom rasa_sdk import Action, Tracker
from rasa_sdk.executor import CollectingDispatcherdef text_date_to_int(text_date):if text_date == "今天":return 0if text_date == "明天":return 1if text_date == "昨天":return -1# in other casereturn Noneweekday_mapping = ["星期一", "星期二", "星期三", "星期四", "星期五", "星期六", "星期天"]def weekday_to_text(weekday):return weekday_mapping[weekday]class ActionQueryTime(Action):def name(self) -> Text:return "action_query_time"def run(self,dispatcher: CollectingDispatcher,tracker: Tracker,domain: Dict[Text, Any],) -> List[Dict[Text, Any]]:current_time = datetime.now().strftime("%H:%M:%S")dispatcher.utter_message(text=current_time)return []class ActionQueryDate(Action):def name(self) -> Text:return "action_query_date"def run(self,dispatcher: CollectingDispatcher,tracker: Tracker,domain: Dict[Text, Any],) -> List[Dict[Text, Any]]:text_date = tracker.get_slot("date") or "今天"int_date = text_date_to_int(text_date)if int_date is not None:delta = timedelta(days=int_date)current_date = datetime.now()target_date = current_date + deltadispatcher.utter_message(text=target_date.strftime("%Y-%m-%d"))else:dispatcher.utter_message(text="系统暂不支持'{}'的日期查询".format(text_date))return []class ActionQueryWeekday(Action):def name(self) -> Text:return "action_query_weekday"def run(self,dispatcher: CollectingDispatcher,tracker: Tracker,domain: Dict[Text, Any],) -> List[Dict[Text, Any]]:text_date = tracker.get_slot("date") or "今天"int_date = text_date_to_int(text_date)if int_date is not None:delta = timedelta(days=int_date)current_date = datetime.now()target_date = current_date + deltadispatcher.utter_message(text=weekday_to_text(target_date.weekday()))else:dispatcher.utter_message(text="系统暂不支持'{}'的星期查询".format(text_date))return []

测试

  • rasa run actions 运行动作服务器
rasa run actions
2022-11-28 09:50:58 INFO     rasa_sdk.endpoint  - Starting action endpoint server...
2022-11-28 09:50:58 INFO     rasa_sdk.executor  - Registered function for 'action_query_time'.
2022-11-28 09:50:58 INFO     rasa_sdk.executor  - Registered function for 'action_query_date'.
2022-11-28 09:50:58 INFO     rasa_sdk.executor  - Registered function for 'action_query_weekday'.
2022-11-28 09:50:58 INFO     rasa_sdk.endpoint  - Action endpoint is up and running on http://0.0.0.0:5055
  • rasa train
  • rasa shell
2022-11-28 21:16:49 INFO     root  - Rasa server is up and running.
Bot loaded. Type a message and press enter (use '/stop' to exit): 
Your input ->  你好呀                                                                               
Building prefix dict from the default dictionary ...
Loading model from cache /tmp/jieba.cache
Loading model cost 1.346 seconds.
Prefix dict has been built successfully.
你好,我是 Silly,我可以帮你查询时间、日期和星期几。你可以对我说「现在几点了?」、「今天几号?」或者「明天星期几?」。
Your input ->  bye                                                                                  
你好,我是 Silly,我可以帮你查询时间、日期和星期几。你可以对我说「现在几点了?」、「今天几号?」或者「明天星期几?」。
Your input ->  拜拜                                                                                 
再见!
Your input ->  现在几点                                                                             
21:18:11
Your input ->  今天是几号                                                                           
2022-11-28
Your input ->  明天是几号                                                                           
2022-11-29
Your input ->  后天是几号                                                                           
系统暂不支持'后天'的日期查询
Your input ->  昨天是几号                                                                           
2022-11-27
Your input ->  今天星期几                                                                           
星期一
Your input ->  明天周几                                                                             
星期二
Your input ->  现在几点了?                                                                         
2022-11-29

修改:
nlu里添加 - [后天](date)的日期
actions.py 添加 if text_date == "后天": return 2

重新训练,测试

Your input ->  后天几号                                                                                                               
2022-11-30
Your input ->  后天星期几                                                                                                             
星期三

相关内容

热门资讯

埃菲尔铁塔在哪 中国仿建埃菲尔... 2019年4月26日,广西南宁市,街头惊现一座巨型山寨版埃菲尔铁塔,高约20米,白色塔身,造型逼真,...
苗族的传统节日 贵州苗族节日有... 【岜沙苗族芦笙节】岜沙,苗语叫“分送”,距从江县城7.5公里,是世界上最崇拜树木并以树为神的枪手部落...
北京的名胜古迹 北京最著名的景... 北京从元代开始,逐渐走上帝国首都的道路,先是成为大辽朝五大首都之一的南京城,随着金灭辽,金代从海陵王...
应用未安装解决办法 平板应用未... ---IT小技术,每天Get一个小技能!一、前言描述苹果IPad2居然不能安装怎么办?与此IPad不...
脚上的穴位图 脚面经络图对应的... 人体穴位作用图解大全更清晰直观的标注了各个人体穴位的作用,包括头部穴位图、胸部穴位图、背部穴位图、胳...
长白山自助游攻略 吉林长白山游... 昨天介绍了西坡的景点详细请看链接:一个人的旅行,据说能看到长白山天池全凭运气,您的运气如何?今日介绍...
世界上最漂亮的人 世界上最漂亮... 此前在某网上,选出了全球265万颜值姣好的女性。从这些数量庞大的女性群体中,人们投票选出了心目中最美...
猫咪吃了塑料袋怎么办 猫咪误食... 你知道吗?塑料袋放久了会长猫哦!要说猫咪对塑料袋的喜爱程度完完全全可以媲美纸箱家里只要一有塑料袋的响...
埃菲尔铁塔在哪 中国仿建埃菲尔... 2019年4月26日,广西南宁市,街头惊现一座巨型山寨版埃菲尔铁塔,高约20米,白色塔身,造型逼真,...
北京的名胜古迹 北京最著名的景... 北京从元代开始,逐渐走上帝国首都的道路,先是成为大辽朝五大首都之一的南京城,随着金灭辽,金代从海陵王...
苗族的传统节日 贵州苗族节日有... 【岜沙苗族芦笙节】岜沙,苗语叫“分送”,距从江县城7.5公里,是世界上最崇拜树木并以树为神的枪手部落...
应用未安装解决办法 平板应用未... ---IT小技术,每天Get一个小技能!一、前言描述苹果IPad2居然不能安装怎么办?与此IPad不...
长白山自助游攻略 吉林长白山游... 昨天介绍了西坡的景点详细请看链接:一个人的旅行,据说能看到长白山天池全凭运气,您的运气如何?今日介绍...