From 8ff8ef21027f6dec7de950e9d592f9e8fcae5102 Mon Sep 17 00:00:00 2001 From: jnjpng Date: Thu, 6 Nov 2025 16:54:06 -0800 Subject: [PATCH] chore: clean up count agent test (#6037) base Co-authored-by: Letta Bot --- tests/sdk_v1/test_sdk_client.py | 33 --------------------------------- 1 file changed, 33 deletions(-) diff --git a/tests/sdk_v1/test_sdk_client.py b/tests/sdk_v1/test_sdk_client.py index e9ac9e79..e08cb5d5 100644 --- a/tests/sdk_v1/test_sdk_client.py +++ b/tests/sdk_v1/test_sdk_client.py @@ -416,39 +416,6 @@ def test_add_and_manage_tags_for_agent(client: LettaSDKClient): client.agents.delete(agent.id) -def test_count_agents(client: LettaSDKClient): - """Test counting agents.""" - # Create a few agents - agent1 = client.agents.create( - memory_blocks=[CreateBlockParam(label="persona", value="test")], - model="openai/gpt-4o-mini", - embedding="openai/text-embedding-3-small", - ) - agent2 = client.agents.create( - memory_blocks=[CreateBlockParam(label="persona", value="test")], - model="openai/gpt-4o-mini", - embedding="openai/text-embedding-3-small", - ) - - try: - # Get initial count - initial_count = client.agents.count() - - # Verify we have at least 2 agents - assert initial_count >= 2, f"Expected at least 2 agents, got {initial_count}" - - # List agents to verify count matches - agents_list = client.agents.list() - # Count should be consistent with list (though list may be paginated) - assert isinstance(initial_count, int), "Count should return an integer" - assert initial_count > 0, "Count should be positive" - - finally: - # Clean up - client.agents.delete(agent_id=agent1.id) - client.agents.delete(agent_id=agent2.id) - - def test_reset_messages(client: LettaSDKClient): """Test resetting messages for an agent.""" # Create an agent