feat: [REMOTE-1538] Support base model on named agents

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-439930897f625c088be1b09e394f4053e45060cd2daab7454dfcf6c775cba1fb.yml
-openapi_spec_hash: f0382bea9210d76f906665e8a825a05b
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/warp-bnavetta/warp-api-e752e75a35d88b84870729ef94b0c32783172983420cbff1b204ca14375553f7.yml
+openapi_spec_hash: 34787afc1e1c84a643431a0f0eb352ae
 config_hash: 5a6e285f6e3a958a887b31b972a3f49c

src/oz_agent_sdk/resources/agent/agent_.py

@@ -50,6 +50,7 @@ def create(
         self,
         *,
         name: str,
+        base_model: Optional[str] | Omit = omit,
         description: Optional[str] | Omit = omit,
         secrets: Iterable[agent_create_params.Secret] | Omit = omit,
         skills: SequenceNotStr[str] | Omit = omit,
@@ -68,6 +69,8 @@ def create(
         Args:
           name: A name for the agent
 
+          base_model: Optional base model for runs executed by this agent.
+
           description: Optional description of the agent
 
           secrets: Optional list of secrets associated with the agent. Duplicate names within a
@@ -94,6 +97,7 @@ def create(
             body=maybe_transform(
                 {
                     "name": name,
+                    "base_model": base_model,
                     "description": description,
                     "secrets": secrets,
                     "skills": skills,
@@ -110,6 +114,7 @@ def update(
         self,
         uid: str,
         *,
+        base_model: Optional[str] | Omit = omit,
         description: Optional[str] | Omit = omit,
         name: str | Omit = omit,
         secrets: Optional[Iterable[agent_update_params.Secret]] | Omit = omit,
@@ -124,9 +129,12 @@ def update(
         """Update an existing agent.
 
         Args:
-          description: Replacement description.
+          base_model: Replacement base model.
+
+        Omit or pass `null` to leave unchanged, or pass an empty
+              string to clear.
 
-        Omit or pass `null` to leave unchanged, or use an empty
+          description: Replacement description. Omit or pass `null` to leave unchanged, or use an empty
               value to clear.
 
           name: The new name for the agent
@@ -151,6 +159,7 @@ def update(
             path_template("/agent/identities/{uid}", uid=uid),
             body=maybe_transform(
                 {
+                    "base_model": base_model,
                     "description": description,
                     "name": name,
                     "secrets": secrets,
@@ -285,6 +294,7 @@ async def create(
         self,
         *,
         name: str,
+        base_model: Optional[str] | Omit = omit,
         description: Optional[str] | Omit = omit,
         secrets: Iterable[agent_create_params.Secret] | Omit = omit,
         skills: SequenceNotStr[str] | Omit = omit,
@@ -303,6 +313,8 @@ async def create(
         Args:
           name: A name for the agent
 
+          base_model: Optional base model for runs executed by this agent.
+
           description: Optional description of the agent
 
           secrets: Optional list of secrets associated with the agent. Duplicate names within a
@@ -329,6 +341,7 @@ async def create(
             body=await async_maybe_transform(
                 {
                     "name": name,
+                    "base_model": base_model,
                     "description": description,
                     "secrets": secrets,
                     "skills": skills,
@@ -345,6 +358,7 @@ async def update(
         self,
         uid: str,
         *,
+        base_model: Optional[str] | Omit = omit,
         description: Optional[str] | Omit = omit,
         name: str | Omit = omit,
         secrets: Optional[Iterable[agent_update_params.Secret]] | Omit = omit,
@@ -359,9 +373,12 @@ async def update(
         """Update an existing agent.
 
         Args:
-          description: Replacement description.
+          base_model: Replacement base model.
+
+        Omit or pass `null` to leave unchanged, or pass an empty
+              string to clear.
 
-        Omit or pass `null` to leave unchanged, or use an empty
+          description: Replacement description. Omit or pass `null` to leave unchanged, or use an empty
               value to clear.
 
           name: The new name for the agent
@@ -386,6 +403,7 @@ async def update(
             path_template("/agent/identities/{uid}", uid=uid),
             body=await async_maybe_transform(
                 {
+                    "base_model": base_model,
                     "description": description,
                     "name": name,
                     "secrets": secrets,

src/oz_agent_sdk/types/agent/agent_create_params.py

@@ -14,6 +14,9 @@ class AgentCreateParams(TypedDict, total=False):
     name: Required[str]
     """A name for the agent"""
 
+    base_model: Optional[str]
+    """Optional base model for runs executed by this agent."""
+
     description: Optional[str]
     """Optional description of the agent"""
 

src/oz_agent_sdk/types/agent/agent_response.py

@@ -37,5 +37,15 @@ class AgentResponse(BaseModel):
     uid: str
     """Unique identifier for the agent"""
 
+    base_model: Optional[str] = None
+    """Base model for runs executed by this agent.
+
+    The precedence order for model resolution is:
+
+    1. The model specified on the run itself
+    2. The agent's base model
+    3. The team's default model
+    """
+
     description: Optional[str] = None
     """Optional description of the agent"""

src/oz_agent_sdk/types/agent/agent_update_params.py

@@ -11,6 +11,12 @@
 
 
 class AgentUpdateParams(TypedDict, total=False):
+    base_model: Optional[str]
+    """Replacement base model.
+
+    Omit or pass `null` to leave unchanged, or pass an empty string to clear.
+    """
+
     description: Optional[str]
     """Replacement description.
 

tests/api_resources/agent/test_agent_.py

@@ -33,6 +33,7 @@ def test_method_create(self, client: OzAPI) -> None:
     def test_method_create_with_all_params(self, client: OzAPI) -> None:
         agent = client.agent.agent.create(
             name="name",
+            base_model="base_model",
             description="description",
             secrets=[{"name": "name"}],
             skills=["string"],
@@ -78,6 +79,7 @@ def test_method_update(self, client: OzAPI) -> None:
     def test_method_update_with_all_params(self, client: OzAPI) -> None:
         agent = client.agent.agent.update(
             uid="uid",
+            base_model="base_model",
             description="description",
             name="name",
             secrets=[{"name": "name"}],
@@ -250,6 +252,7 @@ async def test_method_create(self, async_client: AsyncOzAPI) -> None:
     async def test_method_create_with_all_params(self, async_client: AsyncOzAPI) -> None:
         agent = await async_client.agent.agent.create(
             name="name",
+            base_model="base_model",
             description="description",
             secrets=[{"name": "name"}],
             skills=["string"],
@@ -295,6 +298,7 @@ async def test_method_update(self, async_client: AsyncOzAPI) -> None:
     async def test_method_update_with_all_params(self, async_client: AsyncOzAPI) -> None:
         agent = await async_client.agent.agent.update(
             uid="uid",
+            base_model="base_model",
             description="description",
             name="name",
             secrets=[{"name": "name"}],