This commit is contained in:
Timothy Jaeryang Baek 2026-09-16 00:34:24 -04:00
parent 6d8e63e366
commit e9a0164690

View file

@ -241,6 +241,7 @@ async def update_memories(
operations = validate_memory_operations(form_data)
metadata = getattr(request.state, 'metadata', {}) or {}
model = metadata.get('model')
source = form_data.source or 'tool'
for operation in operations:
if operation.get('action') in {'add', 'replace', 'move'}:
@ -248,7 +249,7 @@ async def update_memories(
'created_by': source,
'chat_id': metadata.get('chat_id'),
'message_id': metadata.get('message_id'),
'model': metadata.get('model'),
'model': model.get('id') if isinstance(model, dict) else None,
}
try:
@ -263,7 +264,7 @@ async def update_memories(
for result in results:
memory = result.get('memory')
if isinstance(memory, MemoryModel):
result = {**result, 'memory': memory.model_dump()}
result = {**result, 'memory': memory.model_dump(exclude={'meta'})}
if result.get('status') in {'created', 'updated'}:
vector = await request.app.state.EMBEDDING_FUNCTION(
memory_vector_text(memory.content, memory.path),
@ -438,7 +439,7 @@ async def read_memory_path(
)
return {
**result,
'memories': [memory.model_dump() for memory in result['memories']],
'memories': [memory.model_dump(exclude={'meta'}) for memory in result['memories']],
}