55 lines
1.7 KiB
Python
Executable File
55 lines
1.7 KiB
Python
Executable File
#!/usr/bin/env python3
|
||
"""sessionStart: inject token-budget context and init turn counter."""
|
||
from __future__ import annotations
|
||
|
||
import json
|
||
import sys
|
||
from pathlib import Path
|
||
|
||
STATE_DIR = Path(__file__).resolve().parent / "state"
|
||
CONTEXT = """[Token Budget · auto]
|
||
省用量模式已开启：
|
||
1) 默认短会话、窄检索，禁止无必要全仓扫描
|
||
2) 日常任务用 Auto/Composer；不要主动升到 High/Grok High
|
||
3) 禁止默认并行多 Agent / Best-of-N
|
||
4) 长会话收口后建议用户新开聊天，先给摘要
|
||
用户说「确认高耗」或「继续本会话」时可放宽。"""
|
||
|
||
|
||
def main() -> None:
|
||
raw = sys.stdin.read().strip()
|
||
payload = json.loads(raw) if raw else {}
|
||
session_id = payload.get("session_id") or payload.get("conversation_id") or "unknown"
|
||
STATE_DIR.mkdir(parents=True, exist_ok=True)
|
||
state = {
|
||
"session_id": session_id,
|
||
"turns": 0,
|
||
"warned_long": False,
|
||
"warned_expensive": False,
|
||
}
|
||
state_path = STATE_DIR / f"session-{session_id}.json"
|
||
state_path.write_text(json.dumps(state, ensure_ascii=False, indent=2), encoding="utf-8")
|
||
(STATE_DIR / "latest-session.json").write_text(
|
||
json.dumps({"session_id": session_id}, ensure_ascii=False),
|
||
encoding="utf-8",
|
||
)
|
||
print(
|
||
json.dumps(
|
||
{
|
||
"env": {
|
||
"ONEOS_TOKEN_BUDGET": "1",
|
||
"ONEOS_TOKEN_BUDGET_SESSION": str(session_id),
|
||
},
|
||
"additional_context": CONTEXT,
|
||
},
|
||
ensure_ascii=False,
|
||
)
|
||
)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
try:
|
||
main()
|
||
except Exception:
|
||
print("{}")
|