satif-ai 0.2.5__py3-none-any.whl → 0.2.6__py3-none-any.whl

satif_ai/code_builders/transformation.py

@@ -1,8 +1,9 @@
 import base64
 import os
 import re
+from collections import defaultdict
 from pathlib import Path
-from typing import Dict, List, Optional, Union
+from typing import Any, Dict, List, Optional, Union
 
 from agents import Agent, Runner, function_tool
 from agents.mcp.server import MCPServer
@@ -33,7 +34,7 @@ async def execute_transformation(code: str) -> str:
     generated_output_path = code_transformer.export(INPUT_SDIF_PATH)
 
     comparisons = []
-    comparator_kwargs = {}
+    comparator_kwargs = {"check_structure_only": True}
     if SCHEMA_ONLY:
         comparator_kwargs["check_structure_only"] = True
 
@@ -112,6 +113,7 @@ class TransformationAsyncCodeBuilder(AsyncCodeBuilder):
         output_sdif: Optional[Path] = None,  # This will now be relative or None
         instructions: Optional[str] = None,
         schema_only: bool = False,
+        representer_options_for_build: Optional[Dict[str, Any]] = None,
     ) -> str:
         global INPUT_SDIF_PATH, OUTPUT_TARGET_FILES, SCHEMA_ONLY
         # INPUT_SDIF_PATH is used by execute_transformation tool, needs to be accessible from where that tool runs.
@@ -179,16 +181,21 @@ class TransformationAsyncCodeBuilder(AsyncCodeBuilder):
         else:
             OUTPUT_TARGET_FILES = {}
 
-        output_representation = {}
+        output_representation = defaultdict(dict)
         if OUTPUT_TARGET_FILES:
             for file_key_abs_path in list(OUTPUT_TARGET_FILES.keys()):
                 agent_facing_name = OUTPUT_TARGET_FILES[file_key_abs_path]
-                print(f"Representing {agent_facing_name} from {file_key_abs_path}!!")
+                print(f"Representing {agent_facing_name} from {file_key_abs_path}")
                 try:
                     # Representer uses the absolute path (file_key_abs_path) to read the example file.
-                    output_representation[agent_facing_name] = get_representer(
-                        file_key_abs_path
-                    ).represent(file_key_abs_path)
+                    representer = get_representer(file_key_abs_path)
+                    representation, used_params = representer.represent(
+                        file_key_abs_path, **(representer_options_for_build or {})
+                    )
+                    output_representation[agent_facing_name] = {
+                        "representation": representation,
+                        "used_params": used_params,
+                    }
                 except Exception as e:
                     print(
                         f"Warning: Could not get representation for {agent_facing_name} (path {file_key_abs_path}): {e}"
@@ -213,10 +220,11 @@ class TransformationAsyncCodeBuilder(AsyncCodeBuilder):
                 "output_schema": output_schema_text,
                 "output_sample": output_sample_text
                 if not SCHEMA_ONLY
-                else "Sample not available. Transform according to the schema only. No data is present in the output example.",
+                else "Sample not available.",
                 "output_representation": str(
                     output_representation
                 ),  # Representation keyed by agent-facing name
+                "instructions": instructions,
             },
         )
         agent = Agent(

{satif_ai-0.2.5.dist-info → satif_ai-0.2.6.dist-info}/METADATA

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

{satif_ai-0.2.5.dist-info → satif_ai-0.2.6.dist-info}/RECORD

@@ -3,15 +3,15 @@ satif_ai/adapters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
 satif_ai/adapters/tidy.py,sha256=2oYj7Gz3vOQtzcpoJI4JbftWlMKvOWL8rdwthjg-zUE,19884
 satif_ai/code_builders/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 satif_ai/code_builders/adaptation.py,sha256=E29YM0S6pMtAfB0uzSUexoeWKwXfF8iJVyYUCKWQz5k,188
-satif_ai/code_builders/transformation.py,sha256=ZYpZpJb28PzN3suNG1lxeF2zB6mWn8gMoOLQov9YJUA,10043
+satif_ai/code_builders/transformation.py,sha256=mO_kGYl6QYvErW1rVaToDYJ2rpE36hUmKC7HjGl4ytI,10432
 satif_ai/plot_builders/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 satif_ai/plot_builders/agent.py,sha256=Ncw7SL9qkpRN0hw76ezSo1K8vVQK6gcXFp8x8VFwqUI,8291
 satif_ai/plot_builders/prompt.py,sha256=m0W1SsxnB9_FhIYRumkthImJbK-7KUm4dygN3kjAXGk,6877
 satif_ai/plot_builders/tool.py,sha256=MeLnG_wFoITSVWZcNFsQLCi157O4L3wItQgolBa4fAw,5994
 satif_ai/standardizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 satif_ai/standardizers/ai_csv.py,sha256=AAeTt7eqFAtayxF2b95Z_K_lnMdwBBnv2Cn-qTEpMp8,29499
-satif_ai-0.2.5.dist-info/LICENSE,sha256=kS8EN6yAaGZd7V5z6GKSn_x3ozcZltrfRky4vMPRCw8,1072
-satif_ai-0.2.5.dist-info/METADATA,sha256=jIyyKOcveIz_98_JTyF6bsfVn5iOVvc3qLFVQZ0wUSo,670
-satif_ai-0.2.5.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-satif_ai-0.2.5.dist-info/entry_points.txt,sha256=Mz2SwYALjktap1bF-Q3EWBgiZVNT6QJCVsCs_fCV33Y,43
-satif_ai-0.2.5.dist-info/RECORD,,
+satif_ai-0.2.6.dist-info/LICENSE,sha256=kS8EN6yAaGZd7V5z6GKSn_x3ozcZltrfRky4vMPRCw8,1072
+satif_ai-0.2.6.dist-info/METADATA,sha256=qM5JQ9OJfC2zTlYlp6XyyrLdzuL1N45hYWsjgUSQxAM,670
+satif_ai-0.2.6.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+satif_ai-0.2.6.dist-info/entry_points.txt,sha256=Mz2SwYALjktap1bF-Q3EWBgiZVNT6QJCVsCs_fCV33Y,43
+satif_ai-0.2.6.dist-info/RECORD,,