From 47be4373569a0b0053f4e56ba985d33afc43010c Mon Sep 17 00:00:00 2001 From: ochafik Date: Wed, 29 Jan 2025 01:51:07 +0000 Subject: [PATCH] Text fireworks v2 template --- examples/server/tests/unit/test_tool_call.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/examples/server/tests/unit/test_tool_call.py b/examples/server/tests/unit/test_tool_call.py index 810bbb9e6..57c053e5d 100644 --- a/examples/server/tests/unit/test_tool_call.py +++ b/examples/server/tests/unit/test_tool_call.py @@ -78,6 +78,8 @@ WEATHER_TOOL = { ("NousResearch-Hermes-3-Llama-3.1-8B-tool_use", PYTHON_TOOL, "code"), ("deepseek-ai-DeepSeek-R1-Distill-Llama-8B", TEST_TOOL, "success"), ("deepseek-ai-DeepSeek-R1-Distill-Llama-8B", PYTHON_TOOL, "code"), + ("fireworks-ai-llama-3-firefunction-v2", TEST_TOOL, "success"), + ("fireworks-ai-llama-3-firefunction-v2", PYTHON_TOOL, "code"), # TODO: fix these ]) def test_completion_with_required_tool_tiny(template_name: str, tool: dict, argument_key: str | None):