Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion agent-memory-client/agent_memory_client/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
memory management capabilities for AI agents and applications.
"""

__version__ = "0.12.5"
__version__ = "0.12.6"

from .client import MemoryAPIClient, MemoryClientConfig, create_memory_client
from .exceptions import (
Expand Down
2 changes: 1 addition & 1 deletion agent-memory-client/agent_memory_client/client.py
Original file line number Diff line number Diff line change
Expand Up @@ -764,7 +764,7 @@ async def search_long_term_memory(
recency: RecencyConfig | None = None,
limit: int = 10,
offset: int = 0,
optimize_query: bool = True,
optimize_query: bool = False,
) -> MemoryRecordResults:
"""
Search long-term memories using semantic search and filters.
Expand Down
22 changes: 11 additions & 11 deletions tests/test_client_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -499,10 +499,10 @@ async def test_memory_prompt_integration(memory_test_client: MemoryAPIClient):


@pytest.mark.asyncio
async def test_search_long_term_memory_with_optimize_query_default_true(
async def test_search_long_term_memory_with_optimize_query_default_false(
memory_test_client: MemoryAPIClient,
):
"""Test that client search_long_term_memory uses optimize_query=True by default."""
"""Test that client search_long_term_memory uses optimize_query=False by default."""
with patch(
"agent_memory_server.long_term_memory.search_long_term_memories"
) as mock_search:
Expand All @@ -519,26 +519,26 @@ async def test_search_long_term_memory_with_optimize_query_default_true(
next_offset=None,
)

# Call search without optimize_query parameter (should default to True)
# Call search without optimize_query parameter (should default to False)
results = await memory_test_client.search_long_term_memory(
text="tell me about my preferences"
)

# Verify search was called with optimize_query=True (default)
# Verify search was called with optimize_query=False (default)
mock_search.assert_called_once()
call_kwargs = mock_search.call_args.kwargs
assert call_kwargs.get("optimize_query") is True
assert call_kwargs.get("optimize_query") is False

# Verify results
assert results.total == 1
assert len(results.memories) == 1


@pytest.mark.asyncio
async def test_search_long_term_memory_with_optimize_query_false_explicit(
async def test_search_long_term_memory_with_optimize_query_true_explicit(
memory_test_client: MemoryAPIClient,
):
"""Test that client search_long_term_memory can use optimize_query=False when explicitly set."""
"""Test that client search_long_term_memory can use optimize_query=True when explicitly set."""
with patch(
"agent_memory_server.long_term_memory.search_long_term_memories"
) as mock_search:
Expand All @@ -555,15 +555,15 @@ async def test_search_long_term_memory_with_optimize_query_false_explicit(
next_offset=None,
)

# Call search with explicit optimize_query=False
# Call search with explicit optimize_query=True
await memory_test_client.search_long_term_memory(
text="tell me about my preferences", optimize_query=False
text="tell me about my preferences", optimize_query=True
)

# Verify search was called with optimize_query=False
# Verify search was called with optimize_query=True
mock_search.assert_called_once()
call_kwargs = mock_search.call_args.kwargs
assert call_kwargs.get("optimize_query") is False
assert call_kwargs.get("optimize_query") is True


@pytest.mark.asyncio
Expand Down