schemez 1.2.3__tar.gz → 1.2.4__tar.gz

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

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

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

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

{schemez-1.2.3 → 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):

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

@@ -4,7 +4,6 @@ from __future__ import annotations
 
 import asyncio
 import importlib
-import os
 from pathlib import Path
 import shutil
 import subprocess
@@ -16,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:

{schemez-1.2.3 → 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.3 → 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.3 → 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]: