mirror of
https://github.com/pewdiepie-archdaemon/odysseus.git
synced 2026-09-13 03:32:21 +02:00
30 lines
1.1 KiB
Python
30 lines
1.1 KiB
Python
from src.agent_loop import (
|
|
_streamed_tool_markup_complete,
|
|
_streamed_tool_markup_starts,
|
|
_strip_incomplete_tool_markup_tail,
|
|
)
|
|
from src.tool_parsing import strip_tool_blocks
|
|
|
|
|
|
def test_dsml_tool_markup_is_complete_only_after_closing_tag():
|
|
start = "<｜｜DSML｜｜tool_calls><｜｜DSML｜｜invoke name=\"host_shell\">"
|
|
assert not _streamed_tool_markup_complete(start)
|
|
assert _streamed_tool_markup_complete(start + "</｜｜DSML｜｜invoke>")
|
|
|
|
|
|
def test_dsml_tool_markup_strips_to_empty_visible_text():
|
|
markup = (
|
|
"<｜｜DSML｜｜tool_calls>"
|
|
"<｜｜DSML｜｜invoke name=\"host_shell\">"
|
|
"<｜｜DSML｜｜parameter name=\"command\" string=\"true\">pwd"
|
|
"</｜｜DSML｜｜parameter>"
|
|
"</｜｜DSML｜｜invoke>"
|
|
"</｜｜DSML｜｜tool_calls>"
|
|
)
|
|
assert strip_tool_blocks(markup, skip_fenced=True) == ""
|
|
|
|
|
|
def test_chunk_split_tool_call_prefix_is_buffered_and_not_persisted():
|
|
assert _streamed_tool_markup_starts('I will search<tool_cal')
|
|
assert _strip_incomplete_tool_markup_tail('I will search<tool_cal') == 'I will search'
|