Merge pull request #2103 from solliancenet/cp-message-history

Fix message history for LangGraphReactAgent and ExternalWorkflow path
solliancenet · Dec 22, 2024 · 789bb4b · 789bb4b
2 parents e1ed2a2 + fed6556
commit 789bb4b
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/src/python/PythonSDK/foundationallm/langchain/agents/langchain_knowledge_management_agent.py b/src/python/PythonSDK/foundationallm/langchain/agents/langchain_knowledge_management_agent.py
@@ -462,7 +462,11 @@ async def invoke_async(self, request: KnowledgeManagementCompletionRequest) -> C
 
             # Define the graph
             graph = create_react_agent(llm, tools=tools, state_modifier=self.prompt.prefix)
-            messages = self._build_conversation_history_message_list(request.message_history, agent.conversation_history_settings.max_history)
+            if agent.conversation_history_settings.enabled:
+                messages = self._build_conversation_history_message_list(request.message_history, agent.conversation_history_settings.max_history)
+            else:
+                messages = []
+
             messages.append(HumanMessage(content=parsed_user_prompt))
 
             response = await graph.ainvoke(
@@ -526,8 +530,11 @@ async def invoke_async(self, request: KnowledgeManagementCompletionRequest) -> C
                 self.user_identity,
                 self.config)
 
-            # Get message history          
-            messages = self._build_conversation_history_message_list(request.message_history, agent.conversation_history_settings.max_history)            
+            # Get message history
+            if agent.conversation_history_settings.enabled:   
+                messages = self._build_conversation_history_message_list(request.message_history, agent.conversation_history_settings.max_history)
+            else:
+                messages = []
 
             response = await workflow.invoke_async(
                 operation_id=request.operation_id,