fix: The extract method logs output only when self.verbose is set to True. (#314)

Co-authored-by: lu4nx <lu4nx@lx-pc>
This commit is contained in:
lu4nx
2024-12-09 17:19:26 +08:00
committed by GitHub
parent e3488da194
commit ba3e808802

View File

@@ -92,8 +92,10 @@ class LLMExtractionStrategy(ExtractionStrategy):
def extract(self, url: str, ix:int, html: str) -> List[Dict[str, Any]]: def extract(self, url: str, ix:int, html: str) -> List[Dict[str, Any]]:
# print("[LOG] Extracting blocks from URL:", url) if self.verbose:
print(f"[LOG] Call LLM for {url} - block index: {ix}") # print("[LOG] Extracting blocks from URL:", url)
print(f"[LOG] Call LLM for {url} - block index: {ix}")
variable_values = { variable_values = {
"URL": url, "URL": url,
"HTML": escape_json_string(sanitize_html(html)), "HTML": escape_json_string(sanitize_html(html)),