From 66c92133c7999869c62c19b5031382aec756f16b Mon Sep 17 00:00:00 2001 From: Yongtao Huang Date: Wed, 19 Mar 2025 00:05:35 +0800 Subject: [PATCH] Remove useless ut codes --- .../core/prompt/test_agent_history_prompt_transform.py | 2 -- .../unit_tests/core/prompt/test_simple_prompt_transform.py | 1 - 2 files changed, 3 deletions(-) diff --git a/api/tests/unit_tests/core/prompt/test_agent_history_prompt_transform.py b/api/tests/unit_tests/core/prompt/test_agent_history_prompt_transform.py index 0fd176e65d..d157a41d2c 100644 --- a/api/tests/unit_tests/core/prompt/test_agent_history_prompt_transform.py +++ b/api/tests/unit_tests/core/prompt/test_agent_history_prompt_transform.py @@ -64,12 +64,10 @@ def test_get_prompt(): transform._calculate_rest_token = MagicMock(return_value=max_token_limit) result = transform.get_prompt() - assert len(result) <= max_token_limit assert len(result) == 4 max_token_limit = 20 transform._calculate_rest_token = MagicMock(return_value=max_token_limit) result = transform.get_prompt() - assert len(result) <= max_token_limit assert len(result) == 12 diff --git a/api/tests/unit_tests/core/prompt/test_simple_prompt_transform.py b/api/tests/unit_tests/core/prompt/test_simple_prompt_transform.py index c32fc2bc34..c822ecbe78 100644 --- a/api/tests/unit_tests/core/prompt/test_simple_prompt_transform.py +++ b/api/tests/unit_tests/core/prompt/test_simple_prompt_transform.py @@ -84,7 +84,6 @@ def test_get_baichuan_completion_app_prompt_template_with_pcq(): query_in_prompt=True, with_memory_prompt=False, ) - print(prompt_template["prompt_template"].template) prompt_rules = prompt_template["prompt_rules"] assert prompt_template["prompt_template"].template == ( prompt_rules["context_prompt"] + pre_prompt + "\n" + prompt_rules["query_prompt"]