wehub-resource-sync
73 lines
2.2 KiB
Python
73 lines
2.2 KiB
Python
from llmai import DeepSeekClientConfig
|
|
from llmai.shared import OpenAIApiType, OpenAIClientConfig
|
|
|
|
from utils.llm_config import get_extra_body, get_llm_config
|
|
|
|
|
|
def test_openai_uses_responses_api_only_for_native_web_search(monkeypatch):
|
|
monkeypatch.setenv("LLM", "openai")
|
|
monkeypatch.setenv("OPENAI_API_KEY", "test-key")
|
|
|
|
regular_config = get_llm_config()
|
|
search_config = get_llm_config(use_openai_responses_api=True)
|
|
|
|
assert regular_config.api_type == OpenAIApiType.COMPLETIONS
|
|
assert search_config.api_type == OpenAIApiType.RESPONSES
|
|
|
|
|
|
def test_deepseek_provider_uses_deepseek_client_config(monkeypatch):
|
|
monkeypatch.setenv("LLM", "deepseek")
|
|
monkeypatch.setenv("DEEPSEEK_API_KEY", "deepseek-key")
|
|
monkeypatch.setenv("DEEPSEEK_BASE_URL", "https://api.deepseek.com/v1")
|
|
|
|
config = get_llm_config()
|
|
|
|
assert isinstance(config, DeepSeekClientConfig)
|
|
assert config.base_url == "https://api.deepseek.com/v1"
|
|
|
|
|
|
def test_custom_provider_uses_openai_client_config(monkeypatch):
|
|
monkeypatch.setenv("LLM", "custom")
|
|
monkeypatch.setenv("CUSTOM_LLM_URL", "http://localhost:11434/v1")
|
|
monkeypatch.setenv("CUSTOM_LLM_API_KEY", "custom-key")
|
|
|
|
config = get_llm_config()
|
|
|
|
assert isinstance(config, OpenAIClientConfig)
|
|
|
|
|
|
def test_deepseek_disable_thinking_uses_deepseek_payload(monkeypatch):
|
|
monkeypatch.setenv("LLM", "deepseek")
|
|
monkeypatch.setenv("DISABLE_THINKING", "true")
|
|
|
|
extra_body = get_extra_body()
|
|
|
|
assert extra_body == {"thinking": {"type": "disabled"}}
|
|
|
|
|
|
def test_deepseek_tool_choice_disables_thinking(monkeypatch):
|
|
monkeypatch.setenv("LLM", "deepseek")
|
|
monkeypatch.setenv("DISABLE_THINKING", "false")
|
|
|
|
extra_body = get_extra_body(uses_tool_choice=True)
|
|
|
|
assert extra_body == {"thinking": {"type": "disabled"}}
|
|
|
|
|
|
def test_deepseek_regular_request_keeps_thinking_default(monkeypatch):
|
|
monkeypatch.setenv("LLM", "deepseek")
|
|
monkeypatch.setenv("DISABLE_THINKING", "false")
|
|
|
|
extra_body = get_extra_body()
|
|
|
|
assert extra_body is None
|
|
|
|
|
|
def test_custom_disable_thinking_uses_legacy_payload(monkeypatch):
|
|
monkeypatch.setenv("LLM", "custom")
|
|
monkeypatch.setenv("DISABLE_THINKING", "true")
|
|
|
|
extra_body = get_extra_body()
|
|
|
|
assert extra_body == {"enable_thinking": False}
|