chore: change input variables

Author: radugheo

samples/generic-rag-agent/input.json

@@ -1,5 +1,4 @@
 {
   "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,
-  "include_hitl": false
+  "add_data_to_index": false
 }

samples/generic-rag-agent/main.py

@@ -1,5 +1,6 @@
 import os
 import time
+from typing import Optional
 
 from llama_index.core import get_response_synthesizer
 from llama_index.core.agent import ReActAgent
@@ -45,7 +46,7 @@
 
 class CustomStartEvent(StartEvent):
     query: str = ""
-    add_data_to_index: bool = False
+    add_data_to_index: Optional[bool] = False
 
 
 class QueryEvent(Event):
@@ -196,13 +197,13 @@ async def add_file_to_index(file_path, index_name, ingest_data):
 
                 print(f"Indexing {len(files)} file(s) from {files_directory} to index '{index_name}'...")
 
-                for i in range(len(files)):
+                for i, file_name in enumerate(files):
                     is_last_config = config_key == list(INDEX_CONFIGS.keys())[-1]
                     is_last_file = i == len(files) - 1
                     ingest_data = is_last_config and is_last_file
 
                     await add_file_to_index(
-                        os.path.join(files_directory, files[i]),
+                        os.path.join(files_directory, file_name),
                         index_name,
                         ingest_data,
                     )
@@ -404,9 +405,11 @@ async def request_human_confirmation(
         print(ev.formatted_answer)
         print("="*80)
 
+        prompt_message = f"{ev.formatted_answer}\n\n{'='*80}\n\nIs this answer satisfactory? (yes/no or provide feedback): "
+
         ctx.write_event_to_stream(
             InputRequiredEvent(
-                prefix="\nIs this answer satisfactory? (yes/no or provide feedback): "
+                prefix=prompt_message
             )
         )
 

samples/generic-rag-agent/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "generic-rag-agent"
-version = "0.0.27"
+version = "0.0.29"
 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,24 +2,22 @@
     "entryPoints": [
         {
             "filePath": "agent",
-            "uniqueId": "739a1d4e-7100-49bb-a866-2daa4ed45b3f",
+            "uniqueId": "e321d542-02cc-47d8-9e28-88b4b638c4d0",
             "type": "agent",
             "input": {
                 "type": "object",
                 "properties": {
                     "query": {
+                        "default": "",
                         "title": "Query",
                         "type": "string"
                     },
                     "add_data_to_index": {
-                        "title": "Add Data To Index",
-                        "type": "boolean"
+                        "type": "boolean",
+                        "nullable": true
                     }
                 },
-                "required": [
-                    "query",
-                    "add_data_to_index"
-                ]
+                "required": []
             },
             "output": {
                 "type": "object",
@@ -40,7 +38,7 @@
         "resources": [
             {
                 "resource": "index",
-                "key": "index_folder_path.index_name",
+                "key": "index_name.index_folder_path",
                 "value": {
                     "name": {
                         "defaultValue": "index_name",
@@ -61,7 +59,7 @@
             },
             {
                 "resource": "index",
-                "key": "index_folder_path.index_name",
+                "key": "index_name.index_folder_path",
                 "value": {
                     "name": {
                         "defaultValue": "index_name",