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

feat: Write tests for search messages [LET-4212] (#4447)

Browse Source 
* Adjust naming

* Add testing and improve message search

* Adjust comments

* Change query text to query

* Fern autogen
This commit is contained in:
Matthew Zhou
2025-09-05 17:52:13 -07:00
committed by GitHub
parent cb7296c81d
commit af302e8ac8
10 changed files with 2678 additions and 2659 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/integration_test_turbopuffer.py
View File

@@ -1937,7 +1937,7 @@ class TestNamespaceTracking:
async def test_agent_namespace_tracking(self, server, default_user, sarah_agent, enable_message_embedding):
"""Test that agent message namespaces are properly tracked in database"""
# Get namespace - should be generated and stored
namespace = await server.agent_manager.get_or_set_vector_db_namespace_async(sarah_agent.id, default_user.organization_id)
namespace = await server.agent_manager.get_or_set_vector_db_namespace_async(default_user.organization_id)
# Should have messages_org_ prefix and environment suffix
expected_prefix = "messages_"
@@ -1947,7 +1947,7 @@ class TestNamespaceTracking:
assert settings.environment.lower() in namespace
# Call again - should return same namespace from database
namespace2 = await server.agent_manager.get_or_set_vector_db_namespace_async(sarah_agent.id, default_user.organization_id)
namespace2 = await server.agent_manager.get_or_set_vector_db_namespace_async(default_user.organization_id)
assert namespace == namespace2
@pytest.mark.asyncio