letta-client 0.1.24__py3-none-any.whl → 0.1.26__py3-none-any.whl

letta_client/agents/client.py

@@ -27,7 +27,7 @@ from ..core.serialization import convert_and_respect_annotation_metadata
 from ..core.jsonable_encoder import jsonable_encoder
 from .types.update_agent_tool_rules_item import UpdateAgentToolRulesItem
 from .types.agents_search_request_search_item import AgentsSearchRequestSearchItem
-from .types.agents_search_request_combinator import AgentsSearchRequestCombinator
+from .types.agents_search_response import AgentsSearchResponse
 from ..core.client_wrapper import AsyncClientWrapper
 from .context.client import AsyncContextClient
 from .tools.client import AsyncToolsClient
@@ -64,6 +64,9 @@ class AgentsClient:
         after: typing.Optional[str] = None,
         limit: typing.Optional[int] = None,
         query_text: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
+        template_id: typing.Optional[str] = None,
+        base_template_id: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.List[AgentState]:
         """
@@ -93,6 +96,15 @@ class AgentsClient:
         query_text : typing.Optional[str]
             Search agents by name
 
+        project_id : typing.Optional[str]
+            Search agents by project id
+
+        template_id : typing.Optional[str]
+            Search agents by template id
+
+        base_template_id : typing.Optional[str]
+            Search agents by base template id
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -121,6 +133,9 @@ class AgentsClient:
                 "after": after,
                 "limit": limit,
                 "query_text": query_text,
+                "project_id": project_id,
+                "template_id": template_id,
+                "base_template_id": base_template_id,
             },
             request_options=request_options,
         )
@@ -177,6 +192,9 @@ class AgentsClient:
         project: typing.Optional[str] = OMIT,
         tool_exec_environment_variables: typing.Optional[typing.Dict[str, typing.Optional[str]]] = OMIT,
         memory_variables: typing.Optional[typing.Dict[str, typing.Optional[str]]] = OMIT,
+        project_id: typing.Optional[str] = OMIT,
+        template_id: typing.Optional[str] = OMIT,
+        base_template_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AgentState:
         """
@@ -262,6 +280,15 @@ class AgentsClient:
         memory_variables : typing.Optional[typing.Dict[str, typing.Optional[str]]]
             The variables that should be set for the agent.
 
+        project_id : typing.Optional[str]
+            The id of the project the agent belongs to.
+
+        template_id : typing.Optional[str]
+            The id of the template the agent belongs to.
+
+        base_template_id : typing.Optional[str]
+            The base template id of the agent.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -319,6 +346,9 @@ class AgentsClient:
                 "project": project,
                 "tool_exec_environment_variables": tool_exec_environment_variables,
                 "memory_variables": memory_variables,
+                "project_id": project_id,
+                "template_id": template_id,
+                "base_template_id": base_template_id,
             },
             headers={
                 "content-type": "application/json",
@@ -481,6 +511,9 @@ class AgentsClient:
         description: typing.Optional[str] = OMIT,
         metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
         tool_exec_environment_variables: typing.Optional[typing.Dict[str, typing.Optional[str]]] = OMIT,
+        project_id: typing.Optional[str] = OMIT,
+        template_id: typing.Optional[str] = OMIT,
+        base_template_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AgentState:
         """
@@ -529,6 +562,15 @@ class AgentsClient:
         tool_exec_environment_variables : typing.Optional[typing.Dict[str, typing.Optional[str]]]
             The environment variables for tool execution specific to this agent.
 
+        project_id : typing.Optional[str]
+            The id of the project the agent belongs to.
+
+        template_id : typing.Optional[str]
+            The id of the template the agent belongs to.
+
+        base_template_id : typing.Optional[str]
+            The base template id of the agent.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -571,6 +613,9 @@ class AgentsClient:
                 "description": description,
                 "metadata": metadata,
                 "tool_exec_environment_variables": tool_exec_environment_variables,
+                "project_id": project_id,
+                "template_id": template_id,
+                "base_template_id": base_template_id,
             },
             headers={
                 "content-type": "application/json",
@@ -675,11 +720,11 @@ class AgentsClient:
         *,
         search: typing.Optional[typing.Sequence[AgentsSearchRequestSearchItem]] = OMIT,
         project_id: typing.Optional[str] = OMIT,
-        combinator: typing.Optional[AgentsSearchRequestCombinator] = OMIT,
+        combinator: typing.Optional[typing.Literal["AND"]] = OMIT,
         limit: typing.Optional[float] = OMIT,
-        offset: typing.Optional[float] = OMIT,
+        after: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> None:
+    ) -> AgentsSearchResponse:
         """
         <Note>This endpoint is only available on Letta Cloud.</Note>
 
@@ -691,18 +736,19 @@ class AgentsClient:
 
         project_id : typing.Optional[str]
 
-        combinator : typing.Optional[AgentsSearchRequestCombinator]
+        combinator : typing.Optional[typing.Literal["AND"]]
 
         limit : typing.Optional[float]
 
-        offset : typing.Optional[float]
+        after : typing.Optional[str]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        None
+        AgentsSearchResponse
+            200
 
         Examples
         --------
@@ -723,7 +769,7 @@ class AgentsClient:
                 "project_id": project_id,
                 "combinator": combinator,
                 "limit": limit,
-                "offset": offset,
+                "after": after,
             },
             headers={
                 "content-type": "application/json",
@@ -733,7 +779,13 @@ class AgentsClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return
+                return typing.cast(
+                    AgentsSearchResponse,
+                    construct_type(
+                        type_=AgentsSearchResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -762,6 +814,9 @@ class AsyncAgentsClient:
         after: typing.Optional[str] = None,
         limit: typing.Optional[int] = None,
         query_text: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
+        template_id: typing.Optional[str] = None,
+        base_template_id: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.List[AgentState]:
         """
@@ -791,6 +846,15 @@ class AsyncAgentsClient:
         query_text : typing.Optional[str]
             Search agents by name
 
+        project_id : typing.Optional[str]
+            Search agents by project id
+
+        template_id : typing.Optional[str]
+            Search agents by template id
+
+        base_template_id : typing.Optional[str]
+            Search agents by base template id
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -827,6 +891,9 @@ class AsyncAgentsClient:
                 "after": after,
                 "limit": limit,
                 "query_text": query_text,
+                "project_id": project_id,
+                "template_id": template_id,
+                "base_template_id": base_template_id,
             },
             request_options=request_options,
         )
@@ -883,6 +950,9 @@ class AsyncAgentsClient:
         project: typing.Optional[str] = OMIT,
         tool_exec_environment_variables: typing.Optional[typing.Dict[str, typing.Optional[str]]] = OMIT,
         memory_variables: typing.Optional[typing.Dict[str, typing.Optional[str]]] = OMIT,
+        project_id: typing.Optional[str] = OMIT,
+        template_id: typing.Optional[str] = OMIT,
+        base_template_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AgentState:
         """
@@ -968,6 +1038,15 @@ class AsyncAgentsClient:
         memory_variables : typing.Optional[typing.Dict[str, typing.Optional[str]]]
             The variables that should be set for the agent.
 
+        project_id : typing.Optional[str]
+            The id of the project the agent belongs to.
+
+        template_id : typing.Optional[str]
+            The id of the template the agent belongs to.
+
+        base_template_id : typing.Optional[str]
+            The base template id of the agent.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -1033,6 +1112,9 @@ class AsyncAgentsClient:
                 "project": project,
                 "tool_exec_environment_variables": tool_exec_environment_variables,
                 "memory_variables": memory_variables,
+                "project_id": project_id,
+                "template_id": template_id,
+                "base_template_id": base_template_id,
             },
             headers={
                 "content-type": "application/json",
@@ -1211,6 +1293,9 @@ class AsyncAgentsClient:
         description: typing.Optional[str] = OMIT,
         metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
         tool_exec_environment_variables: typing.Optional[typing.Dict[str, typing.Optional[str]]] = OMIT,
+        project_id: typing.Optional[str] = OMIT,
+        template_id: typing.Optional[str] = OMIT,
+        base_template_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AgentState:
         """
@@ -1259,6 +1344,15 @@ class AsyncAgentsClient:
         tool_exec_environment_variables : typing.Optional[typing.Dict[str, typing.Optional[str]]]
             The environment variables for tool execution specific to this agent.
 
+        project_id : typing.Optional[str]
+            The id of the project the agent belongs to.
+
+        template_id : typing.Optional[str]
+            The id of the template the agent belongs to.
+
+        base_template_id : typing.Optional[str]
+            The base template id of the agent.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -1309,6 +1403,9 @@ class AsyncAgentsClient:
                 "description": description,
                 "metadata": metadata,
                 "tool_exec_environment_variables": tool_exec_environment_variables,
+                "project_id": project_id,
+                "template_id": template_id,
+                "base_template_id": base_template_id,
             },
             headers={
                 "content-type": "application/json",
@@ -1421,11 +1518,11 @@ class AsyncAgentsClient:
         *,
         search: typing.Optional[typing.Sequence[AgentsSearchRequestSearchItem]] = OMIT,
         project_id: typing.Optional[str] = OMIT,
-        combinator: typing.Optional[AgentsSearchRequestCombinator] = OMIT,
+        combinator: typing.Optional[typing.Literal["AND"]] = OMIT,
         limit: typing.Optional[float] = OMIT,
-        offset: typing.Optional[float] = OMIT,
+        after: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> None:
+    ) -> AgentsSearchResponse:
         """
         <Note>This endpoint is only available on Letta Cloud.</Note>
 
@@ -1437,18 +1534,19 @@ class AsyncAgentsClient:
 
         project_id : typing.Optional[str]
 
-        combinator : typing.Optional[AgentsSearchRequestCombinator]
+        combinator : typing.Optional[typing.Literal["AND"]]
 
         limit : typing.Optional[float]
 
-        offset : typing.Optional[float]
+        after : typing.Optional[str]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        None
+        AgentsSearchResponse
+            200
 
         Examples
         --------
@@ -1477,7 +1575,7 @@ class AsyncAgentsClient:
                 "project_id": project_id,
                 "combinator": combinator,
                 "limit": limit,
-                "offset": offset,
+                "after": after,
             },
             headers={
                 "content-type": "application/json",
@@ -1487,7 +1585,13 @@ class AsyncAgentsClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return
+                return typing.cast(
+                    AgentsSearchResponse,
+                    construct_type(
+                        type_=AgentsSearchResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)

letta_client/agents/templates/__init__.py

@@ -1,5 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from .types import TemplatesMigrateResponse
+from .types import TemplatesCreateResponse, TemplatesMigrateResponse
 
-__all__ = ["TemplatesMigrateResponse"]
+__all__ = ["TemplatesCreateResponse", "TemplatesMigrateResponse"]

letta_client/agents/templates/client.py

@@ -12,6 +12,7 @@ from ...core.api_error import ApiError
 from .types.templates_migrate_response import TemplatesMigrateResponse
 from ...errors.conflict_error import ConflictError
 from ...types.conflict_error_body import ConflictErrorBody
+from .types.templates_create_response import TemplatesCreateResponse
 from ...core.client_wrapper import AsyncClientWrapper
 
 # this is used as the default value for optional parameters
@@ -215,7 +216,7 @@ class TemplatesClient:
         *,
         project: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> None:
+    ) -> TemplatesCreateResponse:
         """
         <Note>This endpoint is only available on Letta Cloud.</Note>
 
@@ -232,7 +233,8 @@ class TemplatesClient:
 
         Returns
         -------
-        None
+        TemplatesCreateResponse
+            201
 
         Examples
         --------
@@ -259,16 +261,12 @@ class TemplatesClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return
-            if _response.status_code == 404:
-                raise NotFoundError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        construct_type(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
+                return typing.cast(
+                    TemplatesCreateResponse,
+                    construct_type(
+                        type_=TemplatesCreateResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
                 )
             if _response.status_code == 500:
                 raise InternalServerError(
@@ -499,7 +497,7 @@ class AsyncTemplatesClient:
         *,
         project: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> None:
+    ) -> TemplatesCreateResponse:
         """
         <Note>This endpoint is only available on Letta Cloud.</Note>
 
@@ -516,7 +514,8 @@ class AsyncTemplatesClient:
 
         Returns
         -------
-        None
+        TemplatesCreateResponse
+            201
 
         Examples
         --------
@@ -551,16 +550,12 @@ class AsyncTemplatesClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return
-            if _response.status_code == 404:
-                raise NotFoundError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        construct_type(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
+                return typing.cast(
+                    TemplatesCreateResponse,
+                    construct_type(
+                        type_=TemplatesCreateResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
                 )
             if _response.status_code == 500:
                 raise InternalServerError(

letta_client/agents/templates/types/__init__.py

@@ -1,5 +1,6 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from .templates_create_response import TemplatesCreateResponse
 from .templates_migrate_response import TemplatesMigrateResponse
 
-__all__ = ["TemplatesMigrateResponse"]
+__all__ = ["TemplatesCreateResponse", "TemplatesMigrateResponse"]

letta_client/agents/templates/types/templates_create_response.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ....core.unchecked_base_model import UncheckedBaseModel
+import typing_extensions
+from ....core.serialization import FieldMetadata
+from ....core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+import pydantic
+
+
+class TemplatesCreateResponse(UncheckedBaseModel):
+    template_name: typing_extensions.Annotated[str, FieldMetadata(alias="templateName")]
+    template_id: typing_extensions.Annotated[str, FieldMetadata(alias="templateId")]
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow