docling-core 2.2.3__py3-none-any.whl → 2.3.1__py3-none-any.whl

docling_core/transforms/chunker/hierarchical_chunker.py

@@ -183,14 +183,15 @@ class HierarchicalChunker(BaseChunker):
                         )
                         list_items = []  # reset
 
-                if isinstance(
-                    item, SectionHeaderItem
-                ) or (  # TODO remove when all captured as SectionHeaderItem:
+                if isinstance(item, SectionHeaderItem) or (
                     isinstance(item, TextItem)
-                    and item.label == DocItemLabel.SECTION_HEADER
+                    and item.label in [DocItemLabel.SECTION_HEADER, DocItemLabel.TITLE]
                 ):
-                    # TODO second branch not needed once cleanup above complete:
-                    level = item.level if isinstance(item, SectionHeaderItem) else 1
+                    level = (
+                        item.level
+                        if isinstance(item, SectionHeaderItem)
+                        else (0 if item.label == DocItemLabel.TITLE else 1)
+                    )
                     heading_by_level[level] = item.text
 
                     # remove headings of higher level as they just went out of scope

docling_core/types/doc/document.py

@@ -100,12 +100,166 @@ class PictureMiscData(BaseModel):
     content: Dict[str, Any]
 
 
+class ChartLine(BaseModel):
+    """Represents a line in a line chart.
+
+    Attributes:
+        label (str): The label for the line.
+        values (List[Tuple[float, float]]): A list of (x, y) coordinate pairs
+            representing the line's data points.
+    """
+
+    label: str
+    values: List[Tuple[float, float]]
+
+
+class ChartBar(BaseModel):
+    """Represents a bar in a bar chart.
+
+    Attributes:
+        label (str): The label for the bar.
+        values (float): The value associated with the bar.
+    """
+
+    label: str
+    values: float
+
+
+class ChartStackedBar(BaseModel):
+    """Represents a stacked bar in a stacked bar chart.
+
+    Attributes:
+        label (List[str]): The labels for the stacked bars. Multiple values are stored
+            in cases where the chart is "double stacked," meaning bars are stacked both
+            horizontally and vertically.
+        values (List[Tuple[str, int]]): A list of values representing different segments
+            of the stacked bar along with their label.
+    """
+
+    label: List[str]
+    values: List[Tuple[str, int]]
+
+
+class ChartSlice(BaseModel):
+    """Represents a slice in a pie chart.
+
+    Attributes:
+        label (str): The label for the slice.
+        value (float): The value represented by the slice.
+    """
+
+    label: str
+    value: float
+
+
+class ChartPoint(BaseModel):
+    """Represents a point in a scatter chart.
+
+    Attributes:
+        value (Tuple[float, float]): A (x, y) coordinate pair representing a point in a
+            chart.
+    """
+
+    value: Tuple[float, float]
+
+
+class PictureChartData(BaseModel):
+    """Base class for picture chart data.
+
+    Attributes:
+        title (str): The title of the chart.
+    """
+
+    title: str
+
+
+class PictureLineChartData(PictureChartData):
+    """Represents data of a line chart.
+
+    Attributes:
+        kind (Literal["line_chart_data"]): The type of the chart.
+        x_axis_label (str): The label for the x-axis.
+        y_axis_label (str): The label for the y-axis.
+        lines (List[ChartLine]): A list of lines in the chart.
+    """
+
+    kind: Literal["line_chart_data"] = "line_chart_data"
+    x_axis_label: str
+    y_axis_label: str
+    lines: List[ChartLine]
+
+
+class PictureBarChartData(PictureChartData):
+    """Represents data of a bar chart.
+
+    Attributes:
+        kind (Literal["bar_chart_data"]): The type of the chart.
+        x_axis_label (str): The label for the x-axis.
+        y_axis_label (str): The label for the y-axis.
+        bars (List[ChartBar]): A list of bars in the chart.
+    """
+
+    kind: Literal["bar_chart_data"] = "bar_chart_data"
+    x_axis_label: str
+    y_axis_label: str
+    bars: List[ChartBar]
+
+
+class PictureStackedBarChartData(PictureChartData):
+    """Represents data of a stacked bar chart.
+
+    Attributes:
+        kind (Literal["stacked_bar_chart_data"]): The type of the chart.
+        x_axis_label (str): The label for the x-axis.
+        y_axis_label (str): The label for the y-axis.
+        stacked_bars (List[ChartStackedBar]): A list of stacked bars in the chart.
+    """
+
+    kind: Literal["stacked_bar_chart_data"] = "stacked_bar_chart_data"
+    x_axis_label: str
+    y_axis_label: str
+    stacked_bars: List[ChartStackedBar]
+
+
+class PicturePieChartData(PictureChartData):
+    """Represents data of a pie chart.
+
+    Attributes:
+        kind (Literal["pie_chart_data"]): The type of the chart.
+        slices (List[ChartSlice]): A list of slices in the pie chart.
+    """
+
+    kind: Literal["pie_chart_data"] = "pie_chart_data"
+    slices: List[ChartSlice]
+
+
+class PictureScatterChartData(PictureChartData):
+    """Represents data of a scatter chart.
+
+    Attributes:
+        kind (Literal["scatter_chart_data"]): The type of the chart.
+        x_axis_label (str): The label for the x-axis.
+        y_axis_label (str): The label for the y-axis.
+        points (List[ChartPoint]): A list of points in the scatter chart.
+    """
+
+    kind: Literal["scatter_chart_data"] = "scatter_chart_data"
+    x_axis_label: str
+    y_axis_label: str
+    points: List[ChartPoint]
+
+
 PictureDataType = Annotated[
     Union[
         PictureClassificationData,
         PictureDescriptionData,
         PictureMoleculeData,
         PictureMiscData,
+        PictureLineChartData,
+        PictureBarChartData,
+        PictureStackedBarChartData,
+        PicturePieChartData,
+        PictureScatterChartData,
     ],
     Field(discriminator="kind"),
 ]

{docling_core-2.2.3.dist-info → docling_core-2.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: docling-core
-Version: 2.2.3
+Version: 2.3.1
 Summary: A python library to define and validate data types in Docling.
 Home-page: https://ds4sd.github.io/
 License: MIT

{docling_core-2.2.3.dist-info → docling_core-2.3.1.dist-info}/RECORD

@@ -16,12 +16,12 @@ docling_core/search/package.py,sha256=Lz2ml2eDy5t0ZimnGTq-DXHAn-f18w0bn4H5xrhs75
 docling_core/transforms/__init__.py,sha256=P81y_oqkiTN4Ld5crh1gQ6BbHqqR6C6nBt9ACDd57ds,106
 docling_core/transforms/chunker/__init__.py,sha256=sSSTnt7ZCt8Og1e0jhApNTtA0pyyHyzwcl8yXFLb2J8,292
 docling_core/transforms/chunker/base.py,sha256=iPouZOJ3cYWvai4P0Gpd3QmsTKQuY5fFUXzTMk_XNmE,1571
-docling_core/transforms/chunker/hierarchical_chunker.py,sha256=uG7nNoUCFqWeQAKydQg731JYJ9sayUe4J48nMF0VHE8,8097
+docling_core/transforms/chunker/hierarchical_chunker.py,sha256=V4FiOYqL0GgBqVB7x6CafAJs3WF5oYifKIiexVggGPE,8086
 docling_core/types/__init__.py,sha256=MVRSgsk5focwGyAplh_TRR3dEecIXpd98g_u3zZ5HXo,260
 docling_core/types/base.py,sha256=PusJskRVL19y-hq0BgXr5e8--QEqSqLnFNJ8UbOqW88,8318
 docling_core/types/doc/__init__.py,sha256=bEL4zKVOG7Wxm6xQrgF58mu-Teds9aSavuEAKVNhrTU,639
 docling_core/types/doc/base.py,sha256=zvx631U_yQCcJam83hNdDanXEYnO3eN-CCw9vDr6S-I,4442
-docling_core/types/doc/document.py,sha256=B56FA5lGAEodjfIUncXSstQclAmyt3GOybMiKEEIc7s,52138
+docling_core/types/doc/document.py,sha256=XF43-v9oflV-E5r2k2quoKvq8qBp5mAB_VunshY9b10,56356
 docling_core/types/doc/labels.py,sha256=A8vWP82VAeXO1rlCO0oDKo_Hb8uDeQe0myOTY3P03hk,1596
 docling_core/types/gen/__init__.py,sha256=C6TuCfvpSnSL5XDOFMcYHUY2-i08vvfOGRcdu6Af0pI,124
 docling_core/types/gen/generic.py,sha256=l4CZ4_Lb8ONG36WNJWbKX5hGKvTh_yU-hXp5hsm7uVU,844
@@ -49,8 +49,8 @@ docling_core/utils/generate_docs.py,sha256=BdKAoduWXOc7YMvcmlhjoJOFlUxij1ybxglj6
 docling_core/utils/generate_jsonschema.py,sha256=uNX1O5XnjyB5nA66XqZXTt3YbGuR2tyi_OhHepHYtZg,1654
 docling_core/utils/validate.py,sha256=3FmnxnKTDZC5J9OGxCL3U3DGRl0t0bBV1NcySXswdas,2031
 docling_core/utils/validators.py,sha256=azcrndLzhNkTWnbFSu9shJ5D3j_znnLrIFA5R8hzmGU,2798
-docling_core-2.2.3.dist-info/LICENSE,sha256=2M9-6EoQ1sxFztTOkXGAtwUDJvnWaAHdB9BYWVwGkIw,1087
-docling_core-2.2.3.dist-info/METADATA,sha256=DlV-TrYKPq-qbI9d0iS4mrOJs_CwV9QZNflqGEy0crE,5432
-docling_core-2.2.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-docling_core-2.2.3.dist-info/entry_points.txt,sha256=jIxlWv3tnO04irlZc0zfhqJIgz1bg9Hha4AkaLWSdUA,177
-docling_core-2.2.3.dist-info/RECORD,,
+docling_core-2.3.1.dist-info/LICENSE,sha256=2M9-6EoQ1sxFztTOkXGAtwUDJvnWaAHdB9BYWVwGkIw,1087
+docling_core-2.3.1.dist-info/METADATA,sha256=mASC44D6AB2bIACFr2oGrsZHtHRzn5e1wjBJyy6ccns,5432
+docling_core-2.3.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+docling_core-2.3.1.dist-info/entry_points.txt,sha256=jIxlWv3tnO04irlZc0zfhqJIgz1bg9Hha4AkaLWSdUA,177
+docling_core-2.3.1.dist-info/RECORD,,