mirror of
https://github.com/NCBM/plyngent.git
synced 2026-07-23 14:14:57 +08:00
NCBM
64d288a9b1
Store provider_name/model on sessions (SQLite migrate); restore on resume and persist on create//model//provider. Startup prefers session selection over re-asking. ask/choose and provider/model pick use readline with Tab completion when available.
50 lines
1.3 KiB
Python
50 lines
1.3 KiB
Python
from __future__ import annotations
|
|
|
|
from datetime import UTC, datetime
|
|
|
|
from plyngent.cli.export import (
|
|
encode_session_export_json,
|
|
format_session_export_md,
|
|
session_export_payload,
|
|
)
|
|
from plyngent.lmproto.openai_compatible.model import AssistantChatMessage, UserChatMessage
|
|
|
|
|
|
def test_format_session_export_md() -> None:
|
|
text = format_session_export_md(
|
|
[
|
|
UserChatMessage(content="hi"),
|
|
AssistantChatMessage(content="yo", reasoning_content="plan"),
|
|
],
|
|
sid=7,
|
|
name="demo",
|
|
workspace="/tmp/ws",
|
|
)
|
|
assert "# Session 7: demo" in text
|
|
assert "## user" in text
|
|
assert "hi" in text
|
|
assert "### reasoning" in text
|
|
assert "plan" in text
|
|
assert "yo" in text
|
|
|
|
|
|
def test_session_export_json_roundtrip() -> None:
|
|
messages = [UserChatMessage(content="a")]
|
|
now = datetime.now(UTC)
|
|
payload = session_export_payload(
|
|
sid=7,
|
|
name="demo",
|
|
workspace="/tmp/ws",
|
|
created_at=now,
|
|
updated_at=now,
|
|
messages=messages,
|
|
provider_name="local",
|
|
model="tiny",
|
|
)
|
|
assert payload.get("provider") == "local"
|
|
assert payload.get("model") == "tiny"
|
|
assert isinstance(payload, dict)
|
|
raw = encode_session_export_json(payload)
|
|
assert "7" in raw
|
|
assert "demo" in raw
|