move objects / history before get_workflow

solliancenet · Jan 15, 2025 · 3042382 · 3042382
1 parent 815def9
commit 3042382
Showing 1 changed file with 2 additions and 2 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
@@ -527,6 +527,8 @@ async def invoke_async(self, request: KnowledgeManagementCompletionRequest) -> C
                 for tool in agent.tools:
                     tools.append(tool_factory.get_tool(tool, request.objects, self.user_identity, self.config))
 
+            request.objects['message_history'] = request.message_history
+
             # create the workflow
             workflow_factory = WorkflowFactory(self.plugin_manager)
             workflow = workflow_factory.get_workflow(
@@ -536,8 +538,6 @@ async def invoke_async(self, request: KnowledgeManagementCompletionRequest) -> C
                 self.user_identity,
                 self.config)
 
-            request.objects['message_history'] = request.message_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)