Preference

初始化与调用

import asyncio
import json
from agentscope.message import Msg
from agentscope.model import OpenAIChatModel
from config_agentscope import init_agentscope
from config import LLM_CONFIG
from context.memory_manager import MemoryManager
from agents.preference_agent import PreferenceAgent

async def save_preference(user_query: str, user_id: str = "default_user", session_id: str = "default"):
    init_agentscope()
    model = OpenAIChatModel(
        model_name=LLM_CONFIG["model_name"],
        api_key=LLM_CONFIG["api_key"],
        client_kwargs={"base_url": LLM_CONFIG["base_url"], "timeout": 60},
        temperature=LLM_CONFIG.get("temperature", 0.7),
        max_tokens=LLM_CONFIG.get("max_tokens", 2000),
    )
    memory_manager = MemoryManager(user_id=user_id, session_id=session_id, llm_model=model)
    agent = PreferenceAgent(name="PreferenceAgent", model=model, memory_manager=memory_manager)
    user_msg = Msg(name="user", content=user_query, role="user")
    result = await agent.reply(user_msg)
    data = json.loads(result.content) if isinstance(result.content, str) else result.content
    if data.get("has_preferences") and data.get("preferences"):
        for item in data["preferences"]:
            pref_type = item.get("type")
            value = item.get("value")
            action = item.get("action", "replace")
            current = memory_manager.long_term.get_preference().get(pref_type)
            if action == "append" and isinstance(current, list):
                memory_manager.long_term.save_preference(pref_type, current + [value])
            else:
                memory_manager.long_term.save_preference(pref_type, value)
    return data

# 使用
data = asyncio.run(save_preference("我还喜欢如家酒店"))
# data: {"preferences": [{"type": "hotel_brand_preference", "value": "如家", "action": "append"}], "has_preferences": true}

返回格式

• preferences: 列表，每项 { "type", "value", "action": "append"|"replace" }
• has_preferences: bool

偏好提取规则（企业差旅画像）

【任务说明】 你需要判断用户的意图：

1. 追加（append）：用户想在已有偏好基础上增加新的选项

   • 关键词：「还」、「也」、「另外」、「以及」
   • 示例："我还喜欢汉庭" → 追加到 hotel_brand_preference
   • 示例："我还接受经济舱" → 追加到 travel_tier（如果该字段为列表）

2. 覆盖（replace）：用户想更新/替换原有的偏好

   • 关键词：「搬家到」、「改成」、「现在是」、「换成」
   • 示例："我常驻地改成上海" → 覆盖 resident_city
   • 示例："我现在喜欢靠窗座位" → 覆盖 seat_preference

3. 首次设置：用户第一次提及某个偏好

   • 如果当前偏好中没有这个字段，默认使用 replace

【常见偏好类型】

• resident_city: 常驻地
• travel_tier: 出行等级（如经济舱/高铁二等座/行政舱）
• seat_preference: 座位偏好
• hotel_brand_preference: 酒店品牌偏好
• reimbursement_preference: 报销偏好（如电子发票优先、按项目归集） （支持自定义新的偏好类型）

【输出格式】(严格JSON) {{ "preferences": [ {{ "type": "hotel_brand_preference", "value": "汉庭", "action": "append" }}, {{ "type": "resident_city", "value": "上海", "action": "replace" }} ], "has_preferences": true }}

【重要规则】

1. action 只能是 "append" 或 "replace"
2. 根据用户的语气和上下文判断 action
3. 如果用户使用「还」、「也」等词，使用 append
4. 如果用户使用「搬家」、「改成」等词，使用 replace
5. 如果是首次提及，使用 replace
6. 如果用户未提及任何偏好，返回 {{"preferences": [], "has_preferences": false}}