@@ -202,6 +202,7 @@ def run_agent(
202202 * ,
203203 message : str ,
204204 session : str | Omit = omit ,
205+ datastores : Iterable [client_run_agent_params .Datastore ] | Omit = omit ,
205206 mcp_servers : Iterable [client_run_agent_params .McpServer ] | Omit = omit ,
206207 mode : Literal ["flash" , "fast" , "max" ] | Omit = omit ,
207208 rules : SequenceNotStr [str ] | Omit = omit ,
@@ -230,6 +231,9 @@ def run_agent(
230231 session: The session ID to continue the agent session conversation. If not provided, a
231232 new session will be created.
232233
234+ datastores: A set of datastores for the agent to utilize. Each object must include a `id`
235+ and `name`.
236+
233237 mcp_servers: A list of MCP server configurations. Each object must include a `serverName` and
234238 `serverUrl`.
235239
@@ -256,6 +260,7 @@ def run_agent(
256260 body = maybe_transform (
257261 {
258262 "message" : message ,
263+ "datastores" : datastores ,
259264 "mcp_servers" : mcp_servers ,
260265 "mode" : mode ,
261266 "rules" : rules ,
@@ -451,6 +456,7 @@ async def run_agent(
451456 * ,
452457 message : str ,
453458 session : str | Omit = omit ,
459+ datastores : Iterable [client_run_agent_params .Datastore ] | Omit = omit ,
454460 mcp_servers : Iterable [client_run_agent_params .McpServer ] | Omit = omit ,
455461 mode : Literal ["flash" , "fast" , "max" ] | Omit = omit ,
456462 rules : SequenceNotStr [str ] | Omit = omit ,
@@ -479,6 +485,9 @@ async def run_agent(
479485 session: The session ID to continue the agent session conversation. If not provided, a
480486 new session will be created.
481487
488+ datastores: A set of datastores for the agent to utilize. Each object must include a `id`
489+ and `name`.
490+
482491 mcp_servers: A list of MCP server configurations. Each object must include a `serverName` and
483492 `serverUrl`.
484493
@@ -505,6 +514,7 @@ async def run_agent(
505514 body = await async_maybe_transform (
506515 {
507516 "message" : message ,
517+ "datastores" : datastores ,
508518 "mcp_servers" : mcp_servers ,
509519 "mode" : mode ,
510520 "rules" : rules ,
0 commit comments