rasa-pro 3.12.0.dev1__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of rasa-pro might be problematic. Click here for more details.
- README.md +41 -0
- rasa/__init__.py +9 -0
- rasa/__main__.py +177 -0
- rasa/anonymization/__init__.py +2 -0
- rasa/anonymization/anonymisation_rule_yaml_reader.py +91 -0
- rasa/anonymization/anonymization_pipeline.py +286 -0
- rasa/anonymization/anonymization_rule_executor.py +260 -0
- rasa/anonymization/anonymization_rule_orchestrator.py +120 -0
- rasa/anonymization/schemas/config.yml +47 -0
- rasa/anonymization/utils.py +118 -0
- rasa/api.py +160 -0
- rasa/cli/__init__.py +5 -0
- rasa/cli/arguments/__init__.py +0 -0
- rasa/cli/arguments/data.py +106 -0
- rasa/cli/arguments/default_arguments.py +207 -0
- rasa/cli/arguments/evaluate.py +65 -0
- rasa/cli/arguments/export.py +51 -0
- rasa/cli/arguments/interactive.py +74 -0
- rasa/cli/arguments/run.py +219 -0
- rasa/cli/arguments/shell.py +17 -0
- rasa/cli/arguments/test.py +211 -0
- rasa/cli/arguments/train.py +279 -0
- rasa/cli/arguments/visualize.py +34 -0
- rasa/cli/arguments/x.py +30 -0
- rasa/cli/data.py +354 -0
- rasa/cli/dialogue_understanding_test.py +251 -0
- rasa/cli/e2e_test.py +259 -0
- rasa/cli/evaluate.py +222 -0
- rasa/cli/export.py +250 -0
- rasa/cli/inspect.py +75 -0
- rasa/cli/interactive.py +166 -0
- rasa/cli/license.py +65 -0
- rasa/cli/llm_fine_tuning.py +403 -0
- rasa/cli/markers.py +78 -0
- rasa/cli/project_templates/__init__.py +0 -0
- rasa/cli/project_templates/calm/actions/__init__.py +0 -0
- rasa/cli/project_templates/calm/actions/action_template.py +27 -0
- rasa/cli/project_templates/calm/actions/add_contact.py +30 -0
- rasa/cli/project_templates/calm/actions/db.py +57 -0
- rasa/cli/project_templates/calm/actions/list_contacts.py +22 -0
- rasa/cli/project_templates/calm/actions/remove_contact.py +35 -0
- rasa/cli/project_templates/calm/config.yml +10 -0
- rasa/cli/project_templates/calm/credentials.yml +33 -0
- rasa/cli/project_templates/calm/data/flows/add_contact.yml +31 -0
- rasa/cli/project_templates/calm/data/flows/list_contacts.yml +14 -0
- rasa/cli/project_templates/calm/data/flows/remove_contact.yml +29 -0
- rasa/cli/project_templates/calm/db/contacts.json +10 -0
- rasa/cli/project_templates/calm/domain/add_contact.yml +39 -0
- rasa/cli/project_templates/calm/domain/list_contacts.yml +17 -0
- rasa/cli/project_templates/calm/domain/remove_contact.yml +38 -0
- rasa/cli/project_templates/calm/domain/shared.yml +10 -0
- rasa/cli/project_templates/calm/e2e_tests/cancelations/user_cancels_during_a_correction.yml +16 -0
- rasa/cli/project_templates/calm/e2e_tests/cancelations/user_changes_mind_on_a_whim.yml +7 -0
- rasa/cli/project_templates/calm/e2e_tests/corrections/user_corrects_contact_handle.yml +20 -0
- rasa/cli/project_templates/calm/e2e_tests/corrections/user_corrects_contact_name.yml +19 -0
- rasa/cli/project_templates/calm/e2e_tests/happy_paths/user_adds_contact_to_their_list.yml +15 -0
- rasa/cli/project_templates/calm/e2e_tests/happy_paths/user_lists_contacts.yml +5 -0
- rasa/cli/project_templates/calm/e2e_tests/happy_paths/user_removes_contact.yml +11 -0
- rasa/cli/project_templates/calm/e2e_tests/happy_paths/user_removes_contact_from_list.yml +12 -0
- rasa/cli/project_templates/calm/endpoints.yml +58 -0
- rasa/cli/project_templates/default/actions/__init__.py +0 -0
- rasa/cli/project_templates/default/actions/actions.py +27 -0
- rasa/cli/project_templates/default/config.yml +44 -0
- rasa/cli/project_templates/default/credentials.yml +33 -0
- rasa/cli/project_templates/default/data/nlu.yml +91 -0
- rasa/cli/project_templates/default/data/rules.yml +13 -0
- rasa/cli/project_templates/default/data/stories.yml +30 -0
- rasa/cli/project_templates/default/domain.yml +34 -0
- rasa/cli/project_templates/default/endpoints.yml +42 -0
- rasa/cli/project_templates/default/tests/test_stories.yml +91 -0
- rasa/cli/project_templates/tutorial/actions/__init__.py +0 -0
- rasa/cli/project_templates/tutorial/actions/actions.py +22 -0
- rasa/cli/project_templates/tutorial/config.yml +12 -0
- rasa/cli/project_templates/tutorial/credentials.yml +33 -0
- rasa/cli/project_templates/tutorial/data/flows.yml +8 -0
- rasa/cli/project_templates/tutorial/data/patterns.yml +11 -0
- rasa/cli/project_templates/tutorial/domain.yml +35 -0
- rasa/cli/project_templates/tutorial/endpoints.yml +55 -0
- rasa/cli/run.py +143 -0
- rasa/cli/scaffold.py +273 -0
- rasa/cli/shell.py +141 -0
- rasa/cli/studio/__init__.py +0 -0
- rasa/cli/studio/download.py +62 -0
- rasa/cli/studio/studio.py +296 -0
- rasa/cli/studio/train.py +59 -0
- rasa/cli/studio/upload.py +62 -0
- rasa/cli/telemetry.py +102 -0
- rasa/cli/test.py +280 -0
- rasa/cli/train.py +278 -0
- rasa/cli/utils.py +484 -0
- rasa/cli/visualize.py +40 -0
- rasa/cli/x.py +206 -0
- rasa/constants.py +45 -0
- rasa/core/__init__.py +17 -0
- rasa/core/actions/__init__.py +0 -0
- rasa/core/actions/action.py +1318 -0
- rasa/core/actions/action_clean_stack.py +59 -0
- rasa/core/actions/action_exceptions.py +24 -0
- rasa/core/actions/action_hangup.py +29 -0
- rasa/core/actions/action_repeat_bot_messages.py +89 -0
- rasa/core/actions/action_run_slot_rejections.py +210 -0
- rasa/core/actions/action_trigger_chitchat.py +31 -0
- rasa/core/actions/action_trigger_flow.py +109 -0
- rasa/core/actions/action_trigger_search.py +31 -0
- rasa/core/actions/constants.py +5 -0
- rasa/core/actions/custom_action_executor.py +191 -0
- rasa/core/actions/direct_custom_actions_executor.py +109 -0
- rasa/core/actions/e2e_stub_custom_action_executor.py +72 -0
- rasa/core/actions/forms.py +741 -0
- rasa/core/actions/grpc_custom_action_executor.py +251 -0
- rasa/core/actions/http_custom_action_executor.py +145 -0
- rasa/core/actions/loops.py +114 -0
- rasa/core/actions/two_stage_fallback.py +186 -0
- rasa/core/agent.py +559 -0
- rasa/core/auth_retry_tracker_store.py +122 -0
- rasa/core/brokers/__init__.py +0 -0
- rasa/core/brokers/broker.py +126 -0
- rasa/core/brokers/file.py +58 -0
- rasa/core/brokers/kafka.py +324 -0
- rasa/core/brokers/pika.py +388 -0
- rasa/core/brokers/sql.py +86 -0
- rasa/core/channels/__init__.py +61 -0
- rasa/core/channels/botframework.py +338 -0
- rasa/core/channels/callback.py +84 -0
- rasa/core/channels/channel.py +456 -0
- rasa/core/channels/console.py +241 -0
- rasa/core/channels/development_inspector.py +197 -0
- rasa/core/channels/facebook.py +419 -0
- rasa/core/channels/hangouts.py +329 -0
- rasa/core/channels/inspector/.eslintrc.cjs +25 -0
- rasa/core/channels/inspector/.gitignore +23 -0
- rasa/core/channels/inspector/README.md +54 -0
- rasa/core/channels/inspector/assets/favicon.ico +0 -0
- rasa/core/channels/inspector/assets/rasa-chat.js +2 -0
- rasa/core/channels/inspector/custom.d.ts +3 -0
- rasa/core/channels/inspector/dist/assets/arc-861ddd57.js +1 -0
- rasa/core/channels/inspector/dist/assets/array-9f3ba611.js +1 -0
- rasa/core/channels/inspector/dist/assets/c4Diagram-d0fbc5ce-921f02db.js +10 -0
- rasa/core/channels/inspector/dist/assets/classDiagram-936ed81e-b436c4f8.js +2 -0
- rasa/core/channels/inspector/dist/assets/classDiagram-v2-c3cb15f1-511a23cb.js +2 -0
- rasa/core/channels/inspector/dist/assets/createText-62fc7601-ef476ecd.js +7 -0
- rasa/core/channels/inspector/dist/assets/edges-f2ad444c-f1878e0a.js +4 -0
- rasa/core/channels/inspector/dist/assets/erDiagram-9d236eb7-fac75185.js +51 -0
- rasa/core/channels/inspector/dist/assets/flowDb-1972c806-201c5bbc.js +6 -0
- rasa/core/channels/inspector/dist/assets/flowDiagram-7ea5b25a-f904ae41.js +4 -0
- rasa/core/channels/inspector/dist/assets/flowDiagram-v2-855bc5b3-b080d6f2.js +1 -0
- rasa/core/channels/inspector/dist/assets/flowchart-elk-definition-abe16c3d-1813da66.js +139 -0
- rasa/core/channels/inspector/dist/assets/ganttDiagram-9b5ea136-872af172.js +266 -0
- rasa/core/channels/inspector/dist/assets/gitGraphDiagram-99d0ae7c-34a0af5a.js +70 -0
- rasa/core/channels/inspector/dist/assets/ibm-plex-mono-v4-latin-regular-128cfa44.ttf +0 -0
- rasa/core/channels/inspector/dist/assets/ibm-plex-mono-v4-latin-regular-21dbcb97.woff +0 -0
- rasa/core/channels/inspector/dist/assets/ibm-plex-mono-v4-latin-regular-222b5e26.svg +329 -0
- rasa/core/channels/inspector/dist/assets/ibm-plex-mono-v4-latin-regular-9ad89b2a.woff2 +0 -0
- rasa/core/channels/inspector/dist/assets/index-2c4b9a3b-42ba3e3d.js +1 -0
- rasa/core/channels/inspector/dist/assets/index-37817b51.js +1317 -0
- rasa/core/channels/inspector/dist/assets/index-3ee28881.css +1 -0
- rasa/core/channels/inspector/dist/assets/infoDiagram-736b4530-6b731386.js +7 -0
- rasa/core/channels/inspector/dist/assets/init-77b53fdd.js +1 -0
- rasa/core/channels/inspector/dist/assets/journeyDiagram-df861f2b-e8579ac6.js +139 -0
- rasa/core/channels/inspector/dist/assets/lato-v14-latin-700-60c05ee4.woff +0 -0
- rasa/core/channels/inspector/dist/assets/lato-v14-latin-700-8335d9b8.svg +438 -0
- rasa/core/channels/inspector/dist/assets/lato-v14-latin-700-9cc39c75.ttf +0 -0
- rasa/core/channels/inspector/dist/assets/lato-v14-latin-700-ead13ccf.woff2 +0 -0
- rasa/core/channels/inspector/dist/assets/lato-v14-latin-regular-16705655.woff2 +0 -0
- rasa/core/channels/inspector/dist/assets/lato-v14-latin-regular-5aeb07f9.woff +0 -0
- rasa/core/channels/inspector/dist/assets/lato-v14-latin-regular-9c459044.ttf +0 -0
- rasa/core/channels/inspector/dist/assets/lato-v14-latin-regular-9e2898a4.svg +435 -0
- rasa/core/channels/inspector/dist/assets/layout-89e6403a.js +1 -0
- rasa/core/channels/inspector/dist/assets/line-dc73d3fc.js +1 -0
- rasa/core/channels/inspector/dist/assets/linear-f5b1d2bc.js +1 -0
- rasa/core/channels/inspector/dist/assets/mindmap-definition-beec6740-82cb74fa.js +109 -0
- rasa/core/channels/inspector/dist/assets/ordinal-ba9b4969.js +1 -0
- rasa/core/channels/inspector/dist/assets/path-53f90ab3.js +1 -0
- rasa/core/channels/inspector/dist/assets/pieDiagram-dbbf0591-bdf5f29b.js +35 -0
- rasa/core/channels/inspector/dist/assets/quadrantDiagram-4d7f4fd6-c7a0cbe4.js +7 -0
- rasa/core/channels/inspector/dist/assets/requirementDiagram-6fc4c22a-7ec5410f.js +52 -0
- rasa/core/channels/inspector/dist/assets/sankeyDiagram-8f13d901-caee5554.js +8 -0
- rasa/core/channels/inspector/dist/assets/sequenceDiagram-b655622a-2935f8db.js +122 -0
- rasa/core/channels/inspector/dist/assets/stateDiagram-59f0c015-8f5d9693.js +1 -0
- rasa/core/channels/inspector/dist/assets/stateDiagram-v2-2b26beab-d565d1de.js +1 -0
- rasa/core/channels/inspector/dist/assets/styles-080da4f6-75ad421d.js +110 -0
- rasa/core/channels/inspector/dist/assets/styles-3dcbcfbf-7e764226.js +159 -0
- rasa/core/channels/inspector/dist/assets/styles-9c745c82-7a4e0e61.js +207 -0
- rasa/core/channels/inspector/dist/assets/svgDrawCommon-4835440b-4019d1bf.js +1 -0
- rasa/core/channels/inspector/dist/assets/timeline-definition-5b62e21b-01ea12df.js +61 -0
- rasa/core/channels/inspector/dist/assets/xychartDiagram-2b33534f-89407137.js +7 -0
- rasa/core/channels/inspector/dist/index.html +42 -0
- rasa/core/channels/inspector/index.html +40 -0
- rasa/core/channels/inspector/jest.config.ts +13 -0
- rasa/core/channels/inspector/package.json +52 -0
- rasa/core/channels/inspector/setupTests.ts +2 -0
- rasa/core/channels/inspector/src/App.tsx +220 -0
- rasa/core/channels/inspector/src/components/Chat.tsx +95 -0
- rasa/core/channels/inspector/src/components/DiagramFlow.tsx +108 -0
- rasa/core/channels/inspector/src/components/DialogueInformation.tsx +187 -0
- rasa/core/channels/inspector/src/components/DialogueStack.tsx +136 -0
- rasa/core/channels/inspector/src/components/ExpandIcon.tsx +16 -0
- rasa/core/channels/inspector/src/components/FullscreenButton.tsx +45 -0
- rasa/core/channels/inspector/src/components/LoadingSpinner.tsx +22 -0
- rasa/core/channels/inspector/src/components/NoActiveFlow.tsx +21 -0
- rasa/core/channels/inspector/src/components/RasaLogo.tsx +32 -0
- rasa/core/channels/inspector/src/components/SaraDiagrams.tsx +39 -0
- rasa/core/channels/inspector/src/components/Slots.tsx +91 -0
- rasa/core/channels/inspector/src/components/Welcome.tsx +54 -0
- rasa/core/channels/inspector/src/helpers/audiostream.ts +191 -0
- rasa/core/channels/inspector/src/helpers/formatters.test.ts +392 -0
- rasa/core/channels/inspector/src/helpers/formatters.ts +306 -0
- rasa/core/channels/inspector/src/helpers/utils.ts +127 -0
- rasa/core/channels/inspector/src/main.tsx +13 -0
- rasa/core/channels/inspector/src/theme/Button/Button.ts +29 -0
- rasa/core/channels/inspector/src/theme/Heading/Heading.ts +31 -0
- rasa/core/channels/inspector/src/theme/Input/Input.ts +27 -0
- rasa/core/channels/inspector/src/theme/Link/Link.ts +10 -0
- rasa/core/channels/inspector/src/theme/Modal/Modal.ts +47 -0
- rasa/core/channels/inspector/src/theme/Table/Table.tsx +38 -0
- rasa/core/channels/inspector/src/theme/Tooltip/Tooltip.ts +12 -0
- rasa/core/channels/inspector/src/theme/base/breakpoints.ts +8 -0
- rasa/core/channels/inspector/src/theme/base/colors.ts +88 -0
- rasa/core/channels/inspector/src/theme/base/fonts/fontFaces.css +29 -0
- rasa/core/channels/inspector/src/theme/base/fonts/ibm-plex-mono-v4-latin/ibm-plex-mono-v4-latin-regular.eot +0 -0
- rasa/core/channels/inspector/src/theme/base/fonts/ibm-plex-mono-v4-latin/ibm-plex-mono-v4-latin-regular.svg +329 -0
- rasa/core/channels/inspector/src/theme/base/fonts/ibm-plex-mono-v4-latin/ibm-plex-mono-v4-latin-regular.ttf +0 -0
- rasa/core/channels/inspector/src/theme/base/fonts/ibm-plex-mono-v4-latin/ibm-plex-mono-v4-latin-regular.woff +0 -0
- rasa/core/channels/inspector/src/theme/base/fonts/ibm-plex-mono-v4-latin/ibm-plex-mono-v4-latin-regular.woff2 +0 -0
- rasa/core/channels/inspector/src/theme/base/fonts/lato-v14-latin/lato-v14-latin-700.eot +0 -0
- rasa/core/channels/inspector/src/theme/base/fonts/lato-v14-latin/lato-v14-latin-700.svg +438 -0
- rasa/core/channels/inspector/src/theme/base/fonts/lato-v14-latin/lato-v14-latin-700.ttf +0 -0
- rasa/core/channels/inspector/src/theme/base/fonts/lato-v14-latin/lato-v14-latin-700.woff +0 -0
- rasa/core/channels/inspector/src/theme/base/fonts/lato-v14-latin/lato-v14-latin-700.woff2 +0 -0
- rasa/core/channels/inspector/src/theme/base/fonts/lato-v14-latin/lato-v14-latin-regular.eot +0 -0
- rasa/core/channels/inspector/src/theme/base/fonts/lato-v14-latin/lato-v14-latin-regular.svg +435 -0
- rasa/core/channels/inspector/src/theme/base/fonts/lato-v14-latin/lato-v14-latin-regular.ttf +0 -0
- rasa/core/channels/inspector/src/theme/base/fonts/lato-v14-latin/lato-v14-latin-regular.woff +0 -0
- rasa/core/channels/inspector/src/theme/base/fonts/lato-v14-latin/lato-v14-latin-regular.woff2 +0 -0
- rasa/core/channels/inspector/src/theme/base/radii.ts +9 -0
- rasa/core/channels/inspector/src/theme/base/shadows.ts +7 -0
- rasa/core/channels/inspector/src/theme/base/sizes.ts +7 -0
- rasa/core/channels/inspector/src/theme/base/space.ts +15 -0
- rasa/core/channels/inspector/src/theme/base/styles.ts +13 -0
- rasa/core/channels/inspector/src/theme/base/typography.ts +24 -0
- rasa/core/channels/inspector/src/theme/base/zIndices.ts +19 -0
- rasa/core/channels/inspector/src/theme/index.ts +101 -0
- rasa/core/channels/inspector/src/types.ts +84 -0
- rasa/core/channels/inspector/src/vite-env.d.ts +1 -0
- rasa/core/channels/inspector/tests/__mocks__/fileMock.ts +1 -0
- rasa/core/channels/inspector/tests/__mocks__/matchMedia.ts +16 -0
- rasa/core/channels/inspector/tests/__mocks__/styleMock.ts +1 -0
- rasa/core/channels/inspector/tests/renderWithProviders.tsx +14 -0
- rasa/core/channels/inspector/tsconfig.json +26 -0
- rasa/core/channels/inspector/tsconfig.node.json +10 -0
- rasa/core/channels/inspector/vite.config.ts +8 -0
- rasa/core/channels/inspector/yarn.lock +6249 -0
- rasa/core/channels/mattermost.py +229 -0
- rasa/core/channels/rasa_chat.py +126 -0
- rasa/core/channels/rest.py +230 -0
- rasa/core/channels/rocketchat.py +174 -0
- rasa/core/channels/slack.py +620 -0
- rasa/core/channels/socketio.py +302 -0
- rasa/core/channels/telegram.py +298 -0
- rasa/core/channels/twilio.py +169 -0
- rasa/core/channels/vier_cvg.py +374 -0
- rasa/core/channels/voice_ready/__init__.py +0 -0
- rasa/core/channels/voice_ready/audiocodes.py +501 -0
- rasa/core/channels/voice_ready/jambonz.py +121 -0
- rasa/core/channels/voice_ready/jambonz_protocol.py +396 -0
- rasa/core/channels/voice_ready/twilio_voice.py +403 -0
- rasa/core/channels/voice_ready/utils.py +37 -0
- rasa/core/channels/voice_stream/__init__.py +0 -0
- rasa/core/channels/voice_stream/asr/__init__.py +0 -0
- rasa/core/channels/voice_stream/asr/asr_engine.py +89 -0
- rasa/core/channels/voice_stream/asr/asr_event.py +18 -0
- rasa/core/channels/voice_stream/asr/azure.py +130 -0
- rasa/core/channels/voice_stream/asr/deepgram.py +90 -0
- rasa/core/channels/voice_stream/audio_bytes.py +8 -0
- rasa/core/channels/voice_stream/browser_audio.py +107 -0
- rasa/core/channels/voice_stream/call_state.py +23 -0
- rasa/core/channels/voice_stream/tts/__init__.py +0 -0
- rasa/core/channels/voice_stream/tts/azure.py +106 -0
- rasa/core/channels/voice_stream/tts/cartesia.py +118 -0
- rasa/core/channels/voice_stream/tts/tts_cache.py +27 -0
- rasa/core/channels/voice_stream/tts/tts_engine.py +58 -0
- rasa/core/channels/voice_stream/twilio_media_streams.py +173 -0
- rasa/core/channels/voice_stream/util.py +57 -0
- rasa/core/channels/voice_stream/voice_channel.py +427 -0
- rasa/core/channels/webexteams.py +134 -0
- rasa/core/concurrent_lock_store.py +210 -0
- rasa/core/constants.py +112 -0
- rasa/core/evaluation/__init__.py +0 -0
- rasa/core/evaluation/marker.py +267 -0
- rasa/core/evaluation/marker_base.py +923 -0
- rasa/core/evaluation/marker_stats.py +293 -0
- rasa/core/evaluation/marker_tracker_loader.py +103 -0
- rasa/core/exceptions.py +29 -0
- rasa/core/exporter.py +284 -0
- rasa/core/featurizers/__init__.py +0 -0
- rasa/core/featurizers/precomputation.py +410 -0
- rasa/core/featurizers/single_state_featurizer.py +421 -0
- rasa/core/featurizers/tracker_featurizers.py +1262 -0
- rasa/core/http_interpreter.py +89 -0
- rasa/core/information_retrieval/__init__.py +7 -0
- rasa/core/information_retrieval/faiss.py +124 -0
- rasa/core/information_retrieval/information_retrieval.py +137 -0
- rasa/core/information_retrieval/milvus.py +59 -0
- rasa/core/information_retrieval/qdrant.py +96 -0
- rasa/core/jobs.py +63 -0
- rasa/core/lock.py +139 -0
- rasa/core/lock_store.py +343 -0
- rasa/core/migrate.py +403 -0
- rasa/core/nlg/__init__.py +3 -0
- rasa/core/nlg/callback.py +146 -0
- rasa/core/nlg/contextual_response_rephraser.py +320 -0
- rasa/core/nlg/generator.py +230 -0
- rasa/core/nlg/interpolator.py +143 -0
- rasa/core/nlg/response.py +155 -0
- rasa/core/nlg/summarize.py +70 -0
- rasa/core/persistor.py +538 -0
- rasa/core/policies/__init__.py +0 -0
- rasa/core/policies/ensemble.py +329 -0
- rasa/core/policies/enterprise_search_policy.py +905 -0
- rasa/core/policies/enterprise_search_prompt_template.jinja2 +25 -0
- rasa/core/policies/enterprise_search_prompt_with_citation_template.jinja2 +60 -0
- rasa/core/policies/flow_policy.py +205 -0
- rasa/core/policies/flows/__init__.py +0 -0
- rasa/core/policies/flows/flow_exceptions.py +44 -0
- rasa/core/policies/flows/flow_executor.py +754 -0
- rasa/core/policies/flows/flow_step_result.py +43 -0
- rasa/core/policies/intentless_policy.py +1031 -0
- rasa/core/policies/intentless_prompt_template.jinja2 +22 -0
- rasa/core/policies/memoization.py +538 -0
- rasa/core/policies/policy.py +725 -0
- rasa/core/policies/rule_policy.py +1273 -0
- rasa/core/policies/ted_policy.py +2169 -0
- rasa/core/policies/unexpected_intent_policy.py +1022 -0
- rasa/core/processor.py +1465 -0
- rasa/core/run.py +342 -0
- rasa/core/secrets_manager/__init__.py +0 -0
- rasa/core/secrets_manager/constants.py +36 -0
- rasa/core/secrets_manager/endpoints.py +391 -0
- rasa/core/secrets_manager/factory.py +241 -0
- rasa/core/secrets_manager/secret_manager.py +262 -0
- rasa/core/secrets_manager/vault.py +584 -0
- rasa/core/test.py +1335 -0
- rasa/core/tracker_store.py +1703 -0
- rasa/core/train.py +105 -0
- rasa/core/training/__init__.py +89 -0
- rasa/core/training/converters/__init__.py +0 -0
- rasa/core/training/converters/responses_prefix_converter.py +119 -0
- rasa/core/training/interactive.py +1744 -0
- rasa/core/training/story_conflict.py +381 -0
- rasa/core/training/training.py +93 -0
- rasa/core/utils.py +366 -0
- rasa/core/visualize.py +70 -0
- rasa/dialogue_understanding/__init__.py +0 -0
- rasa/dialogue_understanding/coexistence/__init__.py +0 -0
- rasa/dialogue_understanding/coexistence/constants.py +4 -0
- rasa/dialogue_understanding/coexistence/intent_based_router.py +196 -0
- rasa/dialogue_understanding/coexistence/llm_based_router.py +327 -0
- rasa/dialogue_understanding/coexistence/router_template.jinja2 +12 -0
- rasa/dialogue_understanding/commands/__init__.py +61 -0
- rasa/dialogue_understanding/commands/can_not_handle_command.py +70 -0
- rasa/dialogue_understanding/commands/cancel_flow_command.py +125 -0
- rasa/dialogue_understanding/commands/change_flow_command.py +44 -0
- rasa/dialogue_understanding/commands/chit_chat_answer_command.py +57 -0
- rasa/dialogue_understanding/commands/clarify_command.py +86 -0
- rasa/dialogue_understanding/commands/command.py +85 -0
- rasa/dialogue_understanding/commands/correct_slots_command.py +297 -0
- rasa/dialogue_understanding/commands/error_command.py +79 -0
- rasa/dialogue_understanding/commands/free_form_answer_command.py +9 -0
- rasa/dialogue_understanding/commands/handle_code_change_command.py +73 -0
- rasa/dialogue_understanding/commands/human_handoff_command.py +66 -0
- rasa/dialogue_understanding/commands/knowledge_answer_command.py +57 -0
- rasa/dialogue_understanding/commands/noop_command.py +54 -0
- rasa/dialogue_understanding/commands/repeat_bot_messages_command.py +60 -0
- rasa/dialogue_understanding/commands/restart_command.py +58 -0
- rasa/dialogue_understanding/commands/session_end_command.py +61 -0
- rasa/dialogue_understanding/commands/session_start_command.py +59 -0
- rasa/dialogue_understanding/commands/set_slot_command.py +160 -0
- rasa/dialogue_understanding/commands/skip_question_command.py +75 -0
- rasa/dialogue_understanding/commands/start_flow_command.py +107 -0
- rasa/dialogue_understanding/commands/user_silence_command.py +59 -0
- rasa/dialogue_understanding/commands/utils.py +45 -0
- rasa/dialogue_understanding/generator/__init__.py +21 -0
- rasa/dialogue_understanding/generator/command_generator.py +464 -0
- rasa/dialogue_understanding/generator/constants.py +27 -0
- rasa/dialogue_understanding/generator/flow_document_template.jinja2 +4 -0
- rasa/dialogue_understanding/generator/flow_retrieval.py +466 -0
- rasa/dialogue_understanding/generator/llm_based_command_generator.py +500 -0
- rasa/dialogue_understanding/generator/llm_command_generator.py +67 -0
- rasa/dialogue_understanding/generator/multi_step/__init__.py +0 -0
- rasa/dialogue_understanding/generator/multi_step/fill_slots_prompt.jinja2 +62 -0
- rasa/dialogue_understanding/generator/multi_step/handle_flows_prompt.jinja2 +38 -0
- rasa/dialogue_understanding/generator/multi_step/multi_step_llm_command_generator.py +920 -0
- rasa/dialogue_understanding/generator/nlu_command_adapter.py +261 -0
- rasa/dialogue_understanding/generator/single_step/__init__.py +0 -0
- rasa/dialogue_understanding/generator/single_step/command_prompt_template.jinja2 +60 -0
- rasa/dialogue_understanding/generator/single_step/single_step_llm_command_generator.py +486 -0
- rasa/dialogue_understanding/patterns/__init__.py +0 -0
- rasa/dialogue_understanding/patterns/cancel.py +111 -0
- rasa/dialogue_understanding/patterns/cannot_handle.py +43 -0
- rasa/dialogue_understanding/patterns/chitchat.py +37 -0
- rasa/dialogue_understanding/patterns/clarify.py +97 -0
- rasa/dialogue_understanding/patterns/code_change.py +41 -0
- rasa/dialogue_understanding/patterns/collect_information.py +90 -0
- rasa/dialogue_understanding/patterns/completed.py +40 -0
- rasa/dialogue_understanding/patterns/continue_interrupted.py +42 -0
- rasa/dialogue_understanding/patterns/correction.py +278 -0
- rasa/dialogue_understanding/patterns/default_flows_for_patterns.yml +301 -0
- rasa/dialogue_understanding/patterns/human_handoff.py +37 -0
- rasa/dialogue_understanding/patterns/internal_error.py +47 -0
- rasa/dialogue_understanding/patterns/repeat.py +37 -0
- rasa/dialogue_understanding/patterns/restart.py +37 -0
- rasa/dialogue_understanding/patterns/search.py +37 -0
- rasa/dialogue_understanding/patterns/session_start.py +37 -0
- rasa/dialogue_understanding/patterns/skip_question.py +38 -0
- rasa/dialogue_understanding/patterns/user_silence.py +37 -0
- rasa/dialogue_understanding/processor/__init__.py +0 -0
- rasa/dialogue_understanding/processor/command_processor.py +720 -0
- rasa/dialogue_understanding/processor/command_processor_component.py +43 -0
- rasa/dialogue_understanding/stack/__init__.py +0 -0
- rasa/dialogue_understanding/stack/dialogue_stack.py +178 -0
- rasa/dialogue_understanding/stack/frames/__init__.py +19 -0
- rasa/dialogue_understanding/stack/frames/chit_chat_frame.py +27 -0
- rasa/dialogue_understanding/stack/frames/dialogue_stack_frame.py +137 -0
- rasa/dialogue_understanding/stack/frames/flow_stack_frame.py +157 -0
- rasa/dialogue_understanding/stack/frames/pattern_frame.py +10 -0
- rasa/dialogue_understanding/stack/frames/search_frame.py +27 -0
- rasa/dialogue_understanding/stack/utils.py +211 -0
- rasa/dialogue_understanding/utils.py +14 -0
- rasa/dialogue_understanding_test/__init__.py +0 -0
- rasa/dialogue_understanding_test/command_metric_calculation.py +12 -0
- rasa/dialogue_understanding_test/constants.py +17 -0
- rasa/dialogue_understanding_test/du_test_case.py +118 -0
- rasa/dialogue_understanding_test/du_test_result.py +11 -0
- rasa/dialogue_understanding_test/du_test_runner.py +93 -0
- rasa/dialogue_understanding_test/io.py +54 -0
- rasa/dialogue_understanding_test/validation.py +22 -0
- rasa/e2e_test/__init__.py +0 -0
- rasa/e2e_test/aggregate_test_stats_calculator.py +134 -0
- rasa/e2e_test/assertions.py +1345 -0
- rasa/e2e_test/assertions_schema.yml +129 -0
- rasa/e2e_test/constants.py +31 -0
- rasa/e2e_test/e2e_config.py +220 -0
- rasa/e2e_test/e2e_config_schema.yml +26 -0
- rasa/e2e_test/e2e_test_case.py +569 -0
- rasa/e2e_test/e2e_test_converter.py +363 -0
- rasa/e2e_test/e2e_test_converter_prompt.jinja2 +70 -0
- rasa/e2e_test/e2e_test_coverage_report.py +364 -0
- rasa/e2e_test/e2e_test_result.py +54 -0
- rasa/e2e_test/e2e_test_runner.py +1192 -0
- rasa/e2e_test/e2e_test_schema.yml +181 -0
- rasa/e2e_test/pykwalify_extensions.py +39 -0
- rasa/e2e_test/stub_custom_action.py +70 -0
- rasa/e2e_test/utils/__init__.py +0 -0
- rasa/e2e_test/utils/e2e_yaml_utils.py +55 -0
- rasa/e2e_test/utils/io.py +598 -0
- rasa/e2e_test/utils/validation.py +178 -0
- rasa/engine/__init__.py +0 -0
- rasa/engine/caching.py +463 -0
- rasa/engine/constants.py +17 -0
- rasa/engine/exceptions.py +14 -0
- rasa/engine/graph.py +642 -0
- rasa/engine/loader.py +48 -0
- rasa/engine/recipes/__init__.py +0 -0
- rasa/engine/recipes/config_files/default_config.yml +41 -0
- rasa/engine/recipes/default_components.py +97 -0
- rasa/engine/recipes/default_recipe.py +1272 -0
- rasa/engine/recipes/graph_recipe.py +79 -0
- rasa/engine/recipes/recipe.py +93 -0
- rasa/engine/runner/__init__.py +0 -0
- rasa/engine/runner/dask.py +250 -0
- rasa/engine/runner/interface.py +49 -0
- rasa/engine/storage/__init__.py +0 -0
- rasa/engine/storage/local_model_storage.py +244 -0
- rasa/engine/storage/resource.py +110 -0
- rasa/engine/storage/storage.py +199 -0
- rasa/engine/training/__init__.py +0 -0
- rasa/engine/training/components.py +176 -0
- rasa/engine/training/fingerprinting.py +64 -0
- rasa/engine/training/graph_trainer.py +256 -0
- rasa/engine/training/hooks.py +164 -0
- rasa/engine/validation.py +1451 -0
- rasa/env.py +14 -0
- rasa/exceptions.py +69 -0
- rasa/graph_components/__init__.py +0 -0
- rasa/graph_components/converters/__init__.py +0 -0
- rasa/graph_components/converters/nlu_message_converter.py +48 -0
- rasa/graph_components/providers/__init__.py +0 -0
- rasa/graph_components/providers/domain_for_core_training_provider.py +87 -0
- rasa/graph_components/providers/domain_provider.py +71 -0
- rasa/graph_components/providers/flows_provider.py +74 -0
- rasa/graph_components/providers/forms_provider.py +44 -0
- rasa/graph_components/providers/nlu_training_data_provider.py +56 -0
- rasa/graph_components/providers/responses_provider.py +44 -0
- rasa/graph_components/providers/rule_only_provider.py +49 -0
- rasa/graph_components/providers/story_graph_provider.py +96 -0
- rasa/graph_components/providers/training_tracker_provider.py +55 -0
- rasa/graph_components/validators/__init__.py +0 -0
- rasa/graph_components/validators/default_recipe_validator.py +550 -0
- rasa/graph_components/validators/finetuning_validator.py +302 -0
- rasa/hooks.py +111 -0
- rasa/jupyter.py +63 -0
- rasa/llm_fine_tuning/__init__.py +0 -0
- rasa/llm_fine_tuning/annotation_module.py +241 -0
- rasa/llm_fine_tuning/conversations.py +144 -0
- rasa/llm_fine_tuning/llm_data_preparation_module.py +178 -0
- rasa/llm_fine_tuning/paraphrasing/__init__.py +0 -0
- rasa/llm_fine_tuning/paraphrasing/conversation_rephraser.py +281 -0
- rasa/llm_fine_tuning/paraphrasing/default_rephrase_prompt_template.jina2 +44 -0
- rasa/llm_fine_tuning/paraphrasing/rephrase_validator.py +121 -0
- rasa/llm_fine_tuning/paraphrasing/rephrased_user_message.py +10 -0
- rasa/llm_fine_tuning/paraphrasing_module.py +128 -0
- rasa/llm_fine_tuning/storage.py +174 -0
- rasa/llm_fine_tuning/train_test_split_module.py +441 -0
- rasa/markers/__init__.py +0 -0
- rasa/markers/marker.py +269 -0
- rasa/markers/marker_base.py +828 -0
- rasa/markers/upload.py +74 -0
- rasa/markers/validate.py +21 -0
- rasa/model.py +118 -0
- rasa/model_manager/__init__.py +0 -0
- rasa/model_manager/config.py +40 -0
- rasa/model_manager/model_api.py +559 -0
- rasa/model_manager/runner_service.py +286 -0
- rasa/model_manager/socket_bridge.py +146 -0
- rasa/model_manager/studio_jwt_auth.py +86 -0
- rasa/model_manager/trainer_service.py +325 -0
- rasa/model_manager/utils.py +87 -0
- rasa/model_manager/warm_rasa_process.py +187 -0
- rasa/model_service.py +112 -0
- rasa/model_testing.py +457 -0
- rasa/model_training.py +596 -0
- rasa/nlu/__init__.py +7 -0
- rasa/nlu/classifiers/__init__.py +3 -0
- rasa/nlu/classifiers/classifier.py +5 -0
- rasa/nlu/classifiers/diet_classifier.py +1881 -0
- rasa/nlu/classifiers/fallback_classifier.py +192 -0
- rasa/nlu/classifiers/keyword_intent_classifier.py +188 -0
- rasa/nlu/classifiers/logistic_regression_classifier.py +253 -0
- rasa/nlu/classifiers/mitie_intent_classifier.py +156 -0
- rasa/nlu/classifiers/regex_message_handler.py +56 -0
- rasa/nlu/classifiers/sklearn_intent_classifier.py +330 -0
- rasa/nlu/constants.py +77 -0
- rasa/nlu/convert.py +40 -0
- rasa/nlu/emulators/__init__.py +0 -0
- rasa/nlu/emulators/dialogflow.py +55 -0
- rasa/nlu/emulators/emulator.py +49 -0
- rasa/nlu/emulators/luis.py +86 -0
- rasa/nlu/emulators/no_emulator.py +10 -0
- rasa/nlu/emulators/wit.py +56 -0
- rasa/nlu/extractors/__init__.py +0 -0
- rasa/nlu/extractors/crf_entity_extractor.py +715 -0
- rasa/nlu/extractors/duckling_entity_extractor.py +206 -0
- rasa/nlu/extractors/entity_synonyms.py +178 -0
- rasa/nlu/extractors/extractor.py +470 -0
- rasa/nlu/extractors/mitie_entity_extractor.py +293 -0
- rasa/nlu/extractors/regex_entity_extractor.py +220 -0
- rasa/nlu/extractors/spacy_entity_extractor.py +95 -0
- rasa/nlu/featurizers/__init__.py +0 -0
- rasa/nlu/featurizers/dense_featurizer/__init__.py +0 -0
- rasa/nlu/featurizers/dense_featurizer/convert_featurizer.py +445 -0
- rasa/nlu/featurizers/dense_featurizer/dense_featurizer.py +57 -0
- rasa/nlu/featurizers/dense_featurizer/lm_featurizer.py +768 -0
- rasa/nlu/featurizers/dense_featurizer/mitie_featurizer.py +170 -0
- rasa/nlu/featurizers/dense_featurizer/spacy_featurizer.py +132 -0
- rasa/nlu/featurizers/featurizer.py +89 -0
- rasa/nlu/featurizers/sparse_featurizer/__init__.py +0 -0
- rasa/nlu/featurizers/sparse_featurizer/count_vectors_featurizer.py +867 -0
- rasa/nlu/featurizers/sparse_featurizer/lexical_syntactic_featurizer.py +571 -0
- rasa/nlu/featurizers/sparse_featurizer/regex_featurizer.py +271 -0
- rasa/nlu/featurizers/sparse_featurizer/sparse_featurizer.py +9 -0
- rasa/nlu/model.py +24 -0
- rasa/nlu/run.py +27 -0
- rasa/nlu/selectors/__init__.py +0 -0
- rasa/nlu/selectors/response_selector.py +987 -0
- rasa/nlu/test.py +1940 -0
- rasa/nlu/tokenizers/__init__.py +0 -0
- rasa/nlu/tokenizers/jieba_tokenizer.py +148 -0
- rasa/nlu/tokenizers/mitie_tokenizer.py +75 -0
- rasa/nlu/tokenizers/spacy_tokenizer.py +72 -0
- rasa/nlu/tokenizers/tokenizer.py +239 -0
- rasa/nlu/tokenizers/whitespace_tokenizer.py +95 -0
- rasa/nlu/utils/__init__.py +35 -0
- rasa/nlu/utils/bilou_utils.py +462 -0
- rasa/nlu/utils/hugging_face/__init__.py +0 -0
- rasa/nlu/utils/hugging_face/registry.py +108 -0
- rasa/nlu/utils/hugging_face/transformers_pre_post_processors.py +311 -0
- rasa/nlu/utils/mitie_utils.py +113 -0
- rasa/nlu/utils/pattern_utils.py +168 -0
- rasa/nlu/utils/spacy_utils.py +310 -0
- rasa/plugin.py +90 -0
- rasa/server.py +1588 -0
- rasa/shared/__init__.py +0 -0
- rasa/shared/constants.py +311 -0
- rasa/shared/core/__init__.py +0 -0
- rasa/shared/core/command_payload_reader.py +109 -0
- rasa/shared/core/constants.py +180 -0
- rasa/shared/core/conversation.py +46 -0
- rasa/shared/core/domain.py +2172 -0
- rasa/shared/core/events.py +2559 -0
- rasa/shared/core/flows/__init__.py +7 -0
- rasa/shared/core/flows/flow.py +562 -0
- rasa/shared/core/flows/flow_path.py +84 -0
- rasa/shared/core/flows/flow_step.py +146 -0
- rasa/shared/core/flows/flow_step_links.py +319 -0
- rasa/shared/core/flows/flow_step_sequence.py +70 -0
- rasa/shared/core/flows/flows_list.py +258 -0
- rasa/shared/core/flows/flows_yaml_schema.json +303 -0
- rasa/shared/core/flows/nlu_trigger.py +117 -0
- rasa/shared/core/flows/steps/__init__.py +24 -0
- rasa/shared/core/flows/steps/action.py +56 -0
- rasa/shared/core/flows/steps/call.py +64 -0
- rasa/shared/core/flows/steps/collect.py +112 -0
- rasa/shared/core/flows/steps/constants.py +5 -0
- rasa/shared/core/flows/steps/continuation.py +36 -0
- rasa/shared/core/flows/steps/end.py +22 -0
- rasa/shared/core/flows/steps/internal.py +44 -0
- rasa/shared/core/flows/steps/link.py +51 -0
- rasa/shared/core/flows/steps/no_operation.py +48 -0
- rasa/shared/core/flows/steps/set_slots.py +50 -0
- rasa/shared/core/flows/steps/start.py +30 -0
- rasa/shared/core/flows/utils.py +39 -0
- rasa/shared/core/flows/validation.py +735 -0
- rasa/shared/core/flows/yaml_flows_io.py +405 -0
- rasa/shared/core/generator.py +908 -0
- rasa/shared/core/slot_mappings.py +526 -0
- rasa/shared/core/slots.py +654 -0
- rasa/shared/core/trackers.py +1183 -0
- rasa/shared/core/training_data/__init__.py +0 -0
- rasa/shared/core/training_data/loading.py +89 -0
- rasa/shared/core/training_data/story_reader/__init__.py +0 -0
- rasa/shared/core/training_data/story_reader/story_reader.py +129 -0
- rasa/shared/core/training_data/story_reader/story_step_builder.py +168 -0
- rasa/shared/core/training_data/story_reader/yaml_story_reader.py +888 -0
- rasa/shared/core/training_data/story_writer/__init__.py +0 -0
- rasa/shared/core/training_data/story_writer/story_writer.py +76 -0
- rasa/shared/core/training_data/story_writer/yaml_story_writer.py +444 -0
- rasa/shared/core/training_data/structures.py +858 -0
- rasa/shared/core/training_data/visualization.html +146 -0
- rasa/shared/core/training_data/visualization.py +603 -0
- rasa/shared/data.py +249 -0
- rasa/shared/engine/__init__.py +0 -0
- rasa/shared/engine/caching.py +26 -0
- rasa/shared/exceptions.py +167 -0
- rasa/shared/importers/__init__.py +0 -0
- rasa/shared/importers/importer.py +770 -0
- rasa/shared/importers/multi_project.py +215 -0
- rasa/shared/importers/rasa.py +108 -0
- rasa/shared/importers/remote_importer.py +196 -0
- rasa/shared/importers/utils.py +36 -0
- rasa/shared/nlu/__init__.py +0 -0
- rasa/shared/nlu/constants.py +53 -0
- rasa/shared/nlu/interpreter.py +10 -0
- rasa/shared/nlu/training_data/__init__.py +0 -0
- rasa/shared/nlu/training_data/entities_parser.py +208 -0
- rasa/shared/nlu/training_data/features.py +492 -0
- rasa/shared/nlu/training_data/formats/__init__.py +10 -0
- rasa/shared/nlu/training_data/formats/dialogflow.py +163 -0
- rasa/shared/nlu/training_data/formats/luis.py +87 -0
- rasa/shared/nlu/training_data/formats/rasa.py +135 -0
- rasa/shared/nlu/training_data/formats/rasa_yaml.py +618 -0
- rasa/shared/nlu/training_data/formats/readerwriter.py +244 -0
- rasa/shared/nlu/training_data/formats/wit.py +52 -0
- rasa/shared/nlu/training_data/loading.py +137 -0
- rasa/shared/nlu/training_data/lookup_tables_parser.py +30 -0
- rasa/shared/nlu/training_data/message.py +490 -0
- rasa/shared/nlu/training_data/schemas/__init__.py +0 -0
- rasa/shared/nlu/training_data/schemas/data_schema.py +85 -0
- rasa/shared/nlu/training_data/schemas/nlu.yml +53 -0
- rasa/shared/nlu/training_data/schemas/responses.yml +70 -0
- rasa/shared/nlu/training_data/synonyms_parser.py +42 -0
- rasa/shared/nlu/training_data/training_data.py +729 -0
- rasa/shared/nlu/training_data/util.py +223 -0
- rasa/shared/providers/__init__.py +0 -0
- rasa/shared/providers/_configs/__init__.py +0 -0
- rasa/shared/providers/_configs/azure_openai_client_config.py +677 -0
- rasa/shared/providers/_configs/client_config.py +59 -0
- rasa/shared/providers/_configs/default_litellm_client_config.py +132 -0
- rasa/shared/providers/_configs/huggingface_local_embedding_client_config.py +236 -0
- rasa/shared/providers/_configs/litellm_router_client_config.py +222 -0
- rasa/shared/providers/_configs/model_group_config.py +173 -0
- rasa/shared/providers/_configs/openai_client_config.py +177 -0
- rasa/shared/providers/_configs/rasa_llm_client_config.py +75 -0
- rasa/shared/providers/_configs/self_hosted_llm_client_config.py +178 -0
- rasa/shared/providers/_configs/utils.py +117 -0
- rasa/shared/providers/_ssl_verification_utils.py +124 -0
- rasa/shared/providers/_utils.py +79 -0
- rasa/shared/providers/constants.py +7 -0
- rasa/shared/providers/embedding/__init__.py +0 -0
- rasa/shared/providers/embedding/_base_litellm_embedding_client.py +243 -0
- rasa/shared/providers/embedding/_langchain_embedding_client_adapter.py +74 -0
- rasa/shared/providers/embedding/azure_openai_embedding_client.py +335 -0
- rasa/shared/providers/embedding/default_litellm_embedding_client.py +126 -0
- rasa/shared/providers/embedding/embedding_client.py +90 -0
- rasa/shared/providers/embedding/embedding_response.py +41 -0
- rasa/shared/providers/embedding/huggingface_local_embedding_client.py +191 -0
- rasa/shared/providers/embedding/litellm_router_embedding_client.py +138 -0
- rasa/shared/providers/embedding/openai_embedding_client.py +172 -0
- rasa/shared/providers/llm/__init__.py +0 -0
- rasa/shared/providers/llm/_base_litellm_client.py +265 -0
- rasa/shared/providers/llm/azure_openai_llm_client.py +415 -0
- rasa/shared/providers/llm/default_litellm_llm_client.py +110 -0
- rasa/shared/providers/llm/litellm_router_llm_client.py +202 -0
- rasa/shared/providers/llm/llm_client.py +78 -0
- rasa/shared/providers/llm/llm_response.py +50 -0
- rasa/shared/providers/llm/openai_llm_client.py +161 -0
- rasa/shared/providers/llm/rasa_llm_client.py +120 -0
- rasa/shared/providers/llm/self_hosted_llm_client.py +276 -0
- rasa/shared/providers/mappings.py +94 -0
- rasa/shared/providers/router/__init__.py +0 -0
- rasa/shared/providers/router/_base_litellm_router_client.py +185 -0
- rasa/shared/providers/router/router_client.py +75 -0
- rasa/shared/utils/__init__.py +0 -0
- rasa/shared/utils/cli.py +102 -0
- rasa/shared/utils/common.py +324 -0
- rasa/shared/utils/constants.py +4 -0
- rasa/shared/utils/health_check/__init__.py +0 -0
- rasa/shared/utils/health_check/embeddings_health_check_mixin.py +31 -0
- rasa/shared/utils/health_check/health_check.py +258 -0
- rasa/shared/utils/health_check/llm_health_check_mixin.py +31 -0
- rasa/shared/utils/io.py +499 -0
- rasa/shared/utils/llm.py +764 -0
- rasa/shared/utils/pykwalify_extensions.py +27 -0
- rasa/shared/utils/schemas/__init__.py +0 -0
- rasa/shared/utils/schemas/config.yml +2 -0
- rasa/shared/utils/schemas/domain.yml +145 -0
- rasa/shared/utils/schemas/events.py +214 -0
- rasa/shared/utils/schemas/model_config.yml +36 -0
- rasa/shared/utils/schemas/stories.yml +173 -0
- rasa/shared/utils/yaml.py +1068 -0
- rasa/studio/__init__.py +0 -0
- rasa/studio/auth.py +270 -0
- rasa/studio/config.py +136 -0
- rasa/studio/constants.py +19 -0
- rasa/studio/data_handler.py +368 -0
- rasa/studio/download.py +489 -0
- rasa/studio/results_logger.py +137 -0
- rasa/studio/train.py +134 -0
- rasa/studio/upload.py +563 -0
- rasa/telemetry.py +1876 -0
- rasa/tracing/__init__.py +0 -0
- rasa/tracing/config.py +355 -0
- rasa/tracing/constants.py +62 -0
- rasa/tracing/instrumentation/__init__.py +0 -0
- rasa/tracing/instrumentation/attribute_extractors.py +765 -0
- rasa/tracing/instrumentation/instrumentation.py +1306 -0
- rasa/tracing/instrumentation/intentless_policy_instrumentation.py +144 -0
- rasa/tracing/instrumentation/metrics.py +294 -0
- rasa/tracing/metric_instrument_provider.py +205 -0
- rasa/utils/__init__.py +0 -0
- rasa/utils/beta.py +83 -0
- rasa/utils/cli.py +28 -0
- rasa/utils/common.py +639 -0
- rasa/utils/converter.py +53 -0
- rasa/utils/endpoints.py +331 -0
- rasa/utils/io.py +252 -0
- rasa/utils/json_utils.py +60 -0
- rasa/utils/licensing.py +542 -0
- rasa/utils/log_utils.py +181 -0
- rasa/utils/mapper.py +210 -0
- rasa/utils/ml_utils.py +147 -0
- rasa/utils/plotting.py +362 -0
- rasa/utils/sanic_error_handler.py +32 -0
- rasa/utils/singleton.py +23 -0
- rasa/utils/tensorflow/__init__.py +0 -0
- rasa/utils/tensorflow/callback.py +112 -0
- rasa/utils/tensorflow/constants.py +116 -0
- rasa/utils/tensorflow/crf.py +492 -0
- rasa/utils/tensorflow/data_generator.py +440 -0
- rasa/utils/tensorflow/environment.py +161 -0
- rasa/utils/tensorflow/exceptions.py +5 -0
- rasa/utils/tensorflow/feature_array.py +366 -0
- rasa/utils/tensorflow/layers.py +1565 -0
- rasa/utils/tensorflow/layers_utils.py +113 -0
- rasa/utils/tensorflow/metrics.py +281 -0
- rasa/utils/tensorflow/model_data.py +798 -0
- rasa/utils/tensorflow/model_data_utils.py +499 -0
- rasa/utils/tensorflow/models.py +935 -0
- rasa/utils/tensorflow/rasa_layers.py +1094 -0
- rasa/utils/tensorflow/transformer.py +640 -0
- rasa/utils/tensorflow/types.py +6 -0
- rasa/utils/train_utils.py +572 -0
- rasa/utils/url_tools.py +53 -0
- rasa/utils/yaml.py +54 -0
- rasa/validator.py +1644 -0
- rasa/version.py +3 -0
- rasa_pro-3.12.0.dev1.dist-info/METADATA +199 -0
- rasa_pro-3.12.0.dev1.dist-info/NOTICE +5 -0
- rasa_pro-3.12.0.dev1.dist-info/RECORD +790 -0
- rasa_pro-3.12.0.dev1.dist-info/WHEEL +4 -0
- rasa_pro-3.12.0.dev1.dist-info/entry_points.txt +3 -0
|
@@ -0,0 +1,754 @@
|
|
|
1
|
+
from __future__ import annotations
|
|
2
|
+
|
|
3
|
+
from typing import Any, Dict, Text, List, Optional
|
|
4
|
+
|
|
5
|
+
import structlog
|
|
6
|
+
from jinja2 import Template
|
|
7
|
+
from pypred import Predicate
|
|
8
|
+
from structlog.contextvars import (
|
|
9
|
+
bound_contextvars,
|
|
10
|
+
)
|
|
11
|
+
|
|
12
|
+
from rasa.core.constants import STEP_ID_METADATA_KEY, ACTIVE_FLOW_METADATA_KEY
|
|
13
|
+
from rasa.core.policies.flows.flow_exceptions import (
|
|
14
|
+
FlowCircuitBreakerTrippedException,
|
|
15
|
+
FlowException,
|
|
16
|
+
NoNextStepInFlowException,
|
|
17
|
+
)
|
|
18
|
+
from rasa.core.policies.flows.flow_step_result import (
|
|
19
|
+
FlowActionPrediction,
|
|
20
|
+
ContinueFlowWithNextStep,
|
|
21
|
+
FlowStepResult,
|
|
22
|
+
PauseFlowReturnPrediction,
|
|
23
|
+
)
|
|
24
|
+
from rasa.dialogue_understanding.commands import CancelFlowCommand
|
|
25
|
+
from rasa.dialogue_understanding.patterns.cancel import CancelPatternFlowStackFrame
|
|
26
|
+
from rasa.dialogue_understanding.patterns.collect_information import (
|
|
27
|
+
CollectInformationPatternFlowStackFrame,
|
|
28
|
+
)
|
|
29
|
+
from rasa.dialogue_understanding.patterns.completed import (
|
|
30
|
+
CompletedPatternFlowStackFrame,
|
|
31
|
+
)
|
|
32
|
+
from rasa.dialogue_understanding.patterns.continue_interrupted import (
|
|
33
|
+
ContinueInterruptedPatternFlowStackFrame,
|
|
34
|
+
)
|
|
35
|
+
from rasa.dialogue_understanding.patterns.human_handoff import (
|
|
36
|
+
HumanHandoffPatternFlowStackFrame,
|
|
37
|
+
)
|
|
38
|
+
from rasa.dialogue_understanding.patterns.internal_error import (
|
|
39
|
+
InternalErrorPatternFlowStackFrame,
|
|
40
|
+
)
|
|
41
|
+
from rasa.dialogue_understanding.patterns.search import SearchPatternFlowStackFrame
|
|
42
|
+
from rasa.dialogue_understanding.stack.dialogue_stack import DialogueStack
|
|
43
|
+
from rasa.dialogue_understanding.stack.frames import (
|
|
44
|
+
BaseFlowStackFrame,
|
|
45
|
+
DialogueStackFrame,
|
|
46
|
+
UserFlowStackFrame,
|
|
47
|
+
)
|
|
48
|
+
from rasa.dialogue_understanding.stack.frames.flow_stack_frame import (
|
|
49
|
+
FlowStackFrameType,
|
|
50
|
+
)
|
|
51
|
+
from rasa.dialogue_understanding.stack.utils import (
|
|
52
|
+
top_user_flow_frame,
|
|
53
|
+
)
|
|
54
|
+
from rasa.shared.constants import RASA_PATTERN_HUMAN_HANDOFF
|
|
55
|
+
from rasa.shared.core.constants import ACTION_LISTEN_NAME, SlotMappingType
|
|
56
|
+
from rasa.shared.core.events import (
|
|
57
|
+
Event,
|
|
58
|
+
FlowCompleted,
|
|
59
|
+
FlowResumed,
|
|
60
|
+
FlowStarted,
|
|
61
|
+
SlotSet,
|
|
62
|
+
)
|
|
63
|
+
from rasa.shared.core.flows import FlowsList
|
|
64
|
+
from rasa.shared.core.flows.flow import (
|
|
65
|
+
END_STEP,
|
|
66
|
+
Flow,
|
|
67
|
+
FlowStep,
|
|
68
|
+
)
|
|
69
|
+
from rasa.shared.core.flows.flow_step_links import (
|
|
70
|
+
StaticFlowStepLink,
|
|
71
|
+
IfFlowStepLink,
|
|
72
|
+
ElseFlowStepLink,
|
|
73
|
+
)
|
|
74
|
+
from rasa.shared.core.flows.steps import (
|
|
75
|
+
ActionFlowStep,
|
|
76
|
+
SetSlotsFlowStep,
|
|
77
|
+
LinkFlowStep,
|
|
78
|
+
ContinueFlowStep,
|
|
79
|
+
EndFlowStep,
|
|
80
|
+
CallFlowStep,
|
|
81
|
+
CollectInformationFlowStep,
|
|
82
|
+
NoOperationFlowStep,
|
|
83
|
+
)
|
|
84
|
+
from rasa.shared.core.flows.steps.collect import SlotRejection
|
|
85
|
+
from rasa.shared.core.slots import Slot
|
|
86
|
+
from rasa.shared.core.trackers import (
|
|
87
|
+
DialogueStateTracker,
|
|
88
|
+
)
|
|
89
|
+
|
|
90
|
+
structlogger = structlog.get_logger()
|
|
91
|
+
|
|
92
|
+
MAX_NUMBER_OF_STEPS = 250
|
|
93
|
+
|
|
94
|
+
|
|
95
|
+
def render_template_variables(text: str, context: Dict[Text, Any]) -> str:
|
|
96
|
+
"""Replace context variables in a text."""
|
|
97
|
+
return Template(text).render(context)
|
|
98
|
+
|
|
99
|
+
|
|
100
|
+
def is_condition_satisfied(
|
|
101
|
+
predicate: Text, context: Dict[str, Any], tracker: DialogueStateTracker
|
|
102
|
+
) -> bool:
|
|
103
|
+
"""Evaluate a predicate condition."""
|
|
104
|
+
# attach context to the predicate evaluation to allow conditions using it
|
|
105
|
+
context = {"context": context}
|
|
106
|
+
|
|
107
|
+
document: Dict[str, Any] = context.copy()
|
|
108
|
+
# add slots namespace to the document
|
|
109
|
+
document["slots"] = tracker.current_slot_values()
|
|
110
|
+
|
|
111
|
+
rendered_condition = render_template_variables(predicate, context)
|
|
112
|
+
p = Predicate(rendered_condition)
|
|
113
|
+
structlogger.debug(
|
|
114
|
+
"flow.predicate.evaluating",
|
|
115
|
+
condition=predicate,
|
|
116
|
+
rendered_condition=rendered_condition,
|
|
117
|
+
)
|
|
118
|
+
try:
|
|
119
|
+
return p.evaluate(document)
|
|
120
|
+
except (TypeError, Exception) as e:
|
|
121
|
+
structlogger.error(
|
|
122
|
+
"flow.predicate.error",
|
|
123
|
+
predicate=predicate,
|
|
124
|
+
document=document,
|
|
125
|
+
error=str(e),
|
|
126
|
+
)
|
|
127
|
+
return False
|
|
128
|
+
|
|
129
|
+
|
|
130
|
+
def is_step_end_of_flow(step: FlowStep) -> bool:
|
|
131
|
+
"""Check if a step is the end of a flow."""
|
|
132
|
+
return (
|
|
133
|
+
step.id == END_STEP
|
|
134
|
+
or
|
|
135
|
+
# not quite at the end but almost, so we'll treat it as the end
|
|
136
|
+
step.id == ContinueFlowStep.continue_step_for_id(END_STEP)
|
|
137
|
+
)
|
|
138
|
+
|
|
139
|
+
|
|
140
|
+
def select_next_step_id(
|
|
141
|
+
current: FlowStep,
|
|
142
|
+
condition_evaluation_context: Dict[str, Any],
|
|
143
|
+
tracker: DialogueStateTracker,
|
|
144
|
+
) -> Optional[Text]:
|
|
145
|
+
"""Selects the next step id based on the current step."""
|
|
146
|
+
next_step = current.next
|
|
147
|
+
if len(next_step.links) == 1 and isinstance(next_step.links[0], StaticFlowStepLink):
|
|
148
|
+
return next_step.links[0].target
|
|
149
|
+
|
|
150
|
+
# evaluate if conditions
|
|
151
|
+
for link in next_step.links:
|
|
152
|
+
if isinstance(link, IfFlowStepLink) and link.condition:
|
|
153
|
+
if is_condition_satisfied(
|
|
154
|
+
link.condition, condition_evaluation_context, tracker
|
|
155
|
+
):
|
|
156
|
+
structlogger.debug(
|
|
157
|
+
"flow.link.if_condition_satisfied",
|
|
158
|
+
current_id=current.id,
|
|
159
|
+
target=link.target,
|
|
160
|
+
)
|
|
161
|
+
return link.target
|
|
162
|
+
|
|
163
|
+
# evaluate else condition
|
|
164
|
+
for link in next_step.links:
|
|
165
|
+
if isinstance(link, ElseFlowStepLink):
|
|
166
|
+
structlogger.debug(
|
|
167
|
+
"flow.link.else_condition_satisfied",
|
|
168
|
+
current_id=current.id,
|
|
169
|
+
target=link.target,
|
|
170
|
+
)
|
|
171
|
+
return link.target
|
|
172
|
+
|
|
173
|
+
if next_step.links:
|
|
174
|
+
structlogger.error(
|
|
175
|
+
"flow.link.failed_to_select_branch",
|
|
176
|
+
current=current,
|
|
177
|
+
links=next_step.links,
|
|
178
|
+
tracker=tracker,
|
|
179
|
+
)
|
|
180
|
+
return None
|
|
181
|
+
|
|
182
|
+
if current.id == END_STEP:
|
|
183
|
+
# we are already at the very end of the flow. There is no next step.
|
|
184
|
+
return None
|
|
185
|
+
elif isinstance(current, LinkFlowStep):
|
|
186
|
+
# link steps don't have a next step, so we'll return the end step
|
|
187
|
+
return END_STEP
|
|
188
|
+
else:
|
|
189
|
+
structlogger.error(
|
|
190
|
+
"flow.step.failed_to_select_next_step",
|
|
191
|
+
step=current,
|
|
192
|
+
tracker=tracker,
|
|
193
|
+
)
|
|
194
|
+
return None
|
|
195
|
+
|
|
196
|
+
|
|
197
|
+
def select_next_step(
|
|
198
|
+
current_step: FlowStep,
|
|
199
|
+
current_flow: Flow,
|
|
200
|
+
stack: DialogueStack,
|
|
201
|
+
tracker: DialogueStateTracker,
|
|
202
|
+
) -> Optional[FlowStep]:
|
|
203
|
+
"""Get the next step to execute."""
|
|
204
|
+
next_id = select_next_step_id(current_step, stack.current_context(), tracker)
|
|
205
|
+
step = current_flow.step_by_id(next_id)
|
|
206
|
+
structlogger.debug(
|
|
207
|
+
"flow.step.next",
|
|
208
|
+
next_id=step.id if step else None,
|
|
209
|
+
current_id=current_step.id,
|
|
210
|
+
flow_id=current_flow.id,
|
|
211
|
+
)
|
|
212
|
+
return step
|
|
213
|
+
|
|
214
|
+
|
|
215
|
+
def update_top_flow_step_id(updated_id: str, stack: DialogueStack) -> DialogueStack:
|
|
216
|
+
"""Update the top flow on the stack."""
|
|
217
|
+
if (top := stack.top()) and isinstance(top, BaseFlowStackFrame):
|
|
218
|
+
top.step_id = updated_id
|
|
219
|
+
return stack
|
|
220
|
+
|
|
221
|
+
|
|
222
|
+
def events_from_set_slots_step(step: SetSlotsFlowStep) -> List[Event]:
|
|
223
|
+
"""Create events from a set slots step."""
|
|
224
|
+
return [SlotSet(slot["key"], slot["value"]) for slot in step.slots]
|
|
225
|
+
|
|
226
|
+
|
|
227
|
+
def events_for_collect_step_execution(
|
|
228
|
+
step: CollectInformationFlowStep, tracker: DialogueStateTracker
|
|
229
|
+
) -> List[Event]:
|
|
230
|
+
"""Create the events needed to prepare for the execution of a collect step."""
|
|
231
|
+
# reset the slots that always need to be explicitly collected
|
|
232
|
+
slot = tracker.slots.get(step.collect, None)
|
|
233
|
+
|
|
234
|
+
if slot and step.ask_before_filling:
|
|
235
|
+
return [SlotSet(step.collect, None)]
|
|
236
|
+
else:
|
|
237
|
+
return []
|
|
238
|
+
|
|
239
|
+
|
|
240
|
+
def trigger_pattern_continue_interrupted(
|
|
241
|
+
current_frame: DialogueStackFrame, stack: DialogueStack, flows: FlowsList
|
|
242
|
+
) -> List[Event]:
|
|
243
|
+
"""Trigger the pattern to continue an interrupted flow if needed."""
|
|
244
|
+
events: List[Event] = []
|
|
245
|
+
|
|
246
|
+
# get previously started user flow that will be continued
|
|
247
|
+
interrupted_user_flow_frame = top_user_flow_frame(stack)
|
|
248
|
+
interrupted_user_flow_step = (
|
|
249
|
+
interrupted_user_flow_frame.step(flows) if interrupted_user_flow_frame else None
|
|
250
|
+
)
|
|
251
|
+
interrupted_user_flow = (
|
|
252
|
+
interrupted_user_flow_frame.flow(flows) if interrupted_user_flow_frame else None
|
|
253
|
+
)
|
|
254
|
+
|
|
255
|
+
if (
|
|
256
|
+
isinstance(current_frame, UserFlowStackFrame)
|
|
257
|
+
and interrupted_user_flow_step is not None
|
|
258
|
+
and interrupted_user_flow is not None
|
|
259
|
+
and current_frame.frame_type == FlowStackFrameType.INTERRUPT
|
|
260
|
+
and not is_step_end_of_flow(interrupted_user_flow_step)
|
|
261
|
+
):
|
|
262
|
+
stack.push(
|
|
263
|
+
ContinueInterruptedPatternFlowStackFrame(
|
|
264
|
+
previous_flow_name=interrupted_user_flow.readable_name(),
|
|
265
|
+
)
|
|
266
|
+
)
|
|
267
|
+
events.append(
|
|
268
|
+
FlowResumed(interrupted_user_flow.id, interrupted_user_flow_step.id)
|
|
269
|
+
)
|
|
270
|
+
|
|
271
|
+
return events
|
|
272
|
+
|
|
273
|
+
|
|
274
|
+
def trigger_pattern_completed(
|
|
275
|
+
current_frame: DialogueStackFrame, stack: DialogueStack, flows: FlowsList
|
|
276
|
+
) -> None:
|
|
277
|
+
"""Trigger the pattern indicating that the stack is empty, if needed."""
|
|
278
|
+
# trigger pattern if the stack is empty and the last frame was either a user flow
|
|
279
|
+
# frame or a search frame
|
|
280
|
+
if stack.is_empty() and (
|
|
281
|
+
isinstance(current_frame, UserFlowStackFrame)
|
|
282
|
+
or isinstance(current_frame, SearchPatternFlowStackFrame)
|
|
283
|
+
):
|
|
284
|
+
completed_flow = current_frame.flow(flows)
|
|
285
|
+
completed_flow_name = completed_flow.readable_name() if completed_flow else None
|
|
286
|
+
stack.push(
|
|
287
|
+
CompletedPatternFlowStackFrame(
|
|
288
|
+
previous_flow_name=completed_flow_name,
|
|
289
|
+
)
|
|
290
|
+
)
|
|
291
|
+
|
|
292
|
+
|
|
293
|
+
def trigger_pattern_ask_collect_information(
|
|
294
|
+
collect: str,
|
|
295
|
+
stack: DialogueStack,
|
|
296
|
+
rejections: List[SlotRejection],
|
|
297
|
+
utter: str,
|
|
298
|
+
collect_action: str,
|
|
299
|
+
) -> None:
|
|
300
|
+
"""Trigger the pattern to ask for a slot value."""
|
|
301
|
+
stack.push(
|
|
302
|
+
CollectInformationPatternFlowStackFrame(
|
|
303
|
+
collect=collect,
|
|
304
|
+
utter=utter,
|
|
305
|
+
collect_action=collect_action,
|
|
306
|
+
rejections=rejections,
|
|
307
|
+
)
|
|
308
|
+
)
|
|
309
|
+
|
|
310
|
+
|
|
311
|
+
def reset_scoped_slots(
|
|
312
|
+
current_frame: DialogueStackFrame, current_flow: Flow, tracker: DialogueStateTracker
|
|
313
|
+
) -> List[Event]:
|
|
314
|
+
"""Reset all scoped slots."""
|
|
315
|
+
|
|
316
|
+
def _reset_slot(slot_name: Text, dialogue_tracker: DialogueStateTracker) -> None:
|
|
317
|
+
slot = dialogue_tracker.slots.get(slot_name, None)
|
|
318
|
+
initial_value = slot.initial_value if slot else None
|
|
319
|
+
events.append(SlotSet(slot_name, initial_value))
|
|
320
|
+
|
|
321
|
+
if (
|
|
322
|
+
isinstance(current_frame, UserFlowStackFrame)
|
|
323
|
+
and current_frame.frame_type == FlowStackFrameType.CALL
|
|
324
|
+
):
|
|
325
|
+
# if a called frame is completed, we don't reset the slots
|
|
326
|
+
# as they are scoped to the called flow. resetting will happen as part
|
|
327
|
+
# of the flow that contained the call step triggering this called flow
|
|
328
|
+
return []
|
|
329
|
+
|
|
330
|
+
events: List[Event] = []
|
|
331
|
+
|
|
332
|
+
not_resettable_slot_names = set()
|
|
333
|
+
flow_persistable_slots = current_flow.persisted_slots
|
|
334
|
+
|
|
335
|
+
for step in current_flow.steps_with_calls_resolved:
|
|
336
|
+
if isinstance(step, CollectInformationFlowStep):
|
|
337
|
+
# reset all slots scoped to the flow
|
|
338
|
+
slot_name = step.collect
|
|
339
|
+
if step.reset_after_flow_ends and slot_name not in flow_persistable_slots:
|
|
340
|
+
_reset_slot(slot_name, tracker)
|
|
341
|
+
else:
|
|
342
|
+
not_resettable_slot_names.add(slot_name)
|
|
343
|
+
|
|
344
|
+
# slots set by the set slots step should be reset after the flow ends
|
|
345
|
+
# unless they are also used in a collect step where `reset_after_flow_ends`
|
|
346
|
+
# is set to `False` or set in the `persisted_slots` list.
|
|
347
|
+
resettable_set_slots = [
|
|
348
|
+
slot["key"]
|
|
349
|
+
for step in current_flow.steps_with_calls_resolved
|
|
350
|
+
if isinstance(step, SetSlotsFlowStep)
|
|
351
|
+
for slot in step.slots
|
|
352
|
+
if slot["key"] not in not_resettable_slot_names
|
|
353
|
+
and slot["key"] not in flow_persistable_slots
|
|
354
|
+
]
|
|
355
|
+
|
|
356
|
+
for name in resettable_set_slots:
|
|
357
|
+
_reset_slot(name, tracker)
|
|
358
|
+
|
|
359
|
+
return events
|
|
360
|
+
|
|
361
|
+
|
|
362
|
+
def advance_flows(
|
|
363
|
+
tracker: DialogueStateTracker, available_actions: List[str], flows: FlowsList
|
|
364
|
+
) -> FlowActionPrediction:
|
|
365
|
+
"""Advance the current flows until the next action.
|
|
366
|
+
|
|
367
|
+
Args:
|
|
368
|
+
tracker: The tracker to get the next action for.
|
|
369
|
+
available_actions: The actions that are available in the domain.
|
|
370
|
+
flows: All flows.
|
|
371
|
+
|
|
372
|
+
Returns:
|
|
373
|
+
The predicted action and the events to run.
|
|
374
|
+
"""
|
|
375
|
+
stack = tracker.stack
|
|
376
|
+
if stack.is_empty():
|
|
377
|
+
# if there are no flows, there is nothing to do
|
|
378
|
+
return FlowActionPrediction(None, 0.0)
|
|
379
|
+
|
|
380
|
+
return advance_flows_until_next_action(tracker, available_actions, flows)
|
|
381
|
+
|
|
382
|
+
|
|
383
|
+
def advance_flows_until_next_action(
|
|
384
|
+
tracker: DialogueStateTracker,
|
|
385
|
+
available_actions: List[str],
|
|
386
|
+
flows: FlowsList,
|
|
387
|
+
) -> FlowActionPrediction:
|
|
388
|
+
"""Advance the flow and select the next action to execute.
|
|
389
|
+
|
|
390
|
+
Advances the current flow and returns the next action to execute. A flow
|
|
391
|
+
is advanced until it is completed or until it predicts an action. If
|
|
392
|
+
the flow is completed, the next flow is popped from the stack and
|
|
393
|
+
advanced. If there are no more flows, the action listen is predicted.
|
|
394
|
+
|
|
395
|
+
Args:
|
|
396
|
+
tracker: The tracker to get the next action for.
|
|
397
|
+
available_actions: The actions that are available in the domain.
|
|
398
|
+
flows: All flows.
|
|
399
|
+
|
|
400
|
+
Returns:
|
|
401
|
+
The next action to execute, the events that should be applied to the
|
|
402
|
+
tracker and the confidence of the prediction.
|
|
403
|
+
"""
|
|
404
|
+
step_result: FlowStepResult = ContinueFlowWithNextStep()
|
|
405
|
+
|
|
406
|
+
tracker = tracker.copy()
|
|
407
|
+
|
|
408
|
+
number_of_initial_events = len(tracker.events)
|
|
409
|
+
|
|
410
|
+
number_of_steps_taken = 0
|
|
411
|
+
|
|
412
|
+
while isinstance(step_result, ContinueFlowWithNextStep):
|
|
413
|
+
number_of_steps_taken += 1
|
|
414
|
+
if number_of_steps_taken > MAX_NUMBER_OF_STEPS:
|
|
415
|
+
raise FlowCircuitBreakerTrippedException(
|
|
416
|
+
tracker.stack, number_of_steps_taken
|
|
417
|
+
)
|
|
418
|
+
|
|
419
|
+
active_frame = tracker.stack.top()
|
|
420
|
+
if not isinstance(active_frame, BaseFlowStackFrame):
|
|
421
|
+
# If there is no current flow, we assume that all flows are done
|
|
422
|
+
# and there is nothing to do. The assumption here is that every
|
|
423
|
+
# flow ends with an action listen.
|
|
424
|
+
step_result = PauseFlowReturnPrediction(
|
|
425
|
+
FlowActionPrediction(ACTION_LISTEN_NAME, 1.0)
|
|
426
|
+
)
|
|
427
|
+
break
|
|
428
|
+
|
|
429
|
+
with bound_contextvars(flow_id=active_frame.flow_id):
|
|
430
|
+
previous_step_id = active_frame.step_id
|
|
431
|
+
structlogger.debug("flow.execution.loop", previous_step_id=previous_step_id)
|
|
432
|
+
current_flow = active_frame.flow(flows)
|
|
433
|
+
next_step = select_next_step(
|
|
434
|
+
active_frame.step(flows), current_flow, tracker.stack, tracker
|
|
435
|
+
)
|
|
436
|
+
|
|
437
|
+
if not next_step:
|
|
438
|
+
raise NoNextStepInFlowException(tracker.stack)
|
|
439
|
+
|
|
440
|
+
tracker.update_stack(update_top_flow_step_id(next_step.id, tracker.stack))
|
|
441
|
+
|
|
442
|
+
with bound_contextvars(step_id=next_step.id):
|
|
443
|
+
step_stack = tracker.stack
|
|
444
|
+
step_result = run_step(
|
|
445
|
+
next_step,
|
|
446
|
+
current_flow,
|
|
447
|
+
step_stack,
|
|
448
|
+
tracker,
|
|
449
|
+
available_actions,
|
|
450
|
+
flows,
|
|
451
|
+
)
|
|
452
|
+
new_events = step_result.events
|
|
453
|
+
if (
|
|
454
|
+
isinstance(step_result, ContinueFlowWithNextStep)
|
|
455
|
+
and step_result.has_flow_ended
|
|
456
|
+
):
|
|
457
|
+
# insert flow completed before flow resumed event
|
|
458
|
+
offset = (
|
|
459
|
+
-1
|
|
460
|
+
if new_events and isinstance(new_events[-1], FlowResumed)
|
|
461
|
+
else 0
|
|
462
|
+
)
|
|
463
|
+
idx = len(new_events) + offset
|
|
464
|
+
new_events.insert(
|
|
465
|
+
idx, FlowCompleted(active_frame.flow_id, previous_step_id)
|
|
466
|
+
)
|
|
467
|
+
tracker.update_stack(step_stack)
|
|
468
|
+
tracker.update_with_events(new_events)
|
|
469
|
+
|
|
470
|
+
gathered_events = list(tracker.events)[number_of_initial_events:]
|
|
471
|
+
if isinstance(step_result, PauseFlowReturnPrediction):
|
|
472
|
+
prediction = step_result.action_prediction
|
|
473
|
+
# make sure we really return all events that got created during the
|
|
474
|
+
# step execution of all steps (not only the last one)
|
|
475
|
+
prediction.events = gathered_events
|
|
476
|
+
prediction.metadata = {
|
|
477
|
+
ACTIVE_FLOW_METADATA_KEY: tracker.active_flow,
|
|
478
|
+
STEP_ID_METADATA_KEY: tracker.current_step_id,
|
|
479
|
+
}
|
|
480
|
+
return prediction
|
|
481
|
+
else:
|
|
482
|
+
structlogger.warning("flow.step.execution.no_action")
|
|
483
|
+
return FlowActionPrediction(None, 0.0, events=gathered_events)
|
|
484
|
+
|
|
485
|
+
|
|
486
|
+
def validate_collect_step(
|
|
487
|
+
step: CollectInformationFlowStep,
|
|
488
|
+
stack: DialogueStack,
|
|
489
|
+
available_actions: List[str],
|
|
490
|
+
slots: Dict[str, Slot],
|
|
491
|
+
flow_name: str,
|
|
492
|
+
) -> bool:
|
|
493
|
+
"""Validate that a collect step can be executed.
|
|
494
|
+
|
|
495
|
+
A collect step can be executed if either the `utter_ask` or the `action_ask` is
|
|
496
|
+
defined in the domain. If neither is defined, the collect step can still be
|
|
497
|
+
executed if the slot has an initial value defined in the domain, which would cause
|
|
498
|
+
the step to be skipped.
|
|
499
|
+
"""
|
|
500
|
+
slot = slots.get(step.collect)
|
|
501
|
+
slot_has_initial_value_defined = slot and slot.initial_value is not None
|
|
502
|
+
if (
|
|
503
|
+
slot_has_initial_value_defined
|
|
504
|
+
or step.utter in available_actions
|
|
505
|
+
or step.collect_action in available_actions
|
|
506
|
+
):
|
|
507
|
+
return True
|
|
508
|
+
|
|
509
|
+
structlogger.error(
|
|
510
|
+
"flow.step.run.collect_missing_utter_or_collect_action",
|
|
511
|
+
slot_name=step.collect,
|
|
512
|
+
)
|
|
513
|
+
|
|
514
|
+
cancel_flow_and_push_internal_error(stack, flow_name)
|
|
515
|
+
|
|
516
|
+
return False
|
|
517
|
+
|
|
518
|
+
|
|
519
|
+
def cancel_flow_and_push_internal_error(stack: DialogueStack, flow_name: str) -> None:
|
|
520
|
+
"""Cancel the top user flow and push the internal error pattern."""
|
|
521
|
+
top_frame = stack.top()
|
|
522
|
+
|
|
523
|
+
if isinstance(top_frame, BaseFlowStackFrame):
|
|
524
|
+
# we need to first cancel the top user flow
|
|
525
|
+
# because we cannot collect one of its slots
|
|
526
|
+
# and therefore should not proceed with the flow
|
|
527
|
+
# after triggering pattern_internal_error
|
|
528
|
+
canceled_frames = CancelFlowCommand.select_canceled_frames(stack)
|
|
529
|
+
stack.push(
|
|
530
|
+
CancelPatternFlowStackFrame(
|
|
531
|
+
canceled_name=flow_name,
|
|
532
|
+
canceled_frames=canceled_frames,
|
|
533
|
+
)
|
|
534
|
+
)
|
|
535
|
+
stack.push(InternalErrorPatternFlowStackFrame())
|
|
536
|
+
|
|
537
|
+
|
|
538
|
+
def validate_custom_slot_mappings(
|
|
539
|
+
step: CollectInformationFlowStep,
|
|
540
|
+
stack: DialogueStack,
|
|
541
|
+
tracker: DialogueStateTracker,
|
|
542
|
+
available_actions: List[str],
|
|
543
|
+
flow_name: str,
|
|
544
|
+
) -> bool:
|
|
545
|
+
"""Validate a slot with custom mappings.
|
|
546
|
+
|
|
547
|
+
If invalid, trigger pattern_internal_error and return False.
|
|
548
|
+
"""
|
|
549
|
+
slot = tracker.slots.get(step.collect, None)
|
|
550
|
+
slot_mappings = slot.mappings if slot else []
|
|
551
|
+
for mapping in slot_mappings:
|
|
552
|
+
if (
|
|
553
|
+
mapping.get("type") == SlotMappingType.CUSTOM.value
|
|
554
|
+
and mapping.get("action") is None
|
|
555
|
+
):
|
|
556
|
+
# this is a slot that must be filled by a custom action
|
|
557
|
+
# check if collect_action exists
|
|
558
|
+
if step.collect_action not in available_actions:
|
|
559
|
+
structlogger.error(
|
|
560
|
+
"flow.step.run.collect_action_not_found_for_custom_slot_mapping",
|
|
561
|
+
action=step.collect_action,
|
|
562
|
+
collect=step.collect,
|
|
563
|
+
)
|
|
564
|
+
cancel_flow_and_push_internal_error(stack, flow_name)
|
|
565
|
+
return False
|
|
566
|
+
|
|
567
|
+
return True
|
|
568
|
+
|
|
569
|
+
|
|
570
|
+
def run_step(
|
|
571
|
+
step: FlowStep,
|
|
572
|
+
flow: Flow,
|
|
573
|
+
stack: DialogueStack,
|
|
574
|
+
tracker: DialogueStateTracker,
|
|
575
|
+
available_actions: List[str],
|
|
576
|
+
flows: FlowsList,
|
|
577
|
+
) -> FlowStepResult:
|
|
578
|
+
"""Run a single step of a flow.
|
|
579
|
+
|
|
580
|
+
Returns the predicted action and a list of events that were generated
|
|
581
|
+
during the step. The predicted action can be `None` if the step
|
|
582
|
+
doesn't generate an action. The list of events can be empty if the
|
|
583
|
+
step doesn't generate any events.
|
|
584
|
+
|
|
585
|
+
Raises a `FlowException` if the step is invalid.
|
|
586
|
+
|
|
587
|
+
Args:
|
|
588
|
+
step: The step to run.
|
|
589
|
+
flow: The flow that the step belongs to.
|
|
590
|
+
stack: The stack that the flow is on.
|
|
591
|
+
tracker: The tracker to run the step on.
|
|
592
|
+
available_actions: The actions that are available in the domain.
|
|
593
|
+
flows: All flows.
|
|
594
|
+
|
|
595
|
+
Returns:
|
|
596
|
+
A result of running the step describing where to transition to.
|
|
597
|
+
"""
|
|
598
|
+
initial_events: List[Event] = []
|
|
599
|
+
if step == flow.first_step_in_flow():
|
|
600
|
+
initial_events.append(FlowStarted(flow.id, metadata=stack.current_context()))
|
|
601
|
+
|
|
602
|
+
if isinstance(step, CollectInformationFlowStep):
|
|
603
|
+
return _run_collect_information_step(
|
|
604
|
+
available_actions,
|
|
605
|
+
initial_events,
|
|
606
|
+
stack,
|
|
607
|
+
step,
|
|
608
|
+
tracker,
|
|
609
|
+
flow.readable_name(),
|
|
610
|
+
)
|
|
611
|
+
|
|
612
|
+
elif isinstance(step, ActionFlowStep):
|
|
613
|
+
if not step.action:
|
|
614
|
+
raise FlowException(f"Action not specified for step {step}")
|
|
615
|
+
return _run_action_step(available_actions, initial_events, stack, step)
|
|
616
|
+
|
|
617
|
+
elif isinstance(step, LinkFlowStep):
|
|
618
|
+
return _run_link_step(initial_events, stack, step)
|
|
619
|
+
|
|
620
|
+
elif isinstance(step, CallFlowStep):
|
|
621
|
+
return _run_call_step(initial_events, stack, step)
|
|
622
|
+
|
|
623
|
+
elif isinstance(step, SetSlotsFlowStep):
|
|
624
|
+
return _run_set_slot_step(initial_events, step)
|
|
625
|
+
|
|
626
|
+
elif isinstance(step, NoOperationFlowStep):
|
|
627
|
+
structlogger.debug("flow.step.run.no_operation")
|
|
628
|
+
return ContinueFlowWithNextStep(events=initial_events)
|
|
629
|
+
|
|
630
|
+
elif isinstance(step, EndFlowStep):
|
|
631
|
+
return _run_end_step(flow, flows, initial_events, stack, tracker)
|
|
632
|
+
|
|
633
|
+
else:
|
|
634
|
+
raise FlowException(f"Unknown flow step type {type(step)}")
|
|
635
|
+
|
|
636
|
+
|
|
637
|
+
def _run_end_step(
|
|
638
|
+
flow: Flow,
|
|
639
|
+
flows: FlowsList,
|
|
640
|
+
initial_events: List[Event],
|
|
641
|
+
stack: DialogueStack,
|
|
642
|
+
tracker: DialogueStateTracker,
|
|
643
|
+
) -> FlowStepResult:
|
|
644
|
+
# this is the end of the flow, so we'll pop it from the stack
|
|
645
|
+
structlogger.debug("flow.step.run.flow_end")
|
|
646
|
+
current_frame = stack.pop()
|
|
647
|
+
trigger_pattern_completed(current_frame, stack, flows)
|
|
648
|
+
resumed_events = trigger_pattern_continue_interrupted(current_frame, stack, flows)
|
|
649
|
+
reset_events: List[Event] = reset_scoped_slots(current_frame, flow, tracker)
|
|
650
|
+
return ContinueFlowWithNextStep(
|
|
651
|
+
events=initial_events + reset_events + resumed_events, has_flow_ended=True
|
|
652
|
+
)
|
|
653
|
+
|
|
654
|
+
|
|
655
|
+
def _run_set_slot_step(
|
|
656
|
+
initial_events: List[Event], step: SetSlotsFlowStep
|
|
657
|
+
) -> FlowStepResult:
|
|
658
|
+
structlogger.debug("flow.step.run.slot")
|
|
659
|
+
slot_events: List[Event] = events_from_set_slots_step(step)
|
|
660
|
+
return ContinueFlowWithNextStep(events=initial_events + slot_events)
|
|
661
|
+
|
|
662
|
+
|
|
663
|
+
def _run_call_step(
|
|
664
|
+
initial_events: List[Event], stack: DialogueStack, step: CallFlowStep
|
|
665
|
+
) -> FlowStepResult:
|
|
666
|
+
structlogger.debug("flow.step.run.call")
|
|
667
|
+
stack.push(
|
|
668
|
+
UserFlowStackFrame(
|
|
669
|
+
flow_id=step.call,
|
|
670
|
+
frame_type=FlowStackFrameType.CALL,
|
|
671
|
+
),
|
|
672
|
+
)
|
|
673
|
+
return ContinueFlowWithNextStep(events=initial_events)
|
|
674
|
+
|
|
675
|
+
|
|
676
|
+
def _run_link_step(
|
|
677
|
+
initial_events: List[Event], stack: DialogueStack, step: LinkFlowStep
|
|
678
|
+
) -> FlowStepResult:
|
|
679
|
+
structlogger.debug("flow.step.run.link")
|
|
680
|
+
|
|
681
|
+
if step.link == RASA_PATTERN_HUMAN_HANDOFF:
|
|
682
|
+
linked_stack_frame: DialogueStackFrame = HumanHandoffPatternFlowStackFrame()
|
|
683
|
+
else:
|
|
684
|
+
linked_stack_frame = UserFlowStackFrame(
|
|
685
|
+
flow_id=step.link,
|
|
686
|
+
frame_type=FlowStackFrameType.LINK,
|
|
687
|
+
)
|
|
688
|
+
|
|
689
|
+
stack.push(
|
|
690
|
+
linked_stack_frame,
|
|
691
|
+
# push this below the current stack frame so that we can
|
|
692
|
+
# complete the current flow first and then continue with the
|
|
693
|
+
# linked flow
|
|
694
|
+
index=-1,
|
|
695
|
+
)
|
|
696
|
+
|
|
697
|
+
return ContinueFlowWithNextStep(events=initial_events)
|
|
698
|
+
|
|
699
|
+
|
|
700
|
+
def _run_action_step(
|
|
701
|
+
available_actions: List[str],
|
|
702
|
+
initial_events: List[Event],
|
|
703
|
+
stack: DialogueStack,
|
|
704
|
+
step: ActionFlowStep,
|
|
705
|
+
) -> FlowStepResult:
|
|
706
|
+
context = {"context": stack.current_context()}
|
|
707
|
+
action_name = render_template_variables(step.action, context)
|
|
708
|
+
|
|
709
|
+
if action_name in available_actions:
|
|
710
|
+
structlogger.debug("flow.step.run.action", context=context)
|
|
711
|
+
return PauseFlowReturnPrediction(
|
|
712
|
+
FlowActionPrediction(action_name, 1.0, events=initial_events)
|
|
713
|
+
)
|
|
714
|
+
else:
|
|
715
|
+
if step.action != "validate_{{context.collect}}":
|
|
716
|
+
# do not log about non-existing validation actions of collect steps
|
|
717
|
+
utter_action_name = render_template_variables("{{context.utter}}", context)
|
|
718
|
+
if utter_action_name not in available_actions:
|
|
719
|
+
structlogger.warning("flow.step.run.action.unknown", action=action_name)
|
|
720
|
+
return ContinueFlowWithNextStep(events=initial_events)
|
|
721
|
+
|
|
722
|
+
|
|
723
|
+
def _run_collect_information_step(
|
|
724
|
+
available_actions: List[str],
|
|
725
|
+
initial_events: List[Event],
|
|
726
|
+
stack: DialogueStack,
|
|
727
|
+
step: CollectInformationFlowStep,
|
|
728
|
+
tracker: DialogueStateTracker,
|
|
729
|
+
flow_name: str,
|
|
730
|
+
) -> FlowStepResult:
|
|
731
|
+
is_step_valid = validate_collect_step(
|
|
732
|
+
step, stack, available_actions, tracker.slots, flow_name
|
|
733
|
+
)
|
|
734
|
+
|
|
735
|
+
if not is_step_valid:
|
|
736
|
+
# if we return any other FlowStepResult, the assistant will stay silent
|
|
737
|
+
# instead of triggering the internal error pattern
|
|
738
|
+
return ContinueFlowWithNextStep(events=initial_events)
|
|
739
|
+
is_mapping_valid = validate_custom_slot_mappings(
|
|
740
|
+
step, stack, tracker, available_actions, flow_name
|
|
741
|
+
)
|
|
742
|
+
|
|
743
|
+
if not is_mapping_valid:
|
|
744
|
+
# if we return any other FlowStepResult, the assistant will stay silent
|
|
745
|
+
# instead of triggering the internal error pattern
|
|
746
|
+
return ContinueFlowWithNextStep(events=initial_events)
|
|
747
|
+
|
|
748
|
+
structlogger.debug("flow.step.run.collect")
|
|
749
|
+
trigger_pattern_ask_collect_information(
|
|
750
|
+
step.collect, stack, step.rejections, step.utter, step.collect_action
|
|
751
|
+
)
|
|
752
|
+
|
|
753
|
+
events: List[Event] = events_for_collect_step_execution(step, tracker)
|
|
754
|
+
return ContinueFlowWithNextStep(events=initial_events + events)
|