feat(memory): agent identity memory store attachments — API layer

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 23
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/warp-bnavetta/warp-api-e752e75a35d88b84870729ef94b0c32783172983420cbff1b204ca14375553f7.yml
-openapi_spec_hash: 34787afc1e1c84a643431a0f0eb352ae
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/warp-bnavetta/warp-api-2783b07289e9f71f07550497c5180824f3dbc92477b5d1d8a80fa6dce4e3b8b6.yml
+openapi_spec_hash: e1d4cff965beed1ec255bf387d55d940
 config_hash: 5a6e285f6e3a958a887b31b972a3f49c

src/oz_agent_sdk/resources/agent/agent_.py

@@ -52,6 +52,7 @@ def create(
         name: str,
         base_model: Optional[str] | Omit = omit,
         description: Optional[str] | Omit = omit,
+        memory_stores: Iterable[agent_create_params.MemoryStore] | Omit = omit,
         secrets: Iterable[agent_create_params.Secret] | Omit = omit,
         skills: SequenceNotStr[str] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -73,6 +74,10 @@ def create(
 
           description: Optional description of the agent
 
+          memory_stores: Optional list of memory stores to attach to the agent. Each store must be
+              team-owned by the same team as the agent. Duplicate UIDs within a single request
+              are rejected.
+
           secrets: Optional list of secrets associated with the agent. Duplicate names within a
               single request are rejected. Each entry is unioned into the run-time secret
               scope when the agent executes.
@@ -99,6 +104,7 @@ def create(
                     "name": name,
                     "base_model": base_model,
                     "description": description,
+                    "memory_stores": memory_stores,
                     "secrets": secrets,
                     "skills": skills,
                 },
@@ -116,6 +122,7 @@ def update(
         *,
         base_model: Optional[str] | Omit = omit,
         description: Optional[str] | Omit = omit,
+        memory_stores: Optional[Iterable[agent_update_params.MemoryStore]] | Omit = omit,
         name: str | Omit = omit,
         secrets: Optional[Iterable[agent_update_params.Secret]] | Omit = omit,
         skills: Optional[SequenceNotStr[str]] | Omit = omit,
@@ -137,6 +144,9 @@ def update(
           description: Replacement description. Omit or pass `null` to leave unchanged, or use an empty
               value to clear.
 
+          memory_stores: Replacement list of memory stores. Omit to leave unchanged, pass an empty array
+              to clear, or pass a non-empty array to replace.
+
           name: The new name for the agent
 
           secrets: Replacement list of secrets. Omit to leave unchanged, pass an empty array to
@@ -161,6 +171,7 @@ def update(
                 {
                     "base_model": base_model,
                     "description": description,
+                    "memory_stores": memory_stores,
                     "name": name,
                     "secrets": secrets,
                     "skills": skills,
@@ -296,6 +307,7 @@ async def create(
         name: str,
         base_model: Optional[str] | Omit = omit,
         description: Optional[str] | Omit = omit,
+        memory_stores: Iterable[agent_create_params.MemoryStore] | Omit = omit,
         secrets: Iterable[agent_create_params.Secret] | Omit = omit,
         skills: SequenceNotStr[str] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -317,6 +329,10 @@ async def create(
 
           description: Optional description of the agent
 
+          memory_stores: Optional list of memory stores to attach to the agent. Each store must be
+              team-owned by the same team as the agent. Duplicate UIDs within a single request
+              are rejected.
+
           secrets: Optional list of secrets associated with the agent. Duplicate names within a
               single request are rejected. Each entry is unioned into the run-time secret
               scope when the agent executes.
@@ -343,6 +359,7 @@ async def create(
                     "name": name,
                     "base_model": base_model,
                     "description": description,
+                    "memory_stores": memory_stores,
                     "secrets": secrets,
                     "skills": skills,
                 },
@@ -360,6 +377,7 @@ async def update(
         *,
         base_model: Optional[str] | Omit = omit,
         description: Optional[str] | Omit = omit,
+        memory_stores: Optional[Iterable[agent_update_params.MemoryStore]] | Omit = omit,
         name: str | Omit = omit,
         secrets: Optional[Iterable[agent_update_params.Secret]] | Omit = omit,
         skills: Optional[SequenceNotStr[str]] | Omit = omit,
@@ -381,6 +399,9 @@ async def update(
           description: Replacement description. Omit or pass `null` to leave unchanged, or use an empty
               value to clear.
 
+          memory_stores: Replacement list of memory stores. Omit to leave unchanged, pass an empty array
+              to clear, or pass a non-empty array to replace.
+
           name: The new name for the agent
 
           secrets: Replacement list of secrets. Omit to leave unchanged, pass an empty array to
@@ -405,6 +426,7 @@ async def update(
                 {
                     "base_model": base_model,
                     "description": description,
+                    "memory_stores": memory_stores,
                     "name": name,
                     "secrets": secrets,
                     "skills": skills,

src/oz_agent_sdk/types/agent/agent_create_params.py

@@ -3,11 +3,11 @@
 from __future__ import annotations
 
 from typing import Iterable, Optional
-from typing_extensions import Required, TypedDict
+from typing_extensions import Literal, Required, TypedDict
 
 from ..._types import SequenceNotStr
 
-__all__ = ["AgentCreateParams", "Secret"]
+__all__ = ["AgentCreateParams", "MemoryStore", "Secret"]
 
 
 class AgentCreateParams(TypedDict, total=False):
@@ -20,6 +20,13 @@ class AgentCreateParams(TypedDict, total=False):
     description: Optional[str]
     """Optional description of the agent"""
 
+    memory_stores: Iterable[MemoryStore]
+    """
+    Optional list of memory stores to attach to the agent. Each store must be
+    team-owned by the same team as the agent. Duplicate UIDs within a single request
+    are rejected.
+    """
+
     secrets: Iterable[Secret]
     """
     Optional list of secrets associated with the agent. Duplicate names within a
@@ -37,6 +44,19 @@ class AgentCreateParams(TypedDict, total=False):
     """
 
 
+class MemoryStore(TypedDict, total=False):
+    """Reference to a memory store to attach to an agent."""
+
+    access: Required[Literal["read_write", "read_only"]]
+    """Access level for the store."""
+
+    instructions: Required[str]
+    """Instructions for how the agent should use this memory store. Must not be empty."""
+
+    uid: Required[str]
+    """UID of the memory store."""
+
+
 class Secret(TypedDict, total=False):
     """Reference to a managed secret by name."""
 

src/oz_agent_sdk/types/agent/agent_response.py

@@ -2,10 +2,24 @@
 
 from typing import List, Optional
 from datetime import datetime
+from typing_extensions import Literal
 
 from ..._models import BaseModel
 
-__all__ = ["AgentResponse", "Secret"]
+__all__ = ["AgentResponse", "MemoryStore", "Secret"]
+
+
+class MemoryStore(BaseModel):
+    """Reference to a memory store to attach to an agent."""
+
+    access: Literal["read_write", "read_only"]
+    """Access level for the store."""
+
+    instructions: str
+    """Instructions for how the agent should use this memory store. Must not be empty."""
+
+    uid: str
+    """UID of the memory store."""
 
 
 class Secret(BaseModel):
@@ -22,6 +36,12 @@ class AgentResponse(BaseModel):
     created_at: datetime
     """When the agent was created (RFC3339)"""
 
+    memory_stores: List[MemoryStore]
+    """
+    Memory stores attached to this agent. Always present; empty when no stores are
+    attached.
+    """
+
     name: str
     """Name of the agent"""
 

src/oz_agent_sdk/types/agent/agent_update_params.py

@@ -3,11 +3,11 @@
 from __future__ import annotations
 
 from typing import Iterable, Optional
-from typing_extensions import Required, TypedDict
+from typing_extensions import Literal, Required, TypedDict
 
 from ..._types import SequenceNotStr
 
-__all__ = ["AgentUpdateParams", "Secret"]
+__all__ = ["AgentUpdateParams", "MemoryStore", "Secret"]
 
 
 class AgentUpdateParams(TypedDict, total=False):
@@ -23,6 +23,13 @@ class AgentUpdateParams(TypedDict, total=False):
     Omit or pass `null` to leave unchanged, or use an empty value to clear.
     """
 
+    memory_stores: Optional[Iterable[MemoryStore]]
+    """Replacement list of memory stores.
+
+    Omit to leave unchanged, pass an empty array to clear, or pass a non-empty array
+    to replace.
+    """
+
     name: str
     """The new name for the agent"""
 
@@ -41,6 +48,19 @@ class AgentUpdateParams(TypedDict, total=False):
     """
 
 
+class MemoryStore(TypedDict, total=False):
+    """Reference to a memory store to attach to an agent."""
+
+    access: Required[Literal["read_write", "read_only"]]
+    """Access level for the store."""
+
+    instructions: Required[str]
+    """Instructions for how the agent should use this memory store. Must not be empty."""
+
+    uid: Required[str]
+    """UID of the memory store."""
+
+
 class Secret(TypedDict, total=False):
     """Reference to a managed secret by name."""
 

tests/api_resources/agent/test_agent_.py

@@ -35,6 +35,13 @@ def test_method_create_with_all_params(self, client: OzAPI) -> None:
             name="name",
             base_model="base_model",
             description="description",
+            memory_stores=[
+                {
+                    "access": "read_write",
+                    "instructions": "instructions",
+                    "uid": "uid",
+                }
+            ],
             secrets=[{"name": "name"}],
             skills=["string"],
         )
@@ -81,6 +88,13 @@ def test_method_update_with_all_params(self, client: OzAPI) -> None:
             uid="uid",
             base_model="base_model",
             description="description",
+            memory_stores=[
+                {
+                    "access": "read_write",
+                    "instructions": "instructions",
+                    "uid": "uid",
+                }
+            ],
             name="name",
             secrets=[{"name": "name"}],
             skills=["string"],
@@ -254,6 +268,13 @@ async def test_method_create_with_all_params(self, async_client: AsyncOzAPI) ->
             name="name",
             base_model="base_model",
             description="description",
+            memory_stores=[
+                {
+                    "access": "read_write",
+                    "instructions": "instructions",
+                    "uid": "uid",
+                }
+            ],
             secrets=[{"name": "name"}],
             skills=["string"],
         )
@@ -300,6 +321,13 @@ async def test_method_update_with_all_params(self, async_client: AsyncOzAPI) ->
             uid="uid",
             base_model="base_model",
             description="description",
+            memory_stores=[
+                {
+                    "access": "read_write",
+                    "instructions": "instructions",
+                    "uid": "uid",
+                }
+            ],
             name="name",
             secrets=[{"name": "name"}],
             skills=["string"],