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
rasa/shared/__init__.py
ADDED
|
File without changes
|
rasa/shared/constants.py
ADDED
|
@@ -0,0 +1,311 @@
|
|
|
1
|
+
from __future__ import annotations
|
|
2
|
+
|
|
3
|
+
from typing import List, Text
|
|
4
|
+
|
|
5
|
+
from rasa.shared.engine.caching import get_local_cache_location
|
|
6
|
+
|
|
7
|
+
DOCS_BASE_URL = "https://rasa.com/docs/rasa-pro"
|
|
8
|
+
DOCS_URL_CONCEPTS = DOCS_BASE_URL + "/concepts"
|
|
9
|
+
DOCS_URL_NLU_BASED = DOCS_BASE_URL + "/nlu-based-assistants"
|
|
10
|
+
|
|
11
|
+
# Docs URLs for CALM assistants
|
|
12
|
+
DOCS_URL_SLOTS = DOCS_URL_CONCEPTS + "/domain#slots"
|
|
13
|
+
DOCS_URL_RESPONSES = DOCS_URL_CONCEPTS + "/responses"
|
|
14
|
+
DOCS_URL_POLICIES = DOCS_URL_CONCEPTS + "/policies/policy-overview"
|
|
15
|
+
DOCS_URL_ACTIONS = DOCS_URL_CONCEPTS + "/actions"
|
|
16
|
+
DOCS_URL_DEFAULT_ACTIONS = DOCS_URL_CONCEPTS + "/default-actions"
|
|
17
|
+
DOCS_URL_COMPONENTS = DOCS_URL_CONCEPTS + "/components/overview"
|
|
18
|
+
DOCS_URL_GRAPH_COMPONENTS = DOCS_URL_CONCEPTS + "/components/custom-graph-components"
|
|
19
|
+
DOCS_URL_GRAPH_RECIPE = DOCS_URL_CONCEPTS + "/components/graph-recipe"
|
|
20
|
+
DOCS_URL_CATEGORICAL_SLOTS = DOCS_URL_CONCEPTS + "/domain#categorical-slot"
|
|
21
|
+
DOCS_URL_DOMAIN = DOCS_URL_CONCEPTS + "/domain"
|
|
22
|
+
|
|
23
|
+
# Docs URLs for NLU-based assistants
|
|
24
|
+
DOCS_URL_TRAINING_DATA = DOCS_URL_NLU_BASED + "/training-data-format"
|
|
25
|
+
DOCS_URL_TRAINING_DATA_NLU = DOCS_URL_TRAINING_DATA + "#nlu-training-data"
|
|
26
|
+
DOCS_URL_DOMAINS = DOCS_URL_NLU_BASED + "/domain"
|
|
27
|
+
DOCS_URL_NLU_BASED_SLOTS = DOCS_URL_DOMAINS + "#slots"
|
|
28
|
+
DOCS_URL_INTENTS = DOCS_URL_DOMAINS + "#intents"
|
|
29
|
+
DOCS_URL_ENTITIES = DOCS_URL_DOMAINS + "#entities"
|
|
30
|
+
DOCS_URL_STORIES = DOCS_URL_NLU_BASED + "/stories"
|
|
31
|
+
DOCS_URL_RULES = DOCS_URL_NLU_BASED + "/rules"
|
|
32
|
+
DOCS_URL_FORMS = DOCS_URL_NLU_BASED + "/forms"
|
|
33
|
+
DOCS_URL_PIPELINE = DOCS_URL_NLU_BASED + "/model-configuration"
|
|
34
|
+
DOCS_URL_NLU_BASED_POLICIES = DOCS_URL_NLU_BASED + "/policies"
|
|
35
|
+
|
|
36
|
+
# Other docs URLs
|
|
37
|
+
DOCS_URL_MARKERS = DOCS_BASE_URL + "/operating/analytics/realtime-markers"
|
|
38
|
+
DOCS_URL_CONNECTORS = DOCS_BASE_URL + "/connectors/messaging-and-voice-channels"
|
|
39
|
+
DOCS_URL_CONNECTORS_SLACK = DOCS_BASE_URL + "/connectors/slack"
|
|
40
|
+
DOCS_URL_EVENT_BROKERS = DOCS_BASE_URL + "/production/event-brokers"
|
|
41
|
+
DOCS_URL_PIKA_EVENT_BROKER = DOCS_URL_EVENT_BROKERS + "#pika-event-broker"
|
|
42
|
+
DOCS_URL_TRACKER_STORES = DOCS_BASE_URL + "/production/tracker-stores"
|
|
43
|
+
DOCS_URL_MIGRATION_GUIDE = DOCS_BASE_URL + "/migration-guide"
|
|
44
|
+
DOCS_URL_TELEMETRY = DOCS_BASE_URL + "/telemetry/telemetry"
|
|
45
|
+
|
|
46
|
+
INTENT_MESSAGE_PREFIX = "/"
|
|
47
|
+
|
|
48
|
+
PACKAGE_NAME = "rasa"
|
|
49
|
+
NEXT_MAJOR_VERSION_FOR_DEPRECATIONS = "4.0.0"
|
|
50
|
+
|
|
51
|
+
MODEL_CONFIG_SCHEMA_FILE = "shared/utils/schemas/model_config.yml"
|
|
52
|
+
CONFIG_SCHEMA_FILE = "shared/utils/schemas/config.yml"
|
|
53
|
+
RESPONSES_SCHEMA_FILE = "shared/nlu/training_data/schemas/responses.yml"
|
|
54
|
+
SCHEMA_EXTENSIONS_FILE = "shared/utils/pykwalify_extensions.py"
|
|
55
|
+
ASSERTIONS_SCHEMA_FILE = "e2e_test/assertions_schema.yml"
|
|
56
|
+
ASSERTIONS_SCHEMA_EXTENSIONS_FILE = "e2e_test/pykwalify_extensions.py"
|
|
57
|
+
LATEST_TRAINING_DATA_FORMAT_VERSION = "3.1"
|
|
58
|
+
|
|
59
|
+
DOMAIN_SCHEMA_FILE = "shared/utils/schemas/domain.yml"
|
|
60
|
+
|
|
61
|
+
DEFAULT_SESSION_EXPIRATION_TIME_IN_MINUTES = 60
|
|
62
|
+
DEFAULT_CARRY_OVER_SLOTS_TO_NEW_SESSION = True
|
|
63
|
+
|
|
64
|
+
DEFAULT_NLU_FALLBACK_INTENT_NAME = "nlu_fallback"
|
|
65
|
+
|
|
66
|
+
DEFAULT_E2E_TESTS_PATH = "."
|
|
67
|
+
TEST_STORIES_FILE_PREFIX = "test_"
|
|
68
|
+
|
|
69
|
+
LOG_LEVEL_NAME_TO_LEVEL = {
|
|
70
|
+
"CRITICAL": 50,
|
|
71
|
+
"ERROR": 40,
|
|
72
|
+
"WARNING": 30,
|
|
73
|
+
"INFO": 20,
|
|
74
|
+
"DEBUG": 10,
|
|
75
|
+
"NOTSET": 0,
|
|
76
|
+
}
|
|
77
|
+
|
|
78
|
+
DEFAULT_LOG_LEVEL = "INFO"
|
|
79
|
+
ENV_LOG_LEVEL = "LOG_LEVEL"
|
|
80
|
+
DEFAULT_LOG_LEVEL_LLM = "DEBUG"
|
|
81
|
+
ENV_LOG_LEVEL_LLM = "LOG_LEVEL_LLM"
|
|
82
|
+
ENV_LOG_LEVEL_LLM_MODULE_NAMES = {
|
|
83
|
+
"LLMCommandGenerator": "LOG_LEVEL_LLM_COMMAND_GENERATOR",
|
|
84
|
+
"SingleStepLLMCommandGenerator": "LOG_LEVEL_LLM_COMMAND_GENERATOR",
|
|
85
|
+
"MultiStepLLMCommandGenerator": "LOG_LEVEL_LLM_COMMAND_GENERATOR",
|
|
86
|
+
"EnterpriseSearchPolicy": "LOG_LEVEL_LLM_ENTERPRISE_SEARCH",
|
|
87
|
+
"IntentlessPolicy": "LOG_LEVEL_LLM_INTENTLESS_POLICY",
|
|
88
|
+
"ContextualResponseRephraser": "LOG_LEVEL_LLM_REPHRASER",
|
|
89
|
+
"NLUCommandAdapter": "LOG_LEVEL_NLU_COMMAND_ADAPTER",
|
|
90
|
+
"LLMBasedRouter": "LOG_LEVEL_LLM_BASED_ROUTER",
|
|
91
|
+
}
|
|
92
|
+
TCP_PROTOCOL = "TCP"
|
|
93
|
+
|
|
94
|
+
DEFAULT_SENDER_ID = "default"
|
|
95
|
+
UTTER_PREFIX = "utter_"
|
|
96
|
+
UTTER_ASK_PREFIX = "utter_ask_"
|
|
97
|
+
ACTION_ASK_PREFIX = "action_ask_"
|
|
98
|
+
FLOW_PREFIX = "flow_"
|
|
99
|
+
|
|
100
|
+
ASSISTANT_ID_KEY = "assistant_id"
|
|
101
|
+
ASSISTANT_ID_DEFAULT_VALUE = "placeholder_default"
|
|
102
|
+
|
|
103
|
+
CONFIG_MANDATORY_COMMON_KEYS = [ASSISTANT_ID_KEY]
|
|
104
|
+
CONFIG_NAME_KEY = "name"
|
|
105
|
+
CONFIG_POLICIES_KEY = "policies"
|
|
106
|
+
CONFIG_PIPELINE_KEY = "pipeline"
|
|
107
|
+
CONFIG_LANGUAGE_KEY = "language"
|
|
108
|
+
CONFIG_RECIPE_KEY = "recipe"
|
|
109
|
+
CONFIG_LLM_KEY = "llm"
|
|
110
|
+
CONFIG_MODEL_NAME_KEY = "model_name"
|
|
111
|
+
CONFIG_TRAIN_SCHEMA = "train_schema"
|
|
112
|
+
CONFIG_PREDICT_SCHEMA = "predict_schema"
|
|
113
|
+
CONFIG_AUTOCONFIGURABLE_KEYS_CORE = [CONFIG_POLICIES_KEY]
|
|
114
|
+
CONFIG_AUTOCONFIGURABLE_KEYS_NLU = [CONFIG_PIPELINE_KEY]
|
|
115
|
+
CONFIG_AUTOCONFIGURABLE_KEYS = (
|
|
116
|
+
CONFIG_AUTOCONFIGURABLE_KEYS_CORE + CONFIG_AUTOCONFIGURABLE_KEYS_NLU
|
|
117
|
+
)
|
|
118
|
+
CONFIG_KEYS_CORE = [CONFIG_POLICIES_KEY] + CONFIG_MANDATORY_COMMON_KEYS
|
|
119
|
+
CONFIG_KEYS_NLU = [
|
|
120
|
+
CONFIG_LANGUAGE_KEY,
|
|
121
|
+
CONFIG_PIPELINE_KEY,
|
|
122
|
+
] + CONFIG_MANDATORY_COMMON_KEYS
|
|
123
|
+
CONFIG_KEYS = CONFIG_KEYS_CORE + CONFIG_KEYS_NLU
|
|
124
|
+
CONFIG_MANDATORY_KEYS_CORE: List[Text] = [] + CONFIG_MANDATORY_COMMON_KEYS
|
|
125
|
+
CONFIG_MANDATORY_KEYS_NLU = [CONFIG_LANGUAGE_KEY] + CONFIG_MANDATORY_COMMON_KEYS
|
|
126
|
+
CONFIG_MANDATORY_KEYS = CONFIG_MANDATORY_KEYS_CORE + CONFIG_MANDATORY_KEYS_NLU
|
|
127
|
+
|
|
128
|
+
# Keys related to Forms (in the Domain)
|
|
129
|
+
REQUIRED_SLOTS_KEY = "required_slots"
|
|
130
|
+
IGNORED_INTENTS = "ignored_intents"
|
|
131
|
+
|
|
132
|
+
# Constants for default Rasa Pro project layout
|
|
133
|
+
DEFAULT_ENDPOINTS_PATH = "endpoints.yml"
|
|
134
|
+
DEFAULT_CREDENTIALS_PATH = "credentials.yml"
|
|
135
|
+
DEFAULT_CONFIG_PATH = "config.yml"
|
|
136
|
+
DEFAULT_DOMAIN_PATHS = ["domain.yml", "domain"]
|
|
137
|
+
DEFAULT_DOMAIN_PATH = DEFAULT_DOMAIN_PATHS[0]
|
|
138
|
+
DEFAULT_ACTIONS_PATH = "actions"
|
|
139
|
+
DEFAULT_MODELS_PATH = "models"
|
|
140
|
+
DEFAULT_CONVERTED_DATA_PATH = "converted_data"
|
|
141
|
+
DEFAULT_DATA_PATH = "data"
|
|
142
|
+
DEFAULT_RESULTS_PATH = "results"
|
|
143
|
+
DEFAULT_NLU_RESULTS_PATH = "nlu_comparison_results"
|
|
144
|
+
DEFAULT_CORE_SUBDIRECTORY_NAME = "core"
|
|
145
|
+
DEFAULT_NLU_SUBDIRECTORY_NAME = "nlu"
|
|
146
|
+
DEFAULT_CONVERSATION_TEST_PATH = "tests"
|
|
147
|
+
DEFAULT_MARKERS_PATH = "markers"
|
|
148
|
+
DEFAULT_MARKERS_CONFIG_PATH = "markers/config"
|
|
149
|
+
DEFAULT_MARKERS_OUTPUT_PATH = "markers/output"
|
|
150
|
+
DEFAULT_MARKERS_STATS_PATH = "markers/stats"
|
|
151
|
+
|
|
152
|
+
DIAGNOSTIC_DATA = "diagnostic_data"
|
|
153
|
+
|
|
154
|
+
RESPONSE_CONDITION = "condition"
|
|
155
|
+
CHANNEL = "channel"
|
|
156
|
+
|
|
157
|
+
API_KEY = "api_key"
|
|
158
|
+
|
|
159
|
+
AZURE_API_KEY_ENV_VAR = "AZURE_API_KEY"
|
|
160
|
+
AZURE_AD_TOKEN_ENV_VAR = "AZURE_AD_TOKEN"
|
|
161
|
+
AZURE_API_BASE_ENV_VAR = "AZURE_API_BASE"
|
|
162
|
+
AZURE_API_VERSION_ENV_VAR = "AZURE_API_VERSION"
|
|
163
|
+
AZURE_API_TYPE_ENV_VAR = "AZURE_API_TYPE"
|
|
164
|
+
AZURE_AD_SCOPES_ENV_VAR = "AZURE_AD_SCOPES"
|
|
165
|
+
AZURE_SPEECH_API_KEY_ENV_VAR = "AZURE_SPEECH_API_KEY"
|
|
166
|
+
|
|
167
|
+
DEEPGRAM_API_KEY_ENV_VAR = "DEEPGRAM_API_KEY"
|
|
168
|
+
CARTESIA_API_KEY_ENV_VAR = "CARTESIA_API_KEY"
|
|
169
|
+
|
|
170
|
+
OPENAI_API_KEY_ENV_VAR = "OPENAI_API_KEY"
|
|
171
|
+
OPENAI_API_TYPE_ENV_VAR = "OPENAI_API_TYPE"
|
|
172
|
+
OPENAI_API_VERSION_ENV_VAR = "OPENAI_API_VERSION"
|
|
173
|
+
OPENAI_API_BASE_ENV_VAR = "OPENAI_API_BASE"
|
|
174
|
+
|
|
175
|
+
OPENAI_API_BASE_CONFIG_KEY = "openai_api_base"
|
|
176
|
+
OPENAI_API_TYPE_CONFIG_KEY = "openai_api_type"
|
|
177
|
+
OPENAI_API_VERSION_CONFIG_KEY = "openai_api_version"
|
|
178
|
+
|
|
179
|
+
AWS_BEDROCK_PROVIDER = "bedrock"
|
|
180
|
+
AWS_SAGEMAKER_PROVIDER = "sagemaker"
|
|
181
|
+
|
|
182
|
+
API_BASE_CONFIG_KEY = "api_base"
|
|
183
|
+
API_TYPE_CONFIG_KEY = "api_type"
|
|
184
|
+
API_VERSION_CONFIG_KEY = "api_version"
|
|
185
|
+
LANGCHAIN_TYPE_CONFIG_KEY = "_type"
|
|
186
|
+
RASA_TYPE_CONFIG_KEY = "type"
|
|
187
|
+
PROVIDER_CONFIG_KEY = "provider"
|
|
188
|
+
|
|
189
|
+
REQUEST_TIMEOUT_CONFIG_KEY = "request_timeout" # deprecated
|
|
190
|
+
TIMEOUT_CONFIG_KEY = "timeout"
|
|
191
|
+
|
|
192
|
+
DEPLOYMENT_NAME_CONFIG_KEY = "deployment_name"
|
|
193
|
+
DEPLOYMENT_CONFIG_KEY = "deployment"
|
|
194
|
+
EMBEDDINGS_CONFIG_KEY = "embeddings"
|
|
195
|
+
ENGINE_CONFIG_KEY = "engine"
|
|
196
|
+
LLM_CONFIG_KEY = "llm"
|
|
197
|
+
MODEL_CONFIG_KEY = "model"
|
|
198
|
+
MODEL_NAME_CONFIG_KEY = "model_name"
|
|
199
|
+
PROMPT_CONFIG_KEY = "prompt"
|
|
200
|
+
PROMPT_TEMPLATE_CONFIG_KEY = "prompt_template"
|
|
201
|
+
STREAM_CONFIG_KEY = "stream"
|
|
202
|
+
N_REPHRASES_CONFIG_KEY = "n"
|
|
203
|
+
USE_CHAT_COMPLETIONS_ENDPOINT_CONFIG_KEY = "use_chat_completions_endpoint"
|
|
204
|
+
|
|
205
|
+
ROUTER_CONFIG_KEY = "router"
|
|
206
|
+
ROUTING_STRATEGY_CONFIG_KEY = "routing_strategy"
|
|
207
|
+
REDIS_HOST_CONFIG_KEY = "redis_host"
|
|
208
|
+
ROUTING_STRATEGIES_REQUIRING_REDIS_CACHE = [
|
|
209
|
+
"cost-based-routing",
|
|
210
|
+
"usage-based-routing",
|
|
211
|
+
]
|
|
212
|
+
ROUTING_STRATEGIES_NOT_REQUIRING_CACHE = [
|
|
213
|
+
"latency-based-routing",
|
|
214
|
+
"least-busy",
|
|
215
|
+
"simple-shuffle",
|
|
216
|
+
]
|
|
217
|
+
VALID_ROUTING_STRATEGIES = (
|
|
218
|
+
ROUTING_STRATEGIES_REQUIRING_REDIS_CACHE + ROUTING_STRATEGIES_NOT_REQUIRING_CACHE
|
|
219
|
+
)
|
|
220
|
+
|
|
221
|
+
MODELS_CONFIG_KEY = "models"
|
|
222
|
+
MODEL_GROUPS_CONFIG_KEY = "model_groups"
|
|
223
|
+
MODEL_GROUP_CONFIG_KEY = "model_group"
|
|
224
|
+
MODEL_GROUP_ID_CONFIG_KEY = "id"
|
|
225
|
+
|
|
226
|
+
EXTRA_PARAMETERS_KEY = "extra_parameters"
|
|
227
|
+
MODEL_GROUP_ID_KEY = "model_group_id"
|
|
228
|
+
MODEL_LIST_KEY = "model_list"
|
|
229
|
+
LITELLM_PARAMS_KEY = "litellm_params"
|
|
230
|
+
|
|
231
|
+
LLM_API_HEALTH_CHECK_ENV_VAR = "LLM_API_HEALTH_CHECK"
|
|
232
|
+
LLM_API_HEALTH_CHECK_DEFAULT_VALUE = "false"
|
|
233
|
+
|
|
234
|
+
AWS_REGION_NAME_CONFIG_KEY = "aws_region_name"
|
|
235
|
+
AWS_ACCESS_KEY_ID_CONFIG_KEY = "aws_access_key_id"
|
|
236
|
+
AWS_SECRET_ACCESS_KEY_CONFIG_KEY = "aws_secret_access_key"
|
|
237
|
+
AWS_SESSION_TOKEN_CONFIG_KEY = "aws_session_token"
|
|
238
|
+
|
|
239
|
+
AWS_ACCESS_KEY_ID_ENV_VAR = "AWS_ACCESS_KEY_ID"
|
|
240
|
+
AWS_SECRET_ACCESS_KEY_ENV_VAR = "AWS_SECRET_ACCESS_KEY"
|
|
241
|
+
AWS_REGION_NAME_ENV_VAR = "AWS_REGION_NAME"
|
|
242
|
+
AWS_SESSION_TOKEN_ENV_VAR = "AWS_SESSION_TOKEN"
|
|
243
|
+
|
|
244
|
+
HUGGINGFACE_MULTIPROCESS_CONFIG_KEY = "multi_process"
|
|
245
|
+
HUGGINGFACE_CACHE_FOLDER_CONFIG_KEY = "cache_folder"
|
|
246
|
+
HUGGINGFACE_SHOW_PROGRESS_CONFIG_KEY = "show_progress"
|
|
247
|
+
HUGGINGFACE_MODEL_KWARGS_CONFIG_KEY = "model_kwargs"
|
|
248
|
+
HUGGINGFACE_ENCODE_KWARGS_CONFIG_KEY = "encode_kwargs"
|
|
249
|
+
HUGGINGFACE_LOCAL_EMBEDDING_CACHING_FOLDER = (
|
|
250
|
+
get_local_cache_location() / "huggingface_local_embeddings"
|
|
251
|
+
)
|
|
252
|
+
|
|
253
|
+
REQUESTS_CA_BUNDLE_ENV_VAR = "REQUESTS_CA_BUNDLE"
|
|
254
|
+
REQUESTS_SSL_CONTEXT_PURPOSE_ENV_VAR = "REQUESTS_SSL_CONTEXT_PURPOSE"
|
|
255
|
+
RASA_CA_BUNDLE_ENV_VAR = "RASA_CA_BUNDLE" # used in verify
|
|
256
|
+
RASA_SSL_CERTIFICATE_ENV_VAR = "RASA_SSL_CERTIFICATE" # used in cert (client side)
|
|
257
|
+
LITELLM_SSL_VERIFY_ENV_VAR = "SSL_VERIFY"
|
|
258
|
+
LITELLM_SSL_CERTIFICATE_ENV_VAR = "SSL_CERTIFICATE"
|
|
259
|
+
|
|
260
|
+
OPENAI_PROVIDER = "openai"
|
|
261
|
+
AZURE_OPENAI_PROVIDER = "azure"
|
|
262
|
+
SELF_HOSTED_PROVIDER = "self-hosted"
|
|
263
|
+
HUGGINGFACE_LOCAL_EMBEDDING_PROVIDER = "huggingface_local"
|
|
264
|
+
RASA_PROVIDER = "rasa"
|
|
265
|
+
|
|
266
|
+
SELF_HOSTED_VLLM_PREFIX = "hosted_vllm"
|
|
267
|
+
SELF_HOSTED_VLLM_API_KEY_ENV_VAR = "HOSTED_VLLM_API_KEY"
|
|
268
|
+
|
|
269
|
+
VALID_PROVIDERS_FOR_API_TYPE_CONFIG_KEY = [
|
|
270
|
+
OPENAI_PROVIDER,
|
|
271
|
+
AZURE_OPENAI_PROVIDER,
|
|
272
|
+
]
|
|
273
|
+
|
|
274
|
+
AZURE_API_TYPE = "azure"
|
|
275
|
+
OPENAI_API_TYPE = "openai"
|
|
276
|
+
|
|
277
|
+
RASA_DEFAULT_FLOW_PATTERN_PREFIX = "pattern_"
|
|
278
|
+
CONTEXT = "context"
|
|
279
|
+
|
|
280
|
+
RASA_PATTERN_INTERNAL_ERROR = "pattern_internal_error"
|
|
281
|
+
RASA_PATTERN_HUMAN_HANDOFF = "pattern_human_handoff"
|
|
282
|
+
|
|
283
|
+
RASA_INTERNAL_ERROR_PREFIX = "rasa_internal_error_"
|
|
284
|
+
RASA_PATTERN_INTERNAL_ERROR_DEFAULT = RASA_INTERNAL_ERROR_PREFIX + "default"
|
|
285
|
+
RASA_PATTERN_INTERNAL_ERROR_USER_INPUT_TOO_LONG = (
|
|
286
|
+
RASA_INTERNAL_ERROR_PREFIX + "user_input_too_long"
|
|
287
|
+
)
|
|
288
|
+
RASA_PATTERN_INTERNAL_ERROR_USER_INPUT_EMPTY = (
|
|
289
|
+
RASA_INTERNAL_ERROR_PREFIX + "user_input_empty"
|
|
290
|
+
)
|
|
291
|
+
|
|
292
|
+
RASA_PATTERN_CANNOT_HANDLE_PREFIX = "cannot_handle_"
|
|
293
|
+
RASA_PATTERN_CANNOT_HANDLE_DEFAULT = RASA_PATTERN_CANNOT_HANDLE_PREFIX + "default"
|
|
294
|
+
RASA_PATTERN_CANNOT_HANDLE_CHITCHAT = RASA_PATTERN_CANNOT_HANDLE_PREFIX + "chitchat"
|
|
295
|
+
RASA_PATTERN_CANNOT_HANDLE_NOT_SUPPORTED = (
|
|
296
|
+
RASA_PATTERN_CANNOT_HANDLE_PREFIX + "not_supported"
|
|
297
|
+
)
|
|
298
|
+
RASA_PATTERN_CANNOT_HANDLE_INVALID_INTENT = (
|
|
299
|
+
RASA_PATTERN_CANNOT_HANDLE_PREFIX + "invalid_intent"
|
|
300
|
+
)
|
|
301
|
+
|
|
302
|
+
ROUTE_TO_CALM_SLOT = "route_session_to_calm"
|
|
303
|
+
|
|
304
|
+
SENSITIVE_DATA = [
|
|
305
|
+
API_KEY,
|
|
306
|
+
AWS_ACCESS_KEY_ID_CONFIG_KEY,
|
|
307
|
+
AWS_SECRET_ACCESS_KEY_CONFIG_KEY,
|
|
308
|
+
AWS_SESSION_TOKEN_CONFIG_KEY,
|
|
309
|
+
]
|
|
310
|
+
|
|
311
|
+
AZURE_AD_KEY = "azure_ad"
|
|
File without changes
|
|
@@ -0,0 +1,109 @@
|
|
|
1
|
+
import copy
|
|
2
|
+
import re
|
|
3
|
+
from typing import List, Optional
|
|
4
|
+
|
|
5
|
+
import structlog
|
|
6
|
+
|
|
7
|
+
from rasa.dialogue_understanding.commands.set_slot_command import SetSlotExtractor
|
|
8
|
+
from rasa.shared.core.domain import Domain
|
|
9
|
+
from rasa.shared.nlu.constants import COMMANDS, SET_SLOT_COMMAND, TEXT
|
|
10
|
+
from rasa.shared.nlu.training_data.message import Message
|
|
11
|
+
|
|
12
|
+
structlogger = structlog.get_logger()
|
|
13
|
+
|
|
14
|
+
SET_SLOTS_PATTERN = r"(?P<slot_name>[^(),=]+)=(?P<slot_value>[^(),]+)"
|
|
15
|
+
MAX_NUMBER_OF_SLOTS = 10
|
|
16
|
+
|
|
17
|
+
|
|
18
|
+
class CommandPayloadReader:
|
|
19
|
+
@staticmethod
|
|
20
|
+
def unpack_regex_message(
|
|
21
|
+
message: Message,
|
|
22
|
+
domain: Optional[Domain] = None,
|
|
23
|
+
entity_extractor_name: Optional[
|
|
24
|
+
str
|
|
25
|
+
] = SetSlotExtractor.COMMAND_PAYLOAD_READER.value,
|
|
26
|
+
) -> Message:
|
|
27
|
+
"""Extracts commands from user text and adds them to the message."""
|
|
28
|
+
user_text = message.get(TEXT).strip()
|
|
29
|
+
|
|
30
|
+
# prevent ReDos attacks by placing a limit on the number of slots
|
|
31
|
+
if CommandPayloadReader.is_above_slot_limit(user_text):
|
|
32
|
+
return message
|
|
33
|
+
|
|
34
|
+
matches = CommandPayloadReader.find_matches(user_text)
|
|
35
|
+
if not matches:
|
|
36
|
+
structlogger.warning(
|
|
37
|
+
"message.parsing.failed", user_text=copy.deepcopy(user_text)
|
|
38
|
+
)
|
|
39
|
+
return message
|
|
40
|
+
|
|
41
|
+
return CommandPayloadReader.extract_commands_from_pattern_matches(
|
|
42
|
+
message, matches, domain, entity_extractor_name
|
|
43
|
+
)
|
|
44
|
+
|
|
45
|
+
@staticmethod
|
|
46
|
+
def find_matches(user_text: str) -> List[re.Match]:
|
|
47
|
+
return re.findall(SET_SLOTS_PATTERN, user_text)
|
|
48
|
+
|
|
49
|
+
@staticmethod
|
|
50
|
+
def extract_commands_from_pattern_matches(
|
|
51
|
+
message: Message,
|
|
52
|
+
matches: List[re.Match],
|
|
53
|
+
domain: Optional[Domain] = None,
|
|
54
|
+
entity_extractor_name: Optional[
|
|
55
|
+
str
|
|
56
|
+
] = SetSlotExtractor.COMMAND_PAYLOAD_READER.value,
|
|
57
|
+
) -> Message:
|
|
58
|
+
"""Extract attributes from the matches and validate them via the domain."""
|
|
59
|
+
commands = []
|
|
60
|
+
|
|
61
|
+
for match in matches:
|
|
62
|
+
slot_name = match[0].strip()
|
|
63
|
+
slot_value = match[1].strip()
|
|
64
|
+
|
|
65
|
+
domain_slot_names = [slot.name for slot in domain.slots] if domain else []
|
|
66
|
+
|
|
67
|
+
if domain and slot_name not in domain_slot_names:
|
|
68
|
+
structlogger.warning(
|
|
69
|
+
"slot.not.found",
|
|
70
|
+
slot_name=slot_name,
|
|
71
|
+
)
|
|
72
|
+
return message
|
|
73
|
+
|
|
74
|
+
slot_value = (
|
|
75
|
+
slot_value
|
|
76
|
+
if slot_value.lower() not in {"none", "null", "undefined"}
|
|
77
|
+
else None
|
|
78
|
+
)
|
|
79
|
+
|
|
80
|
+
# Create new SetSlot commands from the extracted attributes.
|
|
81
|
+
commands.append(
|
|
82
|
+
{
|
|
83
|
+
"command": SET_SLOT_COMMAND,
|
|
84
|
+
"name": slot_name,
|
|
85
|
+
"value": slot_value,
|
|
86
|
+
"extractor": entity_extractor_name,
|
|
87
|
+
}
|
|
88
|
+
)
|
|
89
|
+
|
|
90
|
+
structlogger.debug(
|
|
91
|
+
"slot.set.command.added",
|
|
92
|
+
slot_name=slot_name,
|
|
93
|
+
)
|
|
94
|
+
|
|
95
|
+
# set the command(s) on the Message object
|
|
96
|
+
message.set(COMMANDS, commands, add_to_output=True)
|
|
97
|
+
return message
|
|
98
|
+
|
|
99
|
+
@staticmethod
|
|
100
|
+
def is_above_slot_limit(user_text: str) -> bool:
|
|
101
|
+
"""Prevent ReDoS attacks by limiting the number of slots."""
|
|
102
|
+
if user_text.count("=") > MAX_NUMBER_OF_SLOTS:
|
|
103
|
+
structlogger.warning(
|
|
104
|
+
"too.many.slots",
|
|
105
|
+
user_text=copy.deepcopy(user_text),
|
|
106
|
+
slot_limit=10,
|
|
107
|
+
)
|
|
108
|
+
return True
|
|
109
|
+
return False
|
|
@@ -0,0 +1,180 @@
|
|
|
1
|
+
from __future__ import annotations
|
|
2
|
+
from enum import Enum
|
|
3
|
+
|
|
4
|
+
import rasa.shared.constants as constants
|
|
5
|
+
|
|
6
|
+
|
|
7
|
+
DEFAULT_CATEGORICAL_SLOT_VALUE = "__other__"
|
|
8
|
+
|
|
9
|
+
USER_INTENT_RESTART = "restart"
|
|
10
|
+
USER_INTENT_BACK = "back"
|
|
11
|
+
USER_INTENT_OUT_OF_SCOPE = "out_of_scope"
|
|
12
|
+
USER_INTENT_SESSION_START = "session_start"
|
|
13
|
+
USER_INTENT_SESSION_END = "session_end"
|
|
14
|
+
USER_INTENT_SILENCE_TIMEOUT = "silence_timeout"
|
|
15
|
+
SESSION_START_METADATA_SLOT = "session_started_metadata"
|
|
16
|
+
|
|
17
|
+
DEFAULT_INTENTS = [
|
|
18
|
+
USER_INTENT_RESTART,
|
|
19
|
+
USER_INTENT_BACK,
|
|
20
|
+
USER_INTENT_OUT_OF_SCOPE,
|
|
21
|
+
USER_INTENT_SESSION_START,
|
|
22
|
+
USER_INTENT_SESSION_END,
|
|
23
|
+
USER_INTENT_SILENCE_TIMEOUT,
|
|
24
|
+
constants.DEFAULT_NLU_FALLBACK_INTENT_NAME,
|
|
25
|
+
]
|
|
26
|
+
|
|
27
|
+
LOOP_NAME = "name"
|
|
28
|
+
|
|
29
|
+
ACTION_LISTEN_NAME = "action_listen"
|
|
30
|
+
ACTION_RESTART_NAME = "action_restart"
|
|
31
|
+
ACTION_SEND_TEXT_NAME = "action_send_text"
|
|
32
|
+
ACTION_SESSION_START_NAME = "action_session_start"
|
|
33
|
+
ACTION_DEFAULT_FALLBACK_NAME = "action_default_fallback"
|
|
34
|
+
ACTION_DEACTIVATE_LOOP_NAME = "action_deactivate_loop"
|
|
35
|
+
ACTION_REVERT_FALLBACK_EVENTS_NAME = "action_revert_fallback_events"
|
|
36
|
+
ACTION_DEFAULT_ASK_AFFIRMATION_NAME = "action_default_ask_affirmation"
|
|
37
|
+
ACTION_DEFAULT_ASK_REPHRASE_NAME = "action_default_ask_rephrase"
|
|
38
|
+
ACTION_BACK_NAME = "action_back"
|
|
39
|
+
ACTION_TWO_STAGE_FALLBACK_NAME = "action_two_stage_fallback"
|
|
40
|
+
ACTION_UNLIKELY_INTENT_NAME = "action_unlikely_intent"
|
|
41
|
+
RULE_SNIPPET_ACTION_NAME = "..."
|
|
42
|
+
ACTION_EXTRACT_SLOTS = "action_extract_slots"
|
|
43
|
+
ACTION_VALIDATE_SLOT_MAPPINGS = "action_validate_slot_mappings"
|
|
44
|
+
ACTION_CANCEL_FLOW = "action_cancel_flow"
|
|
45
|
+
ACTION_CLARIFY_FLOWS = "action_clarify_flows"
|
|
46
|
+
ACTION_CORRECT_FLOW_SLOT = "action_correct_flow_slot"
|
|
47
|
+
ACTION_RUN_SLOT_REJECTIONS_NAME = "action_run_slot_rejections"
|
|
48
|
+
ACTION_CLEAN_STACK = "action_clean_stack"
|
|
49
|
+
ACTION_TRIGGER_SEARCH = "action_trigger_search"
|
|
50
|
+
ACTION_TRIGGER_CHITCHAT = "action_trigger_chitchat"
|
|
51
|
+
ACTION_RESET_ROUTING = "action_reset_routing"
|
|
52
|
+
ACTION_HANGUP = "action_hangup"
|
|
53
|
+
ACTION_REPEAT_BOT_MESSAGES = "action_repeat_bot_messages"
|
|
54
|
+
|
|
55
|
+
|
|
56
|
+
DEFAULT_ACTION_NAMES = [
|
|
57
|
+
ACTION_LISTEN_NAME,
|
|
58
|
+
ACTION_RESTART_NAME,
|
|
59
|
+
ACTION_SESSION_START_NAME,
|
|
60
|
+
ACTION_DEFAULT_FALLBACK_NAME,
|
|
61
|
+
ACTION_DEACTIVATE_LOOP_NAME,
|
|
62
|
+
ACTION_REVERT_FALLBACK_EVENTS_NAME,
|
|
63
|
+
ACTION_DEFAULT_ASK_AFFIRMATION_NAME,
|
|
64
|
+
ACTION_DEFAULT_ASK_REPHRASE_NAME,
|
|
65
|
+
ACTION_TWO_STAGE_FALLBACK_NAME,
|
|
66
|
+
ACTION_UNLIKELY_INTENT_NAME,
|
|
67
|
+
ACTION_BACK_NAME,
|
|
68
|
+
ACTION_SEND_TEXT_NAME,
|
|
69
|
+
RULE_SNIPPET_ACTION_NAME,
|
|
70
|
+
ACTION_EXTRACT_SLOTS,
|
|
71
|
+
ACTION_CANCEL_FLOW,
|
|
72
|
+
ACTION_CORRECT_FLOW_SLOT,
|
|
73
|
+
ACTION_CLARIFY_FLOWS,
|
|
74
|
+
ACTION_RUN_SLOT_REJECTIONS_NAME,
|
|
75
|
+
ACTION_CLEAN_STACK,
|
|
76
|
+
ACTION_TRIGGER_SEARCH,
|
|
77
|
+
ACTION_TRIGGER_CHITCHAT,
|
|
78
|
+
ACTION_RESET_ROUTING,
|
|
79
|
+
ACTION_HANGUP,
|
|
80
|
+
ACTION_REPEAT_BOT_MESSAGES,
|
|
81
|
+
]
|
|
82
|
+
|
|
83
|
+
ACTION_SHOULD_SEND_DOMAIN = "send_domain"
|
|
84
|
+
|
|
85
|
+
# rules allow setting a value of slots or active_loops to None;
|
|
86
|
+
# generator substitutes `None`s with this constant to notify rule policy that
|
|
87
|
+
# a value should not be set during prediction to activate a rule
|
|
88
|
+
SHOULD_NOT_BE_SET = "should_not_be_set"
|
|
89
|
+
|
|
90
|
+
PREVIOUS_ACTION = "prev_action"
|
|
91
|
+
ACTIVE_LOOP = "active_loop"
|
|
92
|
+
LOOP_INTERRUPTED = "is_interrupted"
|
|
93
|
+
LOOP_REJECTED = "rejected"
|
|
94
|
+
TRIGGER_MESSAGE = "trigger_message"
|
|
95
|
+
FOLLOWUP_ACTION = "followup_action"
|
|
96
|
+
ACTIVE_FLOW = "active_flow"
|
|
97
|
+
|
|
98
|
+
# start of special user message section
|
|
99
|
+
EXTERNAL_MESSAGE_PREFIX = "EXTERNAL: "
|
|
100
|
+
# Key to access data in the event metadata
|
|
101
|
+
# It specifies if an event was caused by an external entity (e.g. a sensor).
|
|
102
|
+
IS_EXTERNAL = "is_external"
|
|
103
|
+
|
|
104
|
+
ACTION_NAME_SENDER_ID_CONNECTOR_STR = "__sender_id:"
|
|
105
|
+
|
|
106
|
+
REQUESTED_SLOT = "requested_slot"
|
|
107
|
+
FLOW_HASHES_SLOT = "flow_hashes"
|
|
108
|
+
|
|
109
|
+
FLOW_SLOT_NAMES = [FLOW_HASHES_SLOT]
|
|
110
|
+
|
|
111
|
+
# slots for audio timeout
|
|
112
|
+
SLOT_SILENCE_TIMEOUT = "silence_timeout"
|
|
113
|
+
SLOT_CONSECUTIVE_SILENCE_TIMEOUTS = "consecutive_silence_timeouts"
|
|
114
|
+
SILENCE_TIMEOUT_DEFAULT_VALUE = 6.0
|
|
115
|
+
SILENCE_SLOTS = [SLOT_SILENCE_TIMEOUT, SLOT_CONSECUTIVE_SILENCE_TIMEOUTS]
|
|
116
|
+
# slots for knowledge base
|
|
117
|
+
SLOT_LISTED_ITEMS = "knowledge_base_listed_objects"
|
|
118
|
+
SLOT_LAST_OBJECT = "knowledge_base_last_object"
|
|
119
|
+
SLOT_LAST_OBJECT_TYPE = "knowledge_base_last_object_type"
|
|
120
|
+
DEFAULT_KNOWLEDGE_BASE_ACTION = "action_query_knowledge_base"
|
|
121
|
+
|
|
122
|
+
KNOWLEDGE_BASE_SLOT_NAMES = {
|
|
123
|
+
SLOT_LISTED_ITEMS,
|
|
124
|
+
SLOT_LAST_OBJECT,
|
|
125
|
+
SLOT_LAST_OBJECT_TYPE,
|
|
126
|
+
}
|
|
127
|
+
|
|
128
|
+
DEFAULT_SLOT_NAMES = {
|
|
129
|
+
REQUESTED_SLOT,
|
|
130
|
+
SESSION_START_METADATA_SLOT,
|
|
131
|
+
FLOW_HASHES_SLOT,
|
|
132
|
+
}
|
|
133
|
+
|
|
134
|
+
|
|
135
|
+
SLOT_MAPPINGS = "mappings"
|
|
136
|
+
MAPPING_CONDITIONS = "conditions"
|
|
137
|
+
MAPPING_TYPE = "type"
|
|
138
|
+
|
|
139
|
+
|
|
140
|
+
class SlotMappingType(Enum):
|
|
141
|
+
"""Slot mapping types."""
|
|
142
|
+
|
|
143
|
+
FROM_ENTITY = "from_entity"
|
|
144
|
+
FROM_INTENT = "from_intent"
|
|
145
|
+
FROM_TRIGGER_INTENT = "from_trigger_intent"
|
|
146
|
+
FROM_TEXT = "from_text"
|
|
147
|
+
FROM_LLM = "from_llm"
|
|
148
|
+
CUSTOM = "custom"
|
|
149
|
+
|
|
150
|
+
def __str__(self) -> str:
|
|
151
|
+
"""Returns the string representation that should be used in config files."""
|
|
152
|
+
return self.value
|
|
153
|
+
|
|
154
|
+
def is_predefined_type(self) -> bool:
|
|
155
|
+
"""Returns True if the mapping type is NLU-predefined."""
|
|
156
|
+
return not (self == SlotMappingType.CUSTOM or self == SlotMappingType.FROM_LLM)
|
|
157
|
+
|
|
158
|
+
|
|
159
|
+
# the keys for `State` (USER, PREVIOUS_ACTION, SLOTS, ACTIVE_LOOP)
|
|
160
|
+
# represent the origin of a `SubState`
|
|
161
|
+
USER = "user"
|
|
162
|
+
SLOTS = "slots"
|
|
163
|
+
|
|
164
|
+
USE_TEXT_FOR_FEATURIZATION = "use_text_for_featurization"
|
|
165
|
+
ENTITY_LABEL_SEPARATOR = "#"
|
|
166
|
+
|
|
167
|
+
RULE_ONLY_SLOTS = "rule_only_slots"
|
|
168
|
+
RULE_ONLY_LOOPS = "rule_only_loops"
|
|
169
|
+
|
|
170
|
+
# if you add more policy/classifier names, make sure to add a test as well to ensure
|
|
171
|
+
# that the name and the class stay in sync
|
|
172
|
+
POLICY_NAME_TWO_STAGE_FALLBACK = "TwoStageFallbackPolicy"
|
|
173
|
+
POLICY_NAME_MAPPING = "MappingPolicy"
|
|
174
|
+
POLICY_NAME_FALLBACK = "FallbackPolicy"
|
|
175
|
+
POLICY_NAME_FORM = "FormPolicy"
|
|
176
|
+
POLICY_NAME_RULE = "RulePolicy"
|
|
177
|
+
|
|
178
|
+
CLASSIFIER_NAME_FALLBACK = "FallbackClassifier"
|
|
179
|
+
|
|
180
|
+
POLICIES_THAT_EXTRACT_ENTITIES = {"TEDPolicy"}
|
|
@@ -0,0 +1,46 @@
|
|
|
1
|
+
from typing import Dict, List, Text, Any, TYPE_CHECKING
|
|
2
|
+
|
|
3
|
+
import rasa.shared.core.events
|
|
4
|
+
|
|
5
|
+
|
|
6
|
+
if TYPE_CHECKING:
|
|
7
|
+
from rasa.shared.core.events import Event
|
|
8
|
+
|
|
9
|
+
|
|
10
|
+
class Dialogue:
|
|
11
|
+
"""A dialogue comprises a list of Turn objects."""
|
|
12
|
+
|
|
13
|
+
def __init__(self, name: Text, events: List["Event"]) -> None:
|
|
14
|
+
"""This function initialises the dialogue with the dialogue name and the event
|
|
15
|
+
list.
|
|
16
|
+
"""
|
|
17
|
+
self.name = name
|
|
18
|
+
self.events = events
|
|
19
|
+
|
|
20
|
+
def __str__(self) -> Text:
|
|
21
|
+
"""This function returns the dialogue and turns."""
|
|
22
|
+
return "Dialogue with name '{}' and turns:\n{}".format(
|
|
23
|
+
self.name, "\n\n".join([f"\t{t}" for t in self.events])
|
|
24
|
+
)
|
|
25
|
+
|
|
26
|
+
def as_dict(self) -> Dict:
|
|
27
|
+
"""This function returns the dialogue as a dictionary to assist in
|
|
28
|
+
serialization.
|
|
29
|
+
"""
|
|
30
|
+
return {"events": [event.as_dict() for event in self.events], "name": self.name}
|
|
31
|
+
|
|
32
|
+
@classmethod
|
|
33
|
+
def from_parameters(cls, parameters: Dict[Text, Any]) -> "Dialogue":
|
|
34
|
+
"""Create `Dialogue` from parameters.
|
|
35
|
+
|
|
36
|
+
Args:
|
|
37
|
+
parameters: Serialised dialogue, should contain keys 'name' and 'events'.
|
|
38
|
+
|
|
39
|
+
Returns:
|
|
40
|
+
Deserialised `Dialogue`.
|
|
41
|
+
|
|
42
|
+
"""
|
|
43
|
+
return cls(
|
|
44
|
+
parameters.get("name"),
|
|
45
|
+
rasa.shared.core.events.deserialise_events(parameters.get("events", [])),
|
|
46
|
+
)
|