fix: improve maintenance and synchronize live conversations

backend/tests/agents/test_chat_agent_end_chat.py

@@ -65,15 +65,14 @@ def chat_agent(mock_agent_data, mock_db):
     mock_storage = Mock()
     mock_storage.get_session_state.return_value = {"status": "active"}
     
-    with patch('app.agents.chat_agent.get_db') as mock_get_db, \
+    with patch('app.agents.chat_agent.SessionLocal') as mock_session_local, \
          patch('app.agents.chat_agent.JiraRepository') as mock_jira_repo, \
          patch('app.tools.knowledge_search_byagent.KnowledgeSearchByAgent', return_value=mock_knowledge_tool), \
          patch('app.agents.chat_agent.EncryptedPostgresAgentStorage', return_value=mock_storage), \
          patch('app.agents.chat_agent.settings.DATABASE_URL', "mock://test"), \
          patch.dict('os.environ', mock_env, clear=True):
         
-        # Mock database session
-        mock_get_db.return_value = iter([mock_db])
+        mock_session_local.return_value.__enter__.return_value = mock_db
         
         # Mock JiraRepository methods
         mock_jira_repo.return_value.get_agent_with_jira_config.return_value = mock_jira_agent_data
@@ -241,4 +240,4 @@ class TestChatAgentEndChat:
                 
                 # Verify agent's default setting was used
                 assert result.request_rating is True
-                assert "Would you please take a moment to rate your experience?" in result.message 
+                assert "Would you please take a moment to rate your experience?" in result.message

backend/tests/agents/test_chat_agent_lead_capture_prompt.py

@@ -66,7 +66,7 @@ def build_agent(known_email: str = None, known_name: str = None, trusted: bool =
     lead_repo = MagicMock()
     lead_repo.return_value.get_by_agent.return_value = lead_config
 
-    with patch('app.agents.chat_agent.get_db') as mock_get_db, \
+    with patch('app.agents.chat_agent.SessionLocal') as mock_session_local, \
          patch('app.agents.chat_agent.JiraRepository') as mock_jira_repo, \
          patch('app.repositories.customer.CustomerRepository', customer_repo), \
          patch('app.repositories.lead_capture.LeadCaptureConfigRepository', lead_repo), \
@@ -76,7 +76,7 @@ def build_agent(known_email: str = None, known_name: str = None, trusted: bool =
          patch('app.agents.chat_agent.EncryptedPostgresAgentStorage', return_value=storage), \
          patch('app.agents.chat_agent.settings.DATABASE_URL', "mock://test"), \
          patch.dict('os.environ', {}, clear=True):
-        mock_get_db.return_value = iter([MagicMock(spec=Session)])
+        mock_session_local.return_value.__enter__.return_value = MagicMock(spec=Session)
         mock_jira_repo.return_value.get_agent_with_jira_config.return_value = agent_data
 
         return ChatAgent(

backend/tests/agents/test_chat_agent_transfer_prompt.py

@@ -17,7 +17,6 @@ limitations under the License.
 from unittest.mock import MagicMock, Mock, patch
 from uuid import uuid4
 
-import pytest
 from sqlalchemy.orm import Session
 
 from app.agents.chat_agent import ChatAgent
@@ -36,13 +35,13 @@ def build_agent(transfer_to_human: bool) -> ChatAgent:
     storage = Mock()
     storage.get_session_state.return_value = {"status": "active"}
 
-    with patch('app.agents.chat_agent.get_db') as mock_get_db, \
+    with patch('app.agents.chat_agent.SessionLocal') as mock_session_local, \
          patch('app.agents.chat_agent.JiraRepository') as mock_jira_repo, \
          patch('app.tools.knowledge_search_byagent.KnowledgeSearchByAgent', return_value=Mock()), \
          patch('app.agents.chat_agent.EncryptedPostgresAgentStorage', return_value=storage), \
          patch('app.agents.chat_agent.settings.DATABASE_URL', "mock://test"), \
          patch.dict('os.environ', {}, clear=True):
-        mock_get_db.return_value = iter([MagicMock(spec=Session)])
+        mock_session_local.return_value.__enter__.return_value = MagicMock(spec=Session)
         mock_jira_repo.return_value.get_agent_with_jira_config.return_value = agent_data
 
         return ChatAgent(

backend/tests/agents/test_guardrail_policy.py

@@ -26,7 +26,6 @@ from app.agents.guardrail_policy import (
     OPERATOR_CLOSE,
     OPERATOR_OPEN,
     POLICY_HEADER,
-    REFUSAL_MARKER,
     GuardrailContext,
     apply_guardrail_policy,
     build_policy_block,

backend/tests/agents/test_quick_replies.py

@@ -119,13 +119,13 @@ def _build_agent(choice_limits) -> ChatAgent:
     storage = Mock()
     storage.get_session_state.return_value = {"status": "active"}
 
-    with patch('app.agents.chat_agent.get_db') as mock_get_db, \
+    with patch('app.agents.chat_agent.SessionLocal') as mock_session_local, \
          patch('app.agents.chat_agent.JiraRepository') as mock_jira_repo, \
          patch('app.tools.knowledge_search_byagent.KnowledgeSearchByAgent', return_value=Mock()), \
          patch('app.agents.chat_agent.EncryptedPostgresAgentStorage', return_value=storage), \
          patch('app.agents.chat_agent.settings.DATABASE_URL', "mock://test"), \
          patch.dict('os.environ', {}, clear=True):
-        mock_get_db.return_value = iter([MagicMock(spec=Session)])
+        mock_session_local.return_value.__enter__.return_value = MagicMock(spec=Session)
         mock_jira_repo.return_value.get_agent_with_jira_config.return_value = agent_data
         return ChatAgent(api_key="test-key", model_name="gpt-4", model_type="openai",
                          org_id=str(uuid4()), agent_id=str(uuid4()),