mirror of
https://github.com/NCBM/plyngent.git
synced 2026-07-23 14:14:57 +08:00
16 lines
395 B
Python
16 lines
395 B
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from plyngent.agent.budget import truncate_tool_result
|
||
|
|
|
||
|
|
|
||
|
|
def test_truncate_tool_result_short() -> None:
|
||
|
|
assert truncate_tool_result("hello", 100) == "hello"
|
||
|
|
|
||
|
|
|
||
|
|
def test_truncate_tool_result_long() -> None:
|
||
|
|
text = "a" * 50
|
||
|
|
out = truncate_tool_result(text, 20)
|
||
|
|
assert out.startswith("a" * 20)
|
||
|
|
assert "truncated" in out
|
||
|
|
assert "30" in out
|