agent-framework-lib 0.8.7.post6__tar.gz → 0.8.7.post7__tar.gz

{agent_framework_lib-0.8.7.post6/agent_framework_lib.egg-info → agent_framework_lib-0.8.7.post7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-framework-lib
-Version: 0.8.7.post6
+Version: 0.8.7.post7
 Summary: A comprehensive Python framework for building and serving conversational AI agents with FastAPI
 Author-email: Sebastian Pavel <sebastian@cinco.ai>, Elliott Girard <elliott.girard@icloud.com>
 Maintainer-email: Sebastian Pavel <sebastian@cinco.ai>

{agent_framework_lib-0.8.7.post6 → agent_framework_lib-0.8.7.post7}/agent_framework/skills/builtin/skills/image_gen/SKILL.md

@@ -1,6 +1,6 @@
 ---
 name: image_gen
-description: "Generate images using AI image generation APIs (DALL-E 3, DALL-E 2). Create custom illustrations, diagrams, and visual content from text descriptions."
+description: "Generate images using AI image generation APIs (gpt-image-1, DALL-E 3, DALL-E 2). Create custom illustrations, diagrams, and visual content from text descriptions."
 display_name: "Génération d'images par IA"
 metadata:
   config:
@@ -13,6 +13,7 @@ metadata:
       - ai image
       - dall-e
       - dalle
+      - gpt-image
       - image generation
       - create picture
       - generate illustration
@@ -27,9 +28,11 @@ metadata:
 
 ## AI Image Generation
 
-Generate images using OpenAI's DALL-E models by running the `generate_image.py` script via `shell_exec`.
+Generate images using OpenAI's image models by running the `create_image.py` script via `shell_exec`.
 
-The script supports DALL-E 3 (latest, highest quality) and DALL-E 2 (faster, more affordable) with customizable parameters for size, quality, and style.
+The default model is **gpt-image-1** (OpenAI's current image model: best quality, reliable text rendering, transparent backgrounds). **DALL-E 3** and **DALL-E 2** remain available as alternatives.
+
+> **gpt-image-1 requires a verified OpenAI organization.** If the org is not verified the API returns a 403; in that case set `IMAGE_GEN_DEFAULT_MODEL=dall-e-3` (env var) or pass `"model": "dall-e-3"` in the request to fall back.
 
 ---
 
@@ -38,9 +41,11 @@ The script supports DALL-E 3 (latest, highest quality) and DALL-E 2 (faster, mor
 Pipe a JSON generation request to stdin and specify the target filename:
 
 ```bash
-echo '{"prompt": "A futuristic cityscape at sunset", "model": "dall-e-3", "size": "1024x1024"}' | uv run python -m agent_framework.skills.builtin.scripts.create_and_register --skill image_gen --filename "cityscape.png" --output-dir "/path/to/output"
+echo '{"prompt": "A futuristic cityscape at sunset", "size": "1536x1024", "quality": "high"}' | uv run python -m agent_framework.skills.builtin.scripts.create_and_register --skill image_gen --filename "cityscape.png" --output-dir "/path/to/output"
 ```
 
+The default model (`gpt-image-1`) is used when `model` is omitted.
+
 ---
 
 ## Arguments
@@ -58,20 +63,35 @@ Provide a JSON object on stdin with the following structure:
 
 | Field | Type | Required | Description |
 |-------|------|----------|-------------|
-| `prompt` | string | Yes | Text description of the image to generate (max 4000 characters for DALL-E 3, 1000 for DALL-E 2) |
-| `model` | string | No | AI model: `"dall-e-3"` (default), `"dall-e-2"` |
+| `prompt` | string | Yes | Text description of the image to generate (max 32000 chars for gpt-image-1, 4000 for DALL-E 3, 1000 for DALL-E 2) |
+| `model` | string | No | AI model: `"gpt-image-1"` (default), `"dall-e-3"`, `"dall-e-2"` |
 | `size` | string | No | Image size (see Size Options below) |
-| `quality` | string | No | Image quality: `"standard"` (default), `"hd"` (DALL-E 3 only) |
+| `quality` | string | No | gpt-image-1: `low`/`medium`/`high`(default)/`auto` — DALL-E 3: `standard`(default)/`hd` |
 | `style` | string | No | Image style: `"vivid"` (default), `"natural"` (DALL-E 3 only) |
-| `n` | integer | No | Number of images to generate (1-10 for DALL-E 2, always 1 for DALL-E 3) |
+| `n` | integer | No | Number of images to generate (1-10 for gpt-image-1 and DALL-E 2, always 1 for DALL-E 3) |
 
 ---
 
 ## Model Comparison
 
-### DALL-E 3 (Recommended)
+### gpt-image-1 (Default, Recommended)
 
-**Best for**: High-quality, detailed images with better prompt following
+**Best for**: Highest-quality images, accurate text rendering, transparent backgrounds.
+
+| Feature | Value |
+|---------|-------|
+| Quality | Best available (`low`/`medium`/`high`/`auto`) |
+| Prompt adherence | Excellent — handles long, complex prompts (up to 32000 chars) |
+| Available sizes | `1024x1024`, `1536x1024` (landscape), `1024x1536` (portrait), `auto` |
+| Output | Always returned as base64 (handled transparently by the script) |
+| Requirement | **Verified OpenAI organization** (else 403 — fall back to dall-e-3) |
+| Cost / speed | Higher cost, slower than DALL-E 2 |
+
+**Use gpt-image-1 when**: you want the best quality, images with legible text, or transparent backgrounds. This is the default.
+
+### DALL-E 3
+
+**Best for**: High-quality, detailed images with better prompt following (alternative to gpt-image-1).
 
 | Feature | Value |
 |---------|-------|
@@ -83,10 +103,9 @@ Provide a JSON object on stdin with the following structure:
 | Speed | Slower (~10-30 seconds) |
 
 **Use DALL-E 3 when**:
-- You need high-quality, detailed images
+- gpt-image-1 is unavailable (org not verified)
+- You need the `vivid`/`natural` style control
 - Prompt contains complex or specific requirements
-- Image will be used professionally (presentations, marketing)
-- You want HDproperty quality output
 
 ### DALL-E 2
 
@@ -111,6 +130,12 @@ Provide a JSON object on stdin with the following structure:
 
 ## Size Options
 
+### gpt-image-1 Sizes
+- `1024x1024` — Square (default) — General purpose
+- `1536x1024` — Landscape — Presentations, web banners
+- `1024x1536` — Portrait — Mobile screens, posters
+- `auto` — Let the model pick the best size for the prompt
+
 ### DALL-E 3 Sizes
 - `1024x1024` — Square (default) — General purpose
 - `1024x1792` — Portrait (9:16 aspect ratio) — Mobile screens, posters
@@ -123,19 +148,26 @@ Provide a JSON object on stdin with the following structure:
 
 ---
 
-## Quality & Style (DALL-E 3 Only)
+## Quality & Style
+
+### Quality (gpt-image-1)
 
-### Quality
+- `"low"` — Fastest, lowest cost
+- `"medium"` — Balanced
+- `"high"` (default) — Best detail
+- `"auto"` — Let the model decide
+
+### Quality (DALL-E 3)
 
 - `"standard"` (default) — Good quality, faster, lower cost
 - `"hd"` — Highest quality, more detail, slower, higher cost
 
-**Use HD quality for**:
+**Use higher quality for**:
 - Professional presentations and marketing materials
 - Large format prints or displays
 - Images requiring fine details (textures, faces, text)
 
-### Style
+### Style (DALL-E 3 only)
 
 - `"vivid"` (default) — Vibrant, hyper-real, dramatic — Good for eye-catching visuals
 - `"natural"` — More subdued, realistic, natural tones — Good for professional/corporate content
@@ -144,18 +176,18 @@ Provide a JSON object on stdin with the following structure:
 
 ## Complete Example
 
-Generate a high-quality landscape illustration for a presentation:
+Generate a high-quality landscape illustration for a presentation (default model gpt-image-1):
 
 ```bash
 echo '{
   "prompt": "A serene mountain landscape with a crystal-clear lake reflecting snow-capped peaks, surrounded by pine forests, during golden hour. Photorealistic style with soft lighting and vibrant colors.",
-  "model": "dall-e-3",
-  "size": "1792x1024",
-  "quality": "hd",
-  "style": "natural"
+  "size": "1536x1024",
+  "quality": "high"
 }' | uv run python -m agent_framework.skills.builtin.scripts.create_and_register --skill image_gen --filename "mountain_landscape.png"
 ```
 
+To use DALL-E 3 instead, set `"model": "dall-e-3"` (with `quality` `standard`/`hd` and an optional `style`).
+
 ---
 
 ## Output Format (stdout)
@@ -172,15 +204,15 @@ On success, the script prints a JSON object with download information:
   "mime_type": "image/png",
   "local_cleaned": true,
   "metadata": {
-    "model": "dall-e-3",
-    "size": "1792x1024",
-    "quality": "hd",
+    "model": "gpt-image-1",
+    "size": "1536x1024",
+    "quality": "high",
     "revised_prompt": "A tranquil mountain vista featuring..."
   }
 }
 ```
 
-**Note**: The `revised_prompt` field (DALL-E 3 only) shows how OpenAI interpreted and enhanced your original prompt.
+**Note**: The `revised_prompt` field (when provided by the model) shows how OpenAI interpreted and enhanced your original prompt.
 
 Use the `download_url` to provide a download link:
 ```
@@ -209,6 +241,7 @@ On failure, the script prints a JSON error to stderr and exits with non-zero cod
 | 1 | Invalid size | `"Invalid size '2048x2048' for dall-e-3. Must be one of: 1024x1024, 1024x1792, 1792x1024"` |
 | 1 | Missing API key | `"OPENAI_API_KEY environment variable is not set"` |
 | 1 | API error | `"OpenAI API error: Rate limit exceeded"` |
+| 1 | Org not verified (gpt-image-1) | `"model 'gpt-image-1' requires a verified OpenAI organization. Set IMAGE_GEN_DEFAULT_MODEL=dall-e-3 ..."` |
 | 1 | Content policy violation | `"Image generation failed: Content policy violation"` |
 
 ---
@@ -371,7 +404,7 @@ If your prompt violates the content policy, the API will return an error.
 When helping users generate images:
 
 1. **Clarify requirements**: Ask about purpose, style, dimensions, quality needs
-2. **Suggest model**: DALL-E 3 for quality, DALL-E 2 for speed/cost
+2. **Suggest model**: gpt-image-1 (default) for best quality and text rendering; DALL-E 3 if you need vivid/natural style control or the org isn't verified; DALL-E 2 for cheap, fast iterations
 3. **Recommend size**: Based on use case (presentation → landscape, social media → square, poster → portrait)
 4. **Craft detailed prompts**: Include style, mood, lighting, composition
 5. **Set expectations**: Mention generation time (~10-30s for DALL-E 3)

{agent_framework_lib-0.8.7.post6 → agent_framework_lib-0.8.7.post7}/agent_framework/skills/builtin/skills/image_gen/create_image.py

@@ -8,6 +8,7 @@ downloads it, and saves it to the specified output directory.
 This script is designed to be called by create_and_register.py wrapper.
 """
 import argparse
+import base64
 import json
 import os
 import sys
@@ -28,8 +29,21 @@ except ImportError:
     sys.exit(1)
 
 
+# Default model. gpt-image-1 is OpenAI's current image model (best quality,
+# reliable text rendering, transparency). Requires a verified OpenAI org.
+# Override via IMAGE_GEN_DEFAULT_MODEL (e.g. set to "dall-e-3" if the org is
+# not verified).
+DEFAULT_MODEL = os.getenv("IMAGE_GEN_DEFAULT_MODEL", "gpt-image-1")
+
 # Model configurations
 MODEL_CONFIGS = {
+    "gpt-image-1": {
+        "sizes": ["1024x1024", "1536x1024", "1024x1536", "auto"],
+        "qualities": ["low", "medium", "high", "auto"],
+        "max_prompt_length": 32000,
+        "n_images_max": 10,
+        # Always returns base64 (no response_format param accepted).
+    },
     "dall-e-3": {
         "sizes": ["1024x1024", "1024x1792", "1792x1024"],
         "qualities": ["standard", "hd"],
@@ -52,7 +66,7 @@ def validate_request(data: dict[str, Any]) -> tuple[bool, str]:
         return False, "prompt field is required"
 
     prompt = data["prompt"]
-    model = data.get("model", "dall-e-3")
+    model = data.get("model", DEFAULT_MODEL)
 
     # Validate model
     if model not in MODEL_CONFIGS:
@@ -76,13 +90,14 @@ def validate_request(data: dict[str, Any]) -> tuple[bool, str]:
                 f"Invalid size '{size}' for {model}. Must be one of: {', '.join(config['sizes'])}",
             )
 
-    # Validate quality (DALL-E 3 only)
-    if model == "dall-e-3" and "quality" in data:
+    # Validate quality (gpt-image-1 and dall-e-3 have their own scales)
+    if "quality" in data and "qualities" in config:
         quality = data["quality"]
         if quality not in config["qualities"]:
             return (
                 False,
-                f"Invalid quality '{quality}'. Must be one of: {', '.join(config['qualities'])}",
+                f"Invalid quality '{quality}' for {model}. "
+                f"Must be one of: {', '.join(config['qualities'])}",
             )
 
     # Validate style (DALL-E 3 only)
@@ -91,14 +106,14 @@ def validate_request(data: dict[str, Any]) -> tuple[bool, str]:
         if style not in config["styles"]:
             return False, f"Invalid style '{style}'. Must be one of: {', '.join(config['styles'])}"
 
-    # Validate n (DALL-E 2 only)
+    # Validate n
     if "n" in data:
         n = data["n"]
         if model == "dall-e-3" and n != 1:
             return False, "DALL-E 3 can only generate 1 image at a time (n=1)"
-        if model == "dall-e-2":
-            if not isinstance(n, int) or n < 1 or n > config["n_images_max"]:
-                return False, f"n must be between 1 and {config['n_images_max']} for DALL-E 2"
+        max_n = config.get("n_images_max")
+        if max_n is not None and (not isinstance(n, int) or n < 1 or n > max_n):
+            return False, f"n must be between 1 and {max_n} for {model}"
 
     return True, ""
 
@@ -115,14 +130,19 @@ def generate_image(data: dict[str, Any]) -> dict[str, Any]:
 
     # Extract parameters
     prompt = data["prompt"]
-    model = data.get("model", "dall-e-3")
+    model = data.get("model", DEFAULT_MODEL)
     size = data.get("size")
     quality = data.get("quality")
     style = data.get("style")
     n = data.get("n", 1)
 
     # Set defaults based on model
-    if model == "dall-e-3":
+    if model == "gpt-image-1":
+        if not size:
+            size = "1024x1024"
+        if not quality:
+            quality = "high"
+    elif model == "dall-e-3":
         if not size:
             size = "1024x1024"
         if not quality:
@@ -135,16 +155,23 @@ def generate_image(data: dict[str, Any]) -> dict[str, Any]:
             size = "1024x1024"
 
     # Build API request parameters
-    api_params = {
+    api_params: dict[str, Any] = {
         "model": model,
         "prompt": prompt,
         "n": n,
         "size": size,
-        "response_format": "url",  # Get URL instead of base64
     }
 
-    # Add DALL-E 3 specific parameters
-    if model == "dall-e-3":
+    # `response_format` (url | b64_json) is only accepted by the dall-e-* models.
+    # gpt-image-1 rejects it and always returns base64. Only send it for the
+    # dall-e family; otherwise let the API use its default.
+    if model.startswith("dall-e"):
+        api_params["response_format"] = "url"
+
+    # Model-specific parameters
+    if model == "gpt-image-1":
+        api_params["quality"] = quality
+    elif model == "dall-e-3":
         api_params["quality"] = quality
         api_params["style"] = style
 
@@ -162,20 +189,35 @@ def generate_image(data: dict[str, Any]) -> dict[str, Any]:
             raise ValueError("OpenAI API error: Rate limit exceeded. Please try again later.")
         elif "insufficient_quota" in error_message.lower():
             raise ValueError("OpenAI API error: Insufficient quota. Please check your API credits.")
+        elif "must be verified" in error_message.lower() or "organization must be verified" in (
+            error_message.lower()
+        ):
+            raise ValueError(
+                f"OpenAI API error: model '{model}' requires a verified OpenAI organization. "
+                f"Set IMAGE_GEN_DEFAULT_MODEL=dall-e-3 or pass \"model\": \"dall-e-3\" to fall back. "
+                f"({error_message})"
+            )
         else:
             raise ValueError(f"OpenAI API error: {error_message}")
 
-    # Extract response data
+    # Extract response data. The API may return either a hosted URL or inline
+    # base64 (b64_json), depending on the model and whether response_format was
+    # honored. gpt-image-1 always returns base64. Support both.
     image_data = response.data[0]
-    image_url = image_data.url
+    image_url = getattr(image_data, "url", None)
+    image_b64 = getattr(image_data, "b64_json", None)
     revised_prompt = getattr(image_data, "revised_prompt", None)
 
+    if not image_url and not image_b64:
+        raise ValueError("OpenAI API error: response contained neither an image URL nor base64 data")
+
     return {
         "url": image_url,
+        "b64_json": image_b64,
         "revised_prompt": revised_prompt,
         "model": model,
         "size": size,
-        "quality": quality if model == "dall-e-3" else None,
+        "quality": quality if model in ("gpt-image-1", "dall-e-3") else None,
         "style": style if model == "dall-e-3" else None,
     }
 
@@ -202,6 +244,37 @@ def download_image(url: str, output_path: Path) -> int:
         raise ValueError(f"Failed to download image: {e}")
 
 
+def save_image(result: dict[str, Any], output_path: Path) -> int:
+    """Persist a generated image to disk from either a URL or base64 payload.
+
+    Args:
+        result: The dict returned by generate_image (carries 'url' and/or
+            'b64_json').
+        output_path: Path where the image should be saved.
+
+    Returns:
+        Size of the written file in bytes.
+
+    Raises:
+        ValueError: If neither a URL nor base64 data is available, or if
+            decoding/downloading fails.
+    """
+    b64_data = result.get("b64_json")
+    if b64_data:
+        try:
+            image_bytes = base64.b64decode(b64_data)
+        except (ValueError, TypeError) as e:
+            raise ValueError(f"Failed to decode base64 image data: {e}")
+        output_path.write_bytes(image_bytes)
+        return len(image_bytes)
+
+    url = result.get("url")
+    if url:
+        return download_image(url, output_path)
+
+    raise ValueError("No image data to save (neither URL nor base64 returned)")
+
+
 def main() -> None:
     """Main entry point for the image generation script."""
     parser = argparse.ArgumentParser(description="Generate AI images using DALL-E")
@@ -250,8 +323,8 @@ def main() -> None:
         # Generate image via OpenAI API
         result = generate_image(data)
 
-        # Download image to local file
-        file_size = download_image(result["url"], output_path)
+        # Persist image to local file (handles both URL and base64 responses)
+        file_size = save_image(result, output_path)
 
         # Output success result
         output = {

{agent_framework_lib-0.8.7.post6 → agent_framework_lib-0.8.7.post7}/agent_framework/skills/builtin/skills/unified_pdf/SKILL.md

@@ -30,7 +30,7 @@ metadata:
 
 Generate PDF documents from HTML content by running the `create_pdf.py` script via `shell_exec`.
 
-The script uses Playwright to render HTML in a headless Chromium browser and produce an A4 PDF. HTML fragments are automatically wrapped in a minimal document shell.
+The script uses Playwright to render HTML in a headless Chromium browser and produce an A4 PDF. HTML fragments are automatically wrapped in a **professionally styled** document shell (clean typography, styled headings/tables/figures). Images are **automatically scaled to fit the page width** — you never need to set image dimensions or worry about a large chart PNG overflowing the page.
 
 ### Usage
 
@@ -51,8 +51,8 @@ echo '<h1>Hello World</h1><p>This is a PDF.</p>' | uv run python -m agent_framew
 
 Provide raw HTML content on stdin. This can be either:
 
-- A **complete HTML document** (starts with `<!DOCTYPE` or contains `<html`): used as-is.
-- An **HTML fragment** (anything else): automatically wrapped in a minimal HTML document with sensible defaults (sans-serif font, neutral colors, 1.6 line-height).
+- A **complete HTML document** (starts with `<!DOCTYPE` or contains `<html`): used as-is, but print-safety CSS is still injected so oversized images are scaled to the page.
+- An **HTML fragment** (anything else): automatically wrapped in a professionally styled document (clean typography, styled headings, tables, and figures with captions). Recommended for most reports — just provide the body content.
 
 ### Recommended HTML Structure
 

{agent_framework_lib-0.8.7.post6 → agent_framework_lib-0.8.7.post7}/agent_framework/skills/builtin/skills/unified_pdf/create_pdf.py

@@ -21,6 +21,145 @@ import urllib.request
 VIEWPORT_WIDTH = 1280
 VIEWPORT_HEIGHT = 1024
 
+# Critical print-safety CSS. Injected into EVERY document (fragments and complete
+# HTML alike) so that oversized raster images (e.g. 3200px-wide chart PNGs) are
+# scaled to the page width instead of overflowing and spilling across pages.
+_PRINT_SAFETY_CSS = """
+img, svg, canvas, video {
+    max-width: 100% !important;
+    height: auto !important;
+}
+figure {
+    margin: 1.2em 0;
+    text-align: center;
+    page-break-inside: avoid;
+    break-inside: avoid;
+}
+figure img {
+    margin: 0 auto;
+}
+table, pre, blockquote {
+    page-break-inside: avoid;
+    break-inside: avoid;
+}
+h1, h2, h3, h4 {
+    page-break-after: avoid;
+    break-after: avoid;
+}
+"""
+
+# Professional baseline styling applied when wrapping a bare HTML fragment.
+_PROFESSIONAL_CSS = """
+:root {
+    --accent: #2563eb;
+    --ink: #1f2937;
+    --muted: #6b7280;
+    --border: #e5e7eb;
+}
+* { box-sizing: border-box; }
+body {
+    font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto,
+                 "Helvetica Neue", Arial, sans-serif;
+    margin: 0;
+    padding: 0;
+    color: var(--ink);
+    line-height: 1.6;
+    font-size: 11pt;
+}
+h1 {
+    font-size: 24pt;
+    font-weight: 700;
+    margin: 0 0 0.4em;
+    padding-bottom: 0.3em;
+    border-bottom: 3px solid var(--accent);
+}
+h2 {
+    font-size: 16pt;
+    font-weight: 700;
+    margin: 1.4em 0 0.5em;
+    color: var(--ink);
+}
+h3 {
+    font-size: 13pt;
+    font-weight: 600;
+    margin: 1.1em 0 0.4em;
+}
+p { margin: 0.6em 0; }
+a { color: var(--accent); text-decoration: none; }
+ul, ol { margin: 0.6em 0; padding-left: 1.4em; }
+li { margin: 0.25em 0; }
+table {
+    width: 100%;
+    border-collapse: collapse;
+    margin: 1em 0;
+    font-size: 10pt;
+}
+th, td {
+    border: 1px solid var(--border);
+    padding: 8px 10px;
+    text-align: left;
+}
+th {
+    background: #f3f4f6;
+    font-weight: 600;
+}
+tr:nth-child(even) td { background: #fafafa; }
+figure {
+    margin: 1.4em 0;
+    text-align: center;
+}
+figcaption {
+    font-size: 9.5pt;
+    color: var(--muted);
+    margin-top: 0.5em;
+    font-style: italic;
+}
+blockquote {
+    margin: 1em 0;
+    padding: 0.4em 1em;
+    border-left: 4px solid var(--accent);
+    color: var(--muted);
+}
+pre, code {
+    font-family: "SF Mono", Menlo, Consolas, monospace;
+    font-size: 9.5pt;
+}
+pre {
+    background: #f6f8fa;
+    padding: 12px 14px;
+    border-radius: 6px;
+    overflow-x: auto;
+}
+.section { margin-bottom: 1.2em; }
+.break-before { page-break-before: always; break-before: page; }
+"""
+
+
+def _inject_into_head(html: str, style_block: str) -> str:
+    """Insert a <style> block into a complete HTML document.
+
+    Tries to place it just before </head>; falls back to after <html>, or
+    prepends it if neither tag is present. Case-insensitive.
+
+    Args:
+        html: A complete HTML document.
+        style_block: The CSS (without <style> tags) to inject.
+
+    Returns:
+        The document with the style block inserted.
+    """
+    tag = f"<style>{style_block}</style>"
+    lower = html.lower()
+    head_close = lower.find("</head>")
+    if head_close != -1:
+        return html[:head_close] + tag + html[head_close:]
+    html_open = lower.find("<html")
+    if html_open != -1:
+        insert_at = lower.find(">", html_open)
+        if insert_at != -1:
+            return html[: insert_at + 1] + tag + html[insert_at + 1 :]
+    return tag + html
+
 _IMG_SRC_RE = re.compile(r'(<img\b[^>]*?\bsrc\s*=\s*)(["\'])(.*?)\2', re.IGNORECASE)
 _FILES_DOWNLOAD_RE = re.compile(r"/files/[A-Za-z0-9_-]+/download")
 
@@ -168,8 +307,12 @@ async def render_pdf(html_content: str, filename: str, output_dir: str) -> str:
     # would otherwise silently fail to load and the PDF would show no image.
     html_content = _inline_images(html_content)
 
-    # Wrap fragment in a basic HTML document if needed
-    if not _is_complete_html(html_content):
+    # Wrap fragment in a basic HTML document if needed. Complete documents get
+    # the print-safety CSS injected so that oversized images are still scaled
+    # to the page (the #1 cause of charts overflowing / being cropped in PDFs).
+    if _is_complete_html(html_content):
+        html_content = _inject_into_head(html_content, _PRINT_SAFETY_CSS)
+    else:
         html_content = _wrap_html_fragment(html_content)
 
     # Sanitize filename
@@ -223,7 +366,10 @@ def _is_complete_html(content: str) -> bool:
 
 
 def _wrap_html_fragment(fragment: str) -> str:
-    """Wrap an HTML fragment in a minimal HTML document.
+    """Wrap an HTML fragment in a professionally styled HTML document.
+
+    Applies a clean report stylesheet plus the print-safety CSS so that
+    embedded charts and images are scaled to the page width.
 
     Args:
         fragment: HTML fragment to wrap.
@@ -235,16 +381,8 @@ def _wrap_html_fragment(fragment: str) -> str:
 <html>
 <head>
     <meta charset="UTF-8">
-    <style>
-        body {{
-            font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto,
-                         "Helvetica Neue", Arial, sans-serif;
-            margin: 0;
-            padding: 0;
-            color: #333;
-            line-height: 1.6;
-        }}
-    </style>
+    <style>{_PROFESSIONAL_CSS}
+{_PRINT_SAFETY_CSS}</style>
 </head>
 <body>
 {fragment}

{agent_framework_lib-0.8.7.post6 → agent_framework_lib-0.8.7.post7/agent_framework_lib.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-framework-lib
-Version: 0.8.7.post6
+Version: 0.8.7.post7
 Summary: A comprehensive Python framework for building and serving conversational AI agents with FastAPI
 Author-email: Sebastian Pavel <sebastian@cinco.ai>, Elliott Girard <elliott.girard@icloud.com>
 Maintainer-email: Sebastian Pavel <sebastian@cinco.ai>

{agent_framework_lib-0.8.7.post6 → agent_framework_lib-0.8.7.post7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "agent-framework-lib"
-version = "0.8.7post6"
+version = "0.8.7post7"
 description = "A comprehensive Python framework for building and serving conversational AI agents with FastAPI"
 readme = {file = "README.md", content-type = "text/markdown"}
 license = {text = "MIT"}