Skip to content

Commit ed3cdaa

Browse files
switches to new metadata field
1 parent 1672c0e commit ed3cdaa

3 files changed

Lines changed: 7 additions & 7 deletions

File tree

python/packages/autogen-ext/src/autogen_ext/memory/redis/_redis_memory.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -255,7 +255,7 @@ async def add(self, content: MemoryContent, cancellation_token: CancellationToke
255255
metadata = {"mime_type": mime_type}
256256
metadata.update(content.metadata if content.metadata else {})
257257
self.message_history.add_message(
258-
{"role": "user", "content": memory_content, "tool_call_id": serialize(metadata)} # type: ignore[reportArgumentType]
258+
{"role": "user", "content": memory_content, "metadata": serialize(metadata)} # type: ignore[reportArgumentType]
259259
)
260260

261261
async def query(
@@ -324,7 +324,7 @@ async def query(
324324

325325
memories: List[MemoryContent] = []
326326
for result in results: # type: ignore[reportUnkownVariableType]
327-
metadata = deserialize(result["tool_call_id"]) # type: ignore[reportArgumentType]
327+
metadata = deserialize(result["metadata"]) # type: ignore[reportArgumentType]
328328
mime_type = MemoryMimeType(metadata.pop("mime_type"))
329329
if mime_type in (MemoryMimeType.TEXT, MemoryMimeType.MARKDOWN):
330330
memory_content = result["content"] # type: ignore[reportArgumentType]

python/packages/autogen-ext/tests/memory/test_redis_memory.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ async def test_redis_memory_query_with_mock() -> None:
3636
memory = RedisMemory(config=config)
3737

3838
mock_history.get_relevant.return_value = [
39-
{"content": "test content", "tool_call_id": '{"foo": "bar", "mime_type": "text/plain"}'}
39+
{"content": "test content", "metadata": '{"foo": "bar", "mime_type": "text/plain"}'}
4040
]
4141
result = await memory.query("test")
4242
assert len(result.results) == 1

python/uv.lock

Lines changed: 4 additions & 4 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)