fix: get correct message count per turn for conversational agents [JAR-9610] (#746)

Co-authored-by: Maxwell Du <60411452+maxduu@users.noreply.github.com>

Author: andrewwan-uipath

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-langchain"
-version = "0.9.15"
+version = "0.9.16"
 description = "Python SDK that enables developers to build and deploy LangGraph agents to the UiPath Cloud Platform"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"

src/uipath_langchain/agent/react/llm_node.py

@@ -84,7 +84,11 @@ def create_llm_node(
 
     async def llm_node(state: StateT):
         messages: list[AnyMessage] = state.messages
-        agent_ai_messages = sum(1 for msg in messages if isinstance(msg, AIMessage))
+        initial_count = state.inner_state.initial_message_count or 0
+        current_turn_messages = messages[initial_count:]
+        agent_ai_messages = sum(
+            1 for msg in current_turn_messages if isinstance(msg, AIMessage)
+        )
         if agent_ai_messages >= llm_messages_limit:
             raise AgentRuntimeError(
                 code=AgentRuntimeErrorCode.TERMINATION_MAX_ITERATIONS,