fix: remove debug logs

Author: GabrielVasilescu04

packages/uipath-llamaindex/samples/chat-agent/agent.mermaid

@@ -1,6 +1,19 @@
 flowchart TB
   __start__(__start__)
-  chat(chat)
+  aggregate_tool_results(aggregate_tool_results)
+  call_tool(call_tool)
+  init_run(init_run)
+  parse_agent_output(parse_agent_output)
+  run_agent_step(run_agent_step)
+  setup_agent(setup_agent)
   __end__(__end__)
-  __start__ --> |ChatInput|chat
-  chat --> |StopEvent|__end__
+  aggregate_tool_results --> |AgentInput|setup_agent
+  aggregate_tool_results --> |StopEvent|__end__
+  call_tool --> |ToolCallResult|aggregate_tool_results
+  __start__ --> |AgentWorkflowStartEvent|init_run
+  init_run --> |AgentInput|setup_agent
+  parse_agent_output --> |StopEvent|__end__
+  parse_agent_output --> |AgentInput|setup_agent
+  parse_agent_output --> |ToolCall|call_tool
+  run_agent_step --> |AgentOutput|parse_agent_output
+  setup_agent --> |AgentSetup|run_agent_step

packages/uipath-llamaindex/samples/chat-agent/main.py

@@ -1,6 +1,6 @@
 import os
 
-from llama_index.core.agent.workflow import FunctionAgent
+from llama_index.core.agent.workflow import FunctionAgent, AgentWorkflow
 from llama_index.llms.openai import OpenAI
 from llama_index.tools.tavily_research import TavilyToolSpec
 
@@ -24,8 +24,8 @@
     "Remember previous messages and maintain context across the discussion."
 )
 
-agent = FunctionAgent(
-    tools=tavily_tool.to_tool_list(),
+agent = AgentWorkflow.from_tools_or_functions(
+    tools_or_functions=tavily_tool.to_tool_list(),
     llm=llm,
     system_prompt=SYSTEM_PROMPT
 )

packages/uipath-llamaindex/samples/chat-agent/uipath.json

@@ -1,7 +1,7 @@
 {
   "$schema": "https://cloud.uipath.com/draft/2024-12/uipath",
   "runtimeOptions": {
-    "isConversational": false
+    "isConversational": true
   },
   "packOptions": {
     "fileExtensionsIncluded": [],

packages/uipath-llamaindex/src/uipath_llamaindex/runtime/runtime.py

@@ -171,14 +171,8 @@ async def _run_workflow(
         suspended_event: InputRequiredEvent | None = None
         chat = UiPathChatMessagesMapper(runtime_id=self.runtime_id)
 
-        raw_chunks: list[dict] = []
-        mapped_chunks: list[dict] = []
         is_resumed: bool = False
         async for event in event_stream:
-            if not isinstance(event, BreakpointEvent):
-                raw_chunks.append(
-                    {"type": type(event).__name__, "data": json.loads(serialize_json(event))}
-                )
             node_name = self._get_node_name(event)
             if isinstance(event, (AgentOutput, AgentStream, ToolCall, ToolCallResult)):
                 try:
@@ -188,7 +182,6 @@ async def _run_workflow(
                     mapped_events = None
                 if mapped_events:
                     for mapped_event in mapped_events:
-                        mapped_chunks.append(mapped_event.model_dump(by_alias=True, exclude_none=True))
                         if stream_events:
                             yield UiPathRuntimeMessageEvent(payload=mapped_event)
             elif stream_events and not isinstance(event, BreakpointEvent):
@@ -223,12 +216,6 @@ async def _run_workflow(
                     suspended_event = event
                     break
 
-        for filename, data in [("raw_chunks.json", raw_chunks), ("mapped_chunks.json", mapped_chunks)]:
-            output_path = os.path.abspath(filename)
-            with open(output_path, "w") as f:
-                json.dump(data, f, indent=2)
-            print(f"{filename} written to: {output_path}")
-
         if suspended_event is not None:
             await asyncio.sleep(0)  # Yield control to event loop
             await self._save_context()