55 lines
1.9 KiB
Python
55 lines
1.9 KiB
Python
import json
|
|
from unittest import mock
|
|
|
|
import pytest
|
|
|
|
from projektmatch import llm
|
|
|
|
|
|
def fake_post(payloads):
|
|
"""Return a mock for requests.post yielding chat completions."""
|
|
responses = []
|
|
for p in payloads:
|
|
r = mock.Mock()
|
|
r.status_code = 200
|
|
r.json.return_value = {"choices": [{"message": {"content": p}}]}
|
|
responses.append(r)
|
|
return mock.Mock(side_effect=responses)
|
|
|
|
|
|
def test_chat_json_strips_think_and_parses():
|
|
content = "<think>lange Kette</think>{\"a\": 1}"
|
|
with mock.patch("projektmatch.llm.requests.post", fake_post([content])) as p:
|
|
out = llm.chat_json("http://v/v1", "m", [{"role": "user", "content": "x"}],
|
|
{"type": "object"})
|
|
assert out == {"a": 1}
|
|
body = p.call_args.kwargs["json"]
|
|
assert "max_tokens" not in body
|
|
assert body["response_format"]["json_schema"]["schema"] == {"type": "object"}
|
|
|
|
|
|
def test_chat_json_strips_markdown_fences():
|
|
content = "```json\n{\"a\": 2}\n```"
|
|
with mock.patch("projektmatch.llm.requests.post", fake_post([content])):
|
|
out = llm.chat_json("http://v/v1", "m", [], {"type": "object"})
|
|
assert out == {"a": 2}
|
|
|
|
|
|
def test_match_cv_retries_on_missing_nr_then_raises():
|
|
reqs = [{"nr": 1, "text": "Python"}, {"nr": 2, "text": "K8s"}]
|
|
partial = json.dumps({"ratings": [
|
|
{"nr": 1, "rating": "yes", "reason": "ok"}]})
|
|
with mock.patch("projektmatch.llm.requests.post",
|
|
fake_post([partial, partial])):
|
|
with pytest.raises(llm.LlmError, match="unvollständig"):
|
|
llm.match_cv("http://v/v1", "m", "CV", reqs)
|
|
|
|
|
|
def test_match_cv_ok():
|
|
reqs = [{"nr": 1, "text": "Python"}]
|
|
full = json.dumps({"ratings": [
|
|
{"nr": 1, "rating": "unknown", "reason": "Teilevidenz"}]})
|
|
with mock.patch("projektmatch.llm.requests.post", fake_post([full])):
|
|
out = llm.match_cv("http://v/v1", "m", "CV", reqs)
|
|
assert out[0]["rating"] == "unknown"
|