Made few small changes.

Author: s3rius

Cargo.lock

@@ -28,7 +28,7 @@ class Nats:
     async def publish(
         self,
         subject: str,
-        payload: bytes,
+        payload: bytes | str | bytearray | memoryview,
         *,
         headers: dict[str, Any] | None = None,
         reply: str | None = None,

python/natsrpy/_natsrpy_rs/__init__.pyi

@@ -1,5 +1,5 @@
-from natsrpy._natsrpy_rs.js.kv import KeyValue, KVConfig
-from natsrpy._natsrpy_rs.js.stream import Stream, StreamConfig
+from .managers.kv import KVManager
+from .managers.streams import StreamsManager
 
 class JetStream:
     async def publish(
@@ -11,13 +11,7 @@ class JetStream:
         reply: str | None = None,
         err_on_disconnect: bool = False,
     ) -> None: ...
-    # KV
-    async def create_kv(self, config: KVConfig) -> KeyValue: ...
-    async def update_kv(self, config: KVConfig) -> KeyValue: ...
-    async def get_kv(self, bucket: str) -> KeyValue: ...
-    async def delete_kv(self, bucket: str) -> bool: ...
-    # Streams
-    async def create_stream(self, config: StreamConfig) -> Stream: ...
-    async def update_stream(self, config: StreamConfig) -> Stream: ...
-    async def get_stream(self, name: str) -> Stream: ...
-    async def delete_stream(self, name: str) -> bool: ...
+    @property
+    def kv(self) -> KVManager: ...
+    @property
+    def streams(self) -> StreamsManager: ...

python/natsrpy/_natsrpy_rs/js/__init__.pyi

@@ -0,0 +1,19 @@
+from .common import (
+    AckPolicy,
+    DeliverPolicy,
+    PriorityPolicy,
+    ReplayPolicy,
+)
+from .pull import PullConsumer, PullConsumerConfig
+from .push import PushConsumer, PushConsumerConfig
+
+__all__ = [
+    "AckPolicy",
+    "DeliverPolicy",
+    "PriorityPolicy",
+    "PullConsumer",
+    "PullConsumerConfig",
+    "PushConsumer",
+    "PushConsumerConfig",
+    "ReplayPolicy",
+]

python/natsrpy/_natsrpy_rs/js/consumers/__init__.pyi

@@ -0,0 +1,22 @@
+class DeliverPolicy:
+    ALL: DeliverPolicy
+    LAST: DeliverPolicy
+    NEW: DeliverPolicy
+    BY_START_SEQUENCE: DeliverPolicy
+    BY_START_TIME: DeliverPolicy
+    LAST_PER_SUBJECT: DeliverPolicy
+
+class AckPolicy:
+    EXPLICIT: AckPolicy
+    NONE: AckPolicy
+    ALL: AckPolicy
+
+class ReplayPolicy:
+    INSTANT: ReplayPolicy
+    ORIGINAL: ReplayPolicy
+
+class PriorityPolicy:
+    NONE: PriorityPolicy
+    OVERFLOW: PriorityPolicy
+    PINNED_CLIENT: PriorityPolicy
+    PRIORITIZED: PriorityPolicy

python/natsrpy/_natsrpy_rs/js/consumers/common.pyi

@@ -0,0 +1,72 @@
+from datetime import timedelta
+
+from natsrpy._natsrpy_rs.js.consumers.common import (
+    AckPolicy,
+    DeliverPolicy,
+    PriorityPolicy,
+    ReplayPolicy,
+)
+
+class PullConsumerConfig:
+    durable_name: str | None
+    name: str | None
+    description: str | None
+    deliver_policy: DeliverPolicy
+    delivery_start_sequence: int | None
+    delivery_start_time: int | None
+    ack_policy: AckPolicy
+    ack_wait: timedelta
+    max_deliver: int
+    filter_subject: str
+    filter_subjects: list[str]
+    replay_policy: ReplayPolicy
+    rate_limit: int
+    sample_frequency: int
+    max_waiting: int
+    max_ack_pending: int
+    headers_only: bool
+    max_batch: int
+    max_bytes: int
+    max_expires: timedelta
+    inactive_threshold: timedelta
+    num_replicas: int
+    memory_storage: bool
+    metadata: dict[str, str]
+    backoff: list[timedelta]
+    priority_policy: PriorityPolicy
+    priority_groups: list[str]
+    pause_until: int | None
+
+    def __init__(
+        self,
+        durable_name: str | None = None,
+        name: str | None = None,
+        description: str | None = None,
+        deliver_policy: DeliverPolicy | None = None,
+        delivery_start_sequence: int | None = None,
+        delivery_start_time: int | None = None,
+        ack_policy: AckPolicy | None = None,
+        ack_wait: timedelta | None = None,
+        max_deliver: int | None = None,
+        filter_subject: str | None = None,
+        filter_subjects: list[str] | None = None,
+        replay_policy: ReplayPolicy | None = None,
+        rate_limit: int | None = None,
+        sample_frequency: int | None = None,
+        max_waiting: int | None = None,
+        max_ack_pending: int | None = None,
+        headers_only: bool | None = None,
+        max_batch: int | None = None,
+        max_bytes: int | None = None,
+        max_expires: timedelta | None = None,
+        inactive_threshold: timedelta | None = None,
+        num_replicas: int | None = None,
+        memory_storage: bool | None = None,
+        metadata: dict[str, str] | None = None,
+        backoff: list[timedelta] | None = None,
+        priority_policy: PriorityPolicy | None = None,
+        priority_groups: list[str] | None = None,
+        pause_until: int | None = None,
+    ) -> None: ...
+
+class PullConsumer: ...