Fimeg/letta-server
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

feat: make letta_v1_agent default [LET-6153] (#6115)

Browse Source 
* agent change

* fix tests

* fix more tests

* memgpt_v2_agent for cloud-api tests

---------

Co-authored-by: Ari Webb <ari@letta.com>
This commit is contained in:
Ari Webb
2025-11-17 15:39:31 -08:00
committed by Caren Thomas
parent 474f8c1f89
commit b0f9986209
8 changed files with 34 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/helpers/endpoints_helper.py
View File

@@ -64,6 +64,7 @@ async def setup_agent(
request = CreateAgent(
name=agent_uuid,
agent_type="memgpt_v2_agent",
llm_config=llm_config,
embedding_config=embedding_config,
memory_blocks=[

2
tests/integration_test_send_message.py
View File

@@ -815,6 +815,7 @@ def agent_state(client: Letta) -> AgentState:
send_message_tool = client.tools.list(name="send_message")[0]
agent_state_instance = client.agents.create(
name="supervisor",
agent_type="memgpt_v2_agent",
include_base_tools=False,
tool_ids=[send_message_tool.id, dice_tool.id],
model="openai/gpt-4o",
@@ -1878,6 +1879,7 @@ def test_auto_summarize(disable_e2b_api_key: Any, client: Letta, llm_config: LLM
send_message_tool = client.tools.list(name="send_message")[0]
temp_agent_state = client.agents.create(
include_base_tools=False,
agent_type="memgpt_v2_agent",
tool_ids=[send_message_tool.id],
llm_config=pinned_context_window_llm_config,
embedding="letta/letta-free",

1
tests/managers/conftest.py
View File

@@ -363,6 +363,7 @@ async def comprehensive_test_agent_fixture(server: SyncServer, default_user, pri
"""Create a comprehensive test agent with all features."""
memory_blocks = [CreateBlock(label="human", value="BananaBoy"), CreateBlock(label="persona", value="I am a helpful assistant")]
create_agent_request = CreateAgent(
agent_type="memgpt_v2_agent",
system="test system",
memory_blocks=memory_blocks,
llm_config=LLMConfig.default_config("gpt-4o-mini"),

2
tests/sdk_v1/integration/integration_test_send_message.py
View File

@@ -892,6 +892,7 @@ def agent_state(client: Letta) -> AgentState:
send_message_tool = client.tools.list(name="send_message").items[0]
agent_state_instance = client.agents.create(
name="supervisor",
agent_type="memgpt_v2_agent",
include_base_tools=False,
tool_ids=[send_message_tool.id, dice_tool.id],
model="openai/gpt-4o",
@@ -1942,6 +1943,7 @@ def test_auto_summarize(disable_e2b_api_key: Any, client: Letta, llm_config: LLM
send_message_tool = client.tools.list(name="send_message").items[0]
temp_agent_state = client.agents.create(
include_base_tools=False,
agent_type="memgpt_v2_agent",
tool_ids=[send_message_tool.id],
llm_config=pinned_context_window_llm_config,
embedding="letta/letta-free",

1
tests/sdk_v1/test_sdk_client.py
View File

@@ -1463,6 +1463,7 @@ def test_preview_payload(client: LettaSDKClient):
],
model="openai/gpt-4o-mini",
embedding="openai/text-embedding-3-small",
agent_type="memgpt_v2_agent",
)
try:

1
tests/test_letta_agent_batch.py
View File

@@ -123,6 +123,7 @@ async def agents(server, weather_tool):
tags=["test_agents"],
embedding="letta/letta-free",
tool_ids=[weather_tool.id],
agent_type="memgpt_v2_agent",
),
actor=actor,
)

25
tests/test_sdk_client.py
View File

@@ -91,6 +91,7 @@ def agent(client: LettaSDKClient):
],
model="openai/gpt-4o-mini",
embedding="openai/text-embedding-3-small",
agent_type="memgpt_v2_agent",
)
yield agent_state
@@ -285,6 +286,7 @@ def test_shared_blocks(client: LettaSDKClient):
block_ids=[block.id],
model="openai/gpt-4o-mini",
embedding="openai/text-embedding-3-small",
agent_type="memgpt_v2_agent",
)
agent_state2 = client.agents.create(
name="agent2",
@@ -297,6 +299,7 @@ def test_shared_blocks(client: LettaSDKClient):
block_ids=[block.id],
model="openai/gpt-4o-mini",
embedding="openai/text-embedding-3-small",
agent_type="memgpt_v2_agent",
)
# update memory
@@ -343,6 +346,7 @@ def test_read_only_block(client: LettaSDKClient):
],
model="openai/gpt-4o-mini",
embedding="openai/text-embedding-3-small",
agent_type="memgpt_v2_agent",
)
# make sure agent cannot update read-only block
@@ -386,6 +390,7 @@ def test_add_and_manage_tags_for_agent(client: LettaSDKClient):
],
model="openai/gpt-4o-mini",
embedding="openai/text-embedding-3-small",
agent_type="memgpt_v2_agent",
)
assert len(agent.tags) == 0
@@ -439,6 +444,7 @@ def test_agent_tags(client: LettaSDKClient):
model="openai/gpt-4o-mini",
embedding="openai/text-embedding-3-small",
tags=["test", "agent1", "production"],
agent_type="memgpt_v2_agent",
)
agent2 = client.agents.create(
@@ -451,6 +457,7 @@ def test_agent_tags(client: LettaSDKClient):
model="openai/gpt-4o-mini",
embedding="openai/text-embedding-3-small",
tags=["test", "agent2", "development"],
agent_type="memgpt_v2_agent",
)
agent3 = client.agents.create(
@@ -463,6 +470,7 @@ def test_agent_tags(client: LettaSDKClient):
model="openai/gpt-4o-mini",
embedding="openai/text-embedding-3-small",
tags=["test", "agent3", "production"],
agent_type="memgpt_v2_agent",
)
# Test getting all tags
@@ -856,6 +864,7 @@ def test_agent_creation(client: LettaSDKClient):
include_base_tools=False,
tags=["test"],
block_ids=[user_preferences_block.id],
agent_type="memgpt_v2_agent",
)
# Verify the agent was created successfully
@@ -893,6 +902,7 @@ def test_many_blocks(client: LettaSDKClient):
embedding="openai/text-embedding-3-small",
include_base_tools=False,
tags=["test"],
agent_type="memgpt_v2_agent",
)
agent2 = client.agents.create(
name=f"test_agent_{str(uuid.uuid4())}",
@@ -910,6 +920,7 @@ def test_many_blocks(client: LettaSDKClient):
embedding="openai/text-embedding-3-small",
include_base_tools=False,
tags=["test"],
agent_type="memgpt_v2_agent",
)
# Verify the agent was created successfully
@@ -948,6 +959,7 @@ def test_include_return_message_types(client: LettaSDKClient, agent: AgentState,
],
model="letta/letta-free",
embedding="letta/letta-free",
agent_type="memgpt_v2_agent",
)
if message_create == "stream_step":
@@ -1121,6 +1133,7 @@ def test_pydantic_inventory_management_tool(e2b_sandbox_mode, client: LettaSDKCl
embedding="openai/text-embedding-3-small",
tool_ids=[tool.id],
include_base_tools=False,
agent_type="memgpt_v2_agent",
)
response = client.agents.messages.create(
@@ -1211,6 +1224,7 @@ def test_pydantic_task_planning_tool(e2b_sandbox_mode, client: LettaSDKClient):
tool_rules=[
TerminalToolRule(tool_name=tool.name),
],
agent_type="memgpt_v2_agent",
)
response = client.agents.messages.create(
@@ -1299,6 +1313,7 @@ def test_preview_payload(client: LettaSDKClient):
],
model="openai/gpt-4o-mini",
embedding="openai/text-embedding-3-small",
agent_type="memgpt_v2_agent",
)
try:
@@ -1389,6 +1404,7 @@ def test_archive_tags_in_system_prompt(client: LettaSDKClient):
],
model="openai/gpt-4o-mini",
embedding="openai/text-embedding-3-small",
agent_type="memgpt_v2_agent",
)
try:
@@ -1467,6 +1483,7 @@ def test_agent_tools_list(client: LettaSDKClient):
model="openai/gpt-4o-mini",
embedding="openai/text-embedding-3-small",
include_base_tools=True,
agent_type="memgpt_v2_agent",
)
try:
@@ -1507,6 +1524,7 @@ def test_agent_tool_rules_deduplication(client: LettaSDKClient):
embedding="openai/text-embedding-3-small",
tool_rules=duplicate_rules,
include_base_tools=False,
agent_type="memgpt_v2_agent",
)
# Get the agent and check tool rules
@@ -1845,6 +1863,7 @@ def test_import_agent_file_from_disk(
include_base_tools=True,
tags=["test", "export", "import"],
system="You are a helpful assistant specializing in data analysis and mathematical computations.",
agent_type="memgpt_v2_agent",
)
# Add archival memory
@@ -1912,6 +1931,7 @@ def test_agent_serialization_v2(
include_base_tools=True,
tags=["test", "comprehensive", "serialization"],
system="You are a helpful assistant specializing in data analysis and mathematical computations.",
agent_type="memgpt_v2_agent",
)
# Add archival memory
@@ -2075,6 +2095,7 @@ def test_export_import_agent_with_files(client: LettaSDKClient):
model="openai/gpt-4o-mini",
embedding="openai/text-embedding-3-small",
source_ids=[source.id], # Attach the source with files
agent_type="memgpt_v2_agent",
)
# Verify the agent has the source and file blocks
@@ -2243,6 +2264,7 @@ def test_run_list(client: LettaSDKClient):
],
model="openai/gpt-4o-mini",
embedding="openai/text-embedding-3-small",
agent_type="memgpt_v2_agent",
)
# message an agent
@@ -2279,12 +2301,14 @@ async def test_create_batch(client: LettaSDKClient, server: SyncServer):
memory_blocks=[{"label": "persona", "value": "you are agent 1"}],
model="anthropic/claude-3-7-sonnet-20250219",
embedding="letta/letta-free",
agent_type="memgpt_v2_agent",
)
agent2 = client.agents.create(
name="agent2_batch",
memory_blocks=[{"label": "persona", "value": "you are agent 2"}],
model="anthropic/claude-3-7-sonnet-20250219",
embedding="letta/letta-free",
agent_type="memgpt_v2_agent",
)
# create a run
@@ -2353,6 +2377,7 @@ def test_create_agent(client: LettaSDKClient) -> None:
],
model="anthropic/claude-sonnet-4-20250514",
embedding="openai/text-embedding-ada-002",
agent_type="memgpt_v2_agent",
)
assert agent is not None
agents = client.agents.list()