spiderforce4ai 2.5.8__py3-none-any.whl → 2.6__py3-none-any.whl

spiderforce4ai/post_extraction_agent.py

@@ -260,13 +260,34 @@ class PostExtractionAgent:
         """Save individual or combined results synchronously."""
         try:
             if self.config.combine_output and self.config.output_file:
+                # Convert Path to string if needed
+                output_file = str(self.config.output_file) if isinstance(self.config.output_file, Path) else self.config.output_file
+                
+                # Load existing results if file exists
+                if Path(output_file).exists():
+                    try:
+                        with open(output_file, 'r') as f:
+                            self.results = json.load(f)
+                    except json.JSONDecodeError:
+                        self.results = {}
+                
+                # Update results with new data
                 self.results[url] = result
-                # Save combined results
-                with open(self.config.output_file, 'w') as f:
+                
+                # Ensure output directory exists
+                Path(output_file).parent.mkdir(parents=True, exist_ok=True)
+                
+                # Save combined results atomically
+                temp_file = f"{output_file}.tmp"
+                with open(temp_file, 'w') as f:
                     json.dump(self.results, f, indent=2)
                 
-                # Cleanup backup files after successful save
-                for backup_file in self.config.output_file.parent.glob(f"{self.config.output_file.stem}.bak_*"):
+                # Atomic replace
+                Path(temp_file).replace(output_file)
+                logger.info(f"Updated combined results file with {url}")
+                
+                # Cleanup backup files
+                for backup_file in Path(output_file).parent.glob(f"{Path(output_file).stem}.bak_*"):
                     try:
                         backup_file.unlink()
                         logger.info(f"Cleaned up backup file: {backup_file}")
@@ -277,6 +298,7 @@ class PostExtractionAgent:
                 individual_file = self.config.output_file.parent / f"{url.replace('/', '_')}.json"
                 with open(individual_file, 'w') as f:
                     json.dump(result, f, indent=2)
+                logger.info(f"Saved individual result file for {url}")
         except Exception as e:
             logger.error(f"Error saving results for {url}: {str(e)}")
 
@@ -323,14 +345,19 @@ class PostExtractionAgent:
                 
                 # Save result synchronously
                 try:
-                    self._save_result_sync(url, result)
-                    logger.info(f"Saved results for {url}")
+                    # Save both original and transformed result
+                    if self.config.custom_transform_function:
+                        self._save_result_sync(url, transformed_result)
+                        logger.info(f"Saved transformed results for {url}")
+                    else:
+                        self._save_result_sync(url, result)
+                        logger.info(f"Saved original 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
+                return transformed_result if self.config.custom_transform_function else result
             
             # Wait before retry
             if attempt < self.config.max_retries - 1:

{spiderforce4ai-2.5.8.dist-info → spiderforce4ai-2.6.dist-info}/METADATA

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

spiderforce4ai-2.6.dist-info/RECORD

@@ -0,0 +1,7 @@
+spiderforce4ai/__init__.py,sha256=BlrRIrtpDUHjmDedqgXP1KbAAsAH9vwFPncUR5VGGyM,44804
+spiderforce4ai/post_extraction_agent.py,sha256=AysrHIoD-IreqbvWqCDxyN7v8EPSdLOG9yxABamTZSg,17827
+spiderforce4ai-2.6.dist-info/METADATA,sha256=JtIZ1-ojRvfm773-yF1a_M_x6eB5kbnb6WT5XT04KDA,9010
+spiderforce4ai-2.6.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+spiderforce4ai-2.6.dist-info/entry_points.txt,sha256=ibARQxOlDiL1ho12zbDZt4Uq5RKSIk_qk159ZlZ46hc,59
+spiderforce4ai-2.6.dist-info/top_level.txt,sha256=Kth7A21Js7DCp0j5XBBi-FE45SCLouZkeNZU__Yr9Yk,15
+spiderforce4ai-2.6.dist-info/RECORD,,

spiderforce4ai-2.5.8.dist-info/RECORD

@@ -1,7 +0,0 @@
-spiderforce4ai/__init__.py,sha256=BlrRIrtpDUHjmDedqgXP1KbAAsAH9vwFPncUR5VGGyM,44804
-spiderforce4ai/post_extraction_agent.py,sha256=nyc6V21E3laTetFG2aEBMmFiXv5goumo8eHoupp3mKA,16430
-spiderforce4ai-2.5.8.dist-info/METADATA,sha256=eitdlF0yEzoFabYliZWvNSx7hGQMBDVvKwgCh21lU18,9012
-spiderforce4ai-2.5.8.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-spiderforce4ai-2.5.8.dist-info/entry_points.txt,sha256=ibARQxOlDiL1ho12zbDZt4Uq5RKSIk_qk159ZlZ46hc,59
-spiderforce4ai-2.5.8.dist-info/top_level.txt,sha256=Kth7A21Js7DCp0j5XBBi-FE45SCLouZkeNZU__Yr9Yk,15
-spiderforce4ai-2.5.8.dist-info/RECORD,,