Skip to content

Commit 47be437

Browse files
author
ochafik
committed
Text fireworks v2 template
1 parent 4cdbb8c commit 47be437

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

examples/server/tests/unit/test_tool_call.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,8 @@ def create_server():
7878
("NousResearch-Hermes-3-Llama-3.1-8B-tool_use", PYTHON_TOOL, "code"),
7979
("deepseek-ai-DeepSeek-R1-Distill-Llama-8B", TEST_TOOL, "success"),
8080
("deepseek-ai-DeepSeek-R1-Distill-Llama-8B", PYTHON_TOOL, "code"),
81+
("fireworks-ai-llama-3-firefunction-v2", TEST_TOOL, "success"),
82+
("fireworks-ai-llama-3-firefunction-v2", PYTHON_TOOL, "code"),
8183
# TODO: fix these
8284
])
8385
def test_completion_with_required_tool_tiny(template_name: str, tool: dict, argument_key: str | None):

0 commit comments

Comments
 (0)