schemez 1.2.2__tar.gz → 1.2.4__tar.gz

{schemez-1.2.2 → schemez-1.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: schemez
-Version: 1.2.2
+Version: 1.2.4
 Summary: Pydantic shim for config stuff
 Keywords: 
 Author: Philipp Temminghoff

{schemez-1.2.2 → schemez-1.2.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "schemez"
-version = "1.2.2"
+version = "1.2.4"
 description = "Pydantic shim for config stuff"
 readme = "README.md"
 requires-python = ">=3.13"

{schemez-1.2.2 → schemez-1.2.4}/src/schemez/functionschema.py

@@ -12,7 +12,6 @@ import decimal
 import enum
 import inspect
 import ipaddress
-import logging
 from pathlib import Path
 import re
 import types
@@ -23,6 +22,7 @@ from uuid import UUID
 import docstring_parser
 import pydantic
 
+from schemez import log
 from schemez.typedefs import (
     OpenAIFunctionDefinition,
     OpenAIFunctionTool,
@@ -34,7 +34,7 @@ if typing.TYPE_CHECKING:
     from schemez.typedefs import Property
 
 
-logger = logging.getLogger(__name__)
+logger = log.get_logger(__name__)
 
 
 class FunctionType(enum.StrEnum):
@@ -119,6 +119,9 @@ class FunctionSchema(pydantic.BaseModel):
         required = self.parameters.get("required", self.required)
 
         for name, details in properties.items():
+            if name.startswith("_"):  # TODO: kwarg for renaming instead perhaps?
+                logger.debug("Skipping parameter %s due to leading underscore", name)
+                continue
             # Get base type
             if "enum" in details:
                 values = tuple(details["enum"])  # type: ignore
@@ -246,19 +249,14 @@ class FunctionSchema(pydantic.BaseModel):
             RuntimeError: If datamodel-codegen is not available
             subprocess.CalledProcessError: If code generation fails
         """
+        import shutil
         import subprocess
         import tempfile
 
-        try:
-            # Check if datamodel-codegen is available
-            subprocess.run(
-                ["datamodel-codegen", "--version"],
-                check=True,
-                capture_output=True,
-            )
-        except (subprocess.CalledProcessError, FileNotFoundError) as e:
+        # Check if datamodel-codegen is available
+        if not shutil.which("datamodel-codegen"):
             msg = "datamodel-codegen not available"
-            raise RuntimeError(msg) from e
+            raise RuntimeError(msg)
 
         name = class_name or f"{self.name.title()}Response"
 

{schemez-1.2.2 → schemez-1.2.4}/src/schemez/helpers.py

@@ -4,8 +4,8 @@ from __future__ import annotations
 
 import asyncio
 import importlib
-import os
 from pathlib import Path
+import shutil
 import subprocess
 import sys
 import tempfile
@@ -15,7 +15,6 @@ from pydantic import BaseModel
 from pydantic_core import to_json
 
 
-StrPath = str | os.PathLike[str]
 PythonVersion = Literal["3.13", "3.14", "3.15"]
 
 if TYPE_CHECKING:
@@ -177,42 +176,6 @@ def resolve_type_string(type_string: str, safe: bool = True) -> type:
             raise ValueError(msg) from e
 
 
-async def _detect_command(command: str, *, test_flag: str = "--version") -> list[str]:
-    """Detect the correct command prefix for running a command.
-
-    Tries 'uv run' first, then falls back to direct execution.
-
-    Args:
-        command: The command to detect
-        test_flag: Flag to test command availability with
-
-    Returns:
-        Command prefix list (empty for direct execution)
-
-    Raises:
-        RuntimeError: If command is not available
-    """
-    cmd_prefixes = [["uv", "run"], []]
-
-    for prefix in cmd_prefixes:
-        try:
-            proc = await asyncio.create_subprocess_exec(
-                *prefix,
-                command,
-                test_flag,
-                stdout=asyncio.subprocess.PIPE,
-                stderr=asyncio.subprocess.PIPE,
-            )
-            await proc.communicate()
-            if proc.returncode == 0:
-                return prefix
-        except FileNotFoundError:
-            continue
-
-    msg = f"{command} not available (tried both 'uv run' and direct execution)"
-    raise RuntimeError(msg)
-
-
 async def model_to_python_code(
     model: type[BaseModel] | dict[str, Any],
     *,
@@ -236,7 +199,10 @@ async def model_to_python_code(
         RuntimeError: If datamodel-codegen is not available
         subprocess.CalledProcessError: If code generation fails
     """
-    working_prefix = await _detect_command("datamodel-codegen")
+    # Check if datamodel-codegen is available
+    if not shutil.which("datamodel-codegen"):
+        msg = "datamodel-codegen not available"
+        raise RuntimeError(msg)
 
     if isinstance(model, dict):
         schema = model
@@ -271,7 +237,6 @@ async def model_to_python_code(
 
     try:  # Generate model using datamodel-codegen
         proc = await asyncio.create_subprocess_exec(
-            *working_prefix,
             "datamodel-codegen",
             *args,
             stdout=asyncio.subprocess.PIPE,

{schemez-1.2.2 → schemez-1.2.4}/src/schemez/schema.py

@@ -2,7 +2,6 @@
 
 from __future__ import annotations
 
-import os
 from typing import TYPE_CHECKING, Any, Literal, Self
 
 import anyenv
@@ -15,9 +14,9 @@ if TYPE_CHECKING:
 
     from llmling_agent.agent.agent import AgentType
     from llmling_agent.models.content import BaseContent
+    from upath.types import JoinablePathLike
 
 
-StrPath = str | os.PathLike[str]
 SourceType = Literal["pdf", "image"]
 PythonVersion = Literal["3.13", "3.14", "3.15"]
 
@@ -43,7 +42,9 @@ class Schema(BaseModel):
         return merge_models(self, other)
 
     @classmethod
-    def from_yaml(cls, content: str, inherit_path: StrPath | None = None) -> Self:
+    def from_yaml(
+        cls, content: str, inherit_path: JoinablePathLike | None = None
+    ) -> Self:
         """Create from YAML string."""
         import yamling
 
@@ -233,7 +234,7 @@ class Schema(BaseModel):
         )
         return yamling.dump_yaml(text)
 
-    def save(self, path: StrPath, overwrite: bool = False) -> None:
+    def save(self, path: JoinablePathLike, overwrite: bool = False) -> None:
         """Save configuration to a YAML file.
 
         Args:

{schemez-1.2.2 → schemez-1.2.4}/src/schemez/tool_executor/executor.py

@@ -3,7 +3,6 @@
 from __future__ import annotations
 
 import asyncio
-import logging
 from pathlib import Path
 import time
 from typing import TYPE_CHECKING, Any
@@ -12,6 +11,7 @@ from pydantic import BaseModel
 from pydantic_core import from_json
 from upath import UPath
 
+from schemez import log
 from schemez.functionschema import FunctionSchema
 from schemez.tool_executor.helpers import clean_generated_code, generate_input_model
 
@@ -20,11 +20,12 @@ if TYPE_CHECKING:
     from collections.abc import Callable, Sequence
 
     from fastapi import FastAPI
+    from upath.types import JoinablePathLike
 
     from schemez.tool_executor.types import ToolHandler
 
 
-logger = logging.getLogger(__name__)
+logger = log.get_logger(__name__)
 
 
 class HttpToolExecutor:
@@ -32,7 +33,7 @@ class HttpToolExecutor:
 
     def __init__(
         self,
-        schemas: Sequence[dict[str, Any] | Path],
+        schemas: Sequence[dict[str, Any] | UPath],
         handler: ToolHandler,
         base_url: str = "http://localhost:8000",
     ):
@@ -61,7 +62,7 @@ class HttpToolExecutor:
             match schema:
                 case dict():
                     loaded_schemas.append(schema)
-                case str() | Path():
+                case str() | Path() | UPath():
                     text = UPath(schema).read_text("utf-8")
                     loaded_schemas.append(from_json(text))
                 case _:
@@ -264,7 +265,7 @@ from datetime import datetime
         uvicorn.run(app, host=host, port=port)
         return None
 
-    async def save_to_files(self, output_dir: Path) -> dict[str, Path]:
+    async def save_to_files(self, output_dir: JoinablePathLike) -> dict[str, UPath]:
         """Save generated code to files.
 
         Args:
@@ -273,7 +274,7 @@ from datetime import datetime
         Returns:
             Dictionary mapping file types to paths
         """
-        output_dir = Path(output_dir)
+        output_dir = UPath(output_dir)
         output_dir.mkdir(parents=True, exist_ok=True)
 
         saved_files = {}

{schemez-1.2.2 → schemez-1.2.4}/src/schemez/tool_executor/helpers.py

@@ -2,13 +2,13 @@
 
 from __future__ import annotations
 
-import logging
 import time
 
+from schemez import log
 from schemez.helpers import model_to_python_code
 
 
-logger = logging.getLogger(__name__)
+logger = log.get_logger(__name__)
 
 
 async def generate_input_model(schema_dict: dict) -> tuple[str, str]: