scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-13 03:59:12 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 66

chore(mcp): remove unrelated formatting changes

Browse source
This commit is contained in:
Sisyphus 2026-10-07 13:06:52 +08:00
parent 7699926b06
commit 529cf36c15
1 changed files with 13 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
tests/unit/proxy/_experimental/mcp_server/test_openapi_spec_path_url.py
View file

@ -1,6 +1,6 @@
from __future__ import annotations
from typing import Any
from typing import Any, Dict
import httpx
import pytest
@ -33,7 +33,7 @@ class _FakeAsyncHTTPHandler:
def test_load_openapi_spec_supports_http_url(monkeypatch: pytest.MonkeyPatch) -> None:
url = "http://example.local/openapi.json"
expected: dict[str, Any] = {
expected: Dict[str, Any] = {
"openapi": "3.0.0",
"info": {"title": "Test API", "version": "1.0.0"},
"paths": {},
@ -44,7 +44,7 @@ def test_load_openapi_spec_supports_http_url(monkeypatch: pytest.MonkeyPatch) ->
resp = httpx.Response(status_code=200, json=expected, request=req)
calls = {"get_async_httpx_client": 0}
handler_holder: dict[str, Any] = {}
handler_holder: Dict[str, Any] = {}
def fake_get_async_httpx_client(*args, **kwargs):
calls["get_async_httpx_client"] += 1
@ -56,7 +56,9 @@ def test_load_openapi_spec_supports_http_url(monkeypatch: pytest.MonkeyPatch) ->
monkeypatch.setattr(gen, "get_async_httpx_client", fake_get_async_httpx_client)
# Bypass SSRF validation in test (example.local doesn't resolve)
monkeypatch.setattr(gen, "async_safe_get", lambda client, url, **kw: client.get(url))
monkeypatch.setattr(
gen, "async_safe_get", lambda client, url, **kw: client.get(url)
)
# Fail loudly if someone reintroduces direct httpx.get()
def boom(*args, **kwargs):
@ -71,8 +73,10 @@ def test_load_openapi_spec_supports_http_url(monkeypatch: pytest.MonkeyPatch) ->
assert handler_holder["handler"].calls == 1
def test_load_openapi_spec_supports_local_file_path(tmp_path, monkeypatch: pytest.MonkeyPatch) -> None:
expected: dict[str, Any] = {
def test_load_openapi_spec_supports_local_file_path(
tmp_path, monkeypatch: pytest.MonkeyPatch
) -> None:
expected: Dict[str, Any] = {
"openapi": "3.0.0",
"info": {"title": "Local API", "version": "1.0.0"},
"paths": {},
@ -86,7 +90,9 @@ def test_load_openapi_spec_supports_local_file_path(tmp_path, monkeypatch: pytes
# For local files, shared client must NOT be used.
def boom_client(*args, **kwargs):
raise AssertionError("get_async_httpx_client() must not be called for local file paths")
raise AssertionError(
"get_async_httpx_client() must not be called for local file paths"
)
monkeypatch.setattr(gen, "get_async_httpx_client", boom_client)