Fix Unittest and Precommit

shared/prompts/test_time_inference.py

@@ -1,12 +1,13 @@
 import textwrap
 
+
 def intro_prompt() -> str:
     """
     Returns the intro prompt.
     """
 
     intro = textwrap.dedent(
-    """\
+        """\
     You are a world-class expert in analytical reasoning and problem-solving. Your task is to break down complex problems through rigorous step-by-step analysis, carefully examining each aspect before moving forward. For each reasoning step:
 
     OUTPUT FORMAT:
@@ -79,19 +80,21 @@ def intro_prompt() -> str:
 
     return intro
 
+
 def system_acceptance_prompt() -> str:
     """
     Returns the system acceptance prompt.
     """
 
     system_acceptance = textwrap.dedent(
-    """\
+        """\
     I understand. I will now analyze the problem systematically, following the structured reasoning process while maintaining high standards of analytical rigor and self-criticism.
     """
     ).strip()
 
     return system_acceptance
 
+
 def final_answer_prompt() -> str:
     """
     Returns the final answer prompt.

tests/prompting/shared/test_get_prompt.py

@@ -1,5 +1,5 @@
-import pytest
-from shared.prompts.test_time_inference import intro_prompt, system_acceptance_prompt, final_answer_prompt
+from shared.prompts.test_time_inference import final_answer_prompt, intro_prompt, system_acceptance_prompt
+
 
 def test_intro_prompt():
     """Test that intro_prompt returns the correct prompt."""
@@ -11,12 +11,14 @@ def test_intro_prompt():
     assert "REQUIREMENTS:" in prompt
     assert "CRITICAL THINKING CHECKLIST:" in prompt
 
+
 def test_system_acceptance_prompt():
     """Test that system_acceptance_prompt returns the correct prompt."""
     prompt = system_acceptance_prompt()
     assert isinstance(prompt, str)
     assert "I understand. I will now analyze the problem systematically" in prompt
 
+
 def test_final_answer_prompt():
     """Test that final_answer_prompt returns the correct prompt."""
     prompt = final_answer_prompt()

validator_api/test_time_inference.py

@@ -8,7 +8,7 @@
 
 from prompting.llms.apis.llm_messages import LLMMessage, LLMMessages
 from prompting.llms.apis.llm_wrapper import LLMWrapper
-from shared.prompts.test_time_inference import intro_prompt, system_acceptance_prompt, final_answer_prompt
+from shared.prompts.test_time_inference import final_answer_prompt, intro_prompt, system_acceptance_prompt
 from shared.timer import Timer
 from validator_api.chat_completion import chat_completion
 
@@ -199,12 +199,10 @@ async def generate_response(
         step_count += 1
         yield steps, None
 
-    final_answer_prompt = final_answer_prompt()
-
     messages.append(
         {
             "role": "user",
-            "content": final_answer_prompt,
+            "content": final_answer_prompt(),
         }
     )