w -> __extract_json_from_text

Author: whoisarpit

patchwork/steps/SimplifiedLLM/SimplifiedLLM.py

@@ -57,7 +57,7 @@ def __json_loads(json_str: str) -> dict:
             raise e
 
     @staticmethod
-    def w(text: str) -> str:
+    def __extract_json_from_text(text: str) -> str:
         try:
             start = text.find("{")
             end = text.rfind("}")
@@ -79,7 +79,7 @@ def __retry_unit(self, prepare_prompt_outputs, call_llm_inputs, retry_data: Retr
                 try:
                     # For models that don't support JSON mode, extract JSON from the text response first
                     if self.is_json_mode_unsupported:
-                        response = self.w(response)
+                        response = self.__extract_json_from_text(response)
 
                     json_response = self.__json_loads(response)
                     json_responses.append(json_response)