adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor(memory): tighten metadata field types and clean up descriptions

Browse Source 
- Use Literal['set', 'remove'] for MetadataFieldChange.action instead of str
- Simplify field_path description to reflect current schema
- Remove redundant isinstance check in extract_user_metadata_task
This commit is contained in:
lanceyq
2026-04-16 17:29:00 +08:00
parent 73fbc19747
commit 643f69bb90
2 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/app/core/memory/models/metadata_models.py
View File

@@ -4,7 +4,7 @@ Independent from triplet_models.py - these models are used by the
standalone metadata extraction pipeline (post-dedup async Celery task). standalone metadata extraction pipeline (post-dedup async Celery task).
""" """
from typing import List, Optional from typing import List, Literal, Optional
from pydantic import BaseModel, ConfigDict, Field from pydantic import BaseModel, ConfigDict, Field
@@ -35,9 +35,9 @@ class MetadataFieldChange(BaseModel):
model_config = ConfigDict(extra="ignore") model_config = ConfigDict(extra="ignore")
field_path: str = Field( field_path: str = Field(
description="字段路径，用点号分隔，如 'profile.role'、'knowledge_tags'、'behavioral_hints.tone_preference'" description="字段路径，用点号分隔，如 'profile.role'、'profile.expertise'"
) )
action: str = Field( action: Literal["set", "remove"] = Field(
description="操作类型：'set' 表示新增或修改，'remove' 表示移除" description="操作类型：'set' 表示新增或修改，'remove' 表示移除"
) )
value: Optional[str] = Field( value: Optional[str] = Field(

2
api/app/tasks.py
View File

@@ -3157,8 +3157,6 @@ def extract_user_metadata_task(
leaf = parts[-1] leaf = parts[-1]
current_list = target.get(leaf, []) current_list = target.get(leaf, [])
if not isinstance(current_list, list):
current_list = []
if action == "set": if action == "set":
if value not in current_list: if value not in current_list: