updates

xvnpw · Dec 7, 2024 · a9501e3 · a9501e3
1 parent a39f513
commit a9501e3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_agents.py b/tests/test_agents.py
@@ -366,11 +366,11 @@ def test_create_human_prompt(
     assert ret_by_lines[1] == (
         "There will be more files to analyze after this batch." if len(documents) > len(batch) else ""
     )
+    assert ret_by_lines[3] == "CURRENT DESIGN DOCUMENT:"
 
     lines = 0
     if current_description:
-        assert ret_by_lines[3] == "CURRENT DESIGN DOCUMENT:"
         assert ret_by_lines[4] == current_description
-        lines = 3
+        lines = 2
 
-    assert ret_by_lines[3 + lines] == "PROJECT FILES:"
+    assert ret_by_lines[4 + lines] == "PROJECT FILES:"