letta-client 0.1.17__py3-none-any.whl → 0.1.21__py3-none-any.whl

letta_client/__init__.py

@@ -11,37 +11,26 @@ from .types import (
     AppAuthScheme,
     AppAuthSchemeAuthMode,
     AppModel,
-    AssistantFile,
-    AssistantMessageInput,
-    AssistantMessageOutput,
+    AssistantMessage,
+    AssistantMessageContent,
     AuthRequest,
     AuthResponse,
     AuthSchemeField,
     Block,
     BlockUpdate,
-    ChatCompletionRequest,
-    ChatCompletionRequestFunctionCall,
-    ChatCompletionRequestMessagesItem,
-    ChatCompletionRequestStop,
-    ChatCompletionRequestToolChoice,
-    ChatCompletionResponse,
+    ChatCompletionMessageToolCall,
     ChildToolRule,
-    Choice,
     ConditionalToolRule,
     ConflictErrorBody,
     ContextWindowOverview,
-    CreateAssistantFileRequest,
-    CreateAssistantRequest,
     CreateBlock,
-    DeleteAssistantFileResponse,
-    DeleteAssistantResponse,
     E2BSandboxConfig,
     EmbeddingConfig,
     EmbeddingConfigEmbeddingEndpointType,
     FileMetadata,
-    FunctionCallInput,
-    FunctionCallOutput,
-    FunctionSchema,
+    Function,
+    FunctionDefinition,
+    FunctionTool,
     Health,
     HttpValidationError,
     InitToolRule,
@@ -50,45 +39,26 @@ from .types import (
     JobStatus,
     JobType,
     LettaMessageUnion,
-    LettaMessageUnion_AssistantMessage,
-    LettaMessageUnion_ReasoningMessage,
-    LettaMessageUnion_SystemMessage,
-    LettaMessageUnion_ToolCallMessage,
-    LettaMessageUnion_ToolReturnMessage,
-    LettaMessageUnion_UserMessage,
     LettaRequest,
     LettaRequestConfig,
     LettaResponse,
-    LettaSchemasLettaMessageToolCall,
-    LettaSchemasMessageMessage,
-    LettaSchemasOpenaiChatCompletionRequestTool,
-    LettaSchemasOpenaiChatCompletionRequestToolCall,
-    LettaSchemasOpenaiChatCompletionRequestToolCallFunction,
-    LettaSchemasOpenaiChatCompletionResponseMessage,
-    LettaSchemasOpenaiChatCompletionResponseToolCall,
-    LettaSchemasOpenaiChatCompletionsToolCallFunction,
-    LettaSchemasOpenaiChatCompletionsToolCallInput,
-    LettaSchemasOpenaiChatCompletionsToolCallOutput,
-    LettaSchemasToolTool,
     LettaUsageStatistics,
     LlmConfig,
     LlmConfigModelEndpointType,
     LocalSandboxConfig,
-    LogProbToken,
     Memory,
-    MessageContentLogProb,
+    Message,
     MessageCreate,
+    MessageCreateContent,
     MessageCreateRole,
     MessageRole,
     NotFoundErrorBody,
     NotFoundErrorBodyMessage,
-    OpenAiAssistant,
     Organization,
     OrganizationCreate,
     Passage,
     Provider,
     ReasoningMessage,
-    ResponseFormat,
     Run,
     SandboxConfig,
     SandboxConfigCreate,
@@ -100,17 +70,16 @@ from .types import (
     SandboxEnvironmentVariableUpdate,
     SandboxType,
     Source,
-    SystemMessageInput,
-    SystemMessageOutput,
+    SystemMessage,
+    SystemMessageContent,
     TerminalToolRule,
+    TextContent,
+    Tool,
+    ToolCall,
     ToolCallDelta,
-    ToolCallFunctionOutput,
     ToolCallMessage,
     ToolCallMessageToolCall,
     ToolCreate,
-    ToolFunctionChoice,
-    ToolInput,
-    ToolMessage,
     ToolReturnMessage,
     ToolReturnMessageStatus,
     ToolRuleType,
@@ -118,9 +87,8 @@ from .types import (
     UsageStatistics,
     User,
     UserCreate,
-    UserMessageInput,
-    UserMessageInputContent,
-    UserMessageOutput,
+    UserMessage,
+    UserMessageContent,
     UserUpdate,
     ValidationError,
     ValidationErrorLocItem,
@@ -130,17 +98,13 @@ from . import agents, blocks, health, jobs, models, providers, runs, sources, ta
 from .agents import (
     AgentsSearchRequestCombinator,
     AgentsSearchRequestSearchItem,
-    AgentsSearchRequestSearchItemName,
-    AgentsSearchRequestSearchItemNameOperator,
-    AgentsSearchRequestSearchItemOrderBy,
-    AgentsSearchRequestSearchItemOrderByDirection,
-    AgentsSearchRequestSearchItemOrderByValue,
-    AgentsSearchRequestSearchItemTags,
-    AgentsSearchRequestSearchItemVersion,
-    AgentsSearchRequestSearchItem_Name,
-    AgentsSearchRequestSearchItem_OrderBy,
-    AgentsSearchRequestSearchItem_Tags,
-    AgentsSearchRequestSearchItem_Version,
+    AgentsSearchRequestSearchItemDirection,
+    AgentsSearchRequestSearchItemDirectionDirection,
+    AgentsSearchRequestSearchItemDirectionValue,
+    AgentsSearchRequestSearchItemOne,
+    AgentsSearchRequestSearchItemOneOperator,
+    AgentsSearchRequestSearchItemTwo,
+    AgentsSearchRequestSearchItemZero,
     CreateAgentRequestToolRulesItem,
     UpdateAgentToolRulesItem,
 )
@@ -158,54 +122,39 @@ __all__ = [
     "AgentType",
     "AgentsSearchRequestCombinator",
     "AgentsSearchRequestSearchItem",
-    "AgentsSearchRequestSearchItemName",
-    "AgentsSearchRequestSearchItemNameOperator",
-    "AgentsSearchRequestSearchItemOrderBy",
-    "AgentsSearchRequestSearchItemOrderByDirection",
-    "AgentsSearchRequestSearchItemOrderByValue",
-    "AgentsSearchRequestSearchItemTags",
-    "AgentsSearchRequestSearchItemVersion",
-    "AgentsSearchRequestSearchItem_Name",
-    "AgentsSearchRequestSearchItem_OrderBy",
-    "AgentsSearchRequestSearchItem_Tags",
-    "AgentsSearchRequestSearchItem_Version",
+    "AgentsSearchRequestSearchItemDirection",
+    "AgentsSearchRequestSearchItemDirectionDirection",
+    "AgentsSearchRequestSearchItemDirectionValue",
+    "AgentsSearchRequestSearchItemOne",
+    "AgentsSearchRequestSearchItemOneOperator",
+    "AgentsSearchRequestSearchItemTwo",
+    "AgentsSearchRequestSearchItemZero",
     "AppAuthScheme",
     "AppAuthSchemeAuthMode",
     "AppModel",
-    "AssistantFile",
-    "AssistantMessageInput",
-    "AssistantMessageOutput",
+    "AssistantMessage",
+    "AssistantMessageContent",
     "AsyncLetta",
     "AuthRequest",
     "AuthResponse",
     "AuthSchemeField",
     "Block",
     "BlockUpdate",
-    "ChatCompletionRequest",
-    "ChatCompletionRequestFunctionCall",
-    "ChatCompletionRequestMessagesItem",
-    "ChatCompletionRequestStop",
-    "ChatCompletionRequestToolChoice",
-    "ChatCompletionResponse",
+    "ChatCompletionMessageToolCall",
     "ChildToolRule",
-    "Choice",
     "ConditionalToolRule",
     "ConflictError",
     "ConflictErrorBody",
     "ContextWindowOverview",
     "CreateAgentRequestToolRulesItem",
-    "CreateAssistantFileRequest",
-    "CreateAssistantRequest",
     "CreateBlock",
-    "DeleteAssistantFileResponse",
-    "DeleteAssistantResponse",
     "E2BSandboxConfig",
     "EmbeddingConfig",
     "EmbeddingConfigEmbeddingEndpointType",
     "FileMetadata",
-    "FunctionCallInput",
-    "FunctionCallOutput",
-    "FunctionSchema",
+    "Function",
+    "FunctionDefinition",
+    "FunctionTool",
     "Health",
     "HttpValidationError",
     "InitToolRule",
@@ -217,46 +166,27 @@ __all__ = [
     "Letta",
     "LettaEnvironment",
     "LettaMessageUnion",
-    "LettaMessageUnion_AssistantMessage",
-    "LettaMessageUnion_ReasoningMessage",
-    "LettaMessageUnion_SystemMessage",
-    "LettaMessageUnion_ToolCallMessage",
-    "LettaMessageUnion_ToolReturnMessage",
-    "LettaMessageUnion_UserMessage",
     "LettaRequest",
     "LettaRequestConfig",
     "LettaResponse",
-    "LettaSchemasLettaMessageToolCall",
-    "LettaSchemasMessageMessage",
-    "LettaSchemasOpenaiChatCompletionRequestTool",
-    "LettaSchemasOpenaiChatCompletionRequestToolCall",
-    "LettaSchemasOpenaiChatCompletionRequestToolCallFunction",
-    "LettaSchemasOpenaiChatCompletionResponseMessage",
-    "LettaSchemasOpenaiChatCompletionResponseToolCall",
-    "LettaSchemasOpenaiChatCompletionsToolCallFunction",
-    "LettaSchemasOpenaiChatCompletionsToolCallInput",
-    "LettaSchemasOpenaiChatCompletionsToolCallOutput",
-    "LettaSchemasToolTool",
     "LettaUsageStatistics",
     "LlmConfig",
     "LlmConfigModelEndpointType",
     "LocalSandboxConfig",
-    "LogProbToken",
     "Memory",
-    "MessageContentLogProb",
+    "Message",
     "MessageCreate",
+    "MessageCreateContent",
     "MessageCreateRole",
     "MessageRole",
     "NotFoundError",
     "NotFoundErrorBody",
     "NotFoundErrorBodyMessage",
-    "OpenAiAssistant",
     "Organization",
     "OrganizationCreate",
     "Passage",
     "Provider",
     "ReasoningMessage",
-    "ResponseFormat",
     "Run",
     "SandboxConfig",
     "SandboxConfigCreate",
@@ -268,17 +198,16 @@ __all__ = [
     "SandboxEnvironmentVariableUpdate",
     "SandboxType",
     "Source",
-    "SystemMessageInput",
-    "SystemMessageOutput",
+    "SystemMessage",
+    "SystemMessageContent",
     "TerminalToolRule",
+    "TextContent",
+    "Tool",
+    "ToolCall",
     "ToolCallDelta",
-    "ToolCallFunctionOutput",
     "ToolCallMessage",
     "ToolCallMessageToolCall",
     "ToolCreate",
-    "ToolFunctionChoice",
-    "ToolInput",
-    "ToolMessage",
     "ToolReturnMessage",
     "ToolReturnMessageStatus",
     "ToolRuleType",
@@ -288,9 +217,8 @@ __all__ = [
     "UsageStatistics",
     "User",
     "UserCreate",
-    "UserMessageInput",
-    "UserMessageInputContent",
-    "UserMessageOutput",
+    "UserMessage",
+    "UserMessageContent",
     "UserUpdate",
     "ValidationError",
     "ValidationErrorLocItem",

letta_client/agents/__init__.py

@@ -3,59 +3,35 @@
 from .types import (
     AgentsSearchRequestCombinator,
     AgentsSearchRequestSearchItem,
-    AgentsSearchRequestSearchItemName,
-    AgentsSearchRequestSearchItemNameOperator,
-    AgentsSearchRequestSearchItemOrderBy,
-    AgentsSearchRequestSearchItemOrderByDirection,
-    AgentsSearchRequestSearchItemOrderByValue,
-    AgentsSearchRequestSearchItemTags,
-    AgentsSearchRequestSearchItemVersion,
-    AgentsSearchRequestSearchItem_Name,
-    AgentsSearchRequestSearchItem_OrderBy,
-    AgentsSearchRequestSearchItem_Tags,
-    AgentsSearchRequestSearchItem_Version,
+    AgentsSearchRequestSearchItemDirection,
+    AgentsSearchRequestSearchItemDirectionDirection,
+    AgentsSearchRequestSearchItemDirectionValue,
+    AgentsSearchRequestSearchItemOne,
+    AgentsSearchRequestSearchItemOneOperator,
+    AgentsSearchRequestSearchItemTwo,
+    AgentsSearchRequestSearchItemZero,
     CreateAgentRequestToolRulesItem,
     UpdateAgentToolRulesItem,
 )
 from . import archival_memory, context, core_memory, memory_variables, messages, sources, templates, tools
-from .memory_variables import MemoryVariablesGetResponse
-from .messages import (
-    LettaStreamingResponse,
-    LettaStreamingResponse_AssistantMessage,
-    LettaStreamingResponse_ReasoningMessage,
-    LettaStreamingResponse_SystemMessage,
-    LettaStreamingResponse_ToolCallMessage,
-    LettaStreamingResponse_ToolReturnMessage,
-    LettaStreamingResponse_UsageStatistics,
-    LettaStreamingResponse_UserMessage,
-    MessagesListResponse,
-)
+from .memory_variables import MemoryVariablesListResponse
+from .messages import LettaStreamingResponse, MessageUpdateContent, MessagesListResponse
 from .templates import TemplatesMigrateResponse
 
 __all__ = [
     "AgentsSearchRequestCombinator",
     "AgentsSearchRequestSearchItem",
-    "AgentsSearchRequestSearchItemName",
-    "AgentsSearchRequestSearchItemNameOperator",
-    "AgentsSearchRequestSearchItemOrderBy",
-    "AgentsSearchRequestSearchItemOrderByDirection",
-    "AgentsSearchRequestSearchItemOrderByValue",
-    "AgentsSearchRequestSearchItemTags",
-    "AgentsSearchRequestSearchItemVersion",
-    "AgentsSearchRequestSearchItem_Name",
-    "AgentsSearchRequestSearchItem_OrderBy",
-    "AgentsSearchRequestSearchItem_Tags",
-    "AgentsSearchRequestSearchItem_Version",
+    "AgentsSearchRequestSearchItemDirection",
+    "AgentsSearchRequestSearchItemDirectionDirection",
+    "AgentsSearchRequestSearchItemDirectionValue",
+    "AgentsSearchRequestSearchItemOne",
+    "AgentsSearchRequestSearchItemOneOperator",
+    "AgentsSearchRequestSearchItemTwo",
+    "AgentsSearchRequestSearchItemZero",
     "CreateAgentRequestToolRulesItem",
     "LettaStreamingResponse",
-    "LettaStreamingResponse_AssistantMessage",
-    "LettaStreamingResponse_ReasoningMessage",
-    "LettaStreamingResponse_SystemMessage",
-    "LettaStreamingResponse_ToolCallMessage",
-    "LettaStreamingResponse_ToolReturnMessage",
-    "LettaStreamingResponse_UsageStatistics",
-    "LettaStreamingResponse_UserMessage",
-    "MemoryVariablesGetResponse",
+    "MemoryVariablesListResponse",
+    "MessageUpdateContent",
     "MessagesListResponse",
     "TemplatesMigrateResponse",
     "UpdateAgentToolRulesItem",

letta_client/agents/archival_memory/client.py

@@ -65,7 +65,7 @@ class ArchivalMemoryClient:
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/archival_memory",
+            f"v1/agents/{jsonable_encoder(agent_id)}/archival-memory",
             method="GET",
             params={
                 "after": after,
@@ -132,7 +132,7 @@ class ArchivalMemoryClient:
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/archival_memory",
+            f"v1/agents/{jsonable_encoder(agent_id)}/archival-memory",
             method="POST",
             json={
                 "text": text,
@@ -200,7 +200,7 @@ class ArchivalMemoryClient:
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/archival_memory/{jsonable_encoder(memory_id)}",
+            f"v1/agents/{jsonable_encoder(agent_id)}/archival-memory/{jsonable_encoder(memory_id)}",
             method="DELETE",
             request_options=request_options,
         )
@@ -286,7 +286,7 @@ class AsyncArchivalMemoryClient:
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/archival_memory",
+            f"v1/agents/{jsonable_encoder(agent_id)}/archival-memory",
             method="GET",
             params={
                 "after": after,
@@ -361,7 +361,7 @@ class AsyncArchivalMemoryClient:
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/archival_memory",
+            f"v1/agents/{jsonable_encoder(agent_id)}/archival-memory",
             method="POST",
             json={
                 "text": text,
@@ -437,7 +437,7 @@ class AsyncArchivalMemoryClient:
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/archival_memory/{jsonable_encoder(memory_id)}",
+            f"v1/agents/{jsonable_encoder(agent_id)}/archival-memory/{jsonable_encoder(memory_id)}",
             method="DELETE",
             request_options=request_options,
         )

letta_client/agents/client.py

@@ -60,7 +60,8 @@ class AgentsClient:
         name: typing.Optional[str] = None,
         tags: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
         match_all_tags: typing.Optional[bool] = None,
-        cursor: typing.Optional[str] = None,
+        before: typing.Optional[str] = None,
+        after: typing.Optional[str] = None,
         limit: typing.Optional[int] = None,
         query_text: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
@@ -80,7 +81,10 @@ class AgentsClient:
         match_all_tags : typing.Optional[bool]
             If True, only returns agents that match ALL given tags. Otherwise, return agents that have ANY of the passed in tags.
 
-        cursor : typing.Optional[str]
+        before : typing.Optional[str]
+            Cursor for pagination
+
+        after : typing.Optional[str]
             Cursor for pagination
 
         limit : typing.Optional[int]
@@ -113,7 +117,8 @@ class AgentsClient:
                 "name": name,
                 "tags": tags,
                 "match_all_tags": match_all_tags,
-                "cursor": cursor,
+                "before": before,
+                "after": after,
                 "limit": limit,
                 "query_text": query_text,
             },
@@ -304,7 +309,7 @@ class AgentsClient:
                 "include_base_tools": include_base_tools,
                 "include_multi_agent_tools": include_multi_agent_tools,
                 "description": description,
-                "metadata_": metadata,
+                "metadata": metadata,
                 "model": model,
                 "embedding": embedding,
                 "context_window_limit": context_window_limit,
@@ -345,7 +350,7 @@ class AgentsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def get(self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> AgentState:
+    def retrieve(self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> AgentState:
         """
         Get the state of the agent.
 
@@ -368,7 +373,7 @@ class AgentsClient:
         client = Letta(
             token="YOUR_TOKEN",
         )
-        client.agents.get(
+        client.agents.retrieve(
             agent_id="agent_id",
         )
         """
@@ -459,7 +464,7 @@ class AgentsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def update(
+    def modify(
         self,
         agent_id: str,
         *,
@@ -539,7 +544,7 @@ class AgentsClient:
         client = Letta(
             token="YOUR_TOKEN",
         )
-        client.agents.update(
+        client.agents.modify(
             agent_id="agent_id",
         )
         """
@@ -564,7 +569,7 @@ class AgentsClient:
                 ),
                 "message_ids": message_ids,
                 "description": description,
-                "metadata_": metadata,
+                "metadata": metadata,
                 "tool_exec_environment_variables": tool_exec_environment_variables,
             },
             headers={
@@ -753,7 +758,8 @@ class AsyncAgentsClient:
         name: typing.Optional[str] = None,
         tags: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
         match_all_tags: typing.Optional[bool] = None,
-        cursor: typing.Optional[str] = None,
+        before: typing.Optional[str] = None,
+        after: typing.Optional[str] = None,
         limit: typing.Optional[int] = None,
         query_text: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
@@ -773,7 +779,10 @@ class AsyncAgentsClient:
         match_all_tags : typing.Optional[bool]
             If True, only returns agents that match ALL given tags. Otherwise, return agents that have ANY of the passed in tags.
 
-        cursor : typing.Optional[str]
+        before : typing.Optional[str]
+            Cursor for pagination
+
+        after : typing.Optional[str]
             Cursor for pagination
 
         limit : typing.Optional[int]
@@ -814,7 +823,8 @@ class AsyncAgentsClient:
                 "name": name,
                 "tags": tags,
                 "match_all_tags": match_all_tags,
-                "cursor": cursor,
+                "before": before,
+                "after": after,
                 "limit": limit,
                 "query_text": query_text,
             },
@@ -1013,7 +1023,7 @@ class AsyncAgentsClient:
                 "include_base_tools": include_base_tools,
                 "include_multi_agent_tools": include_multi_agent_tools,
                 "description": description,
-                "metadata_": metadata,
+                "metadata": metadata,
                 "model": model,
                 "embedding": embedding,
                 "context_window_limit": context_window_limit,
@@ -1054,7 +1064,7 @@ class AsyncAgentsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def get(self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> AgentState:
+    async def retrieve(self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> AgentState:
         """
         Get the state of the agent.
 
@@ -1082,7 +1092,7 @@ class AsyncAgentsClient:
 
 
         async def main() -> None:
-            await client.agents.get(
+            await client.agents.retrieve(
                 agent_id="agent_id",
             )
 
@@ -1184,7 +1194,7 @@ class AsyncAgentsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def update(
+    async def modify(
         self,
         agent_id: str,
         *,
@@ -1269,7 +1279,7 @@ class AsyncAgentsClient:
 
 
         async def main() -> None:
-            await client.agents.update(
+            await client.agents.modify(
                 agent_id="agent_id",
             )
 
@@ -1297,7 +1307,7 @@ class AsyncAgentsClient:
                 ),
                 "message_ids": message_ids,
                 "description": description,
-                "metadata_": metadata,
+                "metadata": metadata,
                 "tool_exec_environment_variables": tool_exec_environment_variables,
             },
             headers={

letta_client/agents/context/client.py

@@ -17,7 +17,9 @@ class ContextClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    def get(self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> ContextWindowOverview:
+    def retrieve(
+        self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> ContextWindowOverview:
         """
         Retrieve the context window of a specific agent.
 
@@ -40,7 +42,7 @@ class ContextClient:
         client = Letta(
             token="YOUR_TOKEN",
         )
-        client.agents.context.get(
+        client.agents.context.retrieve(
             agent_id="agent_id",
         )
         """
@@ -78,7 +80,7 @@ class AsyncContextClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    async def get(
+    async def retrieve(
         self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None
     ) -> ContextWindowOverview:
         """
@@ -108,7 +110,7 @@ class AsyncContextClient:
 
 
         async def main() -> None:
-            await client.agents.context.get(
+            await client.agents.context.retrieve(
                 agent_id="agent_id",
             )