Fix and new features

scripts/lint.sh

@@ -2,4 +2,4 @@ set -e
 set -x
 
 ruff check src tests examples
-ruff format src tests examples --check
+ruff format src tests examples --check --diff

src/handyllm/__init__.py

@@ -11,9 +11,13 @@
 from .prompt_converter import PromptConverter as PromptConverter
 from .utils import (
     stream_chat_all as stream_chat_all,
+    stream_chat_with_role as stream_chat_with_role,
+    stream_chat_with_reasoning as stream_chat_with_reasoning,
     stream_chat as stream_chat,
     stream_completions as stream_completions,
     astream_chat_all as astream_chat_all,
+    astream_chat_with_role as astream_chat_with_role,
+    astream_chat_with_reasoning as astream_chat_with_reasoning,
     astream_chat as astream_chat,
     astream_completions as astream_completions,
     stream_to_file as stream_to_file,

src/handyllm/_io.py

@@ -3,8 +3,6 @@
 import yaml
 from functools import wraps
 
-from .response import DictProxy
-
 
 # add multi representer for Path, for YAML serialization
 class MySafeDumper(yaml.SafeDumper):
@@ -14,13 +12,10 @@ class MySafeDumper(yaml.SafeDumper):
 MySafeDumper.add_multi_representer(
     Path, lambda dumper, data: dumper.represent_str(str(data))
 )
-MySafeDumper.add_multi_representer(
-    DictProxy, lambda dumper, data: dumper.represent_dict(data)
-)
 
 
 @wraps(yaml.dump)
-def yaml_dump(*args, **kwargs):
+def yaml_dump(*args, **kwargs) -> str:
     kwargs.setdefault("Dumper", MySafeDumper)
     kwargs.setdefault("allow_unicode", True)
     return yaml.dump(*args, **kwargs)

src/handyllm/cache_manager.py

@@ -1,5 +1,3 @@
-__all__ = ["CacheManager"]
-
 from functools import wraps
 from inspect import iscoroutinefunction
 from os import PathLike

src/handyllm/endpoint_manager.py

@@ -1,10 +1,5 @@
 from __future__ import annotations
 
-__all__ = [
-    "Endpoint",
-    "EndpointManager",
-]
-
 import os
 from json import JSONDecodeError
 from threading import Lock