chunkr-ai 0.0.50__py3-none-any.whl → 0.1.0__py3-none-any.whl

chunkr_ai/api/configuration.py

@@ -3,27 +3,34 @@ from enum import Enum
 from typing import Any, List, Optional, Union
 from pydantic import field_validator, field_serializer
 
-class GenerationStrategy(str, Enum):
-    LLM = "LLM"
-    AUTO = "Auto"
-
 class CroppingStrategy(str, Enum):
     ALL = "All"
     AUTO = "Auto"
 
-class EmbedSource(str, Enum):
-    HTML = "HTML"
+class SegmentFormat(str, Enum):
+    HTML = "Html"
     MARKDOWN = "Markdown"
-    LLM = "LLM"
+
+class EmbedSource(str, Enum):
     CONTENT = "Content"
+    HTML = "HTML"  # Deprecated
+    MARKDOWN = "Markdown"  # Deprecated
+    LLM = "LLM"
+
+class GenerationStrategy(str, Enum):
+    LLM = "LLM"
+    AUTO = "Auto"
 
 class GenerationConfig(BaseModel):
-    html: Optional[GenerationStrategy] = None
+    format: Optional[SegmentFormat] = None
+    strategy: Optional[GenerationStrategy] = None
     llm: Optional[str] = None
-    markdown: Optional[GenerationStrategy] = None
     crop_image: Optional[CroppingStrategy] = None
-    embed_sources: Optional[List[EmbedSource]] = Field(default_factory=lambda: [EmbedSource.MARKDOWN])
+    embed_sources: Optional[List[EmbedSource]] = None
     extended_context: Optional[bool] = None
+    # Deprecated fields for backwards compatibility
+    html: Optional[GenerationStrategy] = None  # Deprecated: Use format=SegmentFormat.HTML and strategy instead
+    markdown: Optional[GenerationStrategy] = None  # Deprecated: Use format=SegmentFormat.MARKDOWN and strategy instead
 
 class SegmentProcessing(BaseModel):
     model_config = ConfigDict(populate_by_name=True, alias_generator=str.title)
@@ -246,7 +253,7 @@ class SegmentType(str, Enum):
 
 class Segment(BaseModel):
     bbox: BoundingBox
-    content: str
+    content: str = ""
     page_height: float
     llm: Optional[str] = None
     html: Optional[str] = None
@@ -258,6 +265,7 @@ class Segment(BaseModel):
     segment_id: str
     segment_type: SegmentType
     confidence: Optional[float]
+    text: str = ""
 
 class Chunk(BaseModel):
     chunk_id: str

chunkr_ai/models.py

@@ -15,6 +15,7 @@ from .api.configuration import (
     OcrStrategy,
     OutputResponse,
     Segment,
+    SegmentFormat,
     SegmentProcessing,
     SegmentType,
     SegmentationStrategy,
@@ -42,6 +43,7 @@ __all__ = [
     "OcrStrategy",
     "OutputResponse",
     "Segment",
+    "SegmentFormat",
     "SegmentProcessing",
     "SegmentType",
     "SegmentationStrategy",

{chunkr_ai-0.0.50.dist-info → chunkr_ai-0.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chunkr-ai
-Version: 0.0.50
+Version: 0.1.0
 Summary: Python client for Chunkr: open source document intelligence
 Author-email: Ishaan Kapoor <ishaan@lumina.sh>
 License: MIT License

{chunkr_ai-0.0.50.dist-info → chunkr_ai-0.1.0.dist-info}/RECORD

@@ -1,16 +1,16 @@
 chunkr_ai/__init__.py,sha256=6KpYv2lmD6S5z2kc9pqwuLP5VDHmOuu2qDZArUIhb1s,53
-chunkr_ai/models.py,sha256=L0L9CjY8SgSh9_Fzvo_nJXqKf_2urZHngMWtBVlAQAo,1006
+chunkr_ai/models.py,sha256=1q4l7fSXU7cVfx2ZUcmc5EZ4K0AeUhCbvY9yaBbXL8E,1046
 chunkr_ai/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chunkr_ai/api/auth.py,sha256=0RSNFPvHt4Nrg8qtP2xvA2KbR0J_KUe1B_tKynbq9Fc,436
 chunkr_ai/api/chunkr.py,sha256=uSNYtB_mcs4-QRKsX7wZb8yv6ayXgRrJSDNZ-EbAyvc,3857
 chunkr_ai/api/chunkr_base.py,sha256=8roSPoCADmaXM2r7zz2iHfZzIcY9NopOfa4j-dfk8RA,6310
-chunkr_ai/api/configuration.py,sha256=M0oAXJmGBakOAKywg_Eyoz4GLginBrXuSvCtG0_BznQ,9926
+chunkr_ai/api/configuration.py,sha256=MVAxKe8vTSUMy0AHOPWyJEOgmIL-rPTsAQ8Z83gGXew,10287
 chunkr_ai/api/decorators.py,sha256=w1l_ZEkl99C-BO3qRTbi74sYwHDFspB1Bjt1Arv9lPc,4384
 chunkr_ai/api/misc.py,sha256=AaGLxZlMzNgVPwErskDRKc2UVGkC0JwxLXU-enPwzA0,5354
 chunkr_ai/api/protocol.py,sha256=LjPrYSq52m1afIlAo0yVGXlGZxPRh8J6g7S4PAit3Zo,388
 chunkr_ai/api/task_response.py,sha256=VYa62E08VlZUyjn2YslnY4cohdK9e53HbEzsaYIXKXM,8028
-chunkr_ai-0.0.50.dist-info/licenses/LICENSE,sha256=w3R12yNDyZpMiy2lxy_hvNbsldC75ww79sF0u11rkho,1069
-chunkr_ai-0.0.50.dist-info/METADATA,sha256=ZlM0XHhykHUOLEsNCF52OmMEw2QlvgSUy972FQjgD2s,7053
-chunkr_ai-0.0.50.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-chunkr_ai-0.0.50.dist-info/top_level.txt,sha256=0IZY7PZIiS8bw5r4NUQRUQ-ATi-L_3vLQVq3ZLouOW8,10
-chunkr_ai-0.0.50.dist-info/RECORD,,
+chunkr_ai-0.1.0.dist-info/licenses/LICENSE,sha256=w3R12yNDyZpMiy2lxy_hvNbsldC75ww79sF0u11rkho,1069
+chunkr_ai-0.1.0.dist-info/METADATA,sha256=nRApqY3sX17_8fqSwfgYUH7Sy4Ahg5ybpxU9ZTXvdH4,7052
+chunkr_ai-0.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+chunkr_ai-0.1.0.dist-info/top_level.txt,sha256=0IZY7PZIiS8bw5r4NUQRUQ-ATi-L_3vLQVq3ZLouOW8,10
+chunkr_ai-0.1.0.dist-info/RECORD,,

{chunkr_ai-0.0.50.dist-info → chunkr_ai-0.1.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.8.0)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any