Compare commits
8 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| ca39e5333f | |||
| 741d46ba25 | |||
| 3cd2674f68 | |||
| e25aeecc4c | |||
| 2f8fefd701 | |||
| 0f606bc446 | |||
| 2122c004d9 | |||
| d86a6b93ad |
@@ -0,0 +1,10 @@
|
|||||||
|
BasedOnStyle: LLVM
|
||||||
|
IndentWidth: 4
|
||||||
|
ColumnLimit: 120
|
||||||
|
AllowShortFunctionsOnASingleLine: All
|
||||||
|
AllowShortIfStatementsOnASingleLine: AllIfsAndElse
|
||||||
|
AllowShortLoopsOnASingleLine: true
|
||||||
|
BreakBeforeBraces: Attach
|
||||||
|
PointerAlignment: Right
|
||||||
|
SpaceAfterCStyleCast: false
|
||||||
|
SortIncludes: false
|
||||||
@@ -0,0 +1,24 @@
|
|||||||
|
root = true
|
||||||
|
|
||||||
|
[*]
|
||||||
|
charset = utf-8
|
||||||
|
end_of_line = lf
|
||||||
|
insert_final_newline = true
|
||||||
|
trim_trailing_whitespace = true
|
||||||
|
indent_style = space
|
||||||
|
indent_size = 4
|
||||||
|
|
||||||
|
[*.{c,h,cu}]
|
||||||
|
indent_size = 4
|
||||||
|
|
||||||
|
[*.{ts,tsx,js,json,css}]
|
||||||
|
indent_size = 2
|
||||||
|
|
||||||
|
[Makefile]
|
||||||
|
indent_style = tab
|
||||||
|
|
||||||
|
[*.yml]
|
||||||
|
indent_size = 2
|
||||||
|
|
||||||
|
[*.md]
|
||||||
|
trim_trailing_whitespace = false
|
||||||
@@ -24,7 +24,9 @@ jobs:
|
|||||||
name: windows-x86_64
|
name: windows-x86_64
|
||||||
ext: ".exe"
|
ext: ".exe"
|
||||||
make_args: ""
|
make_args: ""
|
||||||
shell: msys2
|
# GitHub shells are format strings: 'msys2 {0}', never bare 'msys2'
|
||||||
|
# (the v1.0.0 tag build failed on exactly this). Same as ci.yml.
|
||||||
|
shell: "msys2 {0}"
|
||||||
runs-on: ${{ matrix.os }}
|
runs-on: ${{ matrix.os }}
|
||||||
defaults:
|
defaults:
|
||||||
run:
|
run:
|
||||||
@@ -32,11 +34,14 @@ jobs:
|
|||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@v4
|
||||||
|
|
||||||
- if: matrix.shell == 'msys2'
|
- if: matrix.os == 'windows-latest'
|
||||||
uses: msys2/setup-msys2@v2
|
uses: msys2/setup-msys2@v2
|
||||||
with:
|
with:
|
||||||
msystem: UCRT64
|
msystem: UCRT64
|
||||||
update: false
|
update: false
|
||||||
|
# inherit: the Package step calls the runner's 7z.exe, which lives on
|
||||||
|
# the Windows PATH; the default minimal path would not see it.
|
||||||
|
path-type: inherit
|
||||||
install: >-
|
install: >-
|
||||||
make
|
make
|
||||||
mingw-w64-ucrt-x86_64-gcc
|
mingw-w64-ucrt-x86_64-gcc
|
||||||
|
|||||||
@@ -183,6 +183,10 @@ COLI_MODEL=/nvme/glm52_i4 ./coli doctor # read-only readiness check
|
|||||||
The engine at runtime is pure C — python is only used by the one-time converter
|
The engine at runtime is pure C — python is only used by the one-time converter
|
||||||
and the optional API gateway.
|
and the optional API gateway.
|
||||||
|
|
||||||
|
Prefer a `coli` command on your PATH? From a checkout, `pip install -e .`
|
||||||
|
registers it (the engine itself still lives in `c/` — this is an editable
|
||||||
|
install from the clone, not a standalone wheel).
|
||||||
|
|
||||||
### 3. Go deeper
|
### 3. Go deeper
|
||||||
|
|
||||||
| topic | doc |
|
| topic | doc |
|
||||||
|
|||||||
@@ -271,6 +271,14 @@ def parse_tool_calls(reply, tools=None):
|
|||||||
salvaged.append(name)
|
salvaged.append(name)
|
||||||
calls.append({"id": "call_" + uuid.uuid4().hex[:24], "type": "function",
|
calls.append({"id": "call_" + uuid.uuid4().hex[:24], "type": "function",
|
||||||
"function": {"name": name, "arguments": json.dumps(args, ensure_ascii=False)}})
|
"function": {"name": name, "arguments": json.dumps(args, ensure_ascii=False)}})
|
||||||
|
if tools and not calls and re.search(r"</?tool_call>|</?arg_key>|</?arg_value>", reply):
|
||||||
|
# Diagnosi per la #401: il client ha dichiarato i tools e il modello ha PROVATO la
|
||||||
|
# sintassi, ma il parse rigoroso non ha agganciato nulla (tipico output int4 storpiato).
|
||||||
|
# EN: #401 field diagnosis: tools were declared and the model attempted the syntax,
|
||||||
|
# EN: but the strict parse matched nothing (typically quantization-mangled output).
|
||||||
|
sys.stderr.write("[api] tools declared and tool-call markers present, but no call "
|
||||||
|
"parsed -- output may be quantization-mangled; try COLI_TOOL_SALVAGE=1\n")
|
||||||
|
sys.stderr.flush()
|
||||||
text = _BOX_RE.sub("", reply)
|
text = _BOX_RE.sub("", reply)
|
||||||
if THINK_CLOSE in text:
|
if THINK_CLOSE in text:
|
||||||
text = text.split(THINK_CLOSE, 1)[1]
|
text = text.split(THINK_CLOSE, 1)[1]
|
||||||
|
|||||||
@@ -0,0 +1,182 @@
|
|||||||
|
"""End-to-end tool-calling test for the OpenAI gateway (#401).
|
||||||
|
|
||||||
|
Unlike the unit tests in test_openai_server.py (which call parse_tool_calls /
|
||||||
|
render_chat directly), this suite runs openai_server.py as a real subprocess
|
||||||
|
against a mock engine that speaks the actual SERVE wire protocol
|
||||||
|
(READY / SUBMIT / DATA / DONE), then talks to it over real HTTP. It pins down
|
||||||
|
the full path a coding client exercises: tool declaration rendering, marker
|
||||||
|
suppression in streamed deltas (across chunk boundaries), tool_calls in both
|
||||||
|
response shapes, and the <|observation|><tool_response> round trip.
|
||||||
|
"""
|
||||||
|
import json
|
||||||
|
import os
|
||||||
|
import socket
|
||||||
|
import subprocess
|
||||||
|
import sys
|
||||||
|
import tempfile
|
||||||
|
import unittest
|
||||||
|
import urllib.request
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
SERVER = Path(__file__).resolve().parent.parent / "openai_server.py"
|
||||||
|
MODEL_ID = "glm-5.2-colibri"
|
||||||
|
|
||||||
|
# Mock engine: replies are keyed on the prompt so one process covers every case.
|
||||||
|
# Prompts received are appended to MOCK_LOG for assertions on the rendering.
|
||||||
|
MOCK_ENGINE = r'''#!/usr/bin/env python3
|
||||||
|
import sys, os
|
||||||
|
out, inp = sys.stdout.buffer, sys.stdin.buffer
|
||||||
|
out.write(b"\x01\x01READY\x01\x01\n" + b"STAT 0 0 0 0 0\n"); out.flush()
|
||||||
|
|
||||||
|
def reply(rid, text, chunks=1):
|
||||||
|
data = text.encode("utf-8")
|
||||||
|
n = max(1, len(data) // chunks)
|
||||||
|
for i in range(0, len(data), n):
|
||||||
|
part = data[i:i+n]
|
||||||
|
out.write(("DATA %s %d\n" % (rid, len(part))).encode() + part + b"\n"); out.flush()
|
||||||
|
out.write(("DONE %s STAT %d 1.0 50.0 10.0 42 0\n" % (rid, len(text.split()))).encode())
|
||||||
|
out.flush()
|
||||||
|
|
||||||
|
while True:
|
||||||
|
line = inp.readline()
|
||||||
|
if not line: break
|
||||||
|
f = line.decode().strip().split()
|
||||||
|
if not f or f[0] != "SUBMIT": continue
|
||||||
|
rid, plen = f[1], int(f[3])
|
||||||
|
prompt = inp.read(plen).decode("utf-8", "replace"); inp.read(1)
|
||||||
|
with open(os.environ["MOCK_LOG"], "a") as log:
|
||||||
|
log.write(prompt + "\n\x00\n")
|
||||||
|
if "<tool_response>" in prompt:
|
||||||
|
reply(rid, "25 degrees and sunny in Rome.")
|
||||||
|
elif "weather in Rome" in prompt:
|
||||||
|
reply(rid, "<tool_call>get_weather<arg_key>city</arg_key>"
|
||||||
|
"<arg_value>Rome</arg_value></tool_call>")
|
||||||
|
elif "weather in Milan" in prompt:
|
||||||
|
# split across many tiny DATA chunks: streamed marker suppression must
|
||||||
|
# hold even when a marker straddles a chunk boundary
|
||||||
|
reply(rid, "Checking. <tool_call>get_weather<arg_key>city</arg_key>"
|
||||||
|
"<arg_value>Milan</arg_value></tool_call>", chunks=20)
|
||||||
|
else:
|
||||||
|
reply(rid, "Hello from the mock engine.")
|
||||||
|
'''
|
||||||
|
|
||||||
|
TOOLS = [{"type": "function", "function": {
|
||||||
|
"name": "get_weather",
|
||||||
|
"description": "Current weather for a city",
|
||||||
|
"parameters": {"type": "object",
|
||||||
|
"properties": {"city": {"type": "string"}},
|
||||||
|
"required": ["city"]}}}]
|
||||||
|
|
||||||
|
|
||||||
|
@unittest.skipUnless(os.name == "posix",
|
||||||
|
"the mock engine is a shebang script the gateway execs directly; "
|
||||||
|
"Windows CreateProcess cannot run it. The gateway logic under test "
|
||||||
|
"is platform-independent and covered by the POSIX CI jobs.")
|
||||||
|
class ToolCallingE2E(unittest.TestCase):
|
||||||
|
@classmethod
|
||||||
|
def setUpClass(cls):
|
||||||
|
cls.tmp = tempfile.TemporaryDirectory()
|
||||||
|
mock = Path(cls.tmp.name) / "mock_engine.py"
|
||||||
|
mock.write_text(MOCK_ENGINE)
|
||||||
|
mock.chmod(0o755)
|
||||||
|
cls.mock_log = Path(cls.tmp.name) / "prompts.log"
|
||||||
|
cls.mock_log.touch()
|
||||||
|
with socket.socket() as probe: # free port, then hand it to the server
|
||||||
|
probe.bind(("127.0.0.1", 0))
|
||||||
|
cls.port = probe.getsockname()[1]
|
||||||
|
env = dict(os.environ, MOCK_LOG=str(cls.mock_log))
|
||||||
|
cls.server = subprocess.Popen(
|
||||||
|
[sys.executable, str(SERVER), "--model", cls.tmp.name,
|
||||||
|
"--engine", str(mock), "--port", str(cls.port)],
|
||||||
|
env=env, stderr=subprocess.DEVNULL)
|
||||||
|
cls.base = f"http://127.0.0.1:{cls.port}/v1"
|
||||||
|
for _ in range(100):
|
||||||
|
try:
|
||||||
|
urllib.request.urlopen(cls.base + "/models", timeout=2)
|
||||||
|
return
|
||||||
|
except OSError:
|
||||||
|
if cls.server.poll() is not None:
|
||||||
|
raise RuntimeError("gateway exited during startup")
|
||||||
|
import time
|
||||||
|
time.sleep(0.1)
|
||||||
|
raise RuntimeError("gateway did not come up")
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def tearDownClass(cls):
|
||||||
|
cls.server.terminate()
|
||||||
|
cls.server.wait(timeout=5)
|
||||||
|
cls.tmp.cleanup()
|
||||||
|
|
||||||
|
def post(self, body, stream=False):
|
||||||
|
req = urllib.request.Request(
|
||||||
|
self.base + "/chat/completions", json.dumps(body).encode(),
|
||||||
|
{"Content-Type": "application/json"})
|
||||||
|
resp = urllib.request.urlopen(req, timeout=30)
|
||||||
|
if not stream:
|
||||||
|
return json.loads(resp.read())
|
||||||
|
events = []
|
||||||
|
for raw in resp:
|
||||||
|
line = raw.decode().strip()
|
||||||
|
if line.startswith("data: ") and line != "data: [DONE]":
|
||||||
|
events.append(json.loads(line[6:]))
|
||||||
|
return events
|
||||||
|
|
||||||
|
def test_tool_call_non_stream(self):
|
||||||
|
r = self.post({"model": MODEL_ID, "tools": TOOLS,
|
||||||
|
"messages": [{"role": "user",
|
||||||
|
"content": "What is the weather in Rome?"}]})
|
||||||
|
choice = r["choices"][0]
|
||||||
|
self.assertEqual(choice["finish_reason"], "tool_calls")
|
||||||
|
calls = choice["message"]["tool_calls"]
|
||||||
|
self.assertEqual(len(calls), 1)
|
||||||
|
self.assertEqual(calls[0]["function"]["name"], "get_weather")
|
||||||
|
self.assertEqual(json.loads(calls[0]["function"]["arguments"]), {"city": "Rome"})
|
||||||
|
self.assertNotIn("<tool_call>", choice["message"].get("content") or "")
|
||||||
|
|
||||||
|
def test_tool_call_streamed_markers_suppressed(self):
|
||||||
|
events = self.post({"model": MODEL_ID, "tools": TOOLS, "stream": True,
|
||||||
|
"messages": [{"role": "user",
|
||||||
|
"content": "What is the weather in Milan?"}]},
|
||||||
|
stream=True)
|
||||||
|
deltas = [e["choices"][0]["delta"] for e in events if e["choices"]]
|
||||||
|
text = "".join(d.get("content") or "" for d in deltas)
|
||||||
|
self.assertNotIn("<tool_call>", text)
|
||||||
|
self.assertNotIn("<arg_key>", text)
|
||||||
|
calls = [d["tool_calls"] for d in deltas if d.get("tool_calls")]
|
||||||
|
self.assertEqual(len(calls), 1)
|
||||||
|
self.assertEqual(calls[0][0]["function"]["name"], "get_weather")
|
||||||
|
self.assertEqual(json.loads(calls[0][0]["function"]["arguments"]),
|
||||||
|
{"city": "Milan"})
|
||||||
|
finish = [e["choices"][0]["finish_reason"] for e in events
|
||||||
|
if e["choices"] and e["choices"][0].get("finish_reason")]
|
||||||
|
self.assertEqual(finish, ["tool_calls"])
|
||||||
|
|
||||||
|
def test_tool_result_round_trip(self):
|
||||||
|
r = self.post({"model": MODEL_ID, "tools": TOOLS, "messages": [
|
||||||
|
{"role": "user", "content": "What is the weather in Rome?"},
|
||||||
|
{"role": "assistant", "content": None, "tool_calls": [
|
||||||
|
{"id": "call_x", "type": "function",
|
||||||
|
"function": {"name": "get_weather",
|
||||||
|
"arguments": "{\"city\": \"Rome\"}"}}]},
|
||||||
|
{"role": "tool", "tool_call_id": "call_x",
|
||||||
|
"content": "25 degrees, sunny"}]})
|
||||||
|
choice = r["choices"][0]
|
||||||
|
self.assertEqual(choice["finish_reason"], "stop")
|
||||||
|
self.assertFalse(choice["message"].get("tool_calls"))
|
||||||
|
self.assertIn("25 degrees", choice["message"]["content"])
|
||||||
|
rendered = self.mock_log.read_text().split("\x00")[-2]
|
||||||
|
self.assertIn("<|observation|><tool_response>25 degrees, sunny</tool_response>",
|
||||||
|
rendered)
|
||||||
|
self.assertIn("# Tools", rendered)
|
||||||
|
self.assertIn('"get_weather"', rendered)
|
||||||
|
|
||||||
|
def test_no_tools_plain_text(self):
|
||||||
|
r = self.post({"model": MODEL_ID,
|
||||||
|
"messages": [{"role": "user", "content": "Hi!"}]})
|
||||||
|
choice = r["choices"][0]
|
||||||
|
self.assertEqual(choice["finish_reason"], "stop")
|
||||||
|
self.assertIn("mock engine", choice["message"]["content"])
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
unittest.main()
|
||||||
@@ -0,0 +1,5 @@
|
|||||||
|
"""colibrì — tiny engine, immense model."""
|
||||||
|
|
||||||
|
from colibri._version import __version__
|
||||||
|
|
||||||
|
__all__ = ["__version__"]
|
||||||
@@ -0,0 +1,23 @@
|
|||||||
|
"""Version accessor for the pip package.
|
||||||
|
|
||||||
|
The single source of truth is c/version.py (#394): coli --version and the
|
||||||
|
GitHub Release workflow read it, so the pip metadata must read the SAME file
|
||||||
|
instead of carrying a second literal that would drift on the first bump.
|
||||||
|
|
||||||
|
From a checkout (the supported install: `pip install -e .`) the file is read
|
||||||
|
directly. From an installed wheel c/ is not on disk, so fall back to the
|
||||||
|
package metadata that setuptools baked at build time from that same file.
|
||||||
|
"""
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
try:
|
||||||
|
_ns = {}
|
||||||
|
exec((Path(__file__).resolve().parent.parent / "c" / "version.py").read_text(), _ns)
|
||||||
|
__version__ = _ns["__version__"]
|
||||||
|
except OSError:
|
||||||
|
from importlib.metadata import PackageNotFoundError, version
|
||||||
|
|
||||||
|
try:
|
||||||
|
__version__ = version("colibri-engine")
|
||||||
|
except PackageNotFoundError:
|
||||||
|
__version__ = "0.0.0+unknown"
|
||||||
@@ -0,0 +1,30 @@
|
|||||||
|
"""Entry point for `coli` when installed via pip.
|
||||||
|
|
||||||
|
Delegates to the original c/coli script which handles all subcommands.
|
||||||
|
This wrapper exists so `pip install colibri-engine` creates a `coli` console
|
||||||
|
script that works without the user having to add c/ to PATH manually.
|
||||||
|
"""
|
||||||
|
|
||||||
|
import os
|
||||||
|
import sys
|
||||||
|
import runpy
|
||||||
|
|
||||||
|
|
||||||
|
def main():
|
||||||
|
here = os.path.dirname(os.path.abspath(__file__))
|
||||||
|
engine_dir = os.path.join(os.path.dirname(here), "c")
|
||||||
|
coli_script = os.path.join(engine_dir, "coli")
|
||||||
|
|
||||||
|
if not os.path.exists(coli_script):
|
||||||
|
sys.exit(
|
||||||
|
"colibri engine directory not found.\n"
|
||||||
|
"Install from source: git clone + pip install -e ."
|
||||||
|
)
|
||||||
|
|
||||||
|
sys.path.insert(0, engine_dir)
|
||||||
|
sys.argv[0] = coli_script
|
||||||
|
runpy.run_path(coli_script, run_name="__main__")
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
||||||
@@ -0,0 +1,53 @@
|
|||||||
|
[build-system]
|
||||||
|
requires = ["setuptools>=68.0"]
|
||||||
|
build-backend = "setuptools.build_meta"
|
||||||
|
|
||||||
|
[project]
|
||||||
|
name = "colibri-engine"
|
||||||
|
dynamic = ["version"]
|
||||||
|
description = "Tiny engine, immense model — run GLM-5.2 (744B MoE) locally"
|
||||||
|
readme = "README.md"
|
||||||
|
license = "Apache-2.0"
|
||||||
|
requires-python = ">=3.10"
|
||||||
|
authors = [
|
||||||
|
{name = "JustVugg"},
|
||||||
|
]
|
||||||
|
classifiers = [
|
||||||
|
"Development Status :: 4 - Beta",
|
||||||
|
"Environment :: Console",
|
||||||
|
"Intended Audience :: Science/Research",
|
||||||
|
"Operating System :: POSIX :: Linux",
|
||||||
|
"Operating System :: MacOS",
|
||||||
|
"Operating System :: Microsoft :: Windows",
|
||||||
|
"Programming Language :: Python :: 3",
|
||||||
|
"Topic :: Scientific/Engineering :: Artificial Intelligence",
|
||||||
|
]
|
||||||
|
|
||||||
|
[project.optional-dependencies]
|
||||||
|
convert = [
|
||||||
|
"numpy",
|
||||||
|
"huggingface_hub",
|
||||||
|
]
|
||||||
|
oracle = [
|
||||||
|
"torch>=2.0",
|
||||||
|
"transformers>=4.40",
|
||||||
|
"safetensors",
|
||||||
|
]
|
||||||
|
bench = [
|
||||||
|
"tokenizers",
|
||||||
|
"datasets",
|
||||||
|
]
|
||||||
|
|
||||||
|
[project.scripts]
|
||||||
|
coli = "colibri.cli:main"
|
||||||
|
|
||||||
|
[project.urls]
|
||||||
|
Homepage = "https://github.com/JustVugg/colibri"
|
||||||
|
Issues = "https://github.com/JustVugg/colibri/issues"
|
||||||
|
|
||||||
|
[tool.setuptools.dynamic]
|
||||||
|
version = {attr = "colibri._version.__version__"}
|
||||||
|
|
||||||
|
[tool.setuptools.packages.find]
|
||||||
|
where = ["."]
|
||||||
|
include = ["colibri*"]
|
||||||
Reference in New Issue
Block a user