Bypass proxies for loopback engines

This commit is contained in:
2026-04-04 23:50:42 +08:00
parent 7632de8dad
commit 75a9842f1a
2 changed files with 30 additions and 4 deletions

View File

@@ -8,7 +8,7 @@ from pathlib import Path
from unittest import mock
from aituner.cli import main as cli_main
from aituner.http_client import _openai_url
from aituner.http_client import _openai_url, _should_bypass_proxy
from aituner.job import append_job, build_trial_job
from aituner.llm import build_prompt, parse_proposal_text
from aituner.search import ThresholdProbe, binary_search_max_feasible
@@ -833,6 +833,11 @@ class CoreFlowTests(unittest.TestCase):
"http://example.com/v1/chat/completions",
)
def test_loopback_urls_bypass_proxy(self) -> None:
self.assertTrue(_should_bypass_proxy("http://127.0.0.1:8000/v1/models"))
self.assertTrue(_should_bypass_proxy("http://localhost:8000/health"))
self.assertFalse(_should_bypass_proxy("http://example.com/v1/models"))
if __name__ == "__main__":
unittest.main()