iatoolkit 2.75.0__tar.gz → 2.75.2__tar.gz

{iatoolkit-2.75.0/src/iatoolkit.egg-info → iatoolkit-2.75.2}/PKG-INFO

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

{iatoolkit-2.75.0 → iatoolkit-2.75.2}/src/iatoolkit/__init__.py

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

{iatoolkit-2.75.0 → iatoolkit-2.75.2}/src/iatoolkit/services/company_context_service.py

@@ -65,7 +65,6 @@ class CompanyContextService:
         sql_context = ""
         yaml_context = ""
         db_tables = []
-        sql_source_table_scopes = {}
 
         # 1. Context from Markdown (context/*.md)  files
         try:
@@ -85,7 +84,6 @@ class CompanyContextService:
                                                                         allowed_databases=allowed_databases,
                                                                         table_filter=table_filter)
                 sql_context = sql_context.strip()
-                sql_source_table_scopes = self._get_sql_source_table_scopes(company_short_name)
             except Exception as e:
                 logging.warning(f"Could not generate SQL context for '{company_short_name}': {e}")
 
@@ -95,7 +93,6 @@ class CompanyContextService:
                 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}"
@@ -153,17 +150,6 @@ class CompanyContextService:
             logging.warning(f"Could not generate SQL context for '{company_short_name}': {e}")
             return ""
 
-    def _get_sql_source_table_scopes(self, company_short_name: str) -> dict[str, set[str] | None]:
-        scopes: dict[str, set[str] | None] = {}
-        sql_sources = self.sql_source_service.list_sources(company_short_name, include_inactive=False)
-
-        for source in sql_sources:
-            db_name = str(source.get("database") or "").strip()
-            if not db_name:
-                continue
-            scopes[db_name] = self._resolve_allowed_tables_for_source(source)
-        return scopes
-
     def _resolve_allowed_tables_for_source(self, source: dict) -> set[str] | None:
         if bool(source.get("include_all_tables", True)):
             return None
@@ -215,32 +201,25 @@ class CompanyContextService:
         self,
         company_short_name: str,
         db_tables: List[Dict],
-        sql_source_table_scopes: dict[str, set[str] | None] | None = None,
     ) -> str:
         # Get context from .yaml schema files using the repository
         yaml_schema_context = ''
-        sql_source_table_scopes = sql_source_table_scopes or {}
 
         try:
             # 1. List yaml files in the schema "folder"
             schema_files = self.asset_repo.list_files(company_short_name, AssetType.SCHEMA, extension='.yaml')
 
+            # A `<db>-<table>.yaml` of a configured SQL source belongs to the SQL
+            # schema, never to this block: it reaches the model through the
+            # catalogue (map + iat_sql_schema), which applies every filter. Read
+            # here, a table hidden from a person (a knowledge table they may not
+            # use) came back whole as an "additional schema".
+            sql_source_names = self._configured_sql_source_names(company_short_name)
+
             for filename in schema_files:
-                # skip tables that are already in the SQL context
                 if '-' in filename:
-                    dbname, f = filename.split("-", 1)
-                    table_name = f.split('.')[0]
-
-                    exists = any(
-                        item["db_name"] == dbname
-                        and bool(self._table_name_candidates(item["table_name"]) & self._table_name_candidates(table_name))
-                        for item in db_tables
-                    )
-                    if exists:
-                        continue
-
-                    allowed_tables = sql_source_table_scopes.get(dbname)
-                    if dbname in sql_source_table_scopes and not self._table_is_allowed(table_name, allowed_tables):
+                    dbname = self._normalize_identifier(filename.split("-", 1)[0])
+                    if dbname in sql_source_names:
                         continue
 
                 try:
@@ -263,6 +242,15 @@ class CompanyContextService:
 
         return yaml_schema_context
 
+    def _configured_sql_source_names(self, company_short_name: str) -> set[str]:
+        """Every SQL source of the company, active or not, by normalized name."""
+        try:
+            sources = self.sql_source_service.list_sources(company_short_name, include_inactive=True) or []
+        except Exception as e:
+            logging.warning(f"Could not list the SQL sources of '{company_short_name}': {e}")
+            return set()
+        return {self._normalize_identifier(s.get("database")) for s in sources if s.get("database")}
+
     def generate_schema_table(self, schema: dict) -> str:
         if not schema or not isinstance(schema, dict):
             return ""

{iatoolkit-2.75.0 → iatoolkit-2.75.2}/src/iatoolkit/services/context_builder_service.py

@@ -403,8 +403,10 @@ class ContextBuilderService:
             company_short_name,
             allowed_databases=allowed_databases,
             table_filter=access.sql_table_filter(company_short_name, user_identifier),
-            include_sql_context=bool(company_context_policy.get("sql_context")),
-            include_yaml_context=bool(company_context_policy.get("yaml_context")),
+            # Only the Markdown block reaches an agent; its SQL is its bound sources
+            # (sql_context above), never the global map or the residual YAML.
+            include_sql_context=False,
+            include_yaml_context=False,
         )
         markdown_context = (
             company_context_blocks.get("markdown_context")

{iatoolkit-2.75.0 → iatoolkit-2.75.2}/src/iatoolkit/services/prompt_service.py

@@ -62,10 +62,11 @@ class PromptService:
     AGENT_ROLE_CHANNELS = PromptAgentRole.CHANNELS.value
     AGENT_ROLE_OPERATIONS = PromptAgentRole.OPERATIONS.value
     DEFAULT_CATEGORY_LABEL = "General"
+    # An agent's SQL comes only from the sources bound to it: "Global SQL" and
+    # "YAML" were removed, neither ever reached the agent's context. A stored
+    # policy that still names sql_context or yaml_context loads; the key is ignored.
     COMPANY_CONTEXT_BLOCK_KEYS = {
         "markdown_context",
-        "sql_context",
-        "yaml_context",
     }
 
     @inject
@@ -187,14 +188,10 @@ class PromptService:
         if normalized == cls.AGENT_ROLE_WORKSPACE_CHAT:
             blocks = {
                 "markdown_context": True,
-                "sql_context": True,
-                "yaml_context": True,
             }
         else:
             blocks = {
                 "markdown_context": True,
-                "sql_context": False,
-                "yaml_context": False,
             }
         return {
             "company_context_blocks": blocks,

{iatoolkit-2.75.0 → iatoolkit-2.75.2/src/iatoolkit.egg-info}/PKG-INFO

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

{iatoolkit-2.75.0 → iatoolkit-2.75.2}/src/iatoolkit.egg-info/SOURCES.txt

@@ -71,7 +71,6 @@ src/iatoolkit/config/system_prompts/sql_mysql_casting.prompt
 src/iatoolkit/config/system_prompts/sql_mysql_json.prompt
 src/iatoolkit/config/system_prompts/sql_redshift_basics.prompt
 src/iatoolkit/config/system_prompts/sql_redshift_types.prompt
-src/iatoolkit/config/system_prompts/sql_rules.prompt
 src/iatoolkit/config/system_prompts/tool_html_passthrough.prompt
 src/iatoolkit/infra/__init__.py
 src/iatoolkit/infra/brevo_mail_app.py