GithubBackup/MoviePilot
1
0
Fork 0
mirror of https://github.com/jxxghp/MoviePilot.git synced 2026-06-12 11:10:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add LLM proxy toggle

Browse Source
This commit is contained in:
jxxghp
2026-05-27 06:57:09 +08:00
parent 896631d63e
commit 1f7fb304dd
10 changed files with 277 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/test_system_llm_test.py
View File

@@ -147,6 +147,8 @@ class LlmTestEndpointTest(unittest.TestCase):
system_endpoint.settings, "LLM_BASE_URL_PRESET", "deepseek-default"
), patch.object(
system_endpoint.settings, "LLM_USER_AGENT", "MoviePilot-Test/1.0"
), patch.object(
system_endpoint.settings, "LLM_USE_PROXY", True
), patch.object(
system_endpoint.LLMHelper,
"test_current_settings",
@@ -163,6 +165,7 @@ class LlmTestEndpointTest(unittest.TestCase):
base_url="https://api.deepseek.com",
base_url_preset="deepseek-default",
user_agent="MoviePilot-Test/1.0",
use_proxy=True,
)
self.assertTrue(resp.success)
self.assertEqual(resp.data["provider"], "deepseek")
@@ -188,6 +191,7 @@ class LlmTestEndpointTest(unittest.TestCase):
base_url="https://example.com/v1",
base_url_preset="openai-default",
user_agent="MoviePilot-Custom/1.0",
use_proxy=False,
)
with patch.object(system_endpoint.settings, "AI_AGENT_ENABLE", False), patch.object(
@@ -212,6 +216,7 @@ class LlmTestEndpointTest(unittest.TestCase):
base_url="https://example.com/v1",
base_url_preset="openai-default",
user_agent="MoviePilot-Custom/1.0",
use_proxy=False,
)
self.assertTrue(resp.success)
self.assertEqual(resp.data["provider"], "openai")
@@ -234,6 +239,7 @@ class LlmTestEndpointTest(unittest.TestCase):
base_url="https://api.deepseek.com",
base_url_preset="deepseek-default",
user_agent=None,
use_proxy=None,
)
with patch.object(system_endpoint.settings, "AI_AGENT_ENABLE", False), patch.object(
@@ -252,6 +258,7 @@ class LlmTestEndpointTest(unittest.TestCase):
base_url="https://api.deepseek.com",
base_url_preset="deepseek-default",
user_agent=None,
use_proxy=None,
)
self.assertTrue(resp.success)