spiderforce4ai 2.4.9__py3-none-any.whl → 2.5.1__py3-none-any.whl

spiderforce4ai/__init__.py

@@ -5,6 +5,9 @@ import asyncio
 import aiohttp
 import json
 import logging
+
+logger = logging.getLogger(__name__)
+import logging
 from typing import List, Dict, Union, Optional, Tuple, Callable, Any
 from dataclasses import dataclass, asdict
 from urllib.parse import urljoin, urlparse
@@ -746,7 +749,8 @@ class SpiderForce4AI:
                     
                 for result in successful_results:
                     try:
-                        result.extraction_result = asyncio.run(agent.process_content(result.url, result.markdown))
+                        # Process content synchronously since it's not an async method
+                        result.extraction_result = agent.process_content(result.url, result.markdown)
                         progress.update(llm_task, advance=1)
                     except Exception as e:
                         console.print(f"[red]Error in post-extraction processing for {result.url}: {str(e)}[/red]")
@@ -877,7 +881,20 @@ class SpiderForce4AI:
                     
                     for result in successful_results:
                         try:
-                            result.extraction_result = asyncio.run(agent.process_content(result.url, result.markdown))
+                            # Get LLM response
+                            llm_response = agent.process_content(result.url, result.markdown)
+                            if llm_response:
+                                # Add URL to the response before transformation
+                                llm_response['url'] = result.url
+                                # Apply transformation if provided
+                                if config.post_agent_transformer_function:
+                                    try:
+                                        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
+                                else:
+                                    result.extraction_result = llm_response
                             progress.update(llm_task, advance=1)
                         except Exception as e:
                             console.print(f"[red]Error in post-extraction processing for {result.url}: {str(e)}[/red]")

{spiderforce4ai-2.4.9.dist-info → spiderforce4ai-2.5.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: spiderforce4ai
-Version: 2.4.9
+Version: 2.5.1
 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.1.dist-info/RECORD

@@ -0,0 +1,7 @@
+spiderforce4ai/__init__.py,sha256=ipAArcZXJpB-KT4MSs0NReY5mhpO2X0Zsoh-roibWWI,43401
+spiderforce4ai/post_extraction_agent.py,sha256=so5Ze7Vz3konpQ0iT7ZxDGE9kIYeTwPTFyzezRc5oys,15392
+spiderforce4ai-2.5.1.dist-info/METADATA,sha256=VIpmVqwJC6fKufS73tZBYXir8vHaLTScfF3sif28c3o,9012
+spiderforce4ai-2.5.1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+spiderforce4ai-2.5.1.dist-info/entry_points.txt,sha256=ibARQxOlDiL1ho12zbDZt4Uq5RKSIk_qk159ZlZ46hc,59
+spiderforce4ai-2.5.1.dist-info/top_level.txt,sha256=Kth7A21Js7DCp0j5XBBi-FE45SCLouZkeNZU__Yr9Yk,15
+spiderforce4ai-2.5.1.dist-info/RECORD,,

spiderforce4ai-2.4.9.dist-info/RECORD

@@ -1,7 +0,0 @@
-spiderforce4ai/__init__.py,sha256=DUPOKF7-vCVQi7JimsStU1qjk5x3yVUoMnUVOJxOrGk,42360
-spiderforce4ai/post_extraction_agent.py,sha256=so5Ze7Vz3konpQ0iT7ZxDGE9kIYeTwPTFyzezRc5oys,15392
-spiderforce4ai-2.4.9.dist-info/METADATA,sha256=kEq3anAkoe_wpPVzpgaJlsSuAzTQHDgXiDFpirXvUQc,9012
-spiderforce4ai-2.4.9.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-spiderforce4ai-2.4.9.dist-info/entry_points.txt,sha256=ibARQxOlDiL1ho12zbDZt4Uq5RKSIk_qk159ZlZ46hc,59
-spiderforce4ai-2.4.9.dist-info/top_level.txt,sha256=Kth7A21Js7DCp0j5XBBi-FE45SCLouZkeNZU__Yr9Yk,15
-spiderforce4ai-2.4.9.dist-info/RECORD,,