iatoolkit 2.19.0__tar.gz → 2.20.0__tar.gz

{iatoolkit-2.19.0/src/iatoolkit.egg-info → iatoolkit-2.20.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iatoolkit
-Version: 2.19.0
+Version: 2.20.0
 Summary: IAToolkit
 Author: Fernando Libedinsky
 License-Expression: MIT

{iatoolkit-2.19.0 → iatoolkit-2.20.0}/src/iatoolkit/__init__.py

@@ -3,7 +3,7 @@
 #
 # IAToolkit is open source software.
 
-__version__ = "2.19.0"
+__version__ = "2.20.0"
 
 # Expose main classes and functions at the top level of the package
 

{iatoolkit-2.19.0 → iatoolkit-2.20.0}/src/iatoolkit/runtime_logging.py

@@ -19,7 +19,7 @@ def configure_runtime_logging() -> None:
     log_level = _runtime_log_level()
 
     _ensure_stdout_handler(logging.getLogger(), formatter, log_level)
-    _ensure_stdout_handler(logging.getLogger(RUNTIME_LOGGER_NAME), formatter, log_level)
+    logging.getLogger(RUNTIME_LOGGER_NAME).propagate = True
     _preserve_stdout_handler_across_basic_config(formatter)
 
     seen_handlers: set[int] = set()

{iatoolkit-2.19.0 → iatoolkit-2.20.0}/src/iatoolkit/services/company_context_service.py

@@ -50,6 +50,7 @@ class CompanyContextService:
         company_short_name: str,
         *,
         include_sql_context: bool = True,
+        include_yaml_context: bool = True,
         selected_context_files: List[str] | None = None,
     ) -> dict[str, str]:
         """
@@ -83,17 +84,18 @@ class CompanyContextService:
             except Exception as e:
                 logging.warning(f"Could not generate SQL context for '{company_short_name}': {e}")
 
-        # 4. Context from residual yaml (schema/*.yaml) files
-        try:
-            yaml_schema_context = self._get_yaml_schema_context(
-                company_short_name,
-                db_tables,
-                sql_source_table_scopes=sql_source_table_scopes,
-            ).strip()
-            if yaml_schema_context:
-                yaml_context = f"## Esquemas adicionales\n\n{yaml_schema_context}"
-        except Exception as e:
-            logging.warning(f"Could not load Yaml context for '{company_short_name}': {e}")
+        # 3. Context from residual yaml (schema/*.yaml) files
+        if include_yaml_context:
+            try:
+                yaml_schema_context = self._get_yaml_schema_context(
+                    company_short_name,
+                    db_tables,
+                    sql_source_table_scopes=sql_source_table_scopes,
+                ).strip()
+                if yaml_schema_context:
+                    yaml_context = f"## Esquemas adicionales\n\n{yaml_schema_context}"
+            except Exception as e:
+                logging.warning(f"Could not load Yaml context for '{company_short_name}': {e}")
 
         return {
             "markdown_context": md_context,

{iatoolkit-2.19.0 → iatoolkit-2.20.0}/src/iatoolkit/services/configuration_service.py

@@ -247,10 +247,6 @@ class ConfigurationService:
             # 5. Register Knowledge base information
             self._register_knowledge_base(company_short_name, config)
 
-            # 6. Sync SQL source catalog from YAML to DB.
-            # Runtime registration is resolved from DB entries.
-            self._sync_sql_sources(company_short_name, config)
-
         # Keep runtime cache aligned with the latest loaded configuration.
         self._loaded_configs[company_short_name] = config or {}
 
@@ -388,7 +384,6 @@ class ConfigurationService:
                               config: dict = None):
         """
         Resolves runtime SQL registrations from the SQL source catalog table.
-        YAML data_sources are synchronized into the catalog before refresh.
 
         Public method: Can be called externally after initialization (e.g. by Enterprise)
         to re-register sources once new factories (like 'bridge') are available.
@@ -402,8 +397,6 @@ class ConfigurationService:
         if not config:
             return
 
-        self._sync_sql_sources(company_short_name, config or {})
-
         from iatoolkit import current_iatoolkit
         from iatoolkit.services.sql_source_service import SqlSourceService
         sql_source_service = current_iatoolkit().get_injector().get(SqlSourceService)

{iatoolkit-2.19.0 → iatoolkit-2.20.0}/src/iatoolkit/services/context_builder_service.py

@@ -354,7 +354,11 @@ class ContextBuilderService:
                 collection_names=rag_collections,
             )
 
-        company_context_blocks = self.company_context_service.get_company_context_blocks(company_short_name)
+        company_context_blocks = self.company_context_service.get_company_context_blocks(
+            company_short_name,
+            include_sql_context=False,
+            include_yaml_context=False,
+        )
         final_system_context = self._join_context_sections(
             rendered_sections.get("identity"),
             rendered_sections.get("business_context"),
@@ -364,7 +368,6 @@ class ContextBuilderService:
             collection_context,
             rendered_sections.get("data_access_rules"),
             sql_context,
-            company_context_blocks.get("yaml_context"),
             rendered_sections.get("output_contract"),
         )
 

{iatoolkit-2.19.0 → iatoolkit-2.20.0}/src/iatoolkit/services/sql_service.py

@@ -198,13 +198,19 @@ class SqlService:
                 IAToolkitException.ErrorType.DATABASE_ERROR,
                 "SQL query is required.",
             )
-        if ";" in sanitized:
+        sanitized_without_trailing_semicolon = re.sub(r";+\s*$", "", sanitized).strip()
+        if not sanitized_without_trailing_semicolon:
+            raise IAToolkitException(
+                IAToolkitException.ErrorType.DATABASE_ERROR,
+                "SQL query is required.",
+            )
+        if ";" in sanitized_without_trailing_semicolon:
             raise IAToolkitException(
                 IAToolkitException.ErrorType.DATABASE_ERROR,
                 "Only a single read-only SQL statement is allowed.",
             )
 
-        upper_sanitized = sanitized.upper()
+        upper_sanitized = sanitized_without_trailing_semicolon.upper()
         blocked_match = self._BLOCKED_SQL_PATTERN.search(upper_sanitized)
         if blocked_match:
             blocked_keyword = blocked_match.group(0).upper()

{iatoolkit-2.19.0 → iatoolkit-2.20.0/src/iatoolkit.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iatoolkit
-Version: 2.19.0
+Version: 2.20.0
 Summary: IAToolkit
 Author: Fernando Libedinsky
 License-Expression: MIT