satif-ai 0.2.4__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
@@ -15,6 +16,7 @@ from satif_sdk.transformers import CodeTransformer
 # Global variables for transformation
 INPUT_SDIF_PATH: Optional[Path] = None
 OUTPUT_TARGET_FILES: Optional[Dict[Union[str, Path], str]] = None
+SCHEMA_ONLY: Optional[bool] = None
 
 
 @function_tool
@@ -32,6 +34,9 @@ async def execute_transformation(code: str) -> str:
     generated_output_path = code_transformer.export(INPUT_SDIF_PATH)
 
     comparisons = []
+    comparator_kwargs = {"check_structure_only": True}
+    if SCHEMA_ONLY:
+        comparator_kwargs["check_structure_only"] = True
 
     if os.path.isdir(generated_output_path):
         # If it's a directory, compare each file with its corresponding target
@@ -46,7 +51,9 @@ async def execute_transformation(code: str) -> str:
                     generated_output_path, output_target_file_name
                 )
                 comparator = get_comparator(output_target_file_name.split(".")[-1])
-                comparison = comparator.compare(generated_file_path, output_base_file)
+                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}"
                 )
@@ -60,7 +67,9 @@ async def execute_transformation(code: str) -> str:
             output_file = list(OUTPUT_TARGET_FILES.keys())[0]
             output_target_file_name = list(OUTPUT_TARGET_FILES.values())[0]
             comparator = get_comparator(output_file.split(".")[-1])
-            comparison = comparator.compare(generated_output_path, output_file)
+            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}"
             )
@@ -91,7 +100,7 @@ class TransformationAsyncCodeBuilder(AsyncCodeBuilder):
         self,
         mcp_server: MCPServer,
         mcp_session: ClientSession,
-        llm_model: str = "o3-mini",
+        llm_model: str = "o4-mini",
     ):
         self.mcp_server = mcp_server
         self.mcp_session = mcp_session
@@ -103,17 +112,17 @@ class TransformationAsyncCodeBuilder(AsyncCodeBuilder):
         output_target_files: Dict[Union[str, Path], str] | List[Path],
         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
+        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.
         # If execute_transformation runs in the same process as the builder, absolute path is fine.
         # If it were a separate context, this might need adjustment.
         # For now, assume execute_transformation can access absolute paths if needed for its *input SDIF*.
         # However, the sdif for MCP URIs must be relative.
-        INPUT_SDIF_PATH = Path(
-            sdif
-        ).resolve()  # Keep this as absolute for the tool's potential direct access.
-
+        INPUT_SDIF_PATH = Path(sdif).resolve()
+        SCHEMA_ONLY = schema_only
         # Paths for MCP URIs are now expected to be relative to MCP server CWD (project root)
         # So, use them directly as strings.
         input_sdif_mcp_uri_path = base64.b64encode(str(sdif).encode()).decode()
@@ -172,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}"
@@ -204,10 +218,13 @@ class TransformationAsyncCodeBuilder(AsyncCodeBuilder):
                 else "Error reading input sample",
                 "output_files": str(list(OUTPUT_TARGET_FILES.values())),
                 "output_schema": output_schema_text,
-                "output_sample": output_sample_text,
+                "output_sample": output_sample_text
+                if not SCHEMA_ONLY
+                else "Sample not available.",
                 "output_representation": str(
                     output_representation
                 ),  # Representation keyed by agent-facing name
+                "instructions": instructions,
             },
         )
         agent = Agent(

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

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

{satif_ai-0.2.4.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=c0-7JTs5pgiDOYxjWtB-w8Z-nwwqDWSUPCY1kJo3xEs,9638
+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.4.dist-info/LICENSE,sha256=kS8EN6yAaGZd7V5z6GKSn_x3ozcZltrfRky4vMPRCw8,1072
-satif_ai-0.2.4.dist-info/METADATA,sha256=3wCHxt5KwmSGllAwhqSQus7rgtcbteW7DYPmKRxIO-A,670
-satif_ai-0.2.4.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-satif_ai-0.2.4.dist-info/entry_points.txt,sha256=Mz2SwYALjktap1bF-Q3EWBgiZVNT6QJCVsCs_fCV33Y,43
-satif_ai-0.2.4.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,,