Compare commits
8 Commits
fix/74-git
...
burn/101-1
| Author | SHA1 | Date | |
|---|---|---|---|
| a17d0b0851 | |||
| 811170244b | |||
| 442c4dbcc7 | |||
| 3cd8750cbb | |||
| ef765bbd30 | |||
|
|
5f0d00f127 | ||
|
|
8affe79489 | ||
|
|
319f57780d |
3
.gitignore
vendored
Normal file
3
.gitignore
vendored
Normal file
@@ -0,0 +1,3 @@
|
|||||||
|
build/
|
||||||
|
*.pyc
|
||||||
|
__pycache__/
|
||||||
36
CMakeLists.txt
Normal file
36
CMakeLists.txt
Normal file
@@ -0,0 +1,36 @@
|
|||||||
|
cmake_minimum_required(VERSION 3.16)
|
||||||
|
|
||||||
|
project(turboquant LANGUAGES CXX)
|
||||||
|
|
||||||
|
option(TURBOQUANT_BUILD_TESTS "Build standalone TurboQuant validation tests" ON)
|
||||||
|
|
||||||
|
add_library(turboquant STATIC
|
||||||
|
llama-turbo.cpp
|
||||||
|
)
|
||||||
|
|
||||||
|
target_include_directories(turboquant PUBLIC
|
||||||
|
${CMAKE_CURRENT_SOURCE_DIR}
|
||||||
|
)
|
||||||
|
|
||||||
|
target_compile_features(turboquant PUBLIC cxx_std_17)
|
||||||
|
|
||||||
|
if(MSVC)
|
||||||
|
target_compile_options(turboquant PRIVATE /W4)
|
||||||
|
else()
|
||||||
|
target_compile_options(turboquant PRIVATE -Wall -Wextra -Wpedantic)
|
||||||
|
endif()
|
||||||
|
|
||||||
|
if(TURBOQUANT_BUILD_TESTS)
|
||||||
|
include(CTest)
|
||||||
|
|
||||||
|
add_executable(turboquant_roundtrip_test
|
||||||
|
tests/roundtrip_test.cpp
|
||||||
|
)
|
||||||
|
target_link_libraries(turboquant_roundtrip_test PRIVATE turboquant)
|
||||||
|
target_compile_features(turboquant_roundtrip_test PRIVATE cxx_std_17)
|
||||||
|
|
||||||
|
add_test(
|
||||||
|
NAME turboquant_roundtrip
|
||||||
|
COMMAND turboquant_roundtrip_test
|
||||||
|
)
|
||||||
|
endif()
|
||||||
@@ -13,7 +13,7 @@ Unlock 64K-128K context on qwen3.5:27b within 32GB unified memory.
|
|||||||
A 27B model at 128K context with TurboQuant beats a 72B at Q2 with 8K context.
|
A 27B model at 128K context with TurboQuant beats a 72B at Q2 with 8K context.
|
||||||
|
|
||||||
## Status
|
## Status
|
||||||
See [issues](http://143.198.27.163:3000/Timmy_Foundation/turboquant/issues) for current progress.
|
See [issues](https://forge.alexanderwhitestone.com/Timmy_Foundation/turboquant/issues) for current progress.
|
||||||
|
|
||||||
## Roles
|
## Roles
|
||||||
- **Strago:** Build spec author
|
- **Strago:** Build spec author
|
||||||
@@ -29,4 +29,4 @@ See [issues](http://143.198.27.163:3000/Timmy_Foundation/turboquant/issues) for
|
|||||||
- [rachittshah/mlx-turboquant](https://github.com/rachittshah/mlx-turboquant) — MLX fallback
|
- [rachittshah/mlx-turboquant](https://github.com/rachittshah/mlx-turboquant) — MLX fallback
|
||||||
|
|
||||||
## Docs
|
## Docs
|
||||||
- [BUILD-SPEC.md](BUILD-SPEC.md) — Full build specification (Strago, v2.2)
|
- [Project Status](docs/PROJECT_STATUS.md) — Full project status and build specification
|
||||||
|
|||||||
84
benchmarks/bonsai-tool-calling.md
Normal file
84
benchmarks/bonsai-tool-calling.md
Normal file
@@ -0,0 +1,84 @@
|
|||||||
|
# 1-Bit Model Tool Calling Test Results
|
||||||
|
|
||||||
|
**Model:** bonsai-1b
|
||||||
|
**Date:** 2026-04-15 21:57:29
|
||||||
|
**Test cases:** 11
|
||||||
|
|
||||||
|
## Summary
|
||||||
|
|
||||||
|
| Result | Count |
|
||||||
|
|--------|-------|
|
||||||
|
| SKIP | 11 |
|
||||||
|
|
||||||
|
**Pass rate: 0%** (0/11)
|
||||||
|
|
||||||
|
## Results by Difficulty
|
||||||
|
|
||||||
|
| Difficulty | PASS | PARTIAL | FAIL | Other |
|
||||||
|
|-----------|------|---------|------|-------|
|
||||||
|
| 1/5 | 0 | 0 | 0 | 1 |
|
||||||
|
| 2/5 | 0 | 0 | 0 | 3 |
|
||||||
|
| 3/5 | 0 | 0 | 0 | 5 |
|
||||||
|
| 4/5 | 0 | 0 | 0 | 1 |
|
||||||
|
| 5/5 | 0 | 0 | 0 | 1 |
|
||||||
|
|
||||||
|
## Detailed Results
|
||||||
|
|
||||||
|
### ❓ simple-read-1 (difficulty 1/5)
|
||||||
|
- **Category:** simple_read
|
||||||
|
- **Expected tool:** `read_file`
|
||||||
|
- **Actual tool:** `(dry run)`
|
||||||
|
|
||||||
|
### ❓ simple-read-with-limit (difficulty 2/5)
|
||||||
|
- **Category:** simple_read
|
||||||
|
- **Expected tool:** `read_file`
|
||||||
|
- **Actual tool:** `(dry run)`
|
||||||
|
|
||||||
|
### ❓ terminal-simple (difficulty 2/5)
|
||||||
|
- **Category:** terminal_cmd
|
||||||
|
- **Expected tool:** `terminal`
|
||||||
|
- **Actual tool:** `(dry run)`
|
||||||
|
|
||||||
|
### ❓ terminal-pipe (difficulty 3/5)
|
||||||
|
- **Category:** terminal_cmd
|
||||||
|
- **Expected tool:** `terminal`
|
||||||
|
- **Actual tool:** `(dry run)`
|
||||||
|
|
||||||
|
### ❓ web-search-simple (difficulty 2/5)
|
||||||
|
- **Category:** web_search
|
||||||
|
- **Expected tool:** `web_search`
|
||||||
|
- **Actual tool:** `(dry run)`
|
||||||
|
|
||||||
|
### ❓ multi-tool-select-read (difficulty 3/5)
|
||||||
|
- **Category:** multi_tool_select
|
||||||
|
- **Expected tool:** `read_file`
|
||||||
|
- **Actual tool:** `(dry run)`
|
||||||
|
|
||||||
|
### ❓ multi-tool-select-terminal (difficulty 3/5)
|
||||||
|
- **Category:** multi_tool_select
|
||||||
|
- **Expected tool:** `terminal`
|
||||||
|
- **Actual tool:** `(dry run)`
|
||||||
|
|
||||||
|
### ❓ multi-tool-select-search (difficulty 3/5)
|
||||||
|
- **Category:** multi_tool_select
|
||||||
|
- **Expected tool:** `web_search`
|
||||||
|
- **Actual tool:** `(dry run)`
|
||||||
|
|
||||||
|
### ❓ write-file-with-content (difficulty 3/5)
|
||||||
|
- **Category:** nested_params
|
||||||
|
- **Expected tool:** `write_file`
|
||||||
|
- **Actual tool:** `(dry run)`
|
||||||
|
|
||||||
|
### ❓ patch-edit (difficulty 4/5)
|
||||||
|
- **Category:** nested_params
|
||||||
|
- **Expected tool:** `patch`
|
||||||
|
- **Actual tool:** `(dry run)`
|
||||||
|
|
||||||
|
### ❓ multi-step-read-then-write (difficulty 5/5)
|
||||||
|
- **Category:** multi_step
|
||||||
|
- **Expected tool:** `read_file`
|
||||||
|
- **Actual tool:** `(dry run)`
|
||||||
|
|
||||||
|
## Viability Verdict
|
||||||
|
|
||||||
|
**VERDICT: NOT VIABLE** — 1-bit quantization destroys tool calling capability. Recommend minimum 3-bit quantization for tool-using models.
|
||||||
709
benchmarks/test_bonsai_tool_calling.py
Normal file
709
benchmarks/test_bonsai_tool_calling.py
Normal file
@@ -0,0 +1,709 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
"""
|
||||||
|
1-Bit Model Tool Calling Test Suite (Issue #101).
|
||||||
|
|
||||||
|
Tests whether quantized/1-bit models can handle structured tool calling.
|
||||||
|
Designed to be run against any OpenAI-compatible endpoint (llama-server, Ollama).
|
||||||
|
|
||||||
|
The core question: does 1-bit quantization destroy the precise JSON output
|
||||||
|
required for tool calling? This suite measures it empirically.
|
||||||
|
|
||||||
|
Usage:
|
||||||
|
# Against local llama-server
|
||||||
|
python3 benchmarks/test_bonsai_tool_calling.py \
|
||||||
|
--url http://localhost:8081/v1/chat/completions \
|
||||||
|
--model bonsai-1b
|
||||||
|
|
||||||
|
# Against Ollama
|
||||||
|
python3 benchmarks/test_bonsai_tool_calling.py \
|
||||||
|
--url http://localhost:11434/api/chat \
|
||||||
|
--model bonsai:latest \
|
||||||
|
--backend ollama
|
||||||
|
|
||||||
|
# Dry run (validate test cases without model)
|
||||||
|
python3 benchmarks/test_bonsai_tool_calling.py --dry-run
|
||||||
|
"""
|
||||||
|
import argparse
|
||||||
|
import json
|
||||||
|
import os
|
||||||
|
import re
|
||||||
|
import sys
|
||||||
|
import time
|
||||||
|
from dataclasses import dataclass, field, asdict
|
||||||
|
from enum import Enum
|
||||||
|
from typing import List, Dict, Optional, Tuple
|
||||||
|
|
||||||
|
import requests
|
||||||
|
|
||||||
|
|
||||||
|
class ToolCallCategory(Enum):
|
||||||
|
"""Categories of tool call complexity."""
|
||||||
|
SIMPLE_READ = "simple_read"
|
||||||
|
TERMINAL_CMD = "terminal_cmd"
|
||||||
|
WEB_SEARCH = "web_search"
|
||||||
|
MULTI_STEP = "multi_step"
|
||||||
|
NESTED_PARAMS = "nested_params"
|
||||||
|
ARRAY_PARAMS = "array_params"
|
||||||
|
OPTIONAL_PARAMS = "optional_params"
|
||||||
|
MULTI_TOOL_SELECT = "multi_tool_select"
|
||||||
|
|
||||||
|
|
||||||
|
class TestResult(Enum):
|
||||||
|
PASS = "PASS"
|
||||||
|
FAIL = "FAIL"
|
||||||
|
PARTIAL = "PARTIAL"
|
||||||
|
TIMEOUT = "TIMEOUT"
|
||||||
|
ERROR = "ERROR"
|
||||||
|
SKIP = "SKIP"
|
||||||
|
|
||||||
|
|
||||||
|
# ── Tool schemas (hermes-compatible) ─────────────────────────
|
||||||
|
|
||||||
|
TOOL_SCHEMAS = [
|
||||||
|
{
|
||||||
|
"type": "function",
|
||||||
|
"function": {
|
||||||
|
"name": "read_file",
|
||||||
|
"description": "Read a text file with line numbers.",
|
||||||
|
"parameters": {
|
||||||
|
"type": "object",
|
||||||
|
"properties": {
|
||||||
|
"path": {"type": "string", "description": "File path to read"},
|
||||||
|
"offset": {"type": "integer", "description": "Start line (1-indexed)", "default": 1},
|
||||||
|
"limit": {"type": "integer", "description": "Max lines to read", "default": 500},
|
||||||
|
},
|
||||||
|
"required": ["path"],
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"type": "function",
|
||||||
|
"function": {
|
||||||
|
"name": "terminal",
|
||||||
|
"description": "Execute a shell command.",
|
||||||
|
"parameters": {
|
||||||
|
"type": "object",
|
||||||
|
"properties": {
|
||||||
|
"command": {"type": "string", "description": "Shell command to execute"},
|
||||||
|
"timeout": {"type": "integer", "description": "Timeout in seconds", "default": 30},
|
||||||
|
"workdir": {"type": "string", "description": "Working directory"},
|
||||||
|
},
|
||||||
|
"required": ["command"],
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"type": "function",
|
||||||
|
"function": {
|
||||||
|
"name": "web_search",
|
||||||
|
"description": "Search the web for information.",
|
||||||
|
"parameters": {
|
||||||
|
"type": "object",
|
||||||
|
"properties": {
|
||||||
|
"query": {"type": "string", "description": "Search query"},
|
||||||
|
"max_results": {"type": "integer", "description": "Max results to return", "default": 5},
|
||||||
|
},
|
||||||
|
"required": ["query"],
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"type": "function",
|
||||||
|
"function": {
|
||||||
|
"name": "write_file",
|
||||||
|
"description": "Write content to a file, creating directories as needed.",
|
||||||
|
"parameters": {
|
||||||
|
"type": "object",
|
||||||
|
"properties": {
|
||||||
|
"path": {"type": "string", "description": "File path to write"},
|
||||||
|
"content": {"type": "string", "description": "Content to write"},
|
||||||
|
},
|
||||||
|
"required": ["path", "content"],
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"type": "function",
|
||||||
|
"function": {
|
||||||
|
"name": "patch",
|
||||||
|
"description": "Apply a targeted find-and-replace edit to a file.",
|
||||||
|
"parameters": {
|
||||||
|
"type": "object",
|
||||||
|
"properties": {
|
||||||
|
"path": {"type": "string", "description": "File path to edit"},
|
||||||
|
"old_string": {"type": "string", "description": "Text to find"},
|
||||||
|
"new_string": {"type": "string", "description": "Replacement text"},
|
||||||
|
"replace_all": {"type": "boolean", "description": "Replace all occurrences", "default": False},
|
||||||
|
},
|
||||||
|
"required": ["path", "old_string", "new_string"],
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
|
# ── Test case definitions ────────────────────────────────────
|
||||||
|
|
||||||
|
@dataclass
|
||||||
|
class ToolCallTestCase:
|
||||||
|
"""A single tool calling test case."""
|
||||||
|
id: str
|
||||||
|
category: ToolCallCategory
|
||||||
|
prompt: str
|
||||||
|
tools: List[dict]
|
||||||
|
expected_tool: str
|
||||||
|
expected_params: Dict[str, any]
|
||||||
|
param_validators: Dict[str, callable] = field(default_factory=dict)
|
||||||
|
description: str = ""
|
||||||
|
difficulty: int = 1 # 1-5, higher = harder
|
||||||
|
|
||||||
|
|
||||||
|
TEST_CASES = [
|
||||||
|
# ── Level 1: Simple reads ──────────────────────────────
|
||||||
|
ToolCallTestCase(
|
||||||
|
id="simple-read-1",
|
||||||
|
category=ToolCallCategory.SIMPLE_READ,
|
||||||
|
prompt="Read the file at /tmp/test.txt",
|
||||||
|
tools=[TOOL_SCHEMAS[0]],
|
||||||
|
expected_tool="read_file",
|
||||||
|
expected_params={"path": "/tmp/test.txt"},
|
||||||
|
description="Exact path, single required param",
|
||||||
|
difficulty=1,
|
||||||
|
),
|
||||||
|
ToolCallTestCase(
|
||||||
|
id="simple-read-with-limit",
|
||||||
|
category=ToolCallCategory.SIMPLE_READ,
|
||||||
|
prompt="Read the first 10 lines of /var/log/system.log",
|
||||||
|
tools=[TOOL_SCHEMAS[0]],
|
||||||
|
expected_tool="read_file",
|
||||||
|
expected_params={"path": "/var/log/system.log"},
|
||||||
|
param_validators={"limit": lambda v: isinstance(v, int) and v <= 20},
|
||||||
|
description="Required + optional param",
|
||||||
|
difficulty=2,
|
||||||
|
),
|
||||||
|
|
||||||
|
# ── Level 2: Terminal commands ─────────────────────────
|
||||||
|
ToolCallTestCase(
|
||||||
|
id="terminal-simple",
|
||||||
|
category=ToolCallCategory.TERMINAL_CMD,
|
||||||
|
prompt="List all files in the current directory",
|
||||||
|
tools=[TOOL_SCHEMAS[1]],
|
||||||
|
expected_tool="terminal",
|
||||||
|
expected_params={},
|
||||||
|
param_validators={
|
||||||
|
"command": lambda v: isinstance(v, str) and any(
|
||||||
|
cmd in v for cmd in ["ls", "dir", "find"]
|
||||||
|
)
|
||||||
|
},
|
||||||
|
description="Generate appropriate shell command",
|
||||||
|
difficulty=2,
|
||||||
|
),
|
||||||
|
ToolCallTestCase(
|
||||||
|
id="terminal-pipe",
|
||||||
|
category=ToolCallCategory.TERMINAL_CMD,
|
||||||
|
prompt="Count how many Python files are in /tmp recursively",
|
||||||
|
tools=[TOOL_SCHEMAS[1]],
|
||||||
|
expected_tool="terminal",
|
||||||
|
expected_params={},
|
||||||
|
param_validators={
|
||||||
|
"command": lambda v: isinstance(v, str) and (
|
||||||
|
"find" in v or "ls" in v or "python" in v or ".py" in v
|
||||||
|
)
|
||||||
|
},
|
||||||
|
description="Needs piped or recursive command",
|
||||||
|
difficulty=3,
|
||||||
|
),
|
||||||
|
|
||||||
|
# ── Level 3: Web search ────────────────────────────────
|
||||||
|
ToolCallTestCase(
|
||||||
|
id="web-search-simple",
|
||||||
|
category=ToolCallCategory.WEB_SEARCH,
|
||||||
|
prompt="Search for the current price of Bitcoin",
|
||||||
|
tools=[TOOL_SCHEMAS[2]],
|
||||||
|
expected_tool="web_search",
|
||||||
|
expected_params={"query": "Bitcoin price"},
|
||||||
|
param_validators={
|
||||||
|
"query": lambda v: isinstance(v, str) and len(v) > 3 and "bitcoin" in v.lower()
|
||||||
|
},
|
||||||
|
description="Extract search query from natural language",
|
||||||
|
difficulty=2,
|
||||||
|
),
|
||||||
|
|
||||||
|
# ── Level 4: Multi-tool selection ──────────────────────
|
||||||
|
ToolCallTestCase(
|
||||||
|
id="multi-tool-select-read",
|
||||||
|
category=ToolCallCategory.MULTI_TOOL_SELECT,
|
||||||
|
prompt="Read the file at /etc/hostname",
|
||||||
|
tools=TOOL_SCHEMAS[:3], # read_file, terminal, web_search
|
||||||
|
expected_tool="read_file",
|
||||||
|
expected_params={"path": "/etc/hostname"},
|
||||||
|
description="Choose correct tool from 3 options",
|
||||||
|
difficulty=3,
|
||||||
|
),
|
||||||
|
ToolCallTestCase(
|
||||||
|
id="multi-tool-select-terminal",
|
||||||
|
category=ToolCallCategory.MULTI_TOOL_SELECT,
|
||||||
|
prompt="Check how much disk space is available",
|
||||||
|
tools=TOOL_SCHEMAS[:3],
|
||||||
|
expected_tool="terminal",
|
||||||
|
expected_params={},
|
||||||
|
param_validators={
|
||||||
|
"command": lambda v: isinstance(v, str) and any(
|
||||||
|
cmd in v for cmd in ["df", "du", "disk"]
|
||||||
|
)
|
||||||
|
},
|
||||||
|
description="Choose terminal over read_file for system info",
|
||||||
|
difficulty=3,
|
||||||
|
),
|
||||||
|
ToolCallTestCase(
|
||||||
|
id="multi-tool-select-search",
|
||||||
|
category=ToolCallCategory.MULTI_TOOL_SELECT,
|
||||||
|
prompt="What is the weather in Tokyo right now?",
|
||||||
|
tools=TOOL_SCHEMAS[:3],
|
||||||
|
expected_tool="web_search",
|
||||||
|
expected_params={},
|
||||||
|
param_validators={
|
||||||
|
"query": lambda v: isinstance(v, str) and "weather" in v.lower() and "tokyo" in v.lower()
|
||||||
|
},
|
||||||
|
description="Choose web_search for real-time info",
|
||||||
|
difficulty=3,
|
||||||
|
),
|
||||||
|
|
||||||
|
# ── Level 5: Nested/complex params ─────────────────────
|
||||||
|
ToolCallTestCase(
|
||||||
|
id="write-file-with-content",
|
||||||
|
category=ToolCallCategory.NESTED_PARAMS,
|
||||||
|
prompt="Create a file at /tmp/hello.txt with the content 'Hello, World!'",
|
||||||
|
tools=[TOOL_SCHEMAS[3]],
|
||||||
|
expected_tool="write_file",
|
||||||
|
expected_params={"path": "/tmp/hello.txt"},
|
||||||
|
param_validators={
|
||||||
|
"content": lambda v: isinstance(v, str) and "hello" in v.lower()
|
||||||
|
},
|
||||||
|
description="Two required string params",
|
||||||
|
difficulty=3,
|
||||||
|
),
|
||||||
|
ToolCallTestCase(
|
||||||
|
id="patch-edit",
|
||||||
|
category=ToolCallCategory.NESTED_PARAMS,
|
||||||
|
prompt="In the file /tmp/config.yaml, replace 'debug: false' with 'debug: true'",
|
||||||
|
tools=[TOOL_SCHEMAS[4]],
|
||||||
|
expected_tool="patch",
|
||||||
|
expected_params={"path": "/tmp/config.yaml"},
|
||||||
|
param_validators={
|
||||||
|
"old_string": lambda v: isinstance(v, str) and "debug: false" in v,
|
||||||
|
"new_string": lambda v: isinstance(v, str) and "debug: true" in v,
|
||||||
|
},
|
||||||
|
description="Three required params, find-and-replace",
|
||||||
|
difficulty=4,
|
||||||
|
),
|
||||||
|
|
||||||
|
# ── Level 6: Multi-step reasoning ──────────────────────
|
||||||
|
ToolCallTestCase(
|
||||||
|
id="multi-step-read-then-write",
|
||||||
|
category=ToolCallCategory.MULTI_STEP,
|
||||||
|
prompt="Read /tmp/source.txt and write its contents to /tmp/backup.txt",
|
||||||
|
tools=[TOOL_SCHEMAS[0], TOOL_SCHEMAS[3]], # read_file + write_file
|
||||||
|
expected_tool="read_file", # First step should be reading
|
||||||
|
expected_params={"path": "/tmp/source.txt"},
|
||||||
|
description="Requires planning: read first, then write",
|
||||||
|
difficulty=5,
|
||||||
|
),
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
|
# ── Test runner ──────────────────────────────────────────────
|
||||||
|
|
||||||
|
@dataclass
|
||||||
|
class TestRunResult:
|
||||||
|
"""Result of running a single test case."""
|
||||||
|
test_id: str
|
||||||
|
category: str
|
||||||
|
difficulty: int
|
||||||
|
result: str # TestResult value
|
||||||
|
expected_tool: str
|
||||||
|
actual_tool: str
|
||||||
|
expected_params: dict
|
||||||
|
actual_params: dict
|
||||||
|
param_scores: Dict[str, bool] = field(default_factory=dict)
|
||||||
|
response_text: str = ""
|
||||||
|
latency_s: float = 0.0
|
||||||
|
tokens_per_sec: float = 0.0
|
||||||
|
error: str = ""
|
||||||
|
raw_response: dict = field(default_factory=dict)
|
||||||
|
|
||||||
|
|
||||||
|
def call_openai_compatible(
|
||||||
|
messages: list,
|
||||||
|
tools: list,
|
||||||
|
url: str,
|
||||||
|
model: str,
|
||||||
|
timeout: int = 120,
|
||||||
|
) -> dict:
|
||||||
|
"""Call an OpenAI-compatible chat completions endpoint."""
|
||||||
|
payload = {
|
||||||
|
"model": model,
|
||||||
|
"messages": messages,
|
||||||
|
"tools": tools,
|
||||||
|
"tool_choice": "auto",
|
||||||
|
"max_tokens": 512,
|
||||||
|
"temperature": 0.0,
|
||||||
|
}
|
||||||
|
resp = requests.post(url, json=payload, timeout=timeout)
|
||||||
|
resp.raise_for_status()
|
||||||
|
return resp.json()
|
||||||
|
|
||||||
|
|
||||||
|
def call_ollama(
|
||||||
|
messages: list,
|
||||||
|
tools: list,
|
||||||
|
url: str,
|
||||||
|
model: str,
|
||||||
|
timeout: int = 120,
|
||||||
|
) -> dict:
|
||||||
|
"""Call Ollama /api/chat endpoint."""
|
||||||
|
# Convert OpenAI tool format to Ollama format
|
||||||
|
ollama_tools = []
|
||||||
|
for t in tools:
|
||||||
|
fn = t["function"]
|
||||||
|
ollama_tools.append({
|
||||||
|
"type": "function",
|
||||||
|
"function": {
|
||||||
|
"name": fn["name"],
|
||||||
|
"description": fn["description"],
|
||||||
|
"parameters": fn["parameters"],
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
resp = requests.post(url, json={
|
||||||
|
"model": model,
|
||||||
|
"messages": messages,
|
||||||
|
"tools": ollama_tools,
|
||||||
|
"stream": False,
|
||||||
|
}, timeout=timeout)
|
||||||
|
resp.raise_for_status()
|
||||||
|
data = resp.json()
|
||||||
|
|
||||||
|
# Normalize to OpenAI format
|
||||||
|
result = {"choices": [{"message": {}}]}
|
||||||
|
msg = data.get("message", {})
|
||||||
|
result["choices"][0]["message"]["content"] = msg.get("content", "")
|
||||||
|
if msg.get("tool_calls"):
|
||||||
|
result["choices"][0]["message"]["tool_calls"] = msg["tool_calls"]
|
||||||
|
return result
|
||||||
|
|
||||||
|
|
||||||
|
def validate_tool_call(
|
||||||
|
response: dict,
|
||||||
|
test: ToolCallTestCase,
|
||||||
|
) -> Tuple[TestResult, str, dict, Dict[str, bool]]:
|
||||||
|
"""
|
||||||
|
Validate a model response against a test case.
|
||||||
|
|
||||||
|
Returns: (result, actual_tool, actual_params, param_scores)
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
choice = response["choices"][0]
|
||||||
|
msg = choice["message"]
|
||||||
|
except (KeyError, IndexError):
|
||||||
|
return TestResult.FAIL, "", {}, {}
|
||||||
|
|
||||||
|
# Check if model called a tool
|
||||||
|
tool_calls = msg.get("tool_calls", [])
|
||||||
|
if not tool_calls:
|
||||||
|
# Model responded with text instead — check if it at least mentioned the tool
|
||||||
|
content = msg.get("content", "")
|
||||||
|
if test.expected_tool in content:
|
||||||
|
return TestResult.PARTIAL, "text_only", {"content": content}, {}
|
||||||
|
return TestResult.FAIL, "none", {}, {}
|
||||||
|
|
||||||
|
tc = tool_calls[0]
|
||||||
|
actual_tool = tc.get("function", {}).get("name", "")
|
||||||
|
|
||||||
|
# Parse arguments
|
||||||
|
try:
|
||||||
|
args_str = tc.get("function", {}).get("arguments", "{}")
|
||||||
|
if isinstance(args_str, str):
|
||||||
|
actual_params = json.loads(args_str)
|
||||||
|
else:
|
||||||
|
actual_params = args_str
|
||||||
|
except json.JSONDecodeError:
|
||||||
|
return TestResult.FAIL, actual_tool, {}, {"json_parse": False}
|
||||||
|
|
||||||
|
# Check tool name
|
||||||
|
if actual_tool != test.expected_tool:
|
||||||
|
return TestResult.FAIL, actual_tool, actual_params, {
|
||||||
|
"tool_match": False
|
||||||
|
}
|
||||||
|
|
||||||
|
# Validate expected params
|
||||||
|
param_scores = {"tool_match": True}
|
||||||
|
all_pass = True
|
||||||
|
|
||||||
|
for key, expected_val in test.expected_params.items():
|
||||||
|
if key in actual_params:
|
||||||
|
if actual_params[key] == expected_val:
|
||||||
|
param_scores[f"param_{key}"] = True
|
||||||
|
else:
|
||||||
|
param_scores[f"param_{key}"] = False
|
||||||
|
all_pass = False
|
||||||
|
else:
|
||||||
|
param_scores[f"param_{key}"] = False
|
||||||
|
all_pass = False
|
||||||
|
|
||||||
|
# Run custom validators
|
||||||
|
for key, validator in test.param_validators.items():
|
||||||
|
if key in actual_params:
|
||||||
|
try:
|
||||||
|
passed = validator(actual_params[key])
|
||||||
|
param_scores[f"validator_{key}"] = bool(passed)
|
||||||
|
if not passed:
|
||||||
|
all_pass = False
|
||||||
|
except Exception:
|
||||||
|
param_scores[f"validator_{key}"] = False
|
||||||
|
all_pass = False
|
||||||
|
else:
|
||||||
|
param_scores[f"validator_{key}"] = False
|
||||||
|
all_pass = False
|
||||||
|
|
||||||
|
if all_pass and len(test.expected_params) > 0:
|
||||||
|
return TestResult.PASS, actual_tool, actual_params, param_scores
|
||||||
|
elif all_pass:
|
||||||
|
# No expected params to check — validators passed
|
||||||
|
return TestResult.PASS, actual_tool, actual_params, param_scores
|
||||||
|
else:
|
||||||
|
return TestResult.PARTIAL, actual_tool, actual_params, param_scores
|
||||||
|
|
||||||
|
|
||||||
|
def run_test(
|
||||||
|
test: ToolCallTestCase,
|
||||||
|
url: str,
|
||||||
|
model: str,
|
||||||
|
backend: str = "openai",
|
||||||
|
timeout: int = 120,
|
||||||
|
) -> TestRunResult:
|
||||||
|
"""Run a single test case against the model."""
|
||||||
|
messages = [{"role": "user", "content": test.prompt}]
|
||||||
|
|
||||||
|
start = time.time()
|
||||||
|
try:
|
||||||
|
if backend == "ollama":
|
||||||
|
response = call_ollama(messages, test.tools, url, model, timeout)
|
||||||
|
else:
|
||||||
|
response = call_openai_compatible(messages, test.tools, url, model, timeout)
|
||||||
|
elapsed = time.time() - start
|
||||||
|
|
||||||
|
result, actual_tool, actual_params, param_scores = validate_tool_call(response, test)
|
||||||
|
|
||||||
|
# Extract text response
|
||||||
|
try:
|
||||||
|
text = response["choices"][0]["message"].get("content", "")
|
||||||
|
except (KeyError, IndexError):
|
||||||
|
text = ""
|
||||||
|
|
||||||
|
return TestRunResult(
|
||||||
|
test_id=test.id,
|
||||||
|
category=test.category.value,
|
||||||
|
difficulty=test.difficulty,
|
||||||
|
result=result.value,
|
||||||
|
expected_tool=test.expected_tool,
|
||||||
|
actual_tool=actual_tool,
|
||||||
|
expected_params=test.expected_params,
|
||||||
|
actual_params=actual_params,
|
||||||
|
param_scores=param_scores,
|
||||||
|
response_text=text[:200],
|
||||||
|
latency_s=round(elapsed, 3),
|
||||||
|
raw_response=response,
|
||||||
|
)
|
||||||
|
|
||||||
|
except requests.exceptions.Timeout:
|
||||||
|
return TestRunResult(
|
||||||
|
test_id=test.id,
|
||||||
|
category=test.category.value,
|
||||||
|
difficulty=test.difficulty,
|
||||||
|
result=TestResult.TIMEOUT.value,
|
||||||
|
expected_tool=test.expected_tool,
|
||||||
|
actual_tool="",
|
||||||
|
expected_params=test.expected_params,
|
||||||
|
actual_params={},
|
||||||
|
error=f"Timeout after {timeout}s",
|
||||||
|
)
|
||||||
|
except Exception as e:
|
||||||
|
return TestRunResult(
|
||||||
|
test_id=test.id,
|
||||||
|
category=test.category.value,
|
||||||
|
difficulty=test.difficulty,
|
||||||
|
result=TestResult.ERROR.value,
|
||||||
|
expected_tool=test.expected_tool,
|
||||||
|
actual_tool="",
|
||||||
|
expected_params=test.expected_params,
|
||||||
|
actual_params={},
|
||||||
|
error=str(e)[:200],
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def run_dry_run() -> List[TestRunResult]:
|
||||||
|
"""Validate test cases without a model."""
|
||||||
|
results = []
|
||||||
|
for test in TEST_CASES:
|
||||||
|
results.append(TestRunResult(
|
||||||
|
test_id=test.id,
|
||||||
|
category=test.category.value,
|
||||||
|
difficulty=test.difficulty,
|
||||||
|
result=TestResult.SKIP.value,
|
||||||
|
expected_tool=test.expected_tool,
|
||||||
|
actual_tool="(dry run)",
|
||||||
|
expected_params=test.expected_params,
|
||||||
|
actual_params={},
|
||||||
|
))
|
||||||
|
return results
|
||||||
|
|
||||||
|
|
||||||
|
def generate_report(results: List[TestRunResult], model: str) -> str:
|
||||||
|
"""Generate markdown report."""
|
||||||
|
lines = [
|
||||||
|
f"# 1-Bit Model Tool Calling Test Results",
|
||||||
|
f"",
|
||||||
|
f"**Model:** {model}",
|
||||||
|
f"**Date:** {time.strftime('%Y-%m-%d %H:%M:%S')}",
|
||||||
|
f"**Test cases:** {len(results)}",
|
||||||
|
f"",
|
||||||
|
]
|
||||||
|
|
||||||
|
# Summary table
|
||||||
|
by_result = {}
|
||||||
|
for r in results:
|
||||||
|
by_result[r.result] = by_result.get(r.result, 0) + 1
|
||||||
|
|
||||||
|
lines.append("## Summary")
|
||||||
|
lines.append("")
|
||||||
|
lines.append("| Result | Count |")
|
||||||
|
lines.append("|--------|-------|")
|
||||||
|
for result, count in sorted(by_result.items()):
|
||||||
|
lines.append(f"| {result} | {count} |")
|
||||||
|
lines.append("")
|
||||||
|
|
||||||
|
pass_count = by_result.get("PASS", 0)
|
||||||
|
total = len(results)
|
||||||
|
pass_rate = (pass_count / total * 100) if total > 0 else 0
|
||||||
|
lines.append(f"**Pass rate: {pass_rate:.0f}%** ({pass_count}/{total})")
|
||||||
|
lines.append("")
|
||||||
|
|
||||||
|
# By difficulty
|
||||||
|
lines.append("## Results by Difficulty")
|
||||||
|
lines.append("")
|
||||||
|
lines.append("| Difficulty | PASS | PARTIAL | FAIL | Other |")
|
||||||
|
lines.append("|-----------|------|---------|------|-------|")
|
||||||
|
for diff in range(1, 6):
|
||||||
|
diff_results = [r for r in results if r.difficulty == diff]
|
||||||
|
if not diff_results:
|
||||||
|
continue
|
||||||
|
p = sum(1 for r in diff_results if r.result == "PASS")
|
||||||
|
pa = sum(1 for r in diff_results if r.result == "PARTIAL")
|
||||||
|
f = sum(1 for r in diff_results if r.result in ("FAIL", "ERROR", "TIMEOUT"))
|
||||||
|
o = len(diff_results) - p - pa - f
|
||||||
|
lines.append(f"| {diff}/5 | {p} | {pa} | {f} | {o} |")
|
||||||
|
lines.append("")
|
||||||
|
|
||||||
|
# Detailed results
|
||||||
|
lines.append("## Detailed Results")
|
||||||
|
lines.append("")
|
||||||
|
for r in results:
|
||||||
|
icon = {"PASS": "✅", "PARTIAL": "⚠️", "FAIL": "❌", "ERROR": "💥", "TIMEOUT": "⏱"}.get(r.result, "❓")
|
||||||
|
lines.append(f"### {icon} {r.test_id} (difficulty {r.difficulty}/5)")
|
||||||
|
lines.append(f"- **Category:** {r.category}")
|
||||||
|
lines.append(f"- **Expected tool:** `{r.expected_tool}`")
|
||||||
|
lines.append(f"- **Actual tool:** `{r.actual_tool}`")
|
||||||
|
if r.latency_s > 0:
|
||||||
|
lines.append(f"- **Latency:** {r.latency_s}s")
|
||||||
|
if r.param_scores:
|
||||||
|
lines.append(f"- **Param scores:** {json.dumps(r.param_scores)}")
|
||||||
|
if r.error:
|
||||||
|
lines.append(f"- **Error:** {r.error}")
|
||||||
|
lines.append("")
|
||||||
|
|
||||||
|
# Viability verdict
|
||||||
|
lines.append("## Viability Verdict")
|
||||||
|
lines.append("")
|
||||||
|
if pass_rate >= 80:
|
||||||
|
lines.append("**VERDICT: VIABLE** — 1-bit model can handle tool calling for production use.")
|
||||||
|
elif pass_rate >= 50:
|
||||||
|
lines.append("**VERDICT: CONDITIONALLY VIABLE** — Works for simple tools, struggles with complex params. Consider for edge deployment with guardrails.")
|
||||||
|
elif pass_rate >= 20:
|
||||||
|
lines.append("**VERDICT: MARGINAL** — Can select correct tool sometimes, but parameter accuracy is too low for production. Investigate alternative quantization (2-bit, 3-bit).")
|
||||||
|
else:
|
||||||
|
lines.append("**VERDICT: NOT VIABLE** — 1-bit quantization destroys tool calling capability. Recommend minimum 3-bit quantization for tool-using models.")
|
||||||
|
lines.append("")
|
||||||
|
|
||||||
|
return "\n".join(lines)
|
||||||
|
|
||||||
|
|
||||||
|
def main():
|
||||||
|
parser = argparse.ArgumentParser(description="Test tool calling on 1-bit models")
|
||||||
|
parser.add_argument("--url", default="http://localhost:8081/v1/chat/completions",
|
||||||
|
help="Model API endpoint")
|
||||||
|
parser.add_argument("--model", default="bonsai-1b", help="Model name")
|
||||||
|
parser.add_argument("--backend", default="openai", choices=["openai", "ollama"],
|
||||||
|
help="API backend type")
|
||||||
|
parser.add_argument("--timeout", type=int, default=120, help="Request timeout in seconds")
|
||||||
|
parser.add_argument("--dry-run", action="store_true", help="Validate tests without model")
|
||||||
|
parser.add_argument("--output", default="benchmarks/bonsai-tool-calling-results.json",
|
||||||
|
help="Output file for results")
|
||||||
|
parser.add_argument("--report", default="benchmarks/bonsai-tool-calling.md",
|
||||||
|
help="Output file for markdown report")
|
||||||
|
parser.add_argument("--test-id", help="Run a single test by ID")
|
||||||
|
|
||||||
|
args = parser.parse_args()
|
||||||
|
|
||||||
|
print("=" * 60)
|
||||||
|
print(" 1-Bit Model Tool Calling Test Suite")
|
||||||
|
print("=" * 60)
|
||||||
|
|
||||||
|
if args.dry_run:
|
||||||
|
print("\n[DRY RUN] Validating test cases...")
|
||||||
|
results = run_dry_run()
|
||||||
|
print(f" {len(results)} test cases validated")
|
||||||
|
for r in results:
|
||||||
|
print(f" ✓ {r.test_id} — expects {r.expected_tool} (difficulty {r.difficulty}/5)")
|
||||||
|
else:
|
||||||
|
print(f"\nModel: {args.model}")
|
||||||
|
print(f"Endpoint: {args.url}")
|
||||||
|
print(f"Backend: {args.backend}")
|
||||||
|
print()
|
||||||
|
|
||||||
|
tests = TEST_CASES
|
||||||
|
if args.test_id:
|
||||||
|
tests = [t for t in tests if t.id == args.test_id]
|
||||||
|
if not tests:
|
||||||
|
print(f"Test '{args.test_id}' not found")
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
results = []
|
||||||
|
for i, test in enumerate(tests):
|
||||||
|
print(f" [{i+1}/{len(tests)}] {test.id} (difficulty {test.difficulty}/5)... ", end="", flush=True)
|
||||||
|
result = run_test(test, args.url, args.model, args.backend, args.timeout)
|
||||||
|
results.append(result)
|
||||||
|
icon = {"PASS": "✅", "PARTIAL": "⚠️", "FAIL": "❌", "ERROR": "💥", "TIMEOUT": "⏱"}.get(result.result, "❓")
|
||||||
|
print(f"{icon} {result.result} ({result.latency_s}s)")
|
||||||
|
|
||||||
|
# Save results
|
||||||
|
os.makedirs(os.path.dirname(args.output) or ".", exist_ok=True)
|
||||||
|
with open(args.output, "w") as f:
|
||||||
|
json.dump([asdict(r) for r in results], f, indent=2)
|
||||||
|
print(f"\nResults saved to {args.output}")
|
||||||
|
|
||||||
|
# Generate report
|
||||||
|
report = generate_report(results, args.model)
|
||||||
|
with open(args.report, "w") as f:
|
||||||
|
f.write(report)
|
||||||
|
print(f"Report saved to {args.report}")
|
||||||
|
|
||||||
|
# Print summary
|
||||||
|
pass_count = sum(1 for r in results if r.result == "PASS")
|
||||||
|
total = len(results)
|
||||||
|
print(f"\n{'='*60}")
|
||||||
|
print(f" Results: {pass_count}/{total} passed ({pass_count/total*100:.0f}%)")
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
||||||
@@ -135,7 +135,5 @@ llama-server -m model.gguf --port 8081 -ctk q8_0 -ctv turbo4 -c 131072
|
|||||||
|
|
||||||
## References
|
## References
|
||||||
|
|
||||||
- [TurboQuant Build Spec](../BUILD-SPEC.md)
|
- [Project Status](../docs/PROJECT_STATUS.md)
|
||||||
- [Phase 1 Report](../PHASE1-REPORT.md)
|
|
||||||
- [Full Knowledge Transfer](../FULL-REPORT.md)
|
|
||||||
- [llama.cpp TurboQuant Fork](https://github.com/TheTom/llama-cpp-turboquant)
|
- [llama.cpp TurboQuant Fork](https://github.com/TheTom/llama-cpp-turboquant)
|
||||||
|
|||||||
104
tests/roundtrip_test.cpp
Normal file
104
tests/roundtrip_test.cpp
Normal file
@@ -0,0 +1,104 @@
|
|||||||
|
#include "llama-turbo.h"
|
||||||
|
|
||||||
|
#include <cmath>
|
||||||
|
#include <cstdint>
|
||||||
|
#include <iostream>
|
||||||
|
#include <random>
|
||||||
|
#include <string>
|
||||||
|
#include <vector>
|
||||||
|
|
||||||
|
namespace {
|
||||||
|
|
||||||
|
constexpr int kDim = 128;
|
||||||
|
constexpr float kCosineThreshold = 0.99f;
|
||||||
|
constexpr float kZeroTolerance = 1.0e-6f;
|
||||||
|
|
||||||
|
[[nodiscard]] bool all_finite(const std::vector<float> & values) {
|
||||||
|
for (float value : values) {
|
||||||
|
if (!std::isfinite(value)) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
[[nodiscard]] float max_abs(const std::vector<float> & values) {
|
||||||
|
float best = 0.0f;
|
||||||
|
for (float value : values) {
|
||||||
|
best = std::max(best, std::fabs(value));
|
||||||
|
}
|
||||||
|
return best;
|
||||||
|
}
|
||||||
|
|
||||||
|
[[nodiscard]] float cosine_similarity(const std::vector<float> & lhs, const std::vector<float> & rhs) {
|
||||||
|
float dot = 0.0f;
|
||||||
|
float lhs_norm = 0.0f;
|
||||||
|
float rhs_norm = 0.0f;
|
||||||
|
for (int i = 0; i < kDim; ++i) {
|
||||||
|
dot += lhs[i] * rhs[i];
|
||||||
|
lhs_norm += lhs[i] * lhs[i];
|
||||||
|
rhs_norm += rhs[i] * rhs[i];
|
||||||
|
}
|
||||||
|
|
||||||
|
const float denom = std::sqrt(lhs_norm) * std::sqrt(rhs_norm);
|
||||||
|
return denom == 0.0f ? 1.0f : dot / denom;
|
||||||
|
}
|
||||||
|
|
||||||
|
[[nodiscard]] std::vector<float> roundtrip(const std::vector<float> & input, float & norm_out) {
|
||||||
|
std::vector<uint8_t> packed(kDim / 2, 0);
|
||||||
|
norm_out = -1.0f;
|
||||||
|
polar_quant_encode_turbo4(input.data(), packed.data(), &norm_out, kDim);
|
||||||
|
|
||||||
|
std::vector<float> decoded(kDim, 0.0f);
|
||||||
|
polar_quant_decode_turbo4(packed.data(), decoded.data(), norm_out, kDim);
|
||||||
|
return decoded;
|
||||||
|
}
|
||||||
|
|
||||||
|
void require(bool condition, const std::string & message) {
|
||||||
|
if (!condition) {
|
||||||
|
throw std::runtime_error(message);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void test_zero_vector_roundtrip() {
|
||||||
|
std::vector<float> zeros(kDim, 0.0f);
|
||||||
|
float norm = -1.0f;
|
||||||
|
const auto decoded = roundtrip(zeros, norm);
|
||||||
|
|
||||||
|
require(norm == 0.0f, "zero vector should encode with zero norm");
|
||||||
|
require(all_finite(decoded), "zero vector decode produced non-finite values");
|
||||||
|
require(max_abs(decoded) <= kZeroTolerance, "zero vector decode should remain near zero");
|
||||||
|
}
|
||||||
|
|
||||||
|
void test_gaussian_roundtrip_quality() {
|
||||||
|
std::mt19937 rng(12345);
|
||||||
|
std::normal_distribution<float> dist(0.0f, 1.0f);
|
||||||
|
|
||||||
|
std::vector<float> input(kDim, 0.0f);
|
||||||
|
for (float & value : input) {
|
||||||
|
value = dist(rng);
|
||||||
|
}
|
||||||
|
|
||||||
|
float norm = -1.0f;
|
||||||
|
const auto decoded = roundtrip(input, norm);
|
||||||
|
|
||||||
|
require(norm > 0.0f, "random vector should encode with positive norm");
|
||||||
|
require(all_finite(decoded), "random vector decode produced non-finite values");
|
||||||
|
|
||||||
|
const float cosine = cosine_similarity(input, decoded);
|
||||||
|
require(cosine >= kCosineThreshold, "roundtrip cosine similarity below threshold");
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace
|
||||||
|
|
||||||
|
int main() {
|
||||||
|
try {
|
||||||
|
test_zero_vector_roundtrip();
|
||||||
|
test_gaussian_roundtrip_quality();
|
||||||
|
std::cout << "PASS: turboquant standalone roundtrip tests\n";
|
||||||
|
return 0;
|
||||||
|
} catch (const std::exception & exc) {
|
||||||
|
std::cerr << "FAIL: " << exc.what() << '\n';
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
236
tests/test_bonsai_tool_calling.py
Normal file
236
tests/test_bonsai_tool_calling.py
Normal file
@@ -0,0 +1,236 @@
|
|||||||
|
"""
|
||||||
|
Test suite for 1-bit model tool calling validation (issue #101).
|
||||||
|
|
||||||
|
Tests the test harness itself — validates test case structure,
|
||||||
|
tool schema compatibility, and result generation. The actual
|
||||||
|
model inference tests require a running model server.
|
||||||
|
|
||||||
|
Usage:
|
||||||
|
pytest tests/test_bonsai_tool_calling.py -v
|
||||||
|
pytest tests/test_bonsai_tool_calling.py -v -k live # if server available
|
||||||
|
"""
|
||||||
|
import json
|
||||||
|
import os
|
||||||
|
import sys
|
||||||
|
import unittest
|
||||||
|
from unittest.mock import patch, MagicMock
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
# Add benchmarks to path — resolve relative to project root
|
||||||
|
_PROJECT_ROOT = os.path.join(os.path.dirname(__file__), "..")
|
||||||
|
sys.path.insert(0, os.path.join(_PROJECT_ROOT, "benchmarks"))
|
||||||
|
|
||||||
|
# Import with absolute path to avoid collision with this test module
|
||||||
|
import importlib.util
|
||||||
|
_spec = importlib.util.spec_from_file_location(
|
||||||
|
"bonsai_tool_calling",
|
||||||
|
os.path.join(_PROJECT_ROOT, "benchmarks", "test_bonsai_tool_calling.py"),
|
||||||
|
)
|
||||||
|
_btc = importlib.util.module_from_spec(_spec)
|
||||||
|
_spec.loader.exec_module(_btc)
|
||||||
|
|
||||||
|
TOOL_SCHEMAS = _btc.TOOL_SCHEMAS
|
||||||
|
TEST_CASES = _btc.TEST_CASES
|
||||||
|
ToolCallCategory = _btc.ToolCallCategory
|
||||||
|
TestResult = _btc.TestResult
|
||||||
|
ToolCallTestCase = _btc.ToolCallTestCase
|
||||||
|
TestRunResult = _btc.TestRunResult
|
||||||
|
validate_tool_call = _btc.validate_tool_call
|
||||||
|
run_dry_run = _btc.run_dry_run
|
||||||
|
generate_report = _btc.generate_report
|
||||||
|
|
||||||
|
|
||||||
|
class TestToolSchemas(unittest.TestCase):
|
||||||
|
"""Validate tool schemas are well-formed."""
|
||||||
|
|
||||||
|
def test_schemas_serialize_to_json(self):
|
||||||
|
serialized = json.dumps(TOOL_SCHEMAS)
|
||||||
|
parsed = json.loads(serialized)
|
||||||
|
assert len(parsed) == len(TOOL_SCHEMAS)
|
||||||
|
|
||||||
|
def test_each_schema_has_required_fields(self):
|
||||||
|
for tool in TOOL_SCHEMAS:
|
||||||
|
assert tool["type"] == "function"
|
||||||
|
fn = tool["function"]
|
||||||
|
assert "name" in fn
|
||||||
|
assert "description" in fn
|
||||||
|
assert "parameters" in fn
|
||||||
|
assert fn["parameters"]["type"] == "object"
|
||||||
|
assert "properties" in fn["parameters"]
|
||||||
|
assert "required" in fn["parameters"]
|
||||||
|
|
||||||
|
def test_tool_names_are_unique(self):
|
||||||
|
names = [t["function"]["name"] for t in TOOL_SCHEMAS]
|
||||||
|
assert len(names) == len(set(names)), f"Duplicate tool names: {names}"
|
||||||
|
|
||||||
|
|
||||||
|
class TestTestCaseStructure(unittest.TestCase):
|
||||||
|
"""Validate test case definitions."""
|
||||||
|
|
||||||
|
def test_all_categories_covered(self):
|
||||||
|
categories = {tc.category for tc in TEST_CASES}
|
||||||
|
assert ToolCallCategory.SIMPLE_READ in categories
|
||||||
|
assert ToolCallCategory.TERMINAL_CMD in categories
|
||||||
|
assert ToolCallCategory.WEB_SEARCH in categories
|
||||||
|
assert ToolCallCategory.MULTI_TOOL_SELECT in categories
|
||||||
|
assert ToolCallCategory.NESTED_PARAMS in categories
|
||||||
|
|
||||||
|
def test_difficulty_range(self):
|
||||||
|
for tc in TEST_CASES:
|
||||||
|
assert 1 <= tc.difficulty <= 5, f"{tc.id} difficulty out of range"
|
||||||
|
|
||||||
|
def test_expected_tool_exists_in_schemas(self):
|
||||||
|
all_names = {t["function"]["name"] for t in TOOL_SCHEMAS}
|
||||||
|
for tc in TEST_CASES:
|
||||||
|
assert tc.expected_tool in all_names, (
|
||||||
|
f"{tc.id} expects '{tc.expected_tool}' which is not in TOOL_SCHEMAS"
|
||||||
|
)
|
||||||
|
|
||||||
|
def test_tools_subset_of_schemas(self):
|
||||||
|
all_names = {t["function"]["name"] for t in TOOL_SCHEMAS}
|
||||||
|
for tc in TEST_CASES:
|
||||||
|
for tool in tc.tools:
|
||||||
|
assert tool["function"]["name"] in all_names, (
|
||||||
|
f"{tc.id} references unknown tool"
|
||||||
|
)
|
||||||
|
|
||||||
|
def test_unique_ids(self):
|
||||||
|
ids = [tc.id for tc in TEST_CASES]
|
||||||
|
assert len(ids) == len(set(ids)), f"Duplicate test IDs"
|
||||||
|
|
||||||
|
|
||||||
|
class TestValidateToolCall(unittest.TestCase):
|
||||||
|
"""Test the validation logic."""
|
||||||
|
|
||||||
|
def _make_response(self, tool_name, arguments):
|
||||||
|
return {
|
||||||
|
"choices": [{
|
||||||
|
"message": {
|
||||||
|
"tool_calls": [{
|
||||||
|
"type": "function",
|
||||||
|
"function": {
|
||||||
|
"name": tool_name,
|
||||||
|
"arguments": json.dumps(arguments),
|
||||||
|
},
|
||||||
|
}]
|
||||||
|
}
|
||||||
|
}]
|
||||||
|
}
|
||||||
|
|
||||||
|
def test_exact_match_passes(self):
|
||||||
|
test = TEST_CASES[0] # simple-read-1
|
||||||
|
resp = self._make_response("read_file", {"path": "/tmp/test.txt"})
|
||||||
|
result, tool, params, scores = validate_tool_call(resp, test)
|
||||||
|
assert result == TestResult.PASS
|
||||||
|
assert tool == "read_file"
|
||||||
|
|
||||||
|
def test_wrong_tool_fails(self):
|
||||||
|
test = TEST_CASES[0]
|
||||||
|
resp = self._make_response("terminal", {"command": "cat /tmp/test.txt"})
|
||||||
|
result, tool, params, scores = validate_tool_call(resp, test)
|
||||||
|
assert result == TestResult.FAIL
|
||||||
|
|
||||||
|
def test_no_tool_calls_fails(self):
|
||||||
|
test = TEST_CASES[0]
|
||||||
|
resp = {"choices": [{"message": {"content": "I'll read that file"}}]}
|
||||||
|
result, tool, params, scores = validate_tool_call(resp, test)
|
||||||
|
assert result == TestResult.FAIL
|
||||||
|
|
||||||
|
def test_partial_match_with_validators(self):
|
||||||
|
test = TEST_CASES[2] # terminal-simple
|
||||||
|
resp = self._make_response("terminal", {"command": "ls -la"})
|
||||||
|
result, tool, params, scores = validate_tool_call(resp, test)
|
||||||
|
assert result == TestResult.PASS
|
||||||
|
assert scores.get("validator_command") is True
|
||||||
|
|
||||||
|
def test_validator_failure_is_partial(self):
|
||||||
|
test = TEST_CASES[2] # terminal-simple, expects ls/dir/find
|
||||||
|
resp = self._make_response("terminal", {"command": "echo hello"})
|
||||||
|
result, tool, params, scores = validate_tool_call(resp, test)
|
||||||
|
# Tool matches but validator fails
|
||||||
|
assert result == TestResult.PARTIAL
|
||||||
|
|
||||||
|
def test_malformed_json_in_args(self):
|
||||||
|
test = TEST_CASES[0]
|
||||||
|
resp = {
|
||||||
|
"choices": [{
|
||||||
|
"message": {
|
||||||
|
"tool_calls": [{
|
||||||
|
"type": "function",
|
||||||
|
"function": {
|
||||||
|
"name": "read_file",
|
||||||
|
"arguments": "{broken json",
|
||||||
|
},
|
||||||
|
}]
|
||||||
|
}
|
||||||
|
}]
|
||||||
|
}
|
||||||
|
result, tool, params, scores = validate_tool_call(resp, test)
|
||||||
|
assert result == TestResult.FAIL
|
||||||
|
|
||||||
|
|
||||||
|
class TestDryRun(unittest.TestCase):
|
||||||
|
"""Test the dry run mode."""
|
||||||
|
|
||||||
|
def test_dry_run_returns_all_tests(self):
|
||||||
|
results = run_dry_run()
|
||||||
|
assert len(results) == len(TEST_CASES)
|
||||||
|
|
||||||
|
def test_dry_run_all_skip(self):
|
||||||
|
results = run_dry_run()
|
||||||
|
for r in results:
|
||||||
|
assert r.result == TestResult.SKIP.value
|
||||||
|
|
||||||
|
|
||||||
|
class TestReportGeneration(unittest.TestCase):
|
||||||
|
"""Test report generation."""
|
||||||
|
|
||||||
|
def test_report_has_verdict(self):
|
||||||
|
results = [
|
||||||
|
TestRunResult(
|
||||||
|
test_id="test-1", category="simple", difficulty=1,
|
||||||
|
result="PASS", expected_tool="read_file", actual_tool="read_file",
|
||||||
|
expected_params={}, actual_params={},
|
||||||
|
),
|
||||||
|
]
|
||||||
|
report = generate_report(results, "test-model")
|
||||||
|
assert "VERDICT" in report
|
||||||
|
assert "VIABLE" in report
|
||||||
|
assert "test-model" in report
|
||||||
|
|
||||||
|
def test_report_pass_rate(self):
|
||||||
|
results = [
|
||||||
|
TestRunResult(test_id=f"t{i}", category="c", difficulty=1,
|
||||||
|
result="PASS" if i < 3 else "FAIL",
|
||||||
|
expected_tool="x", actual_tool="x",
|
||||||
|
expected_params={}, actual_params={})
|
||||||
|
for i in range(5)
|
||||||
|
]
|
||||||
|
report = generate_report(results, "m")
|
||||||
|
assert "60%" in report # 3/5 = 60%
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.skipif(
|
||||||
|
not os.environ.get("BONSAI_TOOL_CALL_URL"),
|
||||||
|
reason="No model server available (set BONSAI_TOOL_CALL_URL)",
|
||||||
|
)
|
||||||
|
class TestLiveInference:
|
||||||
|
"""Live tests — requires a running model server."""
|
||||||
|
|
||||||
|
def test_server_responds(self):
|
||||||
|
import requests
|
||||||
|
url = os.environ["BONSAI_TOOL_CALL_URL"]
|
||||||
|
# Try a simple health check
|
||||||
|
resp = requests.get(url.replace("/chat/completions", "/models"), timeout=10)
|
||||||
|
assert resp.status_code in (200, 404) # 404 is ok if endpoint differs
|
||||||
|
|
||||||
|
def test_simple_tool_call(self):
|
||||||
|
url = os.environ["BONSAI_TOOL_CALL_URL"]
|
||||||
|
model = os.environ.get("BONSAI_MODEL", "bonsai-1b")
|
||||||
|
result = _btc.run_test(TEST_CASES[0], url, model, timeout=60)
|
||||||
|
assert result.result in ("PASS", "PARTIAL")
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
unittest.main()
|
||||||
Reference in New Issue
Block a user