feat(api): api update (#2)

Author: stainless-app[bot]

.stats.yml

@@ -1,2 +1,2 @@
 configured_endpoints: 9
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/letta-ai%2Fletta-b255d1730fcaf036db43fc05485c8b8a6e766bb0d15504e41b5c2bb0d8d44e3b.yml
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/letta-ai%2Fletta-d32ce280dce8bf09416f5114194c4b05870757f2a9768902b9ccf66fb1ca6919.yml

src/letta/types/agents/message_create_response.py

@@ -21,6 +21,8 @@
     "ToolCallMessageToolCallToolCall",
     "ToolCallMessageToolCallToolCallDelta",
     "ToolReturnMessage",
+    "UsageMessage",
+    "UsageMessageUsage",
     "UserMessage",
 ]
 
@@ -132,6 +134,37 @@ class ToolReturnMessage(BaseModel):
     stdout: Optional[List[str]] = None
 
 
+class UsageMessageUsage(BaseModel):
+    completion_tokens: Optional[int] = None
+    """The number of tokens generated by the agent."""
+
+    prompt_tokens: Optional[int] = None
+    """The number of tokens in the prompt."""
+
+    step_count: Optional[int] = None
+    """The number of steps taken by the agent."""
+
+    total_tokens: Optional[int] = None
+    """The total number of tokens processed by the agent."""
+
+
+class UsageMessage(BaseModel):
+    id: str
+
+    date: datetime
+
+    usage: UsageMessageUsage
+    """Usage statistics for the agent interaction.
+
+    Attributes: completion_tokens (int): The number of tokens generated by the
+    agent. prompt_tokens (int): The number of tokens in the prompt. total_tokens
+    (int): The total number of tokens processed by the agent. step_count (int): The
+    number of steps taken by the agent.
+    """
+
+    message_type: Optional[Literal["usage_message"]] = None
+
+
 class UserMessage(BaseModel):
     id: str
 
@@ -199,6 +232,13 @@ class MessageCreateResponse(BaseModel):
     invocation
     """
 
+    usage_message: Optional[UsageMessage] = FieldInfo(alias="UsageMessage", default=None)
+    """A message representint the usage statistics for the agent interaction.
+
+    Attributes: usage (LettaUsageStatistics): Usage statistics for the agent
+    interaction.
+    """
+
     user_message: Optional[UserMessage] = FieldInfo(alias="UserMessage", default=None)
     """A message sent by the user.