fix(client): prevent feedback telemetry TypeError (#4795)

This commit is contained in:
Asish Kumar
2026-04-12 03:06:05 +05:30
committed by GitHub
parent 9d6b79a14e
commit c239d8a483
2 changed files with 61 additions and 2 deletions
+2 -2
View File
@@ -873,7 +873,7 @@ class MemoryClient:
response = self.client.post("/v1/feedback/", json=data)
response.raise_for_status()
capture_client_event("client.feedback", self, data, {"sync_type": "sync"})
capture_client_event("client.feedback", self, {**data, "sync_type": "sync"})
return response.json()
def _prepare_payload(self, messages: List[Dict[str, str]], kwargs: Dict[str, Any]) -> Dict[str, Any]:
@@ -1743,5 +1743,5 @@ class AsyncMemoryClient:
response = await self.async_client.post("/v1/feedback/", json=data)
response.raise_for_status()
capture_client_event("client.feedback", self, data, {"sync_type": "async"})
capture_client_event("client.feedback", self, {**data, "sync_type": "async"})
return response.json()
+59
View File
@@ -0,0 +1,59 @@
from unittest.mock import AsyncMock, MagicMock, patch
import pytest
from mem0.client.main import AsyncMemoryClient, MemoryClient
def _build_memory_client(response_payload=None):
client = MemoryClient.__new__(MemoryClient)
client.user_email = "user@example.com"
client.client = MagicMock()
response = client.client.post.return_value
response.json.return_value = response_payload or {"message": "Feedback recorded"}
response.raise_for_status.return_value = None
return client
def test_memory_client_feedback_uses_single_telemetry_payload():
client = _build_memory_client()
with patch("mem0.client.main.capture_client_event") as mock_capture:
result = client.feedback("mem_1", feedback="positive", feedback_reason="accurate")
assert result == {"message": "Feedback recorded"}
client.client.post.assert_called_once_with(
"/v1/feedback/",
json={"memory_id": "mem_1", "feedback": "POSITIVE", "feedback_reason": "accurate"},
)
mock_capture.assert_called_once_with(
"client.feedback",
client,
{"memory_id": "mem_1", "feedback": "POSITIVE", "feedback_reason": "accurate", "sync_type": "sync"},
)
@pytest.mark.asyncio
async def test_async_memory_client_feedback_uses_single_telemetry_payload():
client = AsyncMemoryClient.__new__(AsyncMemoryClient)
client.user_email = "user@example.com"
client.async_client = MagicMock()
response = MagicMock()
response.json.return_value = {"message": "Feedback recorded"}
response.raise_for_status.return_value = None
client.async_client.post = AsyncMock(return_value=response)
with patch("mem0.client.main.capture_client_event") as mock_capture:
result = await client.feedback("mem_1", feedback="negative", feedback_reason="outdated")
assert result == {"message": "Feedback recorded"}
client.async_client.post.assert_awaited_once_with(
"/v1/feedback/",
json={"memory_id": "mem_1", "feedback": "NEGATIVE", "feedback_reason": "outdated"},
)
mock_capture.assert_called_once_with(
"client.feedback",
client,
{"memory_id": "mem_1", "feedback": "NEGATIVE", "feedback_reason": "outdated", "sync_type": "async"},
)