1 line
6.1 KiB
JSON
1 line
6.1 KiB
JSON
{"repo": "arto-pihlaja/scrapeset", "files": {"tests/test_llm_client.py": {"40e65a187576231658d33d89e3cf48ec366fee47": {"t": 1785747507.7440138, "n": 1794, "h": "1a2f0efbb1444328", "c": "import pytest\nfrom unittest.mock import MagicMock, patch\nfrom src.llm.utils import format_model_name\nfrom src.config.settings import Settings\n\n@pytest.fixture\ndef mock_settings():\n settings = MagicMock(spec=Settings)\n settings.default_llm_provider = \"openai\"\n settings.default_model = \"gpt-3.5-turbo\"\n settings.openai_api_key = \"sk-...\"\n settings.anthropic_api_key = None\n settings.openrouter_api_key = None\n settings.deepseek_api_key = None\n settings.llm_api_base = None\n settings.llm_api_key = None\n return settings\n\ndef test_format_model_name_default_openai(mock_settings):\n with patch(\"src.llm.utils.settings\", mock_settings):\n assert format_model_name(\"gpt-4\") == \"openai/gpt-4\"\n\ndef test_format_model_name_explicit_deepseek(mock_settings):\n mock_settings.default_llm_provider = \"deepseek\"\n with patch(\"src.llm.utils.settings\", mock_settings):\n assert format_model_name(\"deepseek-chat\") == \"deepseek/deepseek-chat\"\n\ndef test_format_model_name_explicit_openrouter(mock_settings):\n mock_settings.default_llm_provider = \"openrouter\"\n with patch(\"src.llm.utils.settings\", mock_settings):\n assert format_model_name(\"openai/gpt-4o\") == \"openai/gpt-4o\" # Stays as is\n assert format_model_name(\"gpt-4o\") == \"openrouter/gpt-4o\"\n\ndef test_format_model_name_auto_detect_anthropic(mock_settings):\n mock_settings.anthropic_api_key = \"sk-ant-...\"\n with patch(\"src.llm.utils.settings\", mock_settings):\n assert format_model_name(\"claude-3-opus\") == \"anthropic/claude-3-opus\"\n\ndef test_format_model_name_generic_api_base(mock_settings):\n mock_settings.llm_api_base = \"https://my-local-ai.com\"\n with patch(\"src.llm.utils.settings\", mock_settings):\n assert format_model_name(\"my-model\") == \"openai/my-model\"\n"}, "0614320c25f6cd6c0019c7c7f373db70714b513f": {"t": 1785773664.2519982, "n": 1794, "h": "1a2f0efbb1444328", "c": "import pytest\nfrom unittest.mock import MagicMock, patch\nfrom src.llm.utils import format_model_name\nfrom src.config.settings import Settings\n\n@pytest.fixture\ndef mock_settings():\n settings = MagicMock(spec=Settings)\n settings.default_llm_provider = \"openai\"\n settings.default_model = \"gpt-3.5-turbo\"\n settings.openai_api_key = \"sk-...\"\n settings.anthropic_api_key = None\n settings.openrouter_api_key = None\n settings.deepseek_api_key = None\n settings.llm_api_base = None\n settings.llm_api_key = None\n return settings\n\ndef test_format_model_name_default_openai(mock_settings):\n with patch(\"src.llm.utils.settings\", mock_settings):\n assert format_model_name(\"gpt-4\") == \"openai/gpt-4\"\n\ndef test_format_model_name_explicit_deepseek(mock_settings):\n mock_settings.default_llm_provider = \"deepseek\"\n with patch(\"src.llm.utils.settings\", mock_settings):\n assert format_model_name(\"deepseek-chat\") == \"deepseek/deepseek-chat\"\n\ndef test_format_model_name_explicit_openrouter(mock_settings):\n mock_settings.default_llm_provider = \"openrouter\"\n with patch(\"src.llm.utils.settings\", mock_settings):\n assert format_model_name(\"openai/gpt-4o\") == \"openai/gpt-4o\" # Stays as is\n assert format_model_name(\"gpt-4o\") == \"openrouter/gpt-4o\"\n\ndef test_format_model_name_auto_detect_anthropic(mock_settings):\n mock_settings.anthropic_api_key = \"sk-ant-...\"\n with patch(\"src.llm.utils.settings\", mock_settings):\n assert format_model_name(\"claude-3-opus\") == \"anthropic/claude-3-opus\"\n\ndef test_format_model_name_generic_api_base(mock_settings):\n mock_settings.llm_api_base = \"https://my-local-ai.com\"\n with patch(\"src.llm.utils.settings\", mock_settings):\n assert format_model_name(\"my-model\") == \"openai/my-model\"\n"}, "3a605bad535511d3d8ce895f6d66f606687e2a30": {"t": 1785773665.4461904, "n": 1978, "h": "19748e4760dfc153", "c": "import pytest\nfrom unittest.mock import MagicMock, patch\nfrom src.llm import LLMClient\nfrom src.config.settings import Settings\n\n@pytest.fixture\ndef mock_settings():\n settings = MagicMock(spec=Settings)\n settings.default_llm_provider = \"openai\"\n settings.default_model = \"gpt-3.5-turbo\"\n settings.openai_api_key = \"sk-...\"\n settings.anthropic_api_key = None\n settings.openrouter_api_key = None\n settings.deepseek_api_key = None\n settings.llm_api_base = None\n settings.llm_api_key = None\n return settings\n\ndef test_format_model_name_default_openai(mock_settings):\n with patch(\"src.llm.client.settings\", mock_settings):\n client = LLMClient()\n assert client._format_model_name(\"gpt-4\") == \"openai/gpt-4\"\n\ndef test_format_model_name_explicit_deepseek(mock_settings):\n mock_settings.default_llm_provider = \"deepseek\"\n with patch(\"src.llm.client.settings\", mock_settings):\n client = LLMClient()\n assert client._format_model_name(\"deepseek-chat\") == \"deepseek/deepseek-chat\"\n\ndef test_format_model_name_explicit_openrouter(mock_settings):\n mock_settings.default_llm_provider = \"openrouter\"\n with patch(\"src.llm.client.settings\", mock_settings):\n client = LLMClient()\n assert client._format_model_name(\"openai/gpt-4o\") == \"openai/gpt-4o\" # Stays as is\n assert client._format_model_name(\"gpt-4o\") == \"openrouter/gpt-4o\"\n\ndef test_format_model_name_auto_detect_anthropic(mock_settings):\n mock_settings.anthropic_api_key = \"sk-ant-...\"\n with patch(\"src.llm.client.settings\", mock_settings):\n client = LLMClient()\n assert client._format_model_name(\"claude-3-opus\") == \"anthropic/claude-3-opus\"\n\ndef test_format_model_name_generic_api_base(mock_settings):\n mock_settings.llm_api_base = \"https://my-local-ai.com\"\n with patch(\"src.llm.client.settings\", mock_settings):\n client = LLMClient()\n assert client._format_model_name(\"my-model\") == \"openai/my-model\"\n"}}}} |