mirror of
https://github.com/NCBM/plyngent.git
synced 2026-07-23 22:44:59 +08:00
NCBM
b1964951ea
Truncate large tool outputs and run multiple tool calls in one round concurrently when enabled.
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
|