From 7e90a72968afc4af265d68e71f8f6cb95cdab87b Mon Sep 17 00:00:00 2001
From: Matthew Zhou <mattzh1314@gmail.com>
Date: Wed, 9 Jul 2025 19:21:17 -0700
Subject: [PATCH] fix: Fix preview payload tests (#3266)

---
 poetry.lock              |  8 ++++----
 pyproject.toml           |  2 +-
 tests/test_sdk_client.py | 24 +++++++++++++-----------
 3 files changed, 18 insertions(+), 16 deletions(-)

diff --git a/poetry.lock b/poetry.lock
index dd37441d..f455ed90 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -3215,13 +3215,13 @@ pytest = ["pytest (>=7.0.0)", "rich (>=13.9.4,<14.0.0)"]
 
 [[package]]
 name = "letta-client"
-version = "0.1.196"
+version = "0.1.197"
 description = ""
 optional = false
 python-versions = "<4.0,>=3.8"
 files = [
-    {file = "letta_client-0.1.196-py3-none-any.whl", hash = "sha256:78b88f1a08d308e714add3eb14aa5f0489a4d9eac2d1a834aacec45a652c15fb"},
-    {file = "letta_client-0.1.196.tar.gz", hash = "sha256:94f2b2d02d26158ce927ca72a04dcaf37c5dc49394c2cb1228f0e0aac9385b33"},
+    {file = "letta_client-0.1.197-py3-none-any.whl", hash = "sha256:b01eab01ff87a34e79622cd8d3a2f3da56b6bd730312a268d968576671f6cce0"},
+    {file = "letta_client-0.1.197.tar.gz", hash = "sha256:579571623ccec81422087cb7957d5f580fa0b7a53f8495596e5c56d0213220d8"},
 ]
 
 [package.dependencies]
@@ -7657,4 +7657,4 @@ tests = ["wikipedia"]
 [metadata]
 lock-version = "2.0"
 python-versions = "<3.14,>=3.10"
-content-hash = "879ebe61f71d6e677c3c0911440b9623a76ea386f17826f067a5cb9491028630"
+content-hash = "c4fa225d582dac743e5eb8a1338a71c32bd8c8eb8283dec2331c28599bdf7698"
diff --git a/pyproject.toml b/pyproject.toml
index 3c69f2ab..85844190 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -72,7 +72,7 @@ llama-index = "^0.12.2"
 llama-index-embeddings-openai = "^0.3.1"
 e2b-code-interpreter = {version = "^1.0.3", optional = true}
 anthropic = "^0.49.0"
-letta_client = "^0.1.196"
+letta_client = "^0.1.197"
 openai = "^1.60.0"
 opentelemetry-api = "1.30.0"
 opentelemetry-sdk = "1.30.0"
diff --git a/tests/test_sdk_client.py b/tests/test_sdk_client.py
index 2b0b885a..7ff37851 100644
--- a/tests/test_sdk_client.py
+++ b/tests/test_sdk_client.py
@@ -9,7 +9,7 @@ import pytest
 from dotenv import load_dotenv
 from letta_client import CreateBlock
 from letta_client import Letta as LettaSDKClient
-from letta_client import MessageCreate, TextContent
+from letta_client import LettaRequest, MessageCreate, TextContent
 from letta_client.client import BaseTool
 from letta_client.core import ApiError
 from letta_client.types import AgentState, ToolReturnMessage
@@ -972,16 +972,18 @@ def test_create_tool_from_function_with_docstring(e2b_sandbox_mode, client: Lett
 def test_preview_payload(client: LettaSDKClient, agent):
     payload = client.agents.messages.preview_raw_payload(
         agent_id=agent.id,
-        messages=[
-            MessageCreate(
-                role="user",
-                content=[
-                    TextContent(
-                        text="text",
-                    )
-                ],
-            )
-        ],
+        request=LettaRequest(
+            messages=[
+                MessageCreate(
+                    role="user",
+                    content=[
+                        TextContent(
+                            text="text",
+                        )
+                    ],
+                )
+            ],
+        ),
     )
 
     assert isinstance(payload, dict)