firecrawl 4.5.0__tar.gz → 4.6.0__tar.gz

{firecrawl-4.5.0 → firecrawl-4.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: firecrawl
-Version: 4.5.0
+Version: 4.6.0
 Summary: Python SDK for Firecrawl API
 Home-page: https://github.com/firecrawl/firecrawl
 Author: Mendable.ai

{firecrawl-4.5.0 → firecrawl-4.6.0}/firecrawl/__init__.py

@@ -17,7 +17,7 @@ from .v1 import (
     V1ChangeTrackingOptions,
 )
 
-__version__ = "4.5.0"
+__version__ = "4.6.0"
 
 # Define the logger for the Firecrawl project
 logger: logging.Logger = logging.getLogger("firecrawl")

firecrawl-4.6.0/firecrawl/__tests__/unit/v2/methods/test_branding.py

@@ -0,0 +1,214 @@
+import pytest
+from unittest.mock import Mock, MagicMock
+from firecrawl.v2.methods.scrape import scrape
+from firecrawl.v2.types import ScrapeOptions, Document
+
+
+class TestBrandingFormat:
+    """Unit tests for branding format support."""
+
+    def test_scrape_with_branding_format_returns_branding_data(self):
+        """Test that scraping with branding format returns branding data."""
+        mock_response = Mock()
+        mock_response.ok = True
+        mock_response.json.return_value = {
+            "success": True,
+            "data": {
+                "markdown": "# Example",
+                "branding": {
+                    "colorScheme": "light",
+                    "colors": {
+                        "primary": "#E11D48",
+                        "secondary": "#3B82F6",
+                        "accent": "#F59E0B"
+                    },
+                    "typography": {
+                        "fontFamilies": {
+                            "primary": "Inter",
+                            "heading": "Poppins"
+                        },
+                        "fontSizes": {
+                            "h1": "2.5rem",
+                            "body": "1rem"
+                        }
+                    },
+                    "spacing": {
+                        "baseUnit": 8
+                    },
+                    "components": {
+                        "buttonPrimary": {
+                            "background": "#E11D48",
+                            "textColor": "#FFFFFF",
+                            "borderRadius": "0.5rem"
+                        }
+                    }
+                }
+            }
+        }
+
+        mock_client = Mock()
+        mock_client.post.return_value = mock_response
+
+        result = scrape(mock_client, "https://example.com", ScrapeOptions(formats=["branding"]))
+
+        assert result.branding is not None
+        assert result.branding.color_scheme == "light"
+        assert result.branding.colors["primary"] == "#E11D48"
+        assert result.branding.typography["fontFamilies"]["primary"] == "Inter"
+        assert result.branding.spacing["baseUnit"] == 8
+        assert result.branding.components["buttonPrimary"]["background"] == "#E11D48"
+
+    def test_scrape_with_branding_and_markdown_formats_returns_both(self):
+        """Test that scraping with both branding and markdown formats returns both."""
+        mock_response = Mock()
+        mock_response.ok = True
+        mock_response.json.return_value = {
+            "success": True,
+            "data": {
+                "markdown": "# Example Content",
+                "branding": {
+                    "colorScheme": "dark",
+                    "colors": {
+                        "primary": "#10B981"
+                    },
+                    "typography": {
+                        "fontFamilies": {
+                            "primary": "Roboto"
+                        }
+                    }
+                }
+            }
+        }
+
+        mock_client = Mock()
+        mock_client.post.return_value = mock_response
+
+        result = scrape(mock_client, "https://example.com", ScrapeOptions(formats=["markdown", "branding"]))
+
+        assert result.markdown == "# Example Content"
+        assert result.branding is not None
+        assert result.branding.color_scheme == "dark"
+        assert result.branding.colors["primary"] == "#10B981"
+
+    def test_scrape_without_branding_format_does_not_return_branding(self):
+        """Test that scraping without branding format does not return branding."""
+        mock_response = Mock()
+        mock_response.ok = True
+        mock_response.json.return_value = {
+            "success": True,
+            "data": {
+                "markdown": "# Example"
+            }
+        }
+
+        mock_client = Mock()
+        mock_client.post.return_value = mock_response
+
+        result = scrape(mock_client, "https://example.com", ScrapeOptions(formats=["markdown"]))
+
+        assert result.markdown == "# Example"
+        assert result.branding is None
+
+    def test_branding_format_with_all_nested_fields(self):
+        """Test branding format with all nested fields populated."""
+        mock_response = Mock()
+        mock_response.ok = True
+        mock_response.json.return_value = {
+            "success": True,
+            "data": {
+                "branding": {
+                    "colorScheme": "light",
+                    "logo": "https://example.com/logo.png",
+                    "fonts": [
+                        {"family": "Inter", "weight": 400},
+                        {"family": "Poppins", "weight": 700}
+                    ],
+                    "colors": {
+                        "primary": "#E11D48",
+                        "background": "#FFFFFF"
+                    },
+                    "typography": {
+                        "fontFamilies": {"primary": "Inter"},
+                        "fontStacks": {"body": ["Inter", "sans-serif"]},
+                        "fontSizes": {"h1": "2.5rem"},
+                        "lineHeights": {"body": 1.5},
+                        "fontWeights": {"regular": 400}
+                    },
+                    "spacing": {
+                        "baseUnit": 8,
+                        "padding": {"sm": 8, "md": 16}
+                    },
+                    "components": {
+                        "buttonPrimary": {
+                            "background": "#E11D48",
+                            "textColor": "#FFFFFF"
+                        }
+                    },
+                    "icons": {
+                        "style": "outline",
+                        "primaryColor": "#E11D48"
+                    },
+                    "images": {
+                        "logo": "https://example.com/logo.png",
+                        "favicon": "https://example.com/favicon.ico"
+                    },
+                    "animations": {
+                        "transitionDuration": "200ms",
+                        "easing": "ease-in-out"
+                    },
+                    "layout": {
+                        "grid": {"columns": 12, "maxWidth": "1200px"},
+                        "headerHeight": "64px"
+                    },
+                    "tone": {
+                        "voice": "professional",
+                        "emojiUsage": "minimal"
+                    },
+                    "personality": {
+                        "tone": "professional",
+                        "energy": "medium",
+                        "targetAudience": "developers"
+                    }
+                }
+            }
+        }
+
+        mock_client = Mock()
+        mock_client.post.return_value = mock_response
+
+        result = scrape(mock_client, "https://example.com", ScrapeOptions(formats=["branding"]))
+
+        assert result.branding is not None
+        assert result.branding.color_scheme == "light"
+        assert result.branding.logo == "https://example.com/logo.png"
+        assert len(result.branding.fonts) == 2
+        assert result.branding.typography["fontStacks"]["body"] == ["Inter", "sans-serif"]
+        assert result.branding.spacing["padding"] == {"sm": 8, "md": 16}
+        assert result.branding.icons["style"] == "outline"
+        assert result.branding.images["favicon"] == "https://example.com/favicon.ico"
+        assert result.branding.animations["easing"] == "ease-in-out"
+        assert result.branding.layout["grid"]["columns"] == 12
+        assert result.branding.personality["tone"] == "professional"
+
+    def test_branding_colorscheme_normalization(self):
+        """Test that colorScheme is normalized to color_scheme."""
+        mock_response = Mock()
+        mock_response.ok = True
+        mock_response.json.return_value = {
+            "success": True,
+            "data": {
+                "branding": {
+                    "colorScheme": "dark",
+                    "colors": {"primary": "#000000"}
+                }
+            }
+        }
+
+        mock_client = Mock()
+        mock_client.post.return_value = mock_response
+
+        result = scrape(mock_client, "https://example.com", ScrapeOptions(formats=["branding"]))
+
+        assert result.branding is not None
+        assert result.branding.color_scheme == "dark"
+        assert not hasattr(result.branding, "colorScheme")

{firecrawl-4.5.0 → firecrawl-4.6.0}/firecrawl/v2/types.py

@@ -124,6 +124,24 @@ class AttributeResult(BaseModel):
     attribute: str
     values: List[str]
 
+class BrandingProfile(BaseModel):
+    """Branding information extracted from a website."""
+    model_config = {"extra": "allow"}
+    
+    color_scheme: Optional[Literal["light", "dark"]] = None
+    logo: Optional[str] = None
+    fonts: Optional[List[Dict[str, Any]]] = None
+    colors: Optional[Dict[str, str]] = None
+    typography: Optional[Dict[str, Any]] = None
+    spacing: Optional[Dict[str, Any]] = None
+    components: Optional[Dict[str, Any]] = None
+    icons: Optional[Dict[str, str]] = None
+    images: Optional[Dict[str, Optional[str]]] = None
+    animations: Optional[Dict[str, str]] = None
+    layout: Optional[Dict[str, Any]] = None
+    tone: Optional[Dict[str, str]] = None
+    personality: Optional[Dict[str, Any]] = None
+
 class Document(BaseModel):
     """A scraped document."""
     markdown: Optional[str] = None
@@ -138,6 +156,7 @@ class Document(BaseModel):
     actions: Optional[Dict[str, Any]] = None
     warning: Optional[str] = None
     change_tracking: Optional[Dict[str, Any]] = None
+    branding: Optional[BrandingProfile] = None
 
     @property
     def metadata_typed(self) -> DocumentMetadata:
@@ -199,7 +218,7 @@ CategoryOption = Union[str, Category]
 
 FormatString = Literal[
     # camelCase versions (API format)
-    "markdown", "html", "rawHtml", "links",  "images", "screenshot", "summary", "changeTracking", "json", "attributes",
+    "markdown", "html", "rawHtml", "links",  "images", "screenshot", "summary", "changeTracking", "json", "attributes", "branding",
     # snake_case versions (user-friendly)
     "raw_html", "change_tracking"
 ]

{firecrawl-4.5.0 → firecrawl-4.6.0}/firecrawl/v2/utils/normalize.py

@@ -83,6 +83,7 @@ def normalize_document_input(doc: Dict[str, Any]) -> Dict[str, Any]:
     Normalize a raw Document dict from the API into the Python SDK's expected shape:
     - Convert top-level keys rawHtml->raw_html, changeTracking->change_tracking
     - Convert metadata keys from camelCase to snake_case
+    - Convert branding.colorScheme to branding.color_scheme
     """
     normalized = dict(doc)
 
@@ -102,6 +103,12 @@ def normalize_document_input(doc: Dict[str, Any]) -> Dict[str, Any]:
             # Fallback to mapped dict if model construction fails for any reason
             normalized["metadata"] = mapped
 
+    # Normalize branding top-level camelCase keys
+    branding = normalized.get("branding")
+    if isinstance(branding, dict):
+        if "colorScheme" in branding and "color_scheme" not in branding:
+            branding["color_scheme"] = branding.pop("colorScheme")
+
     return normalized
 
 

{firecrawl-4.5.0 → firecrawl-4.6.0}/firecrawl.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: firecrawl
-Version: 4.5.0
+Version: 4.6.0
 Summary: Python SDK for Firecrawl API
 Home-page: https://github.com/firecrawl/firecrawl
 Author: Mendable.ai

{firecrawl-4.5.0 → firecrawl-4.6.0}/firecrawl.egg-info/SOURCES.txt

@@ -30,6 +30,7 @@ firecrawl/__tests__/e2e/v2/aio/test_aio_search.py
 firecrawl/__tests__/e2e/v2/aio/test_aio_usage.py
 firecrawl/__tests__/e2e/v2/aio/test_aio_watcher.py
 firecrawl/__tests__/unit/v2/methods/test_batch_request_preparation.py
+firecrawl/__tests__/unit/v2/methods/test_branding.py
 firecrawl/__tests__/unit/v2/methods/test_crawl_params.py
 firecrawl/__tests__/unit/v2/methods/test_crawl_request_preparation.py
 firecrawl/__tests__/unit/v2/methods/test_crawl_validation.py