mirror of
https://github.com/nesquena/hermes-webui.git
synced 2026-05-27 04:00:37 +00:00
Frank Song
35 lines
1.2 KiB
Python
35 lines
1.2 KiB
Python
from api.streaming import (
|
|
_fallback_title_from_exchange,
|
|
_strip_workspace_prefix,
|
|
_workspace_context_prefix,
|
|
)
|
|
|
|
|
|
def test_workspace_prefix_strips_only_versioned_sentinel():
|
|
assert _strip_workspace_prefix("[Workspace::v1: /tmp/project]\nHello") == "Hello"
|
|
assert _strip_workspace_prefix("[Workspace: /tmp/project]\nHello") == "[Workspace: /tmp/project]\nHello"
|
|
|
|
|
|
def test_workspace_prefix_escapes_paths_with_closing_brackets():
|
|
prefix = _workspace_context_prefix("/tmp/proj-[wip]/src")
|
|
|
|
assert prefix == "[Workspace::v1: /tmp/proj-[wip\\]/src]\n"
|
|
assert _strip_workspace_prefix(f"{prefix}Continue") == "Continue"
|
|
|
|
|
|
def test_legacy_workspace_prefix_only_strips_for_compatibility_callers():
|
|
legacy = "[Workspace: /tmp/project]\nContinue"
|
|
|
|
assert _strip_workspace_prefix(legacy) == legacy
|
|
assert _strip_workspace_prefix(legacy, include_legacy=True) == "Continue"
|
|
|
|
|
|
def test_user_typed_legacy_workspace_prefix_survives_fallback_title():
|
|
title = _fallback_title_from_exchange(
|
|
"[Workspace: /tmp/project]\nExplain this literal prefix",
|
|
"Sure",
|
|
)
|
|
|
|
assert title is not None
|
|
assert title.startswith("Workspace tmp/project")
|