mirror of
https://github.com/open-webui/open-webui.git
synced 2026-09-17 23:52:29 +00:00
refac
This commit is contained in:
parent
6d8e63e366
commit
e9a0164690
1 changed files with 4 additions and 3 deletions
|
|
@ -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']],
|
||||
}
|
||||
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue