spiderforce4ai 2.5.2__py3-none-any.whl → 2.5.4__py3-none-any.whl

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/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.dist-info → spiderforce4ai-2.5.4.dist-info}/METADATA

@@ -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.4.dist-info/RECORD

@@ -0,0 +1,7 @@
+spiderforce4ai/__init__.py,sha256=YgzX9AKw_q2Jp_jZq17UgOuoeJeBRfWz9I73TOBerSs,44100
+spiderforce4ai/post_extraction_agent.py,sha256=vQcna7GMYXQBRaU6uCeGi4-uZYDVAaW0dugFFEscs6o,16059
+spiderforce4ai-2.5.4.dist-info/METADATA,sha256=_oVetBxWh-pkOdK1BNKfMJgPv3uDGgSlKeiIfDR2gvo,9012
+spiderforce4ai-2.5.4.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+spiderforce4ai-2.5.4.dist-info/entry_points.txt,sha256=ibARQxOlDiL1ho12zbDZt4Uq5RKSIk_qk159ZlZ46hc,59
+spiderforce4ai-2.5.4.dist-info/top_level.txt,sha256=Kth7A21Js7DCp0j5XBBi-FE45SCLouZkeNZU__Yr9Yk,15
+spiderforce4ai-2.5.4.dist-info/RECORD,,

spiderforce4ai-2.5.2.dist-info/RECORD

@@ -1,7 +0,0 @@
-spiderforce4ai/__init__.py,sha256=ipAArcZXJpB-KT4MSs0NReY5mhpO2X0Zsoh-roibWWI,43401
-spiderforce4ai/post_extraction_agent.py,sha256=r8xRq_uwwTxAM-1FVxYRlJMg7CIwQs710plw_10P7eg,15502
-spiderforce4ai-2.5.2.dist-info/METADATA,sha256=vVB8EjRQ9BjfU6Rw-6KG_Ce5eIk3Tp6SHSnw6pLtN5Y,9012
-spiderforce4ai-2.5.2.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-spiderforce4ai-2.5.2.dist-info/entry_points.txt,sha256=ibARQxOlDiL1ho12zbDZt4Uq5RKSIk_qk159ZlZ46hc,59
-spiderforce4ai-2.5.2.dist-info/top_level.txt,sha256=Kth7A21Js7DCp0j5XBBi-FE45SCLouZkeNZU__Yr9Yk,15
-spiderforce4ai-2.5.2.dist-info/RECORD,,