Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion .release-please-manifest.json
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
{
".": "0.2.0"
".": "0.3.0"
}
4 changes: 2 additions & 2 deletions .stats.yml
Original file line number Diff line number Diff line change
@@ -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
8 changes: 8 additions & 0 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
@@ -1,5 +1,13 @@
# Changelog

## 0.3.0 (2025-10-03)

Full Changelog: [v0.2.0...v0.3.0](https://github.com/AgentbaseHQ/agentbase-python/compare/v0.2.0...v0.3.0)

### Features

* **api:** manual updates ([1356607](https://github.com/AgentbaseHQ/agentbase-python/commit/1356607bd114f47701ad219980afd39ad510c07f))

## 0.2.0 (2025-10-03)

Full Changelog: [v0.1.1...v0.2.0](https://github.com/AgentbaseHQ/agentbase-python/compare/v0.1.1...v0.2.0)
Expand Down
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
[project]
name = "agentbase-sdk"
version = "0.2.0"
version = "0.3.0"
description = "The official Python library for the agentbase API"
dynamic = ["readme"]
license = "Apache-2.0"
Expand Down
10 changes: 10 additions & 0 deletions src/agentbase/_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down Expand Up @@ -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`.

Expand All @@ -256,6 +260,7 @@ def run_agent(
body=maybe_transform(
{
"message": message,
"datastores": datastores,
"mcp_servers": mcp_servers,
"mode": mode,
"rules": rules,
Expand Down Expand Up @@ -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,
Expand Down Expand Up @@ -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`.

Expand All @@ -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,
Expand Down
2 changes: 1 addition & 1 deletion src/agentbase/_version.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.

__title__ = "agentbase"
__version__ = "0.2.0" # x-release-please-version
__version__ = "0.3.0" # x-release-please-version
16 changes: 15 additions & 1 deletion src/agentbase/types/client_run_agent_params.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
from .._types import SequenceNotStr
from .._utils import PropertyInfo

__all__ = ["ClientRunAgentParams", "McpServer"]
__all__ = ["ClientRunAgentParams", "Datastore", "McpServer"]


class ClientRunAgentParams(TypedDict, total=False):
Expand All @@ -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.

Expand All @@ -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."""
Expand Down
12 changes: 12 additions & 0 deletions tests/api_resources/test_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down Expand Up @@ -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",
Expand Down