satif-ai 0.2.6__tar.gz → 0.2.7__tar.gz

{satif_ai-0.2.6 → satif_ai-0.2.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: satif-ai
-Version: 0.2.6
+Version: 0.2.7
 Summary: AI Agents for Satif
 License: MIT
 Author: Bryan Djafer

{satif_ai-0.2.6 → satif_ai-0.2.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "satif-ai"
-version = "0.2.6"
+version = "0.2.7"
 description = "AI Agents for Satif"
 authors = [
     {name = "Bryan Djafer", email = "bryan.djafer@syncpulse.fr"}

{satif_ai-0.2.6 → satif_ai-0.2.7}/satif_ai/code_builders/transformation.py

@@ -19,6 +19,37 @@ OUTPUT_TARGET_FILES: Optional[Dict[Union[str, Path], str]] = None
 SCHEMA_ONLY: Optional[bool] = None
 
 
+def _format_comparison_output(
+    comparison_result: Dict[str, Any],
+    schema_only_mode: Optional[bool],
+    source_file_display_name: str,
+    target_file_display_name: str,
+) -> str:
+    """
+    Formats the comparison result string, with special handling for schema_only mode
+    where files are equivalent due to being empty.
+    """
+    base_message_prefix = f"Comparison for {source_file_display_name} [SOURCE] with {target_file_display_name} [TARGET]:"
+
+    if schema_only_mode is True and comparison_result.get("are_equivalent") is True:
+        details = comparison_result.get("details", {})
+        row_comparison = details.get("row_comparison", {})
+
+        row_count1 = row_comparison.get("row_count1")
+        row_count2 = row_comparison.get("row_count2")
+
+        if (
+            isinstance(row_count1, (int, float))
+            and row_count1 == 0
+            and isinstance(row_count2, (int, float))
+            and row_count2 == 0
+        ):
+            return f"{base_message_prefix} Files have the same headers but are both empty (no data rows). This should not happen. Please verify the instructions and try again."
+
+    # Default formatting if the special condition isn't met
+    return f"{base_message_prefix} {comparison_result}"
+
+
 @function_tool
 async def execute_transformation(code: str) -> str:
     """Executes the transformation code on the input and returns the
@@ -54,9 +85,13 @@ async def execute_transformation(code: str) -> str:
                 comparison = comparator.compare(
                     generated_file_path, output_base_file, **comparator_kwargs
                 )
-                comparisons.append(
-                    f"Comparison for {generated_file_path} [SOURCE] with {output_target_file_name} [TARGET]: {comparison}"
+                formatted_message = _format_comparison_output(
+                    comparison,
+                    SCHEMA_ONLY,
+                    generated_file_path,
+                    output_target_file_name,
                 )
+                comparisons.append(formatted_message)
             else:
                 comparisons.append(
                     f"Error: {output_target_file_name} not found in the generated output"
@@ -70,9 +105,13 @@ async def execute_transformation(code: str) -> str:
             comparison = comparator.compare(
                 generated_output_path, output_file, **comparator_kwargs
             )
-            comparisons.append(
-                f"Comparison for {generated_output_path} [SOURCE] with {output_target_file_name} [TARGET]: {comparison}"
+            formatted_message = _format_comparison_output(
+                comparison,
+                SCHEMA_ONLY,
+                str(generated_output_path),
+                output_target_file_name,
             )
+            comparisons.append(formatted_message)
         else:
             comparisons.append(
                 "Error: Single output file generated but multiple target files expected"