Merge pull request #578 from NVIDIA/feature/improve-bot-action-generation

Feature/improve bot action generation

Author: drazvan

nemoguardrails/actions/llm/utils.py

@@ -508,12 +508,23 @@ def get_first_bot_intent(strings: List[str]) -> Optional[str]:
 
 def get_first_bot_action(strings: List[str]) -> Optional[str]:
     """Returns first bot action."""
-    action: Optional[str] = None
+    action_started = False
+    action: str = ""
     for string in strings:
         if string.startswith("bot action: "):
-            action = string.replace("bot action: ", "")
-        elif action and (string.startswith("  and") or string.startswith("  or")):
+            if action != "":
+                action += "\n"
+            action += string.replace("bot action: ", "")
+            action_started = True
+        elif (
+            string.startswith("  and") or string.startswith("  or")
+        ) and action_started:
             action = action + string
+        elif string == "":
+            action_started = False
+            continue
+        elif action != "":
+            return action
     return action
 
 

nemoguardrails/actions/v2_x/generation.py

@@ -569,6 +569,7 @@ async def create_flow(
         events: List[dict],
         name: str,
         body: str,
+        decorators: Optional[str] = None,
     ) -> dict:
         """Create a new flow during runtime."""
 
@@ -578,10 +579,14 @@ async def create_flow(
         flow_name = f"_dynamic_{uuid} {name}"
         # TODO: parse potential parameters from flow name with a regex
 
+        body = f"flow {flow_name}\n  " + body
+        if decorators:
+            body = decorators + "\n" + body
+
         return {
             "name": flow_name,
             "parameters": [],
-            "body": f"flow {flow_name}\n  " + body,
+            "body": body,
         }
 
     @action(name="GenerateValueAction", is_system_action=True, execute_async=True)

nemoguardrails/colang/v2_x/library/llm.co

@@ -104,19 +104,18 @@ flow continuation on unhandled user utterance
     # Create and start new flow or an existing on with same name as bot intent
     $flow_exists = False
     if $bot_intent is None
-      $bot_intent = "bot custom action"
+      $bot_intent = "bot reacts accordingly"
     else
       $flow_exists = await CheckValidFlowExistsAction(flow_id=$bot_intent)
 
     if $flow_exists == False and $bot_action is not None
-      $flow_info = await CreateFlowAction(name=$bot_intent, body=$bot_action)
+      $flow_info = await CreateFlowAction(name=$bot_intent, body=$bot_action, decorators='@meta(bot_intent="{$bot_intent}")\n')
       $flows = await AddFlowsAction(config=$flow_info.body)
       if len($flows) == 0
         print "Failed parsing created bot action flow!"
         return
-      # Start
-      await await_flow_by_name $flow_info.name
-      await RemoveFlowsAction(flow_ids=[$flow_info.name])
+    await await_flow_by_name $flow_info.name
+    await RemoveFlowsAction(flow_ids=[$flow_info.name])
 
 
 flow unhandled user intent -> $intent