mirror of
https://github.com/pewdiepie-archdaemon/odysseus.git
synced 2026-10-06 06:52:20 +02:00
77 lines
3.4 KiB
Python
77 lines
3.4 KiB
Python
from src import agent_loop, llm_core
|
|
from src.agent_tools import admin_tools
|
|
from src import settings
|
|
from routes.chat_routes import _resolve_prompt_thinking_mode
|
|
|
|
|
|
def test_kimi_thinking_off_translates_to_provider_payload():
|
|
assert llm_core._detect_provider("https://api.kimi.com/coding/v1") == "kimi-code"
|
|
payload = {}
|
|
llm_core._apply_hosted_thinking_mode(
|
|
payload, "kimi-code", "kimi-for-coding", "off"
|
|
)
|
|
assert payload["thinking"] == {"type": "disabled"}
|
|
|
|
|
|
def test_kimi_thinking_default_does_not_add_provider_field():
|
|
payload = {}
|
|
llm_core._apply_hosted_thinking_mode(
|
|
payload, "kimi-code", "kimi-for-coding", None
|
|
)
|
|
assert "thinking" not in payload
|
|
|
|
|
|
def test_openrouter_thinking_off_uses_unified_reasoning_control():
|
|
payload = {}
|
|
llm_core._apply_hosted_thinking_mode(
|
|
payload, "openrouter", "moonshotai/kimi-k3", "off"
|
|
)
|
|
assert payload["reasoning"] == {"effort": "none"}
|
|
|
|
|
|
def test_openrouter_does_not_disable_mandatory_grok_45_reasoning():
|
|
payload = {"reasoning": {"effort": "none"}}
|
|
llm_core._apply_hosted_thinking_mode(
|
|
payload, "openrouter", "x-ai/grok-4.5", "off"
|
|
)
|
|
assert "reasoning" not in payload
|
|
|
|
|
|
def test_stream_transport_only_keeps_required_protocol_reasoning_when_off():
|
|
import ast
|
|
source = open(llm_core.__file__, encoding="utf-8").read()
|
|
condition = next(node.test for node in ast.walk(ast.parse(source))
|
|
if isinstance(node, ast.If) and isinstance(node.test, ast.BoolOp)
|
|
and isinstance(node.test.values[0], ast.Name)
|
|
and node.test.values[0].id == "reasoning")
|
|
expression = compile(ast.Expression(condition), "reasoning-policy", "eval")
|
|
for model, mode, expected in [("generic", "off", False), ("generic", "on", True),
|
|
("deepseek-v4", "off", True)]:
|
|
assert bool(eval(expression, {"reasoning": "analysis", "thinking_mode": mode,
|
|
"model": model, "_normalize_thinking_mode": llm_core._normalize_thinking_mode})) is expected
|
|
|
|
|
|
def test_response_cache_is_partitioned_by_thinking_mode():
|
|
args = ("https://api.kimi.com/coding/v1", "kimi-for-coding", [{"role": "user", "content": "Hi"}], 1.0, 100)
|
|
assert llm_core._get_cache_key(*args, thinking_mode="off") != llm_core._get_cache_key(*args, thinking_mode="on")
|
|
|
|
|
|
def test_chat_backend_reads_thinking_mode_from_active_prompt_preset():
|
|
manager = type("PresetManager", (), {"presets": {"custom": {"enabled": True, "thinking_mode": "off"}}})()
|
|
assert _resolve_prompt_thinking_mode(None, "custom", manager) == "off"
|
|
assert _resolve_prompt_thinking_mode("on", "custom", manager) == "on"
|
|
|
|
|
|
def test_email_writing_style_is_agent_manageable_and_routable():
|
|
assert settings.DEFAULT_SETTINGS["email_writing_style"] == ""
|
|
assert settings.DEFAULT_SETTINGS["document_writing_style"] == ""
|
|
source = open(admin_tools.__file__, encoding="utf-8").read()
|
|
assert '"writing style": "document_writing_style"' in source
|
|
assert '"email writing style": "email_writing_style"' in source
|
|
assert any(
|
|
"writing style" in triggers and "manage_settings" in tools
|
|
for triggers, tools in agent_loop._QWEN38_ROUTER_KEYWORD_TOOLS
|
|
)
|
|
index_source = open("src/tool_index.py", encoding="utf-8").read()
|
|
assert '"writing style", "reply style", "email writing style"' in index_source
|