chore: include hitl flag for generic agent

Author: radugheo

samples/generic-rag-agent/input.json

@@ -1,4 +1,5 @@
 {
   "query": "What is the company policy on travel spending? What is the price of GPUs in 2025? What is 5+6+7+8+(6*10)?",
-  "add_data_to_index": false
+  "add_data_to_index": false,
+  "include_hitl": false
 }

samples/generic-rag-agent/main.py

@@ -47,6 +47,7 @@
 class CustomStartEvent(StartEvent):
     query: str = ""
     add_data_to_index: Optional[bool] = False
+    include_hitl: bool = False
 
 
 class QueryEvent(Event):
@@ -165,6 +166,7 @@ async def workflow_entrypoint(
         await ctx.store.set("original_query", ev.query)
         await ctx.store.set("iteration_count", 0)
         await ctx.store.set("feedback_history", [])
+        await ctx.store.set("include_hitl", ev.include_hitl)
 
         if ev.add_data_to_index:
             return AddDataToIndexEvent()
@@ -396,8 +398,17 @@ async def request_human_confirmation(
     ) -> OutputEvent | QueryEvent:
         """
         Present the formatted answer to the user and request confirmation.
+        If include_hitl is False, skip human confirmation and return the output directly.
         """
-        await ctx.store.set("last_formatted_answer", ev.formatted_answer)
+        include_hitl = await ctx.store.get("include_hitl", False)
+
+        if not include_hitl:
+            print("\n" + "="*80)
+            print("FORMATTED ANSWER (HITL disabled):")
+            print("="*80)
+            print(ev.formatted_answer)
+            print("="*80)
+            return OutputEvent(output=ev.formatted_answer)
 
         print("\n" + "="*80)
         print("FORMATTED ANSWER:")

samples/generic-rag-agent/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "generic-rag-agent"
-version = "0.0.29"
+version = "0.0.30"
 description = "A generic RAG agent that combines multiple knowledge bases with web search capabilities and MCP tool support"
 authors = [{ name = "John Doe", email = "john.doe@myemail.com" }]
 readme = { file = "README.md", content-type = "text/markdown" }

samples/generic-rag-agent/uipath.json

@@ -2,7 +2,7 @@
     "entryPoints": [
         {
             "filePath": "agent",
-            "uniqueId": "e321d542-02cc-47d8-9e28-88b4b638c4d0",
+            "uniqueId": "de56dd1d-85cc-4b3d-b794-e8ddccae291a",
             "type": "agent",
             "input": {
                 "type": "object",
@@ -15,6 +15,11 @@
                     "add_data_to_index": {
                         "type": "boolean",
                         "nullable": true
+                    },
+                    "include_hitl": {
+                        "default": false,
+                        "title": "Include Hitl",
+                        "type": "boolean"
                     }
                 },
                 "required": []