letta-client 1.0.0a11__py3-none-any.whl → 1.0.0a12__py3-none-any.whl

letta_client/_client.py

@@ -49,6 +49,7 @@ from .resources.batches import batches
 from .resources.folders import folders
 from .resources.templates import templates
 from .resources.identities import identities
+from .resources.mcp_servers import mcp_servers
 from .types.health_response import HealthResponse
 
 __all__ = [
@@ -83,11 +84,13 @@ class Letta(SyncAPIClient):
     tags: tags.TagsResource
     batches: batches.BatchesResource
     templates: templates.TemplatesResource
+    mcp_servers: mcp_servers.McpServersResource
     with_raw_response: LettaWithRawResponse
     with_streaming_response: LettaWithStreamedResponse
 
     # client options
     api_key: str
+    project_id: str | None
 
     _environment: Literal["cloud", "local"] | NotGiven
 
@@ -95,6 +98,7 @@ class Letta(SyncAPIClient):
         self,
         *,
         api_key: str | None = None,
+        project_id: str | None = None,
         environment: Literal["cloud", "local"] | NotGiven = not_given,
         base_url: str | httpx.URL | None | NotGiven = not_given,
         timeout: float | Timeout | None | NotGiven = not_given,
@@ -127,6 +131,8 @@ class Letta(SyncAPIClient):
             )
         self.api_key = api_key
 
+        self.project_id = project_id
+
         self._environment = environment
 
         base_url_env = os.environ.get("LETTA_BASE_URL")
@@ -179,6 +185,7 @@ class Letta(SyncAPIClient):
         self.tags = tags.TagsResource(self)
         self.batches = batches.BatchesResource(self)
         self.templates = templates.TemplatesResource(self)
+        self.mcp_servers = mcp_servers.McpServersResource(self)
         self.with_raw_response = LettaWithRawResponse(self)
         self.with_streaming_response = LettaWithStreamedResponse(self)
 
@@ -199,6 +206,7 @@ class Letta(SyncAPIClient):
         return {
             **super().default_headers,
             "X-Stainless-Async": "false",
+            "X-Project": self.project_id if self.project_id is not None else Omit(),
             **self._custom_headers,
         }
 
@@ -206,6 +214,7 @@ class Letta(SyncAPIClient):
         self,
         *,
         api_key: str | None = None,
+        project_id: str | None = None,
         environment: Literal["cloud", "local"] | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: float | Timeout | None | NotGiven = not_given,
@@ -241,6 +250,7 @@ class Letta(SyncAPIClient):
         http_client = http_client or self._client
         return self.__class__(
             api_key=api_key or self.api_key,
+            project_id=project_id or self.project_id,
             base_url=base_url or self.base_url,
             environment=environment or self._environment,
             timeout=self.timeout if isinstance(timeout, NotGiven) else timeout,
@@ -322,11 +332,13 @@ class AsyncLetta(AsyncAPIClient):
     tags: tags.AsyncTagsResource
     batches: batches.AsyncBatchesResource
     templates: templates.AsyncTemplatesResource
+    mcp_servers: mcp_servers.AsyncMcpServersResource
     with_raw_response: AsyncLettaWithRawResponse
     with_streaming_response: AsyncLettaWithStreamedResponse
 
     # client options
     api_key: str
+    project_id: str | None
 
     _environment: Literal["cloud", "local"] | NotGiven
 
@@ -334,6 +346,7 @@ class AsyncLetta(AsyncAPIClient):
         self,
         *,
         api_key: str | None = None,
+        project_id: str | None = None,
         environment: Literal["cloud", "local"] | NotGiven = not_given,
         base_url: str | httpx.URL | None | NotGiven = not_given,
         timeout: float | Timeout | None | NotGiven = not_given,
@@ -366,6 +379,8 @@ class AsyncLetta(AsyncAPIClient):
             )
         self.api_key = api_key
 
+        self.project_id = project_id
+
         self._environment = environment
 
         base_url_env = os.environ.get("LETTA_BASE_URL")
@@ -418,6 +433,7 @@ class AsyncLetta(AsyncAPIClient):
         self.tags = tags.AsyncTagsResource(self)
         self.batches = batches.AsyncBatchesResource(self)
         self.templates = templates.AsyncTemplatesResource(self)
+        self.mcp_servers = mcp_servers.AsyncMcpServersResource(self)
         self.with_raw_response = AsyncLettaWithRawResponse(self)
         self.with_streaming_response = AsyncLettaWithStreamedResponse(self)
 
@@ -438,6 +454,7 @@ class AsyncLetta(AsyncAPIClient):
         return {
             **super().default_headers,
             "X-Stainless-Async": f"async:{get_async_library()}",
+            "X-Project": self.project_id if self.project_id is not None else Omit(),
             **self._custom_headers,
         }
 
@@ -445,6 +462,7 @@ class AsyncLetta(AsyncAPIClient):
         self,
         *,
         api_key: str | None = None,
+        project_id: str | None = None,
         environment: Literal["cloud", "local"] | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: float | Timeout | None | NotGiven = not_given,
@@ -480,6 +498,7 @@ class AsyncLetta(AsyncAPIClient):
         http_client = http_client or self._client
         return self.__class__(
             api_key=api_key or self.api_key,
+            project_id=project_id or self.project_id,
             base_url=base_url or self.base_url,
             environment=environment or self._environment,
             timeout=self.timeout if isinstance(timeout, NotGiven) else timeout,
@@ -562,6 +581,7 @@ class LettaWithRawResponse:
         self.tags = tags.TagsResourceWithRawResponse(client.tags)
         self.batches = batches.BatchesResourceWithRawResponse(client.batches)
         self.templates = templates.TemplatesResourceWithRawResponse(client.templates)
+        self.mcp_servers = mcp_servers.McpServersResourceWithRawResponse(client.mcp_servers)
 
         self.health = to_raw_response_wrapper(
             client.health,
@@ -583,6 +603,7 @@ class AsyncLettaWithRawResponse:
         self.tags = tags.AsyncTagsResourceWithRawResponse(client.tags)
         self.batches = batches.AsyncBatchesResourceWithRawResponse(client.batches)
         self.templates = templates.AsyncTemplatesResourceWithRawResponse(client.templates)
+        self.mcp_servers = mcp_servers.AsyncMcpServersResourceWithRawResponse(client.mcp_servers)
 
         self.health = async_to_raw_response_wrapper(
             client.health,
@@ -604,6 +625,7 @@ class LettaWithStreamedResponse:
         self.tags = tags.TagsResourceWithStreamingResponse(client.tags)
         self.batches = batches.BatchesResourceWithStreamingResponse(client.batches)
         self.templates = templates.TemplatesResourceWithStreamingResponse(client.templates)
+        self.mcp_servers = mcp_servers.McpServersResourceWithStreamingResponse(client.mcp_servers)
 
         self.health = to_streamed_response_wrapper(
             client.health,
@@ -625,6 +647,7 @@ class AsyncLettaWithStreamedResponse:
         self.tags = tags.AsyncTagsResourceWithStreamingResponse(client.tags)
         self.batches = batches.AsyncBatchesResourceWithStreamingResponse(client.batches)
         self.templates = templates.AsyncTemplatesResourceWithStreamingResponse(client.templates)
+        self.mcp_servers = mcp_servers.AsyncMcpServersResourceWithStreamingResponse(client.mcp_servers)
 
         self.health = async_to_streamed_response_wrapper(
             client.health,

letta_client/_streaming.py

@@ -76,9 +76,8 @@ class Stream(Generic[_T]):
             if sse.event is None:
                 yield process_data(data=sse.json(), cast_to=cast_to, response=response)
 
-        # Ensure the entire stream is consumed
-        for _sse in iterator:
-            ...
+        # As we might not fully consume the response stream, we need to close it explicitly
+        response.close()
 
     def __enter__(self) -> Self:
         return self
@@ -159,9 +158,8 @@ class AsyncStream(Generic[_T]):
             if sse.event is None:
                 yield process_data(data=sse.json(), cast_to=cast_to, response=response)
 
-        # Ensure the entire stream is consumed
-        async for _sse in iterator:
-            ...
+        # As we might not fully consume the response stream, we need to close it explicitly
+        await response.aclose()
 
     async def __aenter__(self) -> Self:
         return self

letta_client/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "letta_client"
-__version__ = "1.0.0-alpha.11"  # x-release-please-version
+__version__ = "1.0.0-alpha.12"  # x-release-please-version

letta_client/resources/__init__.py

@@ -104,6 +104,14 @@ from .identities import (
     IdentitiesResourceWithStreamingResponse,
     AsyncIdentitiesResourceWithStreamingResponse,
 )
+from .mcp_servers import (
+    McpServersResource,
+    AsyncMcpServersResource,
+    McpServersResourceWithRawResponse,
+    AsyncMcpServersResourceWithRawResponse,
+    McpServersResourceWithStreamingResponse,
+    AsyncMcpServersResourceWithStreamingResponse,
+)
 
 __all__ = [
     "ArchivesResource",
@@ -184,4 +192,10 @@ __all__ = [
     "AsyncTemplatesResourceWithRawResponse",
     "TemplatesResourceWithStreamingResponse",
     "AsyncTemplatesResourceWithStreamingResponse",
+    "McpServersResource",
+    "AsyncMcpServersResource",
+    "McpServersResourceWithRawResponse",
+    "AsyncMcpServersResourceWithRawResponse",
+    "McpServersResourceWithStreamingResponse",
+    "AsyncMcpServersResourceWithStreamingResponse",
 ]

letta_client/resources/agents/agents.py

@@ -191,7 +191,6 @@ class AgentsResource(SyncAPIResource):
         tool_ids: Optional[SequenceNotStr[str]] | Omit = omit,
         tool_rules: Optional[Iterable[agent_create_params.ToolRule]] | Omit = omit,
         tools: Optional[SequenceNotStr[str]] | Omit = omit,
-        x_project: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -306,8 +305,6 @@ class AgentsResource(SyncAPIResource):
 
           tools: The tools used by the agent.
 
-          x_project: The project slug to associate with the agent (cloud only).
-
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -316,7 +313,6 @@ class AgentsResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        extra_headers = {**strip_not_given({"X-Project": x_project}), **(extra_headers or {})}
         return self._post(
             "/v1/agents/",
             body=maybe_transform(
@@ -1026,7 +1022,6 @@ class AsyncAgentsResource(AsyncAPIResource):
         tool_ids: Optional[SequenceNotStr[str]] | Omit = omit,
         tool_rules: Optional[Iterable[agent_create_params.ToolRule]] | Omit = omit,
         tools: Optional[SequenceNotStr[str]] | Omit = omit,
-        x_project: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -1141,8 +1136,6 @@ class AsyncAgentsResource(AsyncAPIResource):
 
           tools: The tools used by the agent.
 
-          x_project: The project slug to associate with the agent (cloud only).
-
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -1151,7 +1144,6 @@ class AsyncAgentsResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        extra_headers = {**strip_not_given({"X-Project": x_project}), **(extra_headers or {})}
         return await self._post(
             "/v1/agents/",
             body=await async_maybe_transform(

letta_client/resources/groups/groups.py

@@ -9,7 +9,7 @@ import httpx
 
 from ...types import ManagerType, group_list_params, group_create_params, group_modify_params
 from ..._types import Body, Omit, Query, Headers, NotGiven, SequenceNotStr, omit, not_given
-from ..._utils import maybe_transform, strip_not_given, async_maybe_transform
+from ..._utils import maybe_transform, async_maybe_transform
 from .messages import (
     MessagesResource,
     AsyncMessagesResource,
@@ -68,7 +68,6 @@ class GroupsResource(SyncAPIResource):
         manager_config: group_create_params.ManagerConfig | Omit = omit,
         project_id: Optional[str] | Omit = omit,
         shared_block_ids: SequenceNotStr[str] | Omit = omit,
-        x_project: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -92,8 +91,6 @@ class GroupsResource(SyncAPIResource):
 
           shared_block_ids
 
-          x_project: The project slug to associate with the group (cloud only).
-
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -102,7 +99,6 @@ class GroupsResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        extra_headers = {**strip_not_given({"X-Project": x_project}), **(extra_headers or {})}
         return self._post(
             "/v1/groups/",
             body=maybe_transform(
@@ -290,7 +286,6 @@ class GroupsResource(SyncAPIResource):
         manager_config: Optional[group_modify_params.ManagerConfig] | Omit = omit,
         project_id: Optional[str] | Omit = omit,
         shared_block_ids: Optional[SequenceNotStr[str]] | Omit = omit,
-        x_project: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -314,8 +309,6 @@ class GroupsResource(SyncAPIResource):
 
           shared_block_ids
 
-          x_project: The project slug to associate with the group (cloud only).
-
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -326,7 +319,6 @@ class GroupsResource(SyncAPIResource):
         """
         if not group_id:
             raise ValueError(f"Expected a non-empty value for `group_id` but received {group_id!r}")
-        extra_headers = {**strip_not_given({"X-Project": x_project}), **(extra_headers or {})}
         return self._patch(
             f"/v1/groups/{group_id}",
             body=maybe_transform(
@@ -379,7 +371,6 @@ class AsyncGroupsResource(AsyncAPIResource):
         manager_config: group_create_params.ManagerConfig | Omit = omit,
         project_id: Optional[str] | Omit = omit,
         shared_block_ids: SequenceNotStr[str] | Omit = omit,
-        x_project: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -403,8 +394,6 @@ class AsyncGroupsResource(AsyncAPIResource):
 
           shared_block_ids
 
-          x_project: The project slug to associate with the group (cloud only).
-
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -413,7 +402,6 @@ class AsyncGroupsResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        extra_headers = {**strip_not_given({"X-Project": x_project}), **(extra_headers or {})}
         return await self._post(
             "/v1/groups/",
             body=await async_maybe_transform(
@@ -601,7 +589,6 @@ class AsyncGroupsResource(AsyncAPIResource):
         manager_config: Optional[group_modify_params.ManagerConfig] | Omit = omit,
         project_id: Optional[str] | Omit = omit,
         shared_block_ids: Optional[SequenceNotStr[str]] | Omit = omit,
-        x_project: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -625,8 +612,6 @@ class AsyncGroupsResource(AsyncAPIResource):
 
           shared_block_ids
 
-          x_project: The project slug to associate with the group (cloud only).
-
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -637,7 +622,6 @@ class AsyncGroupsResource(AsyncAPIResource):
         """
         if not group_id:
             raise ValueError(f"Expected a non-empty value for `group_id` but received {group_id!r}")
-        extra_headers = {**strip_not_given({"X-Project": x_project}), **(extra_headers or {})}
         return await self._patch(
             f"/v1/groups/{group_id}",
             body=await async_maybe_transform(

letta_client/resources/identities/identities.py

@@ -31,7 +31,7 @@ from ...types import (
     identity_upsert_params,
 )
 from ..._types import Body, Omit, Query, Headers, NotGiven, SequenceNotStr, omit, not_given
-from ..._utils import maybe_transform, strip_not_given, async_maybe_transform
+from ..._utils import maybe_transform, async_maybe_transform
 from ..._compat import cached_property
 from .properties import (
     PropertiesResource,
@@ -100,7 +100,6 @@ class IdentitiesResource(SyncAPIResource):
         block_ids: Optional[SequenceNotStr[str]] | Omit = omit,
         project_id: Optional[str] | Omit = omit,
         properties: Optional[Iterable[IdentityPropertyParam]] | Omit = omit,
-        x_project: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -126,8 +125,6 @@ class IdentitiesResource(SyncAPIResource):
 
           properties: List of properties associated with the identity.
 
-          x_project: The project slug to associate with the identity (cloud only).
-
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -136,7 +133,6 @@ class IdentitiesResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        extra_headers = {**strip_not_given({"X-Project": x_project}), **(extra_headers or {})}
         return self._post(
             "/v1/identities/",
             body=maybe_transform(
@@ -391,7 +387,6 @@ class IdentitiesResource(SyncAPIResource):
         block_ids: Optional[SequenceNotStr[str]] | Omit = omit,
         project_id: Optional[str] | Omit = omit,
         properties: Optional[Iterable[IdentityPropertyParam]] | Omit = omit,
-        x_project: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -417,8 +412,6 @@ class IdentitiesResource(SyncAPIResource):
 
           properties: List of properties associated with the identity.
 
-          x_project: The project slug to associate with the identity (cloud only).
-
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -427,7 +420,6 @@ class IdentitiesResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        extra_headers = {**strip_not_given({"X-Project": x_project}), **(extra_headers or {})}
         return self._put(
             "/v1/identities/",
             body=maybe_transform(
@@ -491,7 +483,6 @@ class AsyncIdentitiesResource(AsyncAPIResource):
         block_ids: Optional[SequenceNotStr[str]] | Omit = omit,
         project_id: Optional[str] | Omit = omit,
         properties: Optional[Iterable[IdentityPropertyParam]] | Omit = omit,
-        x_project: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -517,8 +508,6 @@ class AsyncIdentitiesResource(AsyncAPIResource):
 
           properties: List of properties associated with the identity.
 
-          x_project: The project slug to associate with the identity (cloud only).
-
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -527,7 +516,6 @@ class AsyncIdentitiesResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        extra_headers = {**strip_not_given({"X-Project": x_project}), **(extra_headers or {})}
         return await self._post(
             "/v1/identities/",
             body=await async_maybe_transform(
@@ -782,7 +770,6 @@ class AsyncIdentitiesResource(AsyncAPIResource):
         block_ids: Optional[SequenceNotStr[str]] | Omit = omit,
         project_id: Optional[str] | Omit = omit,
         properties: Optional[Iterable[IdentityPropertyParam]] | Omit = omit,
-        x_project: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -808,8 +795,6 @@ class AsyncIdentitiesResource(AsyncAPIResource):
 
           properties: List of properties associated with the identity.
 
-          x_project: The project slug to associate with the identity (cloud only).
-
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -818,7 +803,6 @@ class AsyncIdentitiesResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        extra_headers = {**strip_not_given({"X-Project": x_project}), **(extra_headers or {})}
         return await self._put(
             "/v1/identities/",
             body=await async_maybe_transform(

letta_client/resources/mcp_servers/__init__.py

@@ -0,0 +1,47 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from .tools import (
+    ToolsResource,
+    AsyncToolsResource,
+    ToolsResourceWithRawResponse,
+    AsyncToolsResourceWithRawResponse,
+    ToolsResourceWithStreamingResponse,
+    AsyncToolsResourceWithStreamingResponse,
+)
+from .refresh import (
+    RefreshResource,
+    AsyncRefreshResource,
+    RefreshResourceWithRawResponse,
+    AsyncRefreshResourceWithRawResponse,
+    RefreshResourceWithStreamingResponse,
+    AsyncRefreshResourceWithStreamingResponse,
+)
+from .mcp_servers import (
+    McpServersResource,
+    AsyncMcpServersResource,
+    McpServersResourceWithRawResponse,
+    AsyncMcpServersResourceWithRawResponse,
+    McpServersResourceWithStreamingResponse,
+    AsyncMcpServersResourceWithStreamingResponse,
+)
+
+__all__ = [
+    "ToolsResource",
+    "AsyncToolsResource",
+    "ToolsResourceWithRawResponse",
+    "AsyncToolsResourceWithRawResponse",
+    "ToolsResourceWithStreamingResponse",
+    "AsyncToolsResourceWithStreamingResponse",
+    "RefreshResource",
+    "AsyncRefreshResource",
+    "RefreshResourceWithRawResponse",
+    "AsyncRefreshResourceWithRawResponse",
+    "RefreshResourceWithStreamingResponse",
+    "AsyncRefreshResourceWithStreamingResponse",
+    "McpServersResource",
+    "AsyncMcpServersResource",
+    "McpServersResourceWithRawResponse",
+    "AsyncMcpServersResourceWithRawResponse",
+    "McpServersResourceWithStreamingResponse",
+    "AsyncMcpServersResourceWithStreamingResponse",
+]