mirror of
https://github.com/run-llama/fs-explorer.git
synced 2026-07-21 03:05:23 -04:00
Clelia (Astra) Bertelli
42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
from fs_explorer.models import (
|
|
ToolCallAction,
|
|
Action,
|
|
ToolCallArg,
|
|
GoDeeperAction,
|
|
StopAction,
|
|
)
|
|
|
|
|
|
def test_tool_call_action_to_tool_args() -> None:
|
|
tool_call_action = ToolCallAction(
|
|
tool_name="glob",
|
|
tool_input=[
|
|
ToolCallArg(parameter_name="directory", parameter_value="tests/testfiles"),
|
|
ToolCallArg(parameter_name="pattern", parameter_value="file?.*"),
|
|
],
|
|
)
|
|
assert tool_call_action.to_fn_args() == {
|
|
"directory": "tests/testfiles",
|
|
"pattern": "file?.*",
|
|
}
|
|
|
|
|
|
def test_action_to_action_type() -> None:
|
|
action = Action(
|
|
action=ToolCallAction(
|
|
tool_name="glob",
|
|
tool_input=[
|
|
ToolCallArg(
|
|
parameter_name="directory", parameter_value="tests/testfiles"
|
|
),
|
|
ToolCallArg(parameter_name="pattern", parameter_value="file?.*"),
|
|
],
|
|
),
|
|
reason="",
|
|
)
|
|
assert action.to_action_type() == "toolcall"
|
|
action = Action(action=GoDeeperAction(directory="tests/testfiles/last"), reason="")
|
|
assert action.to_action_type() == "godeeper"
|
|
action = Action(action=StopAction(final_result="hello"), reason="")
|
|
assert action.to_action_type() == "stop"
|