fix formatting

Author: rbotnari

rlm/core/rlm.py

@@ -265,7 +265,9 @@ def _spawn_completion_context(self, prompt: str | dict[str, Any]):
             if self.tool_code and self.tool_code.strip():
                 existing_setup_code = env_kwargs.get("setup_code")
                 if existing_setup_code and existing_setup_code.strip():
-                    env_kwargs["setup_code"] = existing_setup_code.rstrip() + "\n\n" + self.tool_code.lstrip()
+                    env_kwargs["setup_code"] = (
+                        existing_setup_code.rstrip() + "\n\n" + self.tool_code.lstrip()
+                    )
                 else:
                     env_kwargs["setup_code"] = self.tool_code
 

rlm/utils/prompts.py

@@ -16,6 +16,7 @@ def append_prompt_sections(base_prompt: str, extra_sections: str | list[str] | N
     # Keep the existing prompt verbatim; just append extra sections.
     return base_prompt.rstrip() + "\n\n" + "\n\n".join(cleaned_sections) + "\n"
 
+
 # System prompt for the REPL environment with explicit final answer checking
 RLM_SYSTEM_PROMPT = textwrap.dedent(
     """You are tasked with answering a query with associated context. You can access, transform, and analyze this context interactively in a REPL environment that can recursively query sub-LLMs, which you are strongly encouraged to use as much as possible. You will be queried iteratively until you provide a final answer.