From 74ee665af617d57f26dce1c949fea4fd091c6a1f Mon Sep 17 00:00:00 2001 From: wangxiaolei Date: Thu, 9 Jul 2026 19:26:16 +0800 Subject: [PATCH] fix: fix miss session param (#38612) Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> --- api/core/app/apps/agent_app/app_generator.py | 1 + api/core/app/apps/agent_chat/app_generator.py | 1 + .../app/apps/agent_chat/test_agent_chat_app_generator.py | 5 ++++- 3 files changed, 6 insertions(+), 1 deletion(-) diff --git a/api/core/app/apps/agent_app/app_generator.py b/api/core/app/apps/agent_app/app_generator.py index 4c10004fc7e..c316c3ebac7 100644 --- a/api/core/app/apps/agent_app/app_generator.py +++ b/api/core/app/apps/agent_app/app_generator.py @@ -333,6 +333,7 @@ class AgentAppGenerator(MessageBasedAppGenerator): target=self._generate_worker, kwargs={ "flask_app": current_app._get_current_object(), # type: ignore + "session": db.session(), "context": context, "application_generate_entity": application_generate_entity, "queue_manager": queue_manager, diff --git a/api/core/app/apps/agent_chat/app_generator.py b/api/core/app/apps/agent_chat/app_generator.py index a3cc913abf3..1fbef2a4092 100644 --- a/api/core/app/apps/agent_chat/app_generator.py +++ b/api/core/app/apps/agent_chat/app_generator.py @@ -205,6 +205,7 @@ class AgentChatAppGenerator(MessageBasedAppGenerator): target=self._generate_worker, kwargs={ "flask_app": current_app._get_current_object(), # type: ignore + "session": db.session(), "context": context, "application_generate_entity": application_generate_entity, "queue_manager": queue_manager, diff --git a/api/tests/unit_tests/core/app/apps/agent_chat/test_agent_chat_app_generator.py b/api/tests/unit_tests/core/app/apps/agent_chat/test_agent_chat_app_generator.py index eb4bf9bc166..60297b45181 100644 --- a/api/tests/unit_tests/core/app/apps/agent_chat/test_agent_chat_app_generator.py +++ b/api/tests/unit_tests/core/app/apps/agent_chat/test_agent_chat_app_generator.py @@ -116,10 +116,12 @@ class TestAgentChatAppGeneratorGenerate: ) thread_obj = mocker.MagicMock() - mocker.patch( + thread_constructor = mocker.patch( "core.app.apps.agent_chat.app_generator.threading.Thread", return_value=thread_obj, ) + session = mocker.MagicMock() + mocker.patch("core.app.apps.agent_chat.app_generator.db.session", return_value=session) mocker.patch( "core.app.apps.agent_chat.app_generator.AgentChatAppGenerateResponseConverter.convert", @@ -144,6 +146,7 @@ class TestAgentChatAppGeneratorGenerate: assert result == {"result": "ok"} assert generate_entity.call_args.kwargs["extras"]["trace_session_id"] == "session-1" + assert thread_constructor.call_args.kwargs["kwargs"]["session"] is session thread_obj.start.assert_called_once() def test_generate_without_file_config(self, generator, mocker: MockerFixture):