diff --git a/.gitignore b/.gitignore index 3dcb474..eeeb8eb 100644 --- a/.gitignore +++ b/.gitignore @@ -8,4 +8,7 @@ tts.wav keys.yml dist/* build/* -*.egg-info \ No newline at end of file +*.egg-info +ros2/build +ros2/install +ros2/log diff --git a/config/hello_world.yml b/config/hello_world.yml index 0e12ef6..74351fc 100644 --- a/config/hello_world.yml +++ b/config/hello_world.yml @@ -2,5 +2,5 @@ module: core config: import: - ravestate_conio - - ravestate_hello_world + - ravestate_wildtalk - ravestate_nlp \ No newline at end of file diff --git a/config/telegram_hello_world.yml b/config/telegram_hello_world.yml index 0476e79..f816d58 100644 --- a/config/telegram_hello_world.yml +++ b/config/telegram_hello_world.yml @@ -13,5 +13,5 @@ config: import: - ravestate_telegramio - ravestate_conio - - ravestate_hello_world + - ravestate_wildtalk - ravestate_nlp \ No newline at end of file diff --git a/modules/ravestate_wildtalk/__init__.py b/modules/ravestate_wildtalk/__init__.py new file mode 100644 index 0000000..eaf04b3 --- /dev/null +++ b/modules/ravestate_wildtalk/__init__.py @@ -0,0 +1,14 @@ + +from ravestate.state import state +from ravestate import registry + +from roboy_parlai import wildtalk +import ravestate_rawio + + +@state(read="rawio:in", write="rawio:out") +def wildtalk_state(ctx): + ctx["rawio:out"] = wildtalk(ctx["rawio:in"]) + + +registry.register(name="wildtalk", states=(wildtalk_state,)) diff --git a/requirements.txt b/requirements.txt index 8999e63..7ea6fb4 100644 --- a/requirements.txt +++ b/requirements.txt @@ -6,5 +6,6 @@ PyAudio Flask python-telegram-bot pyyaml-include +roboy_parlai>=0.1.post3 spacy https://github.com/explosion/spacy-models/releases/download/en_core_web_sm-2.0.0/en_core_web_sm-2.0.0.tar.gz \ No newline at end of file