chunkr-ai 0.0.9__tar.gz → 0.0.10__tar.gz

{chunkr_ai-0.0.9/src/chunkr_ai.egg-info → chunkr_ai-0.0.10}/PKG-INFO

@@ -1,12 +1,12 @@
 Metadata-Version: 2.2
 Name: chunkr-ai
-Version: 0.0.9
+Version: 0.0.10
 Summary: Python client for Chunkr: open source document intelligence
 Author-email: Ishaan Kapoor <ishaan@lumina.sh>
 Project-URL: Homepage, https://chunkr.ai
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: httpx>=0.24.0
+Requires-Dist: httpx>=0.25.0
 Requires-Dist: pillow>=10.0.0
 Requires-Dist: pydantic>=2.0.0
 Requires-Dist: pytest-asyncio>=0.21.0

{chunkr_ai-0.0.9 → chunkr_ai-0.0.10}/pyproject.toml

@@ -4,14 +4,14 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "chunkr-ai"
-version = "0.0.9"
+version = "0.0.10"
 authors = [{"name" = "Ishaan Kapoor", "email" = "ishaan@lumina.sh"}]
 description = "Python client for Chunkr: open source document intelligence"
 readme = "README.md"
 license = {"file" = "LICENSE"}
 urls = {Homepage = "https://chunkr.ai"} 
 dependencies = [
-    "httpx>=0.24.0",
+    "httpx>=0.25.0",
     "pillow>=10.0.0",
     "pydantic>=2.0.0",
     "pytest-asyncio>=0.21.0",

{chunkr_ai-0.0.9 → chunkr_ai-0.0.10}/src/chunkr_ai/api/chunkr.py

@@ -1,4 +1,4 @@
-from .base import ChunkrBase
+from .chunkr_base import ChunkrBase
 from .config import Configuration
 from .task import TaskResponse
 from pathlib import Path
@@ -163,5 +163,3 @@ class Chunkr(ChunkrBase):
             headers=self._headers()
         )
         r.raise_for_status()
-
-  

{chunkr_ai-0.0.9 → chunkr_ai-0.0.10}/src/chunkr_ai/api/chunkr_async.py

@@ -1,4 +1,4 @@
-from .base import ChunkrBase
+from .chunkr_base import ChunkrBase
 from .task import TaskResponse
 from .config import Configuration
 import httpx

{chunkr_ai-0.0.9 → chunkr_ai-0.0.10}/src/chunkr_ai/api/config.py

@@ -40,7 +40,6 @@ class ChunkProcessing(BaseModel):
 
 class Property(BaseModel):
     name: str
-    title: Optional[str] = None
     prop_type: str
     description: Optional[str] = None
     default: Optional[str] = None

{chunkr_ai-0.0.9 → chunkr_ai-0.0.10}/src/chunkr_ai/api/misc.py

@@ -1,11 +1,11 @@
+from .config import Configuration, Property, JsonSchema
 import io
 import json
 from pathlib import Path
 from PIL import Image
 import requests
 from typing import Union, Tuple, BinaryIO, Optional
-from .config import Configuration
-
+from pydantic import BaseModel
 
 def prepare_file(
     file: Union[str, Path, BinaryIO, Image.Image]
@@ -15,8 +15,31 @@ def prepare_file(
     if isinstance(file, str) and (file.startswith('http://') or file.startswith('https://')):
         response = requests.get(file)
         response.raise_for_status()
+        
+        # Try to get filename from Content-Disposition header first
+        filename = None
+        content_disposition = response.headers.get('Content-Disposition')
+        if content_disposition and 'filename=' in content_disposition:
+            filename = content_disposition.split('filename=')[-1].strip('"\'')
+        
+        # If no Content-Disposition, try to get clean filename from URL path
+        if not filename:
+            from urllib.parse import urlparse, unquote
+            parsed_url = urlparse(file)
+            path = unquote(parsed_url.path)
+            filename = Path(path).name if path else None
+        
+        # Fallback to default name if we couldn't extract one
+        filename = filename or 'downloaded_file'
+        
+        # Sanitize filename: remove invalid characters and limit length
+        import re
+        filename = re.sub(r'[<>:"/\\|?*%]', '_', filename)  # Replace invalid chars with underscore
+        filename = re.sub(r'\s+', '_', filename)            # Replace whitespace with underscore
+        filename = filename.strip('._')                     # Remove leading/trailing dots and underscores
+        filename = filename[:255]                           # Limit length to 255 characters
+        
         file_obj = io.BytesIO(response.content)
-        filename = Path(file.split('/')[-1]).name or 'downloaded_file'
         return filename, file_obj
 
     # Handle base64 strings
@@ -104,3 +127,33 @@ def prepare_upload_data(
             files[key] = (None, json.dumps(value), 'application/json')
                 
     return files
+
+def from_pydantic(pydantic: BaseModel) -> dict:
+    """Convert a Pydantic model to a Chunk json schema.
+    
+    Args:
+        pydantic: A Pydantic BaseModel class or instance
+        
+    Returns:
+        dict: A JSON schema compatible with Chunk's format
+    """
+    model = pydantic if isinstance(pydantic, type) else pydantic.__class__
+    schema = model.model_json_schema()
+    print(schema)
+    properties = []
+    for name, details in schema.get('properties', {}).items():
+        prop = Property(
+            name=name,
+            title=details.get('title'),
+            prop_type=details.get('type', 'string'),
+            description=details.get('description'),
+            default=str(details.get('default')) if details.get('default') is not None else None
+        )
+        properties.append(prop)
+    
+    json_schema = JsonSchema(
+        title=schema.get('title', model.__name__),
+        properties=properties
+    )
+    
+    return json_schema.model_dump(mode="json", exclude_none=True)

{chunkr_ai-0.0.9 → chunkr_ai-0.0.10/src/chunkr_ai.egg-info}/PKG-INFO

@@ -1,12 +1,12 @@
 Metadata-Version: 2.2
 Name: chunkr-ai
-Version: 0.0.9
+Version: 0.0.10
 Summary: Python client for Chunkr: open source document intelligence
 Author-email: Ishaan Kapoor <ishaan@lumina.sh>
 Project-URL: Homepage, https://chunkr.ai
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: httpx>=0.24.0
+Requires-Dist: httpx>=0.25.0
 Requires-Dist: pillow>=10.0.0
 Requires-Dist: pydantic>=2.0.0
 Requires-Dist: pytest-asyncio>=0.21.0

{chunkr_ai-0.0.9 → chunkr_ai-0.0.10}/src/chunkr_ai.egg-info/SOURCES.txt

@@ -10,11 +10,10 @@ src/chunkr_ai.egg-info/dependency_links.txt
 src/chunkr_ai.egg-info/requires.txt
 src/chunkr_ai.egg-info/top_level.txt
 src/chunkr_ai/api/__init__.py
-src/chunkr_ai/api/api.py
 src/chunkr_ai/api/auth.py
-src/chunkr_ai/api/base.py
 src/chunkr_ai/api/chunkr.py
 src/chunkr_ai/api/chunkr_async.py
+src/chunkr_ai/api/chunkr_base.py
 src/chunkr_ai/api/config.py
 src/chunkr_ai/api/misc.py
 src/chunkr_ai/api/protocol.py

{chunkr_ai-0.0.9 → chunkr_ai-0.0.10}/src/chunkr_ai.egg-info/requires.txt

@@ -1,4 +1,4 @@
-httpx>=0.24.0
+httpx>=0.25.0
 pillow>=10.0.0
 pydantic>=2.0.0
 pytest-asyncio>=0.21.0