spiderforce4ai 2.4.5__tar.gz → 2.4.7__tar.gz

{spiderforce4ai-2.4.5 → spiderforce4ai-2.4.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: spiderforce4ai
-Version: 2.4.5
+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.5 → spiderforce4ai-2.4.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "spiderforce4ai"
-version = "2.4.5"
+version = "2.4.7"
 description = "Python wrapper for SpiderForce4AI HTML-to-Markdown conversion service with LLM post-processing"
 readme = "README.md"
 authors = [

{spiderforce4ai-2.4.5 → spiderforce4ai-2.4.7}/setup.py

@@ -7,7 +7,7 @@ with open("README.md", encoding="utf-8") as f:
 
 setup(
     name="spiderforce4ai",
-    version="2.4.5",
+    version="2.4.7",
     author="Piotr Tamulewicz",
     author_email="pt@petertam.pro",
     description="Python wrapper for SpiderForce4AI HTML-to-Markdown conversion service with LLM post-processing",

{spiderforce4ai-2.4.5 → spiderforce4ai-2.4.7}/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__)
 
@@ -190,17 +193,45 @@ class PostExtractionAgent:
                         api_base=self.config.base_url
                     )
                     
-                    # Parse response
-                    extracted_data = json.loads(response.choices[0].message.content)
-                    self.buffer.remove_request(url)  # Remove from buffer if successful
-                    return extracted_data
+                    # Log raw response for debugging
+                    raw_content = response.choices[0].message.content
+                    logger.debug(f"Raw LLM response for {url}: {raw_content}")
                     
-                except json.JSONDecodeError as e:
-                    last_error = f"Invalid JSON response from LLM: {e}"
-                    if attempt < max_retries - 1:
-                        time.sleep(retry_delay * (attempt + 1))
+                    try:
+                        # 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 Exception as e:
+                        error_msg = (
+                            f"Error processing LLM response for {url}:\n"
+                            f"Error: {str(e)}\n"
+                            f"Raw content: {raw_content[:500]}..."  # First 500 chars of response
+                        )
+                        logger.error(error_msg)
+                        last_error = error_msg
+                        if attempt < max_retries - 1:
+                            time.sleep(retry_delay * (attempt + 1))
+                            
                 except Exception as e:
-                    last_error = str(e)
+                    error_msg = f"LLM processing error for {url}: {str(e)}"
+                    logger.error(error_msg)
+                    last_error = error_msg
                     if attempt < max_retries - 1:
                         time.sleep(retry_delay * (attempt + 1))
             
@@ -242,24 +273,42 @@ class PostExtractionAgent:
 
     def process_content(self, url: str, content: str) -> Optional[Dict]:
         """Process content with retry mechanism."""
+        logger.info(f"Starting content processing for {url}")
+        
         for attempt in range(self.config.max_retries):
+            logger.info(f"Processing attempt {attempt + 1}/{self.config.max_retries} for {url}")
+            
             result = self._process_single_content(url, content)
             if result:
+                logger.info(f"Successfully processed content for {url}")
+                
                 # Apply custom transformation if provided
                 if self.config.custom_transform_function:
                     try:
                         result = self.config.custom_transform_function(result)
+                        logger.info(f"Applied custom transformation for {url}")
                     except Exception as e:
-                        logger.error(f"Error in custom transform for {url}: {str(e)}")
+                        error_msg = f"Error in custom transform for {url}: {str(e)}"
+                        logger.error(error_msg)
+                        console.print(f"[red]{error_msg}[/red]")
                 
                 # Save result synchronously
-                self._save_result_sync(url, result)
+                try:
+                    self._save_result_sync(url, result)
+                    logger.info(f"Saved results for {url}")
+                except Exception as e:
+                    error_msg = f"Error saving results for {url}: {str(e)}"
+                    logger.error(error_msg)
+                    console.print(f"[red]{error_msg}[/red]")
+                
                 return result
             
             # Wait before retry
             if attempt < self.config.max_retries - 1:
+                logger.info(f"Attempt {attempt + 1} failed for {url}, waiting {self.config.retry_delay}s before retry")
                 time.sleep(self.config.retry_delay)
         
+        logger.error(f"All processing attempts failed for {url}")
         return None
 
     async def process_bulk_content(self, content_map: Dict[str, str]) -> Dict[str, Optional[Dict]]:

{spiderforce4ai-2.4.5 → spiderforce4ai-2.4.7}/spiderforce4ai.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: spiderforce4ai
-Version: 2.4.5
+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