spiderforce4ai 2.4.6__py3-none-any.whl → 2.4.7__py3-none-any.whl

spiderforce4ai/post_extraction_agent.py

@@ -12,6 +12,9 @@ from pydantic import BaseModel, Field
 import logging
 from datetime import datetime
 import re
+from rich.console import Console
+
+console = Console()
 
 logger = logging.getLogger(__name__)
 
@@ -195,12 +198,28 @@ class PostExtractionAgent:
                     logger.debug(f"Raw LLM response for {url}: {raw_content}")
                     
                     try:
-                        extracted_data = json.loads(raw_content)
+                        # First try direct JSON parsing
+                        try:
+                            extracted_data = json.loads(raw_content)
+                        except json.JSONDecodeError:
+                            # Look for JSON in markdown code blocks
+                            json_match = re.search(r'```(?:json)?\s*\n([\s\S]*?)\n```', raw_content)
+                            if json_match:
+                                json_content = json_match.group(1).strip()
+                                extracted_data = json.loads(json_content)
+                            else:
+                                # If no JSON found, try to extract structured data in any format
+                                extracted_data = {
+                                    "raw_content": raw_content,
+                                    "format": "text",
+                                    "timestamp": datetime.now().isoformat()
+                                }
+                        
                         self.buffer.remove_request(url)  # Remove from buffer if successful
                         return extracted_data
-                    except json.JSONDecodeError as e:
+                    except Exception as e:
                         error_msg = (
-                            f"Invalid JSON response from LLM for {url}:\n"
+                            f"Error processing LLM response for {url}:\n"
                             f"Error: {str(e)}\n"
                             f"Raw content: {raw_content[:500]}..."  # First 500 chars of response
                         )

{spiderforce4ai-2.4.6.dist-info → spiderforce4ai-2.4.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: spiderforce4ai
-Version: 2.4.6
+Version: 2.4.7
 Summary: Python wrapper for SpiderForce4AI HTML-to-Markdown conversion service with LLM post-processing
 Home-page: https://petertam.pro
 Author: Piotr Tamulewicz

spiderforce4ai-2.4.7.dist-info/RECORD

@@ -0,0 +1,7 @@
+spiderforce4ai/__init__.py,sha256=PPpJLowJhgoRijsF2ebmdkFbIriI_yIFlCi1wL6hSP8,42267
+spiderforce4ai/post_extraction_agent.py,sha256=q2ohsqw_F1e5rT2H9eSzCWzstJLbwGyCtwLsC6eMufs,14560
+spiderforce4ai-2.4.7.dist-info/METADATA,sha256=r273h2ogI76aXTd8XN9b81EWtQLuhdSjZkXD2Ks8GnM,9012
+spiderforce4ai-2.4.7.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+spiderforce4ai-2.4.7.dist-info/entry_points.txt,sha256=ibARQxOlDiL1ho12zbDZt4Uq5RKSIk_qk159ZlZ46hc,59
+spiderforce4ai-2.4.7.dist-info/top_level.txt,sha256=Kth7A21Js7DCp0j5XBBi-FE45SCLouZkeNZU__Yr9Yk,15
+spiderforce4ai-2.4.7.dist-info/RECORD,,

spiderforce4ai-2.4.6.dist-info/RECORD

@@ -1,7 +0,0 @@
-spiderforce4ai/__init__.py,sha256=PPpJLowJhgoRijsF2ebmdkFbIriI_yIFlCi1wL6hSP8,42267
-spiderforce4ai/post_extraction_agent.py,sha256=Ty9-Ai3_RVT86RrPUxKEzf4oUh-Wr7bk2aM87Je_WvE,13580
-spiderforce4ai-2.4.6.dist-info/METADATA,sha256=7FaEgAHdD-8a0XmuDMkpAUjAQ7ZmFTD89IqQM17nllI,9012
-spiderforce4ai-2.4.6.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-spiderforce4ai-2.4.6.dist-info/entry_points.txt,sha256=ibARQxOlDiL1ho12zbDZt4Uq5RKSIk_qk159ZlZ46hc,59
-spiderforce4ai-2.4.6.dist-info/top_level.txt,sha256=Kth7A21Js7DCp0j5XBBi-FE45SCLouZkeNZU__Yr9Yk,15
-spiderforce4ai-2.4.6.dist-info/RECORD,,