agent-Specialization/server/chat_flow_runner_helpers.py

58 lines
1.8 KiB
Python

from __future__ import annotations
import re
from typing import Any, Dict, List, Optional
def extract_intent_from_partial(arg_str: str) -> Optional[str]:
"""从不完整的JSON字符串中粗略提取 intent 字段,容错用于流式阶段。"""
if not arg_str or "intent" not in arg_str:
return None
match = re.search(r'"intent"\s*:\s*"([^"]{0,128})', arg_str, re.IGNORECASE | re.DOTALL)
if match:
return match.group(1)
return None
def resolve_monitor_path(args: Dict[str, Any], fallback: Optional[str] = None) -> Optional[str]:
candidates = [
args.get('path'),
args.get('target_path'),
args.get('file_path'),
args.get('destination_path'),
fallback,
]
for candidate in candidates:
if isinstance(candidate, str):
trimmed = candidate.strip()
if trimmed:
return trimmed
return None
def resolve_monitor_memory(entries: Any, entry_limit: int) -> Optional[List[str]]:
if isinstance(entries, list):
return [str(item) for item in entries][:entry_limit]
return None
def capture_monitor_snapshot(file_manager, path: Optional[str], snapshot_char_limit: int, debug_logger) -> Optional[Dict[str, Any]]:
if not path:
return None
try:
read_result = file_manager.read_file(path)
except Exception as exc:
debug_logger(f"[MonitorSnapshot] 读取文件失败: {path} ({exc})")
return None
if not isinstance(read_result, dict) or not read_result.get('success'):
return None
content = read_result.get('content')
if not isinstance(content, str):
content = ''
if len(content) > snapshot_char_limit:
content = content[:snapshot_char_limit]
return {
'path': read_result.get('path') or path,
'content': content,
}