fix bugs

2026-05-04 17:38:04 +08:00 · 2024-02-25 21:55:39 +08:00
parent 98cb17e79e
commit 34ed5e428c
2 changed files with 25 additions and 10 deletions
--- a/api/tests/unit_tests/core/prompt/test_advanced_prompt_transform.py
+++ b/api/tests/unit_tests/core/prompt/test_advanced_prompt_transform.py
@ -50,6 +50,7 @@ def test__get_completion_model_prompt_messages():
    prompt_messages = prompt_transform._get_completion_model_prompt_messages(
        prompt_template_entity=prompt_template_entity,
        inputs=inputs,
+        query=None,
        files=files,
        context=context,
        memory=memory,