mirror of
https://github.com/open-webui/open-webui.git
synced 2026-09-27 17:56:26 -04:00
refac
This commit is contained in:
@@ -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']],
|
||||
}
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user