mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-08-28 05:25:04 +00:00
8b82ff88d0
* feat(evolve): enhance agent reply processing and logging capabilities - Add agent_reply_result_text function to extract final user-visible text from agent replies - Implement comprehensive logging throughout auto_memory, auto_resource, and dream modules - Add max_units configuration option to limit extracted memory units - Improve error handling and validation in auto_resource step - Refactor dream extract step to respect max_units limit during processing - Enhance summary rendering in dream finish step with detailed breakdown - Add result hook functionality for embedding hosts integration - Implement loose resource filename handling for root-level resources - Update test cases to reflect new functionality and improved error messages * test(background-steps): update fake upsert function to include created parameter - Modified fake_upsert function to accept 'created' parameter instead of '_created' - Added 'created' field to captured dictionary in fake_upsert function - Included 'created': True in the expected response dictionary for test case - Updated test assertion to match new parameter structure
42 lines
1.5 KiB
Python
42 lines
1.5 KiB
Python
"""Shared helpers for evolve steps."""
|
|
|
|
import datetime
|
|
import zoneinfo
|
|
|
|
from agentscope.message import Msg
|
|
|
|
|
|
def now(timezone: str | None = None) -> datetime.datetime:
|
|
"""Return current datetime in the given IANA timezone, falling back to local."""
|
|
if not timezone:
|
|
return datetime.datetime.now()
|
|
try:
|
|
return datetime.datetime.now(zoneinfo.ZoneInfo(timezone))
|
|
except Exception:
|
|
return datetime.datetime.now()
|
|
|
|
|
|
def format_history(messages: list[Msg], include_timestamp: bool = True) -> str:
|
|
"""Render a conversation slice as a human-readable transcript."""
|
|
lines: list[str] = []
|
|
for msg in messages:
|
|
text = (msg.get_text_content() or "").strip()
|
|
if not text:
|
|
continue
|
|
speaker = msg.name or msg.role or "?"
|
|
header = f"[{speaker} @ {msg.created_at}]" if include_timestamp else f"[{speaker}]"
|
|
lines.append(f"{header}\n{text}")
|
|
return "\n\n".join(lines) or "(empty)"
|
|
|
|
|
|
def agent_reply_result_text(reply_result: dict) -> str:
|
|
"""Return the final user-visible text block from an agent reply result."""
|
|
last_message = reply_result.get("last_message") or {}
|
|
content = last_message.get("content") if isinstance(last_message, dict) else None
|
|
if isinstance(content, list):
|
|
for block in reversed(content):
|
|
if isinstance(block, dict) and block.get("type") == "text":
|
|
text = str(block.get("text") or "").strip()
|
|
if text:
|
|
return text
|
|
return str(reply_result.get("result") or "").strip()
|