further stubs for outlines

Author: lars20070

stubs/mlx/__init__.pyi

@@ -4,5 +4,3 @@ from . import nn
 
 # mlx is imported as a package, primarily for mlx.nn
 __all__: list[str] = []
-
-nn: Any

stubs/outlines/inputs.pyi

@@ -0,0 +1,18 @@
+from __future__ import annotations
+
+from collections.abc import Sequence
+from typing import Any
+
+from PIL.Image import Image as PILImage
+
+class Chat:
+    def __init__(self, messages: list[dict[str, Any]] | None = None) -> None: ...
+    def add_system_message(self, content: str) -> None: ...
+    def add_user_message(self, content: str | Sequence[str | Image]) -> None: ...
+    def add_assistant_message(self, content: str | list[str | Image]) -> None: ...
+    def extend(self, messages: list[dict[str, Any]]) -> None: ...
+    def append(self, message: dict[str, Any]) -> None: ...
+    def pop(self) -> dict[str, Any] | None: ...
+
+class Image:
+    def __init__(self, image: PILImage.Image) -> None: ...

stubs/outlines/models/base.pyi

@@ -5,6 +5,6 @@ class Model:
     def __call__(self, *args: Any, **kwargs: Any) -> Any: ...
     def stream(self, *args: Any, **kwargs: Any) -> Iterable[Any]: ...
 
-class AsyncModel(Model):
+class AsyncModel:
     async def __call__(self, *args: Any, **kwargs: Any) -> Any: ...
     def stream(self, *args: Any, **kwargs: Any) -> AsyncIterable[Any]: ...

stubs/outlines/types/__init__.pyi

@@ -0,0 +1,3 @@
+from typing import Any
+
+__all__: list[str] = []

stubs/outlines/types/dsl.pyi

@@ -0,0 +1,4 @@
+from typing import Any
+
+class JsonSchema:
+    def __init__(self, schema: dict[str, Any]) -> None: ...