Supplementary unit tests

Signed-off-by: Zhi Wang <[email protected]>

Author: Zhi-a

test/backend/services/test_agent_service.py

@@ -966,6 +966,73 @@ def mock_get_model(model_id):
     mock_get_model_by_model_id.assert_any_call(789)
 
 
+@patch('backend.services.agent_service.get_model_by_model_id')
+@patch('backend.services.agent_service.query_sub_agents_id_list')
+@patch('backend.services.agent_service.search_tools_for_sub_agent')
+@patch('backend.services.agent_service.search_agent_info_by_agent_id')
+@pytest.mark.asyncio
+async def test_get_agent_info_impl_with_business_logic_model_none(mock_search_agent_info, mock_search_tools, mock_query_sub_agents_id, mock_get_model_by_model_id):
+    """
+    Test get_agent_info_impl with business_logic_model_id but get_model_by_model_id returns None.
+
+    This test verifies that:
+    1. The function correctly handles when business_logic_model_id is not None but get_model_by_model_id returns None
+    2. It sets business_logic_model_name to None when model_info is None
+    """
+    # Setup
+    mock_agent_info = {
+        "agent_id": 123,
+        "model_id": 456,
+        "business_logic_model_id": 789,
+        "business_description": "Test agent"
+    }
+    mock_search_agent_info.return_value = mock_agent_info
+
+    mock_tools = [{"tool_id": 1, "name": "Tool 1"}]
+    mock_search_tools.return_value = mock_tools
+
+    mock_sub_agent_ids = [101, 102]
+    mock_query_sub_agents_id.return_value = mock_sub_agent_ids
+
+    # Mock model info for main model
+    mock_main_model_info = {
+        "model_id": 456,
+        "display_name": "GPT-4",
+        "provider": "openai"
+    }
+    
+    # Mock get_model_by_model_id to return None for business_logic_model_id
+    def mock_get_model(model_id):
+        if model_id == 456:
+            return mock_main_model_info
+        elif model_id == 789:
+            return None  # Business logic model not found
+        return None
+    
+    mock_get_model_by_model_id.side_effect = mock_get_model
+
+    # Execute
+    result = await get_agent_info_impl(agent_id=123, tenant_id="test_tenant")
+
+    # Assert
+    expected_result = {
+        "agent_id": 123,
+        "model_id": 456,
+        "business_logic_model_id": 789,
+        "business_description": "Test agent",
+        "tools": mock_tools,
+        "sub_agent_id_list": mock_sub_agent_ids,
+        "model_name": "GPT-4",
+        "business_logic_model_name": None  # Should be None when model info is not found
+    }
+    assert result == expected_result
+    
+    # Verify both models were looked up
+    assert mock_get_model_by_model_id.call_count == 2
+    mock_get_model_by_model_id.assert_any_call(456)
+    mock_get_model_by_model_id.assert_any_call(789)
+
+
 async def test_list_all_agent_info_impl_success():
     """
     Test successful retrieval of all agent information.