mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-09 05:50:44 +08:00
oliver
ba3836f00f
- 在 oclaw/ 下重新初始化 Git 仓库 - 补齐子仓库 .gitignore，避免提交本地运行态数据（_local、node_modules、logs 等） - 提交当前工程代码与配置 Made-with: Cursor
66 lines
2.1 KiB
Python
66 lines
2.1 KiB
Python
from __future__ import annotations
|
|
|
|
import json
|
|
from typing import Any
|
|
|
|
from oclaw.platform.persistence.sqlite_store import ChatMessage, SqliteStore
|
|
|
|
|
|
def export_session_json(store: SqliteStore, session_id: str, message_limit: int = 50_000) -> str:
|
|
title = ""
|
|
s = store.get_session(session_id)
|
|
if s:
|
|
title = s.title
|
|
msgs = store.get_messages(session_id=session_id, limit=message_limit)
|
|
payload: dict[str, Any] = {
|
|
"session_id": session_id,
|
|
"title": title,
|
|
"messages": [_message_to_dict(m) for m in msgs],
|
|
}
|
|
return json.dumps(payload, ensure_ascii=False, indent=2)
|
|
|
|
|
|
def export_session_markdown(store: SqliteStore, session_id: str, message_limit: int = 50_000) -> str:
|
|
s = store.get_session(session_id)
|
|
title = (s.title if s else "") or session_id[:12]
|
|
lines = [f"# {title}", "", f"Session: `{session_id}`", ""]
|
|
for m in store.get_messages(session_id=session_id, limit=message_limit):
|
|
lines.extend(_message_to_markdown_lines(m))
|
|
lines.append("")
|
|
return "\n".join(lines).rstrip() + "\n"
|
|
|
|
|
|
def _message_to_dict(m: ChatMessage) -> dict[str, Any]:
|
|
return {
|
|
"id": m.id,
|
|
"role": m.role,
|
|
"content": m.content,
|
|
"tool_calls": m.tool_calls,
|
|
"attachments": m.attachments,
|
|
"timestamp": m.timestamp,
|
|
}
|
|
|
|
|
|
def _message_to_markdown_lines(m: ChatMessage) -> list[str]:
|
|
role = m.role.upper()
|
|
ts = m.timestamp or ""
|
|
head = f"## {role} · `{m.id}` · {ts}"
|
|
body = m.content or ""
|
|
if m.role == "tool" and m.tool_calls:
|
|
head = f"## TOOL · `{m.id}` · {ts}"
|
|
lines = [head, ""]
|
|
if body.strip():
|
|
lines.append(body)
|
|
lines.append("")
|
|
if m.tool_calls and m.role != "tool":
|
|
lines.append("```json")
|
|
lines.append(m.tool_calls)
|
|
lines.append("```")
|
|
elif m.tool_calls and m.role == "tool":
|
|
lines.append(f"_meta_: `{m.tool_calls}`")
|
|
if m.attachments:
|
|
lines.append("_attachments_: see JSON export for full data.")
|
|
return lines
|
|
|
|
|
|
__all__ = ["export_session_json", "export_session_markdown"]
|