wuyang/zk-data-agent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add assistant-ui data agent frontend

Browse Source
This commit is contained in:
武阳
2026-05-06 16:18:32 +08:00
parent d6a2359dc1
commit 7d4ae3e7ba
119 changed files with 14330 additions and 14420 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/session_store.py
+7 -2
View File
@@ -72,14 +72,19 @@ class StoredAgentSession:
def save_agent_session(session: StoredAgentSession, directory: Path | None = None) -> Path:
target_dir = directory or DEFAULT_AGENT_SESSION_DIR
target_dir.mkdir(parents=True, exist_ok=True)
path = target_dir / f'{session.session_id}.json'
session_dir = target_dir / session.session_id
session_dir.mkdir(parents=True, exist_ok=True)
path = session_dir / 'session.json'
path.write_text(json.dumps(asdict(session), indent=2), encoding='utf-8')
return path
def load_agent_session(session_id: str, directory: Path | None = None) -> StoredAgentSession:
target_dir = directory or DEFAULT_AGENT_SESSION_DIR
data = json.loads((target_dir / f'{session_id}.json').read_text(encoding='utf-8'))
path = target_dir / session_id / 'session.json'
if not path.exists():
path = target_dir / f'{session_id}.json'
data = json.loads(path.read_text(encoding='utf-8'))
return StoredAgentSession(
session_id=data['session_id'],
model_config=dict(data['model_config']),