fix: conv agent tool calling

Author: mike-deem-uipath

src/uipath_langchain/agent/react/agent.py

@@ -157,6 +157,9 @@ def create_agent(
         target_node_names,
     )
 
+    if config.is_conversational:
+        target_node_names.append(AgentGraphNode.AGENT)
+
     for tool_name in tool_node_names:
         builder.add_conditional_edges(tool_name, route_agent, target_node_names)
     builder.add_edge(AgentGraphNode.TERMINATE, END)

src/uipath_langchain/runtime/factory.py

@@ -264,16 +264,17 @@ async def _create_runtime_instance(
         Returns:
             Configured runtime instance
         """
+        memory = await self._get_memory()
+        storage = SqliteResumableStorage(memory)
+        trigger_manager = UiPathResumeTriggerHandler()
+
         base_runtime = UiPathLangGraphRuntime(
             graph=compiled_graph,
             runtime_id=runtime_id,
             entrypoint=entrypoint,
+            storage=storage,
         )
 
-        memory = await self._get_memory()
-        storage = SqliteResumableStorage(memory)
-        trigger_manager = UiPathResumeTriggerHandler()
-
         return UiPathResumableRuntime(
             delegate=base_runtime,
             storage=storage,