From 921735dbcbcda59170e275b60bb340b19cf5664d Mon Sep 17 00:00:00 2001 From: lda Date: Tue, 1 Sep 2026 03:09:38 +0700 Subject: [PATCH] fix: remove unused agent arg from App.poll --- src/agentmsgs/core/app.py | 2 +- src/agentmsgs/mcp/server.py | 4 +--- tests/test_app.py | 2 +- 3 files changed, 3 insertions(+), 5 deletions(-) diff --git a/src/agentmsgs/core/app.py b/src/agentmsgs/core/app.py index 40d667a..f29ab9a 100644 --- a/src/agentmsgs/core/app.py +++ b/src/agentmsgs/core/app.py @@ -41,7 +41,7 @@ class App: def append(self, tid: uuid.UUID, sender: Agent, content: str): return ops.append_message(self.store, tid, sender, content) - def poll(self, tid: uuid.UUID, agent: Agent, after_seq: int = 0): + def poll(self, tid: uuid.UUID, after_seq: int = 0) -> list: return ops.list_messages(self.store, tid, after_seq) def mark_read(self, tid: uuid.UUID, agent: Agent, seq: int) -> None: diff --git a/src/agentmsgs/mcp/server.py b/src/agentmsgs/mcp/server.py index 9d97df1..38a9c9f 100644 --- a/src/agentmsgs/mcp/server.py +++ b/src/agentmsgs/mcp/server.py @@ -71,9 +71,7 @@ def poll(thread_id: str, after_seq: int = 0) -> list[dict]: """Poll messages in a thread after seq.""" from uuid import UUID - # poll doesn't need agent for cursor; use raw store list for simplicity - msgs = _app.poll(UUID(thread_id), _app.get_or_create_agent("__poll__"), after_seq) - # filter out the dummy poll agent if it ever sent (it never does) + msgs = _app.poll(UUID(thread_id), after_seq) return [ { "seq": m.seq, diff --git a/tests/test_app.py b/tests/test_app.py index dec047c..2f2a8f4 100644 --- a/tests/test_app.py +++ b/tests/test_app.py @@ -9,7 +9,7 @@ def test_app_facade_with_memory(): t = app.create_thread(alice, bob) app.append(t.id, alice, "hello") assert app.has_unread(t.id, bob) is True - msgs = app.poll(t.id, bob) + msgs = app.poll(t.id) assert len(msgs) == 1 app.mark_read(t.id, bob, msgs[0].seq) assert app.has_unread(t.id, bob) is False