stackshift-agent 0.1.0__tar.gz → 0.1.1__tar.gz

{stackshift_agent-0.1.0 → stackshift_agent-0.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: stackshift-agent
-Version: 0.1.0
+Version: 0.1.1
 Summary: Run an agent on StackShift: events, governed actions, StackShift calls, state and model access.
 Author-email: StackShift <support@stackshift.cloud>
 Project-URL: Homepage, https://stackshift.cloud

{stackshift_agent-0.1.0 → stackshift_agent-0.1.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "stackshift-agent"
-version = "0.1.0"
+version = "0.1.1"
 description = "Run an agent on StackShift: events, governed actions, StackShift calls, state and model access."
 readme = "README.md"
 requires-python = ">=3.9"

{stackshift_agent-0.1.0 → stackshift_agent-0.1.1}/stackshift_agent/inbox.py

@@ -18,8 +18,11 @@ class Inbox:
             raise StackShiftAgentError("This agent has no inbox. Add an inbox to the agent block in stackshift.yaml.", status=409, code="no_inbox")
         return inbox
 
+    async def _account(self) -> str:
+        return (await self._self())["agent"]["account_id"]
+
     async def info(self) -> Any:
-        return await self._platform.inspect("mail.agent_inboxes.get", {"inbox_id": await self.id()})
+        return await self._platform.inspect("mail.agent_inboxes.get", {"inbox_id": await self.id()}, resource_id=await self._account())
 
     async def send(
         self,
@@ -41,6 +44,7 @@ class Inbox:
             "html": html,
             "reply_to_message_id": reply_to_message_id,
         }
+        options.setdefault("resource_id", await self._account())
         return await self._platform.call("mail.agent_inbox.send", {key: value for key, value in configuration.items() if value is not None}, **options)
 
     async def reply(self, message: dict[str, Any], text: str, **options: Any) -> dict[str, Any]:

{stackshift_agent-0.1.0 → stackshift_agent-0.1.1}/stackshift_agent.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: stackshift-agent
-Version: 0.1.0
+Version: 0.1.1
 Summary: Run an agent on StackShift: events, governed actions, StackShift calls, state and model access.
 Author-email: StackShift <support@stackshift.cloud>
 Project-URL: Homepage, https://stackshift.cloud

{stackshift_agent-0.1.0 → stackshift_agent-0.1.1}/tests/test_platform.py

@@ -84,7 +84,7 @@ class PlatformTest(AgentTestCase):
     async def test_the_inbox_sends_and_replies_through_the_registry_actions_using_the_agent_inbox_id(self):
         self.mcp(
             lambda name, arguments: reply({"id": "op_3", "state": "succeeded"}),
-            lambda call: {"body": {"agent": {"id": "agt_1", "handle": "supplies", "name": "Supplies", "inbox_mailbox_id": "inbox_1"}, "actions": [], "platform": [], "models": []}} if call["path"] == "/api/v1/agents/self" else None,
+            lambda call: {"body": {"agent": {"id": "agt_1", "account_id": "acct_1", "handle": "supplies", "name": "Supplies", "inbox_mailbox_id": "inbox_1"}, "actions": [], "platform": [], "models": []}} if call["path"] == "/api/v1/agents/self" else None,
         )
 
         async def work():
@@ -93,7 +93,8 @@ class PlatformTest(AgentTestCase):
 
         await self.in_run(work)
         sends = [call["arguments"]["request"] for call in self.tool_calls()]
-        self.assertEqual(sends[0], {"action": "mail.agent_inbox.send", "configuration": {"inbox_id": "inbox_1", "to": ["buyer@example.com"], "subject": "Quote", "text": "Here it is"}})
+        self.assertEqual(sends[0], {"action": "mail.agent_inbox.send", "configuration": {"inbox_id": "inbox_1", "to": ["buyer@example.com"], "subject": "Quote", "text": "Here it is"}, "resource_id": "acct_1"})
+        self.assertEqual(sends[1]["resource_id"], "acct_1")
         self.assertEqual(sends[1]["configuration"], {"inbox_id": "inbox_1", "to": ["ada@example.com"], "subject": "Re: Quote", "text": "Thanks", "reply_to_message_id": "m1"})
 
     async def test_an_agent_without_an_inbox_says_so(self):
@@ -115,7 +116,7 @@ class PlatformTest(AgentTestCase):
     async def test_agreements_use_the_agreements_registry_actions(self):
         self.mcp(
             lambda name, arguments: reply({"threads": []} if name == "inspect" else {"id": "op_4", "state": "succeeded"}),
-            lambda call: {"body": {"agent": {"id": "agt_1", "handle": "supplies", "name": "Supplies", "inbox_mailbox_id": "inbox_1"}, "actions": [], "platform": [], "models": []}} if call["path"] == "/api/v1/agents/self" else None,
+            lambda call: {"body": {"agent": {"id": "agt_1", "account_id": "acct_1", "handle": "supplies", "name": "Supplies", "inbox_mailbox_id": "inbox_1"}, "actions": [], "platform": [], "models": []}} if call["path"] == "/api/v1/agents/self" else None,
         )
 
         async def work():