feat(api): manual updates

added datastores

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 3
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/agentbase%2Fagentbase-ef0ae966d97c74423a09430a8a0c419b5cb398b5f1c6e35ca01363ca3151d548.yml
-openapi_spec_hash: 46fa0bf1eb5482a17c4ddfe277253e9f
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/agentbase%2Fagentbase-b248631a2800e6d0f22a253717a63a551d9305e4c990dceb2f5c5361c519fa24.yml
+openapi_spec_hash: 4c06f6155fb614add9b30cd294266e11
 config_hash: 2ad1177ae9ac1a57dc61c0f7e142fc06

src/agentbase/_client.py

@@ -202,6 +202,7 @@ def run_agent(
         *,
         message: str,
         session: str | Omit = omit,
+        datastores: Iterable[client_run_agent_params.Datastore] | Omit = omit,
         mcp_servers: Iterable[client_run_agent_params.McpServer] | Omit = omit,
         mode: Literal["flash", "fast", "max"] | Omit = omit,
         rules: SequenceNotStr[str] | Omit = omit,
@@ -230,6 +231,9 @@ def run_agent(
           session: The session ID to continue the agent session conversation. If not provided, a
               new session will be created.
 
+          datastores: A set of datastores for the agent to utilize. Each object must include a `id`
+              and `name`.
+
           mcp_servers: A list of MCP server configurations. Each object must include a `serverName` and
               `serverUrl`.
 
@@ -256,6 +260,7 @@ def run_agent(
             body=maybe_transform(
                 {
                     "message": message,
+                    "datastores": datastores,
                     "mcp_servers": mcp_servers,
                     "mode": mode,
                     "rules": rules,
@@ -451,6 +456,7 @@ async def run_agent(
         *,
         message: str,
         session: str | Omit = omit,
+        datastores: Iterable[client_run_agent_params.Datastore] | Omit = omit,
         mcp_servers: Iterable[client_run_agent_params.McpServer] | Omit = omit,
         mode: Literal["flash", "fast", "max"] | Omit = omit,
         rules: SequenceNotStr[str] | Omit = omit,
@@ -479,6 +485,9 @@ async def run_agent(
           session: The session ID to continue the agent session conversation. If not provided, a
               new session will be created.
 
+          datastores: A set of datastores for the agent to utilize. Each object must include a `id`
+              and `name`.
+
           mcp_servers: A list of MCP server configurations. Each object must include a `serverName` and
               `serverUrl`.
 
@@ -505,6 +514,7 @@ async def run_agent(
             body=await async_maybe_transform(
                 {
                     "message": message,
+                    "datastores": datastores,
                     "mcp_servers": mcp_servers,
                     "mode": mode,
                     "rules": rules,

src/agentbase/types/client_run_agent_params.py

@@ -8,7 +8,7 @@
 from .._types import SequenceNotStr
 from .._utils import PropertyInfo
 
-__all__ = ["ClientRunAgentParams", "McpServer"]
+__all__ = ["ClientRunAgentParams", "Datastore", "McpServer"]
 
 
 class ClientRunAgentParams(TypedDict, total=False):
@@ -21,6 +21,12 @@ class ClientRunAgentParams(TypedDict, total=False):
     If not provided, a new session will be created.
     """
 
+    datastores: Iterable[Datastore]
+    """A set of datastores for the agent to utilize.
+
+    Each object must include a `id` and `name`.
+    """
+
     mcp_servers: Iterable[McpServer]
     """A list of MCP server configurations.
 
@@ -43,6 +49,14 @@ class ClientRunAgentParams(TypedDict, total=False):
     """A system prompt to provide system information to the agent."""
 
 
+class Datastore(TypedDict, total=False):
+    id: Required[str]
+    """The ID of the datastore."""
+
+    name: Required[str]
+    """The name of the datastore."""
+
+
 class McpServer(TypedDict, total=False):
     server_name: Required[Annotated[str, PropertyInfo(alias="serverName")]]
     """Name of the MCP server."""

tests/api_resources/test_client.py

@@ -29,6 +29,12 @@ def test_method_run_agent_with_all_params(self, client: Agentbase) -> None:
         client_stream = client.run_agent(
             message="message",
             session="session",
+            datastores=[
+                {
+                    "id": "id",
+                    "name": "name",
+                }
+            ],
             mcp_servers=[
                 {
                     "server_name": "serverName",
@@ -87,6 +93,12 @@ async def test_method_run_agent_with_all_params(self, async_client: AsyncAgentba
         client_stream = await async_client.run_agent(
             message="message",
             session="session",
+            datastores=[
+                {
+                    "id": "id",
+                    "name": "name",
+                }
+            ],
             mcp_servers=[
                 {
                     "server_name": "serverName",