mirror of
https://github.com/infiniflow/ragflow.git
synced 2025-12-19 12:06:42 +08:00
Feat/memory (#11812)
### What problem does this PR solve? Manage and display memory datasets. ### Type of change - [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
54
api/utils/memory_utils.py
Normal file
54
api/utils/memory_utils.py
Normal file
@ -0,0 +1,54 @@
|
||||
#
|
||||
# Copyright 2025 The InfiniFlow Authors. All Rights Reserved.
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
#
|
||||
from typing import List
|
||||
from common.constants import MemoryType
|
||||
|
||||
def format_ret_data_from_memory(memory):
|
||||
return {
|
||||
"id": memory.id,
|
||||
"name": memory.name,
|
||||
"avatar": memory.avatar,
|
||||
"tenant_id": memory.tenant_id,
|
||||
"owner_name": memory.owner_name if hasattr(memory, "owner_name") else None,
|
||||
"memory_type": get_memory_type_human(memory.memory_type),
|
||||
"storage_type": memory.storage_type,
|
||||
"embd_id": memory.embd_id,
|
||||
"llm_id": memory.llm_id,
|
||||
"permissions": memory.permissions,
|
||||
"description": memory.description,
|
||||
"memory_size": memory.memory_size,
|
||||
"forgetting_policy": memory.forgetting_policy,
|
||||
"temperature": memory.temperature,
|
||||
"system_prompt": memory.system_prompt,
|
||||
"user_prompt": memory.user_prompt,
|
||||
"create_time": memory.create_time,
|
||||
"create_date": memory.create_date,
|
||||
"update_time": memory.update_time,
|
||||
"update_date": memory.update_date
|
||||
}
|
||||
|
||||
|
||||
def get_memory_type_human(memory_type: int) -> List[str]:
|
||||
return [mem_type.name.lower() for mem_type in MemoryType if memory_type & mem_type.value]
|
||||
|
||||
|
||||
def calculate_memory_type(memory_type_name_list: List[str]) -> int:
|
||||
memory_type = 0
|
||||
type_value_map = {mem_type.name.lower(): mem_type.value for mem_type in MemoryType}
|
||||
for mem_type in memory_type_name_list:
|
||||
if mem_type in type_value_map:
|
||||
memory_type |= type_value_map[mem_type]
|
||||
return memory_type
|
||||
Reference in New Issue
Block a user