make pylint happy in tests

Author: maorfr

tests/unit/app/endpoints/test_query.py

@@ -1077,7 +1077,7 @@ def test_get_agent_cache_hit(prepare_agent_mocks):
     # Set up cache with existing agent
     conversation_id = "test_conversation_id"
 
-    result_agent, result_conversation_id = get_agent(
+    result_agent, result_conversation_id, _ = get_agent(
         client=mock_client,
         model_id="test_model",
         system_prompt="test_prompt",
@@ -1118,7 +1118,7 @@ def test_get_agent_cache_miss_with_conversation_id(
     mocker.patch("app.endpoints.query.configuration", setup_configuration)
 
     # Call function with conversation_id but no cached agent
-    result_agent, result_conversation_id = get_agent(
+    result_agent, result_conversation_id, _ = get_agent(
         client=mock_client,
         model_id="test_model",
         system_prompt="test_prompt",
@@ -1168,7 +1168,7 @@ def test_get_agent_no_conversation_id(setup_configuration, prepare_agent_mocks,
     mocker.patch("app.endpoints.query.configuration", setup_configuration)
 
     # Call function with None conversation_id
-    result_agent, result_conversation_id = get_agent(
+    result_agent, result_conversation_id, _ = get_agent(
         client=mock_client,
         model_id="test_model",
         system_prompt="test_prompt",
@@ -1218,7 +1218,7 @@ def test_get_agent_empty_shields(setup_configuration, prepare_agent_mocks, mocke
     mocker.patch("app.endpoints.query.configuration", setup_configuration)
 
     # Call function with empty shields list
-    result_agent, result_conversation_id = get_agent(
+    result_agent, result_conversation_id, _ = get_agent(
         client=mock_client,
         model_id="test_model",
         system_prompt="test_prompt",
@@ -1272,7 +1272,7 @@ def test_get_agent_multiple_mcp_servers(
     mocker.patch("app.endpoints.query.configuration", setup_configuration)
 
     # Call function
-    result_agent, result_conversation_id = get_agent(
+    result_agent, result_conversation_id, _ = get_agent(
         client=mock_client,
         model_id="test_model",
         system_prompt="test_prompt",
@@ -1579,7 +1579,7 @@ def test_get_agent_no_tools_no_parser(setup_configuration, prepare_agent_mocks,
     mocker.patch("app.endpoints.query.configuration", setup_configuration)
 
     # Call function with no_tools=True
-    result_agent, result_conversation_id = get_agent(
+    result_agent, result_conversation_id, _ = get_agent(
         client=mock_client,
         model_id="test_model",
         system_prompt="test_prompt",
@@ -1637,7 +1637,7 @@ def test_get_agent_no_tools_false_preserves_parser(
     mocker.patch("app.endpoints.query.configuration", setup_configuration)
 
     # Call function with no_tools=False
-    result_agent, result_conversation_id = get_agent(
+    result_agent, result_conversation_id, _ = get_agent(
         client=mock_client,
         model_id="test_model",
         system_prompt="test_prompt",