From bfa754b16665a60401d9dfde383fbf03bc89da5f Mon Sep 17 00:00:00 2001 From: Pavel Tisnovsky Date: Fri, 26 Sep 2025 09:13:22 +0200 Subject: [PATCH] LCORE-726: use proper user ID and conversation ID in unit tests --- tests/unit/app/endpoints/test_query.py | 137 +++++++++++++----- .../app/endpoints/test_streaming_query.py | 102 +++++++++---- 2 files changed, 178 insertions(+), 61 deletions(-) diff --git a/tests/unit/app/endpoints/test_query.py b/tests/unit/app/endpoints/test_query.py index c15c1225..5e1f6363 100644 --- a/tests/unit/app/endpoints/test_query.py +++ b/tests/unit/app/endpoints/test_query.py @@ -38,7 +38,13 @@ ) from utils.types import ToolCallSummary, TurnSummary -MOCK_AUTH = ("mock_user_id", "mock_username", False, "mock_token") +# User ID must be proper UUID +MOCK_AUTH = ( + "00000001-0001-0001-0001-000000000001", + "mock_username", + False, + "mock_token", +) @pytest.fixture @@ -93,6 +99,9 @@ def setup_configuration_fixture(): }, "mcp_servers": [], "customization": None, + "conversation_cache": { + "type": "noop", + }, } cfg = AppConfig() cfg.init_from_dict(config_dict) @@ -172,7 +181,7 @@ async def _test_query_endpoint_handler( ) ], ) - conversation_id = "fake_conversation_id" + conversation_id = "00000000-0000-0000-0000-000000000000" query = "What is OpenStack?" referenced_documents = [] @@ -209,7 +218,7 @@ async def _test_query_endpoint_handler( # Assert the store_transcript function is called if transcripts are enabled if store_transcript_to_file: mock_transcript.assert_called_once_with( - user_id="mock_user_id", + user_id="00000001-0001-0001-0001-000000000001", conversation_id=conversation_id, model_id="fake_model_id", provider_id="fake_provider_id", @@ -458,7 +467,11 @@ async def test_retrieve_response_no_returned_message(prepare_agent_mocks, mocker mocker.patch("app.endpoints.query.configuration", mock_config) mocker.patch( "app.endpoints.query.get_agent", - return_value=(mock_agent, "fake_conversation_id", "fake_session_id"), + return_value=( + mock_agent, + "00000000-0000-0000-0000-000000000000", + "fake_session_id", + ), ) mock_metrics(mocker) @@ -490,7 +503,11 @@ async def test_retrieve_response_message_without_content(prepare_agent_mocks, mo mocker.patch("app.endpoints.query.configuration", mock_config) mocker.patch( "app.endpoints.query.get_agent", - return_value=(mock_agent, "fake_conversation_id", "fake_session_id"), + return_value=( + mock_agent, + "00000000-0000-0000-0000-000000000000", + "fake_session_id", + ), ) mock_metrics(mocker) @@ -523,7 +540,11 @@ async def test_retrieve_response_vector_db_available(prepare_agent_mocks, mocker mocker.patch("app.endpoints.query.configuration", mock_config) mocker.patch( "app.endpoints.query.get_agent", - return_value=(mock_agent, "fake_conversation_id", "fake_session_id"), + return_value=( + mock_agent, + "00000000-0000-0000-0000-000000000000", + "fake_session_id", + ), ) mock_metrics(mocker) @@ -538,7 +559,7 @@ async def test_retrieve_response_vector_db_available(prepare_agent_mocks, mocker # Assert that the metric for validation errors is NOT incremented mock_metric.inc.assert_not_called() assert summary.llm_response == "LLM answer" - assert conversation_id == "fake_conversation_id" + assert conversation_id == "00000000-0000-0000-0000-000000000000" mock_agent.create_turn.assert_called_once_with( messages=[UserMessage(content="What is OpenStack?", role="user")], session_id="fake_session_id", @@ -562,7 +583,11 @@ async def test_retrieve_response_no_available_shields(prepare_agent_mocks, mocke mocker.patch("app.endpoints.query.configuration", mock_config) mocker.patch( "app.endpoints.query.get_agent", - return_value=(mock_agent, "fake_conversation_id", "fake_session_id"), + return_value=( + mock_agent, + "00000000-0000-0000-0000-000000000000", + "fake_session_id", + ), ) mock_metrics(mocker) @@ -575,7 +600,7 @@ async def test_retrieve_response_no_available_shields(prepare_agent_mocks, mocke ) assert summary.llm_response == "LLM answer" - assert conversation_id == "fake_conversation_id" + assert conversation_id == "00000000-0000-0000-0000-000000000000" mock_agent.create_turn.assert_called_once_with( messages=[UserMessage(content="What is OpenStack?", role="user")], session_id="fake_session_id", @@ -612,7 +637,11 @@ def __repr__(self): mocker.patch("app.endpoints.query.configuration", mock_config) mocker.patch( "app.endpoints.query.get_agent", - return_value=(mock_agent, "fake_conversation_id", "fake_session_id"), + return_value=( + mock_agent, + "00000000-0000-0000-0000-000000000000", + "fake_session_id", + ), ) mock_metrics(mocker) @@ -625,7 +654,7 @@ def __repr__(self): ) assert summary.llm_response == "LLM answer" - assert conversation_id == "fake_conversation_id" + assert conversation_id == "00000000-0000-0000-0000-000000000000" mock_agent.create_turn.assert_called_once_with( messages=[UserMessage(content="What is OpenStack?", role="user")], session_id="fake_session_id", @@ -665,7 +694,11 @@ def __repr__(self): mocker.patch("app.endpoints.query.configuration", mock_config) mocker.patch( "app.endpoints.query.get_agent", - return_value=(mock_agent, "fake_conversation_id", "fake_session_id"), + return_value=( + mock_agent, + "00000000-0000-0000-0000-000000000000", + "fake_session_id", + ), ) mock_metrics(mocker) @@ -678,7 +711,7 @@ def __repr__(self): ) assert summary.llm_response == "LLM answer" - assert conversation_id == "fake_conversation_id" + assert conversation_id == "00000000-0000-0000-0000-000000000000" mock_agent.create_turn.assert_called_once_with( messages=[UserMessage(content="What is OpenStack?", role="user")], session_id="fake_session_id", @@ -720,7 +753,11 @@ def __repr__(self): mocker.patch("app.endpoints.query.configuration", mock_config) mock_get_agent = mocker.patch( "app.endpoints.query.get_agent", - return_value=(mock_agent, "fake_conversation_id", "fake_session_id"), + return_value=( + mock_agent, + "00000000-0000-0000-0000-000000000000", + "fake_session_id", + ), ) mock_metrics(mocker) @@ -733,7 +770,7 @@ def __repr__(self): ) assert summary.llm_response == "LLM answer" - assert conversation_id == "fake_conversation_id" + assert conversation_id == "00000000-0000-0000-0000-000000000000" # Verify get_agent was called with the correct parameters mock_get_agent.assert_called_once_with( @@ -777,7 +814,11 @@ async def test_retrieve_response_with_one_attachment(prepare_agent_mocks, mocker ] mocker.patch( "app.endpoints.query.get_agent", - return_value=(mock_agent, "fake_conversation_id", "fake_session_id"), + return_value=( + mock_agent, + "00000000-0000-0000-0000-000000000000", + "fake_session_id", + ), ) mock_metrics(mocker) @@ -790,7 +831,7 @@ async def test_retrieve_response_with_one_attachment(prepare_agent_mocks, mocker ) assert summary.llm_response == "LLM answer" - assert conversation_id == "fake_conversation_id" + assert conversation_id == "00000000-0000-0000-0000-000000000000" mock_agent.create_turn.assert_called_once_with( messages=[UserMessage(content="What is OpenStack?", role="user")], session_id="fake_session_id", @@ -832,7 +873,11 @@ async def test_retrieve_response_with_two_attachments(prepare_agent_mocks, mocke ] mocker.patch( "app.endpoints.query.get_agent", - return_value=(mock_agent, "fake_conversation_id", "fake_session_id"), + return_value=( + mock_agent, + "00000000-0000-0000-0000-000000000000", + "fake_session_id", + ), ) mock_metrics(mocker) @@ -845,7 +890,7 @@ async def test_retrieve_response_with_two_attachments(prepare_agent_mocks, mocke ) assert summary.llm_response == "LLM answer" - assert conversation_id == "fake_conversation_id" + assert conversation_id == "00000000-0000-0000-0000-000000000000" mock_agent.create_turn.assert_called_once_with( messages=[UserMessage(content="What is OpenStack?", role="user")], session_id="fake_session_id", @@ -1005,7 +1050,11 @@ async def test_retrieve_response_with_mcp_servers(prepare_agent_mocks, mocker): mocker.patch("app.endpoints.query.configuration", mock_config) mock_get_agent = mocker.patch( "app.endpoints.query.get_agent", - return_value=(mock_agent, "fake_conversation_id", "fake_session_id"), + return_value=( + mock_agent, + "00000000-0000-0000-0000-000000000000", + "fake_session_id", + ), ) mock_metrics(mocker) @@ -1018,7 +1067,7 @@ async def test_retrieve_response_with_mcp_servers(prepare_agent_mocks, mocker): ) assert summary.llm_response == "LLM answer" - assert conversation_id == "fake_conversation_id" + assert conversation_id == "00000000-0000-0000-0000-000000000000" # Verify get_agent was called with the correct parameters mock_get_agent.assert_called_once_with( @@ -1075,7 +1124,11 @@ async def test_retrieve_response_with_mcp_servers_empty_token( mocker.patch("app.endpoints.query.configuration", mock_config) mock_get_agent = mocker.patch( "app.endpoints.query.get_agent", - return_value=(mock_agent, "fake_conversation_id", "fake_session_id"), + return_value=( + mock_agent, + "00000000-0000-0000-0000-000000000000", + "fake_session_id", + ), ) mock_metrics(mocker) @@ -1088,7 +1141,7 @@ async def test_retrieve_response_with_mcp_servers_empty_token( ) assert summary.llm_response == "LLM answer" - assert conversation_id == "fake_conversation_id" + assert conversation_id == "00000000-0000-0000-0000-000000000000" # Verify get_agent was called with the correct parameters mock_get_agent.assert_called_once_with( @@ -1137,7 +1190,11 @@ async def test_retrieve_response_with_mcp_servers_and_mcp_headers( mocker.patch("app.endpoints.query.configuration", mock_config) mock_get_agent = mocker.patch( "app.endpoints.query.get_agent", - return_value=(mock_agent, "fake_conversation_id", "fake_session_id"), + return_value=( + mock_agent, + "00000000-0000-0000-0000-000000000000", + "fake_session_id", + ), ) mock_metrics(mocker) @@ -1164,7 +1221,7 @@ async def test_retrieve_response_with_mcp_servers_and_mcp_headers( ) assert summary.llm_response == "LLM answer" - assert conversation_id == "fake_conversation_id" + assert conversation_id == "00000000-0000-0000-0000-000000000000" # Verify get_agent was called with the correct parameters mock_get_agent.assert_called_once_with( @@ -1234,7 +1291,11 @@ async def test_retrieve_response_shield_violation(prepare_agent_mocks, mocker): mocker.patch("app.endpoints.query.configuration", mock_config) mocker.patch( "app.endpoints.query.get_agent", - return_value=(mock_agent, "fake_conversation_id", "fake_session_id"), + return_value=( + mock_agent, + "00000000-0000-0000-0000-000000000000", + "fake_session_id", + ), ) mock_metrics(mocker) @@ -1247,7 +1308,7 @@ async def test_retrieve_response_shield_violation(prepare_agent_mocks, mocker): # Assert that the metric for validation errors is incremented mock_metric.inc.assert_called_once() - assert conversation_id == "fake_conversation_id" + assert conversation_id == "00000000-0000-0000-0000-000000000000" mock_agent.create_turn.assert_called_once_with( messages=[UserMessage(content="What is OpenStack?", role="user")], session_id="fake_session_id", @@ -1325,7 +1386,7 @@ async def test_auth_tuple_unpacking_in_query_endpoint_handler(mocker, dummy_requ ) mock_retrieve_response = mocker.patch( "app.endpoints.query.retrieve_response", - return_value=(summary, "test_conversation_id", []), + return_value=(summary, "00000000-0000-0000-0000-000000000000", []), ) mocker.patch( @@ -1371,7 +1432,7 @@ async def test_query_endpoint_handler_no_tools_true(mocker, dummy_request): ) ], ) - conversation_id = "fake_conversation_id" + conversation_id = "00000000-0000-0000-0000-000000000000" query = "What is OpenStack?" referenced_documents = [] @@ -1423,7 +1484,7 @@ async def test_query_endpoint_handler_no_tools_false(mocker, dummy_request): ) ], ) - conversation_id = "fake_conversation_id" + conversation_id = "00000000-0000-0000-0000-000000000000" query = "What is OpenStack?" referenced_documents = [] @@ -1473,7 +1534,11 @@ async def test_retrieve_response_no_tools_bypasses_mcp_and_rag( mocker.patch("app.endpoints.query.configuration", mock_config) mocker.patch( "app.endpoints.query.get_agent", - return_value=(mock_agent, "fake_conversation_id", "fake_session_id"), + return_value=( + mock_agent, + "00000000-0000-0000-0000-000000000000", + "fake_session_id", + ), ) mock_metrics(mocker) @@ -1486,7 +1551,7 @@ async def test_retrieve_response_no_tools_bypasses_mcp_and_rag( ) assert summary.llm_response == "LLM answer" - assert conversation_id == "fake_conversation_id" + assert conversation_id == "00000000-0000-0000-0000-000000000000" # Verify that agent.extra_headers is empty (no MCP headers) assert mock_agent.extra_headers == {} @@ -1524,7 +1589,11 @@ async def test_retrieve_response_no_tools_false_preserves_functionality( mocker.patch("app.endpoints.query.configuration", mock_config) mocker.patch( "app.endpoints.query.get_agent", - return_value=(mock_agent, "fake_conversation_id", "fake_session_id"), + return_value=( + mock_agent, + "00000000-0000-0000-0000-000000000000", + "fake_session_id", + ), ) mock_metrics(mocker) @@ -1537,7 +1606,7 @@ async def test_retrieve_response_no_tools_false_preserves_functionality( ) assert summary.llm_response == "LLM answer" - assert conversation_id == "fake_conversation_id" + assert conversation_id == "00000000-0000-0000-0000-000000000000" # Verify that agent.extra_headers contains MCP headers expected_extra_headers = { diff --git a/tests/unit/app/endpoints/test_streaming_query.py b/tests/unit/app/endpoints/test_streaming_query.py index ae732ed8..773ab66a 100644 --- a/tests/unit/app/endpoints/test_streaming_query.py +++ b/tests/unit/app/endpoints/test_streaming_query.py @@ -48,7 +48,12 @@ from models.responses import RAGChunk from utils.types import ToolCallSummary, TurnSummary -MOCK_AUTH = ("mock_user_id", "mock_username", False, "mock_token") +MOCK_AUTH = ( + "017adfa4-7cc6-46e4-b663-3653e1ae69df", + "mock_username", + False, + "mock_token", +) def mock_database_operations(mocker): @@ -119,6 +124,9 @@ def setup_configuration_fixture(): "transcripts_enabled": False, }, "mcp_servers": [], + "conversation_cache": { + "type": "noop", + }, } cfg = AppConfig() cfg.init_from_dict(config_dict) @@ -272,7 +280,7 @@ async def _test_streaming_query_endpoint_handler(mocker, store_transcript=False) query = "What is OpenStack?" mocker.patch( "app.endpoints.streaming_query.retrieve_response", - return_value=(mock_streaming_response, "test_conversation_id"), + return_value=(mock_streaming_response, "00000000-0000-0000-0000-000000000000"), ) mocker.patch( "app.endpoints.streaming_query.select_model_and_provider_id", @@ -327,8 +335,8 @@ async def _test_streaming_query_endpoint_handler(mocker, store_transcript=False) # Assert the store_transcript function is called if transcripts are enabled if store_transcript: mock_transcript.assert_called_once_with( - user_id="mock_user_id", - conversation_id="test_conversation_id", + user_id="017adfa4-7cc6-46e4-b663-3653e1ae69df", + conversation_id="00000000-0000-0000-0000-000000000000", model_id="fake_model_id", provider_id="fake_provider_id", query_is_valid=True, @@ -389,7 +397,11 @@ async def test_retrieve_response_vector_db_available(prepare_agent_mocks, mocker mocker.patch("app.endpoints.streaming_query.configuration", mock_config) mocker.patch( "app.endpoints.streaming_query.get_agent", - return_value=(mock_agent, "test_conversation_id", "test_session_id"), + return_value=( + mock_agent, + "00000000-0000-0000-0000-000000000000", + "test_session_id", + ), ) query_request = QueryRequest(query="What is OpenStack?") @@ -403,7 +415,7 @@ async def test_retrieve_response_vector_db_available(prepare_agent_mocks, mocker # For streaming, the response should be the streaming object and # conversation_id should be returned assert response is not None - assert conversation_id == "test_conversation_id" + assert conversation_id == "00000000-0000-0000-0000-000000000000" mock_agent.create_turn.assert_called_once_with( messages=[UserMessage(role="user", content="What is OpenStack?")], session_id="test_session_id", @@ -426,7 +438,11 @@ async def test_retrieve_response_no_available_shields(prepare_agent_mocks, mocke mocker.patch("app.endpoints.streaming_query.configuration", mock_config) mocker.patch( "app.endpoints.streaming_query.get_agent", - return_value=(mock_agent, "test_conversation_id", "test_session_id"), + return_value=( + mock_agent, + "00000000-0000-0000-0000-000000000000", + "test_session_id", + ), ) query_request = QueryRequest(query="What is OpenStack?") @@ -440,7 +456,7 @@ async def test_retrieve_response_no_available_shields(prepare_agent_mocks, mocke # For streaming, the response should be the streaming object and # conversation_id should be returned assert response is not None - assert conversation_id == "test_conversation_id" + assert conversation_id == "00000000-0000-0000-0000-000000000000" mock_agent.create_turn.assert_called_once_with( messages=[UserMessage(role="user", content="What is OpenStack?")], session_id="test_session_id", @@ -476,7 +492,11 @@ def __repr__(self): mocker.patch("app.endpoints.streaming_query.configuration", mock_config) mocker.patch( "app.endpoints.streaming_query.get_agent", - return_value=(mock_agent, "test_conversation_id", "test_session_id"), + return_value=( + mock_agent, + "00000000-0000-0000-0000-000000000000", + "test_session_id", + ), ) query_request = QueryRequest(query="What is OpenStack?") @@ -488,7 +508,7 @@ def __repr__(self): ) assert response is not None - assert conversation_id == "test_conversation_id" + assert conversation_id == "00000000-0000-0000-0000-000000000000" mock_agent.create_turn.assert_called_once_with( messages=[UserMessage(role="user", content="What is OpenStack?")], session_id="test_session_id", @@ -527,7 +547,11 @@ def __repr__(self): mocker.patch("app.endpoints.streaming_query.configuration", mock_config) mocker.patch( "app.endpoints.streaming_query.get_agent", - return_value=(mock_agent, "test_conversation_id", "test_session_id"), + return_value=( + mock_agent, + "00000000-0000-0000-0000-000000000000", + "test_session_id", + ), ) query_request = QueryRequest(query="What is OpenStack?") @@ -539,7 +563,7 @@ def __repr__(self): ) assert response is not None - assert conversation_id == "test_conversation_id" + assert conversation_id == "00000000-0000-0000-0000-000000000000" mock_agent.create_turn.assert_called_once_with( messages=[UserMessage(role="user", content="What is OpenStack?")], session_id="test_session_id", @@ -580,7 +604,11 @@ def __repr__(self): mocker.patch("app.endpoints.streaming_query.configuration", mock_config) mock_get_agent = mocker.patch( "app.endpoints.streaming_query.get_agent", - return_value=(mock_agent, "test_conversation_id", "test_session_id"), + return_value=( + mock_agent, + "00000000-0000-0000-0000-000000000000", + "test_session_id", + ), ) query_request = QueryRequest(query="What is OpenStack?") @@ -592,7 +620,7 @@ def __repr__(self): ) assert response is not None - assert conversation_id == "test_conversation_id" + assert conversation_id == "00000000-0000-0000-0000-000000000000" # Verify get_agent was called with the correct parameters mock_get_agent.assert_called_once_with( @@ -635,7 +663,11 @@ async def test_retrieve_response_with_one_attachment(prepare_agent_mocks, mocker ] mocker.patch( "app.endpoints.streaming_query.get_agent", - return_value=(mock_agent, "test_conversation_id", "test_session_id"), + return_value=( + mock_agent, + "00000000-0000-0000-0000-000000000000", + "test_session_id", + ), ) query_request = QueryRequest(query="What is OpenStack?", attachments=attachments) @@ -647,7 +679,7 @@ async def test_retrieve_response_with_one_attachment(prepare_agent_mocks, mocker ) assert response is not None - assert conversation_id == "test_conversation_id" + assert conversation_id == "00000000-0000-0000-0000-000000000000" mock_agent.create_turn.assert_called_once_with( messages=[UserMessage(role="user", content="What is OpenStack?")], session_id="test_session_id", @@ -688,7 +720,11 @@ async def test_retrieve_response_with_two_attachments(prepare_agent_mocks, mocke ] mocker.patch( "app.endpoints.streaming_query.get_agent", - return_value=(mock_agent, "test_conversation_id", "test_session_id"), + return_value=( + mock_agent, + "00000000-0000-0000-0000-000000000000", + "test_session_id", + ), ) query_request = QueryRequest(query="What is OpenStack?", attachments=attachments) @@ -700,7 +736,7 @@ async def test_retrieve_response_with_two_attachments(prepare_agent_mocks, mocke ) assert response is not None - assert conversation_id == "test_conversation_id" + assert conversation_id == "00000000-0000-0000-0000-000000000000" mock_agent.create_turn.assert_called_once_with( messages=[UserMessage(role="user", content="What is OpenStack?")], session_id="test_session_id", @@ -1089,7 +1125,11 @@ async def test_retrieve_response_with_mcp_servers(prepare_agent_mocks, mocker): mocker.patch("app.endpoints.streaming_query.configuration", mock_config) mock_get_agent = mocker.patch( "app.endpoints.streaming_query.get_agent", - return_value=(mock_agent, "test_conversation_id", "test_session_id"), + return_value=( + mock_agent, + "00000000-0000-0000-0000-000000000000", + "test_session_id", + ), ) query_request = QueryRequest(query="What is OpenStack?") @@ -1101,7 +1141,7 @@ async def test_retrieve_response_with_mcp_servers(prepare_agent_mocks, mocker): ) assert response is not None - assert conversation_id == "test_conversation_id" + assert conversation_id == "00000000-0000-0000-0000-000000000000" # Verify get_agent was called with the correct parameters mock_get_agent.assert_called_once_with( @@ -1157,7 +1197,11 @@ async def test_retrieve_response_with_mcp_servers_empty_token( mocker.patch("app.endpoints.streaming_query.configuration", mock_config) mock_get_agent = mocker.patch( "app.endpoints.streaming_query.get_agent", - return_value=(mock_agent, "test_conversation_id", "test_session_id"), + return_value=( + mock_agent, + "00000000-0000-0000-0000-000000000000", + "test_session_id", + ), ) query_request = QueryRequest(query="What is OpenStack?") @@ -1169,7 +1213,7 @@ async def test_retrieve_response_with_mcp_servers_empty_token( ) assert response is not None - assert conversation_id == "test_conversation_id" + assert conversation_id == "00000000-0000-0000-0000-000000000000" # Verify get_agent was called with the correct parameters mock_get_agent.assert_called_once_with( @@ -1222,7 +1266,11 @@ async def test_retrieve_response_with_mcp_servers_and_mcp_headers(mocker): mocker.patch("app.endpoints.streaming_query.configuration", mock_config) mock_get_agent = mocker.patch( "app.endpoints.streaming_query.get_agent", - return_value=(mock_agent, "test_conversation_id", "test_session_id"), + return_value=( + mock_agent, + "00000000-0000-0000-0000-000000000000", + "test_session_id", + ), ) query_request = QueryRequest(query="What is OpenStack?") @@ -1248,7 +1296,7 @@ async def test_retrieve_response_with_mcp_servers_and_mcp_headers(mocker): ) assert response is not None - assert conversation_id == "test_conversation_id" + assert conversation_id == "00000000-0000-0000-0000-000000000000" # Verify get_agent was called with the correct parameters mock_get_agent.assert_called_once_with( @@ -1306,7 +1354,7 @@ async def test_auth_tuple_unpacking_in_streaming_query_endpoint_handler(mocker): mock_streaming_response.__aiter__.return_value = iter([]) mock_retrieve_response = mocker.patch( "app.endpoints.streaming_query.retrieve_response", - return_value=(mock_streaming_response, "test_conversation_id"), + return_value=(mock_streaming_response, "00000000-0000-0000-0000-000000000000"), ) mocker.patch( @@ -1353,7 +1401,7 @@ async def test_streaming_query_endpoint_handler_no_tools_true(mocker): mocker.patch( "app.endpoints.streaming_query.retrieve_response", - return_value=(mock_streaming_response, "test_conversation_id"), + return_value=(mock_streaming_response, "00000000-0000-0000-0000-000000000000"), ) mocker.patch( "app.endpoints.streaming_query.select_model_and_provider_id", @@ -1400,7 +1448,7 @@ async def test_streaming_query_endpoint_handler_no_tools_false(mocker): mocker.patch( "app.endpoints.streaming_query.retrieve_response", - return_value=(mock_streaming_response, "test_conversation_id"), + return_value=(mock_streaming_response, "00000000-0000-0000-0000-000000000000"), ) mocker.patch( "app.endpoints.streaming_query.select_model_and_provider_id",