rasa-pro 3.10.7.dev3__py3-none-any.whl → 3.10.7.dev5__py3-none-any.whl

rasa/model_manager/model_api.py

@@ -53,26 +53,30 @@ def prepare_working_directories() -> None:
 def cleanup_training_processes() -> None:
     """Terminate all running training processes."""
     structlogger.debug("model_trainer.cleanup_processes.started")
-    for training in trainings.values():
+    running = list(trainings.values())
+    for training in running:
         terminate_training(training)
 
 
 def cleanup_bot_processes() -> None:
     """Terminate all running bot processes."""
     structlogger.debug("model_runner.cleanup_processes.started")
-    for bot in running_bots.values():
+    running = list(running_bots.values())
+    for bot in running:
         terminate_bot(bot)
 
 
 def update_status_of_all_trainings() -> None:
     """Update the status of all training processes."""
-    for training in trainings.values():
+    running = list(trainings.values())
+    for training in running:
         update_training_status(training)
 
 
 async def update_status_of_all_bots() -> None:
     """Update the status of all bot processes."""
-    for bot in running_bots.values():
+    running = list(running_bots.values())
+    for bot in running:
         await update_bot_status(bot)
 
 
@@ -127,9 +131,12 @@ async def continuously_update_process_status() -> None:
     structlogger.debug("model_api.update_process_status.started")
 
     while True:
-        update_status_of_all_trainings()
-        await update_status_of_all_bots()
-        await asyncio.sleep(1)
+        try:
+            update_status_of_all_trainings()
+            await update_status_of_all_bots()
+            await asyncio.sleep(1)
+        except Exception as e:
+            structlogger.error("model_api.update_process_status.error", error=str(e))
 
 
 def internal_blueprint() -> Blueprint:
@@ -360,6 +367,33 @@ def external_blueprint() -> Blueprint:
     sio = AsyncServer(async_mode="sanic", cors_allowed_origins=[])
     bp = SocketBlueprint(sio, "", "model_api_external")
 
+    @bp.get("/health")
+    async def health(request: Request) -> response.HTTPResponse:
+        return json(
+            {
+                "status": "ok",
+                "bots": [
+                    {
+                        "deployment_id": bot.deployment_id,
+                        "status": bot.status,
+                        "internal_url": bot.internal_url,
+                        "url": bot.url,
+                    }
+                    for bot in running_bots.values()
+                ],
+                "trainings": [
+                    {
+                        "training_id": training.training_id,
+                        "assistant_id": training.assistant_id,
+                        "client_id": training.client_id,
+                        "progress": training.progress,
+                        "status": training.status,
+                    }
+                    for training in trainings.values()
+                ],
+            }
+        )
+
     @bp.route("/logs/<path:path>")
     async def get_training_logs(request: Request, path: str) -> response.HTTPResponse:
         try:

rasa/version.py

@@ -1,3 +1,3 @@
 # this file will automatically be changed,
 # do not add anything but the version number here!
-__version__ = "3.10.7.dev3"
+__version__ = "3.10.7.dev5"

{rasa_pro-3.10.7.dev3.dist-info → rasa_pro-3.10.7.dev5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rasa-pro
-Version: 3.10.7.dev3
+Version: 3.10.7.dev5
 Summary: State-of-the-art open-core Conversational AI framework for Enterprises that natively leverages generative AI for effortless assistant development.
 Home-page: https://rasa.com
 Keywords: nlp,machine-learning,machine-learning-library,bot,bots,botkit,rasa conversational-agents,conversational-ai,chatbot,chatbot-framework,bot-framework

{rasa_pro-3.10.7.dev3.dist-info → rasa_pro-3.10.7.dev5.dist-info}/RECORD

@@ -484,7 +484,7 @@ rasa/markers/validate.py,sha256=YypXKRS87xxrMMEz9HpAQzYJUwg0OPbczMXBRNjlJq4,709
 rasa/model.py,sha256=GH1-N6Po3gL3nwfa9eGoN2bMRNMrn4f3mi17-osW3T0,3491
 rasa/model_manager/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rasa/model_manager/config.py,sha256=LCzRoV78bLX_FREW7hfQLsQepupoTs1m-CnK3u_D01s,349
-rasa/model_manager/model_api.py,sha256=_rj1VfYRbFj7TbPG1mryR_SVXLgTu6gVfXAiEdbbqwM,15187
+rasa/model_manager/model_api.py,sha256=6BHK8Qme8gpMwzQ31DTdMBGhXzfe3DE2cVd7tbJGyPE,16403
 rasa/model_manager/runner_service.py,sha256=aMupR7t0EEy2yOGT5cR_6kluApnwOiiGkjyIVYxdsMk,5553
 rasa/model_manager/socket_bridge.py,sha256=f-dhsV8xYaENXVZT5OOzkMdpSPPPxFEzsQqNJAuktxQ,1390
 rasa/model_manager/trainer_service.py,sha256=7LP_EsLQt-P2oqO1AgqbgIcTC515eCW58BamDkDjJjw,7856
@@ -727,9 +727,9 @@ rasa/utils/train_utils.py,sha256=f1NWpp5y6al0dzoQyyio4hc4Nf73DRoRSHDzEK6-C4E,212
 rasa/utils/url_tools.py,sha256=JQcHL2aLqLHu82k7_d9imUoETCm2bmlHaDpOJ-dKqBc,1218
 rasa/utils/yaml.py,sha256=KjbZq5C94ZP7Jdsw8bYYF7HASI6K4-C_kdHfrnPLpSI,2000
 rasa/validator.py,sha256=HM0ZIWjo3JRt2FMIfgNI_s932OABOSXkflm-rFTNkvY,62608
-rasa/version.py,sha256=DlTtggo2Cpv3vfmdgNMdjJQ8QS2-yQJIFm9LME4uBLY,122
-rasa_pro-3.10.7.dev3.dist-info/METADATA,sha256=JUZzFADQy5AKkpzwfIMNb2968TjBMJDEpzRFeFklz6A,28217
-rasa_pro-3.10.7.dev3.dist-info/NOTICE,sha256=7HlBoMHJY9CL2GlYSfTQ-PZsVmLmVkYmMiPlTjhuCqA,218
-rasa_pro-3.10.7.dev3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-rasa_pro-3.10.7.dev3.dist-info/entry_points.txt,sha256=ckJ2SfEyTPgBqj_I6vm_tqY9dZF_LAPJZA335Xp0Q9U,43
-rasa_pro-3.10.7.dev3.dist-info/RECORD,,
+rasa/version.py,sha256=nZYg5eop2_tsKPRLWnWE9JxkDMeo0ObpKxBEAtArcGk,122
+rasa_pro-3.10.7.dev5.dist-info/METADATA,sha256=jXD7sJZgCCyQGZyBYOcr-1JKPeLGX9eoN0QRJmeFSAY,28217
+rasa_pro-3.10.7.dev5.dist-info/NOTICE,sha256=7HlBoMHJY9CL2GlYSfTQ-PZsVmLmVkYmMiPlTjhuCqA,218
+rasa_pro-3.10.7.dev5.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+rasa_pro-3.10.7.dev5.dist-info/entry_points.txt,sha256=ckJ2SfEyTPgBqj_I6vm_tqY9dZF_LAPJZA335Xp0Q9U,43
+rasa_pro-3.10.7.dev5.dist-info/RECORD,,