update

Author: 0xrohitgarg

context/docs/process_docs.py

@@ -165,7 +165,7 @@ def process(self):
         with open(refined_prompt_path, "r", encoding="utf-8") as f:
             refined_prompt = f.read()
         refined_content = simplify_content_with_llm(refined_prompt, output, self.llm)
-        return refined_content
+        return refined_content["response"]
 
 
 if __name__ == "__main__":

context/examples/process_examples.py

@@ -163,7 +163,7 @@ def process(self):
         with open(refined_prompt_path, "r", encoding="utf-8") as f:
             refined_prompt = f.read()
         refined_content = simplify_content_with_llm(refined_prompt, combined_content, self.llm)
-        return refined_content
+        return refined_content["response"]
 
 
 if __name__ == "__main__":