feat: add endpoint to generate memory from session

Merge https://github.com/google/adk-python/pull/2900

In relation with #2416

COPYBARA_INTEGRATE_REVIEW=https://github.com/google/adk-python/pull/2900 from guillaumeblaquiere:add-session-to-memory 0507de43021c62f9223167dca8f53b536227ad04
PiperOrigin-RevId: 808658162
This commit is contained in:
guillaume blaquiere
2025-09-18 11:13:21 -07:00
committed by Copybara-Service
parent 6b49391546
commit 25958242db
2 changed files with 58 additions and 1 deletions
+15 -1
View File
@@ -22,6 +22,7 @@ import tempfile
import time
from typing import Any
from typing import Optional
from unittest.mock import AsyncMock
from unittest.mock import MagicMock
from unittest.mock import patch
@@ -344,7 +345,7 @@ def mock_artifact_service():
@pytest.fixture
def mock_memory_service():
"""Create a mock memory service."""
return MagicMock()
return AsyncMock()
@pytest.fixture
@@ -939,5 +940,18 @@ def test_a2a_disabled_by_default(test_app):
logger.info("A2A disabled by default test passed")
def test_patch_memory(test_app, create_test_session, mock_memory_service):
"""Test adding a session to memory."""
info = create_test_session
url = f"/apps/{info['app_name']}/users/{info['user_id']}/memory"
payload = {"session_id": info["session_id"]}
response = test_app.patch(url, json=payload)
# Verify the response
assert response.status_code == 200
mock_memory_service.add_session_to_memory.assert_called_once()
logger.info("Add session to memory test completed successfully")
if __name__ == "__main__":
pytest.main(["-xvs", __file__])