mirror of
https://github.com/agent0ai/agent-zero.git
synced 2026-05-26 15:45:45 +00:00
6a248a57cc
* Add human-in-the-loop survey helper Co-authored-by: nic <nicsins@users.noreply.github.com> * Make survey helper launcher robust and add CLI fallback Co-authored-by: nic <nicsins@users.noreply.github.com> --------- Co-authored-by: Cursor Agent <cursoragent@cursor.com> Co-authored-by: nic <nicsins@users.noreply.github.com>
68 lines
2.3 KiB
Python
68 lines
2.3 KiB
Python
from __future__ import annotations
|
|
|
|
import json
|
|
from dataclasses import dataclass, field
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
|
|
DEFAULT_PROFILE_PATH = Path("memory") / "survey_profile.json"
|
|
|
|
|
|
@dataclass
|
|
class SurveyProfile:
|
|
"""
|
|
A user-owned profile used to answer surveys honestly and consistently.
|
|
|
|
This is not a "made-up persona" generator. It's meant to store real, user-approved facts
|
|
and preferences that can be reused across surveys.
|
|
"""
|
|
|
|
demographics: dict[str, Any] = field(default_factory=dict)
|
|
preferences: dict[str, Any] = field(default_factory=dict)
|
|
writing_style: dict[str, Any] = field(default_factory=dict)
|
|
misc: dict[str, Any] = field(default_factory=dict)
|
|
|
|
@classmethod
|
|
def load(cls, path: str | Path = DEFAULT_PROFILE_PATH) -> "SurveyProfile":
|
|
p = Path(path)
|
|
if not p.exists():
|
|
return cls()
|
|
data = json.loads(p.read_text(encoding="utf-8"))
|
|
if not isinstance(data, dict):
|
|
return cls()
|
|
return cls(
|
|
demographics=dict(data.get("demographics") or {}),
|
|
preferences=dict(data.get("preferences") or {}),
|
|
writing_style=dict(data.get("writing_style") or {}),
|
|
misc=dict(data.get("misc") or {}),
|
|
)
|
|
|
|
def save(self, path: str | Path = DEFAULT_PROFILE_PATH) -> None:
|
|
p = Path(path)
|
|
p.parent.mkdir(parents=True, exist_ok=True)
|
|
payload = {
|
|
"demographics": self.demographics,
|
|
"preferences": self.preferences,
|
|
"writing_style": self.writing_style,
|
|
"misc": self.misc,
|
|
}
|
|
p.write_text(json.dumps(payload, indent=2, ensure_ascii=False) + "\n", encoding="utf-8")
|
|
|
|
def as_dict(self) -> dict[str, Any]:
|
|
return {
|
|
"demographics": self.demographics,
|
|
"preferences": self.preferences,
|
|
"writing_style": self.writing_style,
|
|
"misc": self.misc,
|
|
}
|
|
|
|
def merge_updates(self, updates: dict[str, Any]) -> None:
|
|
"""
|
|
Shallow-merge updates into the profile. Intended for user-approved updates.
|
|
"""
|
|
for key in ("demographics", "preferences", "writing_style", "misc"):
|
|
val = updates.get(key)
|
|
if isinstance(val, dict):
|
|
getattr(self, key).update(val)
|
|
|