agstack 1.17.0__tar.gz → 1.17.1__tar.gz

{agstack-1.17.0 → agstack-1.17.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agstack
-Version: 1.17.0
+Version: 1.17.1
 Summary: Production-ready toolkit for building FastAPI and LLM applications
 Author-email: XtraVisions <gitadmin@xtravisions.com>, Chen Hao <chenhao@xtravisions.com>
 Maintainer-email: XtraVisions <gitadmin@xtravisions.com>, Chen Hao <chenhao@xtravisions.com>

{agstack-1.17.0 → agstack-1.17.1}/agstack/messagebus/redis.py

@@ -2,6 +2,7 @@
 
 import asyncio
 from collections.abc import AsyncIterator
+from typing import Self
 
 import redis.asyncio as aioredis
 from redis.asyncio.client import PubSub
@@ -17,15 +18,21 @@ class RedisSubscription(Subscription):
         self._channels = channels
         self._task: asyncio.Task[None] | None = None
         self._queue: asyncio.Queue[tuple[str, bytes]] = asyncio.Queue()
+        self._started = False
         self._closed = False
 
     async def _start(self) -> None:
-        """订阅 channel 并启动后台监听任务"""
+        if self._started:
+            return
+        self._started = True
         await self._pubsub.subscribe(*self._channels)
         self._task = asyncio.create_task(self._listen())
 
+    async def __aenter__(self) -> Self:
+        await self._start()
+        return self
+
     async def _listen(self) -> None:
-        """后台监听 Redis Pub/Sub 消息"""
         try:
             async for msg in self._pubsub.listen():
                 if self._closed:
@@ -47,6 +54,8 @@ class RedisSubscription(Subscription):
         return self
 
     async def __anext__(self) -> tuple[str, bytes]:
+        if not self._started:
+            await self._start()
         if self._closed and self._queue.empty():
             raise StopAsyncIteration
         try:
@@ -79,9 +88,7 @@ class RedisMessageBus(MessageBus):
 
     def subscribe(self, *channels: str) -> RedisSubscription:
         pubsub = self._client.pubsub()
-        sub = RedisSubscription(pubsub, channels)
-        # 调用方需要在 async with 或手动 await _start() 后使用
-        return sub
+        return RedisSubscription(pubsub, channels)
 
     async def close(self) -> None:
         await self._client.aclose()

{agstack-1.17.0 → agstack-1.17.1}/agstack.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agstack
-Version: 1.17.0
+Version: 1.17.1
 Summary: Production-ready toolkit for building FastAPI and LLM applications
 Author-email: XtraVisions <gitadmin@xtravisions.com>, Chen Hao <chenhao@xtravisions.com>
 Maintainer-email: XtraVisions <gitadmin@xtravisions.com>, Chen Hao <chenhao@xtravisions.com>

{agstack-1.17.0 → agstack-1.17.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agstack"
-version = "1.17.0"
+version = "1.17.1"
 description = "Production-ready toolkit for building FastAPI and LLM applications"
 readme = "README.md"
 license = "MIT"

{agstack-1.17.0 → agstack-1.17.1}/tests/test_messagebus_redis.py

@@ -26,40 +26,44 @@ async def bus():
 
 class TestPublishSubscribe:
     async def test_single_channel(self, bus: RedisMessageBus) -> None:
-        sub = bus.subscribe("test:ch1")
-        await sub._start()
-        async with sub:
+        async with bus.subscribe("test:ch1") as sub:
             await asyncio.sleep(0.1)
             await bus.publish("test:ch1", b"hello")
-            channel, message = await asyncio.wait_for(sub.__anext__(), timeout=3)
+            channel, message = await asyncio.wait_for(anext(sub), timeout=3)
             assert channel == "test:ch1"
             assert message == b"hello"
 
     async def test_multi_channel(self, bus: RedisMessageBus) -> None:
-        sub = bus.subscribe("test:ch1", "test:ch2")
-        await sub._start()
-        async with sub:
+        async with bus.subscribe("test:ch1", "test:ch2") as sub:
             await asyncio.sleep(0.1)
             await bus.publish("test:ch1", b"msg1")
             await bus.publish("test:ch2", b"msg2")
             results = []
             for _ in range(2):
-                ch, msg = await asyncio.wait_for(sub.__anext__(), timeout=3)
+                ch, msg = await asyncio.wait_for(anext(sub), timeout=3)
                 results.append((ch, msg))
             channels = [r[0] for r in results]
             assert "test:ch1" in channels
             assert "test:ch2" in channels
 
+    async def test_lazy_start_without_context_manager(self, bus: RedisMessageBus) -> None:
+        sub = bus.subscribe("test:ch1")
+        await asyncio.sleep(0.1)
+        await bus.publish("test:ch1", b"world")
+        channel, message = await asyncio.wait_for(anext(sub), timeout=3)
+        assert channel == "test:ch1"
+        assert message == b"world"
+        await sub.close()
+
     async def test_publish_no_subscribers(self, bus: RedisMessageBus) -> None:
         await bus.publish("test:nobody", b"lost message")
 
 
 class TestSubscriptionClose:
     async def test_close(self, bus: RedisMessageBus) -> None:
-        sub = bus.subscribe("test:ch1")
-        await sub._start()
-        await asyncio.sleep(0.1)
-        await sub.close()
+        async with bus.subscribe("test:ch1") as sub:
+            await asyncio.sleep(0.1)
+        assert sub._closed
 
 
 class TestBusClose: