databricks-tellr-app 0.3.11.dev6__tar.gz → 0.3.11.dev7__tar.gz

{databricks_tellr_app-0.3.11.dev6 → databricks_tellr_app-0.3.11.dev7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: databricks-tellr-app
-Version: 0.3.11.dev6
+Version: 0.3.11.dev7
 Summary: Tellr application package for Databricks Apps
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown

{databricks_tellr_app-0.3.11.dev6 → databricks_tellr_app-0.3.11.dev7}/databricks_tellr_app.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: databricks-tellr-app
-Version: 0.3.11.dev6
+Version: 0.3.11.dev7
 Summary: Tellr application package for Databricks Apps
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown

{databricks_tellr_app-0.3.11.dev6 → databricks_tellr_app-0.3.11.dev7}/databricks_tellr_app.egg-info/SOURCES.txt

@@ -33,6 +33,7 @@ setup.py
 ./src/api/routes/settings/__init__.py
 ./src/api/routes/settings/contributors.py
 ./src/api/routes/settings/deck_prompts.py
+./src/api/routes/settings/design_systems.py
 ./src/api/routes/settings/identities.py
 ./src/api/routes/settings/slide_styles.py
 ./src/api/schemas/__init__.py
@@ -73,6 +74,7 @@ setup.py
 ./src/database/__init__.py
 ./src/database/models/__init__.py
 ./src/database/models/deck_contributor.py
+./src/database/models/design_system.py
 ./src/database/models/feedback.py
 ./src/database/models/genie_space.py
 ./src/database/models/google_global_credentials.py
@@ -95,6 +97,9 @@ setup.py
 ./src/services/agent_factory.py
 ./src/services/config_service.py
 ./src/services/config_validator.py
+./src/services/design_system_compiler.py
+./src/services/design_system_service.py
+./src/services/design_system_templates.py
 ./src/services/drive_uploader.py
 ./src/services/genie_service.py
 ./src/services/google_slides_auth.py
@@ -119,12 +124,14 @@ setup.py
 ./src/services/identity_providers/workspace_provider.py
 ./src/services/tools/__init__.py
 ./src/services/tools/agent_bricks_tool.py
+./src/services/tools/ds_asset_tool.py
 ./src/services/tools/genie_tool.py
 ./src/services/tools/mcp_tool.py
 ./src/services/tools/model_endpoint_tool.py
 ./src/services/tools/vector_tool.py
 ./src/utils/__init__.py
 ./src/utils/css_utils.py
+./src/utils/ds_asset_utils.py
 ./src/utils/error_handling.py
 ./src/utils/html_safety.py
 ./src/utils/html_utils.py
@@ -172,6 +179,7 @@ src/api/routes/version.py
 src/api/routes/settings/__init__.py
 src/api/routes/settings/contributors.py
 src/api/routes/settings/deck_prompts.py
+src/api/routes/settings/design_systems.py
 src/api/routes/settings/identities.py
 src/api/routes/settings/slide_styles.py
 src/api/schemas/__init__.py
@@ -212,6 +220,7 @@ src/core/user_context.py
 src/database/__init__.py
 src/database/models/__init__.py
 src/database/models/deck_contributor.py
+src/database/models/design_system.py
 src/database/models/feedback.py
 src/database/models/genie_space.py
 src/database/models/google_global_credentials.py
@@ -234,6 +243,9 @@ src/services/agent.py
 src/services/agent_factory.py
 src/services/config_service.py
 src/services/config_validator.py
+src/services/design_system_compiler.py
+src/services/design_system_service.py
+src/services/design_system_templates.py
 src/services/drive_uploader.py
 src/services/genie_service.py
 src/services/google_slides_auth.py
@@ -258,12 +270,14 @@ src/services/identity_providers/local_provider.py
 src/services/identity_providers/workspace_provider.py
 src/services/tools/__init__.py
 src/services/tools/agent_bricks_tool.py
+src/services/tools/ds_asset_tool.py
 src/services/tools/genie_tool.py
 src/services/tools/mcp_tool.py
 src/services/tools/model_endpoint_tool.py
 src/services/tools/vector_tool.py
 src/utils/__init__.py
 src/utils/css_utils.py
+src/utils/ds_asset_utils.py
 src/utils/error_handling.py
 src/utils/html_safety.py
 src/utils/html_utils.py

{databricks_tellr_app-0.3.11.dev6 → databricks_tellr_app-0.3.11.dev7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "databricks-tellr-app"
-version = "0.3.11.dev6"
+version = "0.3.11.dev7"
 description = "Tellr application package for Databricks Apps"
 readme = "README.md"
 requires-python = ">=3.10"

{databricks_tellr_app-0.3.11.dev6 → databricks_tellr_app-0.3.11.dev7}/src/api/main.py

@@ -27,6 +27,7 @@ from src.core.permission_context import (
 from src.api.routes.settings import (
     contributors_router,
     deck_prompts_router,
+    design_systems_router,
     identities_router,
     slide_styles_router,
 )
@@ -434,6 +435,7 @@ app.include_router(contributors_router, prefix="/api/settings", tags=["settings"
 app.include_router(deck_prompts_router, prefix="/api/settings", tags=["settings"])
 app.include_router(identities_router, prefix="/api/settings", tags=["settings"])
 app.include_router(slide_styles_router, prefix="/api/settings", tags=["settings"])
+app.include_router(design_systems_router, prefix="/api/settings", tags=["settings"])
 
 # MCP server — mount the FastMCP streamable-HTTP ASGI app at /mcp.
 # Must be registered before the SPA catch-all (which is added lazily by

{databricks_tellr_app-0.3.11.dev6 → databricks_tellr_app-0.3.11.dev7}/src/api/routes/agent_config.py

@@ -21,10 +21,61 @@ router = APIRouter(prefix="/api/sessions/{session_id}/agent-config", tags=["agen
 
 
 def _validate_references(config: AgentConfig) -> None:
-    """Validate that slide_style_id and deck_prompt_id reference existing library entries."""
-    from src.database.models import SlideStyleLibrary, SlideDeckPromptLibrary
+    """Validate the library references and SANITIZE the template pin, in place.
+
+    ``design_system_id``, ``slide_style_id`` and ``deck_prompt_id`` stay STRICT
+    (422 on a missing/inactive entry — the existing convention). The template
+    pin is SELF-HEALING instead: template ids are re-minted by the sanctioned
+    delete+re-upload workflow, so a persisted config can legitimately hold a
+    stale pin — rejecting it would wedge EVERY later config update. When
+    ``template_id`` does not resolve to a template belonging to the selected
+    design system (pin with no design system, another system's template, or a
+    missing/re-materialized template), it is cleared in place with a warning;
+    the caller persists and returns the sanitized config so the frontend state
+    syncs. This mirrors the deliberately lenient generation path.
+    """
+    from src.database.models import (
+        DesignSystem,
+        DesignSystemTemplate,
+        SlideDeckPromptLibrary,
+        SlideStyleLibrary,
+    )
 
     with get_db_session() as db:
+        if config.design_system_id is not None:
+            design_system = (
+                db.query(DesignSystem)
+                .filter(
+                    DesignSystem.id == config.design_system_id,
+                    DesignSystem.is_active == True,  # noqa: E712
+                )
+                .first()
+            )
+            if not design_system:
+                raise HTTPException(
+                    status_code=422,
+                    detail=f"design_system_id {config.design_system_id} not found",
+                )
+        if config.template_id is not None:
+            template = None
+            if config.design_system_id is not None:
+                template = (
+                    db.query(DesignSystemTemplate)
+                    .filter(
+                        DesignSystemTemplate.id == config.template_id,
+                        DesignSystemTemplate.design_system_id == config.design_system_id,
+                    )
+                    .first()
+                )
+            if template is None:
+                logger.warning(
+                    "Clearing stale template pin %s (design_system_id=%s): the "
+                    "template does not exist or does not belong to the selected "
+                    "design system; saving the config without it",
+                    config.template_id,
+                    config.design_system_id,
+                )
+                config.template_id = None
         if config.slide_style_id is not None:
             style = (
                 db.query(SlideStyleLibrary)
@@ -100,7 +151,9 @@ async def get_agent_config(session_id: str):
 async def put_agent_config(session_id: str, config: AgentConfig):
     """Replace the full agent config for a session."""
     # Pydantic already validated duplicates via model_validator.
-    # Now validate foreign-key references.
+    # Now validate foreign-key references; a stale template pin is sanitized
+    # (cleared in place) rather than rejected, so what is persisted and
+    # returned below is the effective config.
     _validate_references(config)
 
     try:

{databricks_tellr_app-0.3.11.dev6 → databricks_tellr_app-0.3.11.dev7}/src/api/routes/export.py

@@ -405,12 +405,14 @@ async def export_to_pptx(request: ExportPPTXRequest):
         if not slide_deck or not slide_deck.get("slides"):
             raise HTTPException(status_code=404, detail="No slides available")
 
-        # Substitute {{image:ID}} placeholders with base64 data URIs
-        # so the PPTX converter can extract and embed the actual images
+        # Substitute {{image:ID}} + {{ds-asset:ID}} placeholders with base64 data URIs
+        # so the PPTX converter can extract and embed the actual images/brand assets
+        from src.utils.ds_asset_utils import substitute_deck_dict_ds_assets
         from src.utils.image_utils import substitute_deck_dict_images
         from src.core.database import get_db_session
         with get_db_session() as db:
             substitute_deck_dict_images(slide_deck, db)
+            substitute_deck_dict_ds_assets(slide_deck, db)
         
         slide_count = len(slide_deck.get("slides", []))
         log_msg = (

{databricks_tellr_app-0.3.11.dev6 → databricks_tellr_app-0.3.11.dev7}/src/api/routes/google_slides.py

@@ -276,11 +276,13 @@ async def start_google_slides_export(
     if not slide_deck or not slide_deck.get("slides"):
         raise HTTPException(status_code=404, detail="No slides available")
 
-    # Substitute {{image:ID}} placeholders with base64 data URIs
+    # Substitute {{image:ID}} + {{ds-asset:ID}} placeholders with base64 data URIs
+    from src.utils.ds_asset_utils import substitute_deck_dict_ds_assets
     from src.utils.image_utils import substitute_deck_dict_images
     from src.core.database import get_db_session
     with get_db_session() as db:
         substitute_deck_dict_images(slide_deck, db)
+        substitute_deck_dict_ds_assets(slide_deck, db)
 
     slides_data = slide_deck.get("slides", [])
     total = len(slides_data)
@@ -502,9 +504,11 @@ async def export_google_slides_from_huashu(
     # legacy /export route). Use a fresh session because the request's `db`
     # is held by the route's transaction context.
     from src.core.database import get_db_session
+    from src.utils.ds_asset_utils import substitute_deck_dict_ds_assets
     from src.utils.image_utils import substitute_deck_dict_images
     with get_db_session() as imdb:
         substitute_deck_dict_images(slide_deck, imdb)
+        substitute_deck_dict_ds_assets(slide_deck, imdb)
 
     title = slide_deck.get("title") or "Presentation"
     slides_data = slide_deck.get("slides") or []

{databricks_tellr_app-0.3.11.dev6 → databricks_tellr_app-0.3.11.dev7}/src/api/routes/sessions.py

@@ -149,10 +149,12 @@ def _check_deck_permission_for_session(
 def _substitute_deck_images(deck_dict: dict) -> None:
     """Substitute {{image:ID}} placeholders in a deck dict with base64 data URIs."""
     from src.core.database import get_db_session
+    from src.utils.ds_asset_utils import substitute_deck_dict_ds_assets
     from src.utils.image_utils import substitute_deck_dict_images
 
     with get_db_session() as db:
         substitute_deck_dict_images(deck_dict, db)
+        substitute_deck_dict_ds_assets(deck_dict, db)
 
 
 @router.post("")

{databricks_tellr_app-0.3.11.dev6 → databricks_tellr_app-0.3.11.dev7}/src/api/routes/settings/__init__.py

@@ -1,12 +1,14 @@
 """Configuration API routes."""
 from src.api.routes.settings.contributors import router as contributors_router
 from src.api.routes.settings.deck_prompts import router as deck_prompts_router
+from src.api.routes.settings.design_systems import router as design_systems_router
 from src.api.routes.settings.identities import router as identities_router
 from src.api.routes.settings.slide_styles import router as slide_styles_router
 
 __all__ = [
     "contributors_router",
     "deck_prompts_router",
+    "design_systems_router",
     "identities_router",
     "slide_styles_router",
 ]