@@ -51,7 +51,7 @@ def test_parse_single_global(self):
5151 from seclab_taskflow_agent .__main__ import parse_prompt_args
5252 available_tools = AvailableTools ()
5353
54- p , t , l , cli_globals , user_prompt = parse_prompt_args (
54+ p , t , l , cli_globals , user_prompt , _ = parse_prompt_args (
5555 available_tools , "-t example -g fruit=apples" )
5656
5757 assert t == "example"
@@ -64,7 +64,7 @@ def test_parse_multiple_globals(self):
6464 from seclab_taskflow_agent .__main__ import parse_prompt_args
6565 available_tools = AvailableTools ()
6666
67- p , t , l , cli_globals , user_prompt = parse_prompt_args (
67+ p , t , l , cli_globals , user_prompt , _ = parse_prompt_args (
6868 available_tools , "-t example -g fruit=apples -g color=red" )
6969
7070 assert t == "example"
@@ -77,7 +77,7 @@ def test_parse_global_with_spaces(self):
7777 from seclab_taskflow_agent .__main__ import parse_prompt_args
7878 available_tools = AvailableTools ()
7979
80- p , t , l , cli_globals , user_prompt = parse_prompt_args (
80+ p , t , l , cli_globals , user_prompt , _ = parse_prompt_args (
8181 available_tools , "-t example -g message=hello world" )
8282
8383 assert t == "example"
@@ -90,7 +90,7 @@ def test_parse_global_with_equals_in_value(self):
9090 from seclab_taskflow_agent .__main__ import parse_prompt_args
9191 available_tools = AvailableTools ()
9292
93- p , t , l , cli_globals , user_prompt = parse_prompt_args (
93+ p , t , l , cli_globals , user_prompt , _ = parse_prompt_args (
9494 available_tools , "-t example -g equation=x=5" )
9595
9696 assert t == "example"
0 commit comments