spiderforce4ai 2.5.2__tar.gz → 2.5.4__tar.gz

{spiderforce4ai-2.5.2 → spiderforce4ai-2.5.4}/PKG-INFO

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

{spiderforce4ai-2.5.2 → spiderforce4ai-2.5.4}/pyproject.toml

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

{spiderforce4ai-2.5.2 → spiderforce4ai-2.5.4}/setup.py

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

{spiderforce4ai-2.5.2 → spiderforce4ai-2.5.4}/spiderforce4ai/__init__.py

@@ -890,6 +890,14 @@ class SpiderForce4AI:
                                 if config.post_agent_transformer_function:
                                     try:
                                         result.extraction_result = config.post_agent_transformer_function(llm_response)
+                                    except KeyError as e:
+                                        # Log missing field but continue with transformation
+                                        missing_field = str(e).strip("'")
+                                        logger.warning(f"Missing field '{missing_field}' in LLM response for {result.url}")
+                                        # Add missing field with empty value
+                                        llm_response[missing_field] = ""
+                                        # Retry transformation with added field
+                                        result.extraction_result = config.post_agent_transformer_function(llm_response)
                                     except Exception as e:
                                         logger.error(f"Transformer error for {result.url}: {str(e)}")
                                         result.extraction_result = llm_response  # Use original response if transform fails

{spiderforce4ai-2.5.2 → spiderforce4ai-2.5.4}/spiderforce4ai/post_extraction_agent.py

@@ -298,12 +298,20 @@ class PostExtractionAgent:
                     try:
                         # Add URL to result before transformation
                         result['url'] = url
+                        
+                        # Check for required fields and set to None if missing
+                        required_fields = ['ContactInformation', 'CallToAction', 'KeyPoints', 'Title', 'Description', 'CanonicalUrl']
+                        for field in required_fields:
+                            if field not in result:
+                                logger.warning(f"Missing field '{field}' in LLM response for {url}, setting to None")
+                                result[field] = None
+                        
                         result = self.config.custom_transform_function(result)
                         logger.info(f"Applied custom transformation for {url}")
                     except Exception as e:
-                        error_msg = f"Error in custom transform for {url}: {str(e)}"
-                        logger.error(error_msg)
-                        console.print(f"[red]{error_msg}[/red]")
+                        error_msg = f"Warning: Issue in custom transform for {url}: {str(e)}"
+                        logger.warning(error_msg)
+                        console.print(f"[yellow]{error_msg}[/yellow]")
                 
                 # Save result synchronously
                 try:

{spiderforce4ai-2.5.2 → spiderforce4ai-2.5.4}/spiderforce4ai.egg-info/PKG-INFO

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