scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-25 01:02:15 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 80

test(mcp): check catalog prompt contract via model schema, not a FastAPI app
Some checks are pending
LiteLLM Rust / rust-lint (push) Waiting to run
Details
LiteLLM Rust / rust-test (push) Waiting to run
Details
LiteLLM Rust / rust-wheel (push) Waiting to run
Details
Terraform Provider / gofmt, vet, build, test (push) Waiting to run
Details
Terraform Provider / Provider endpoints vs proxy OpenAPI schema (push) Waiting to run
Details

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
joshua 2026-09-21 23:38:40 +00:00
parent 779e4bd433
commit 1aaf259ed0
1 changed files with 10 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
tests/test_litellm/proxy/_experimental/mcp_server/test_rest_endpoints.py
View file

@ -2563,29 +2563,20 @@ class TestListPromptsAndResourcesRestAPI:
"""The catalog response reuses the MCP SDK prompt type, which shares its class name with the
prompt-management request model, so the two must land as separate OpenAPI components:
POST /prompts still requires prompt_id + litellm_params while the catalog item requires name."""
from fastapi import FastAPI
from litellm.proxy.prompts.prompt_endpoints import Prompt as PromptManagementRequest
from litellm.proxy.prompts.prompt_endpoints import router as prompt_router
app = FastAPI()
app.include_router(rest_endpoints.router)
app.include_router(prompt_router)
spec = app.openapi()
schemas = spec["components"]["schemas"]
def component(ref: Dict[str, Any]) -> Dict[str, Any]:
return schemas[ref["$ref"].rsplit("/", 1)[1]]
create_prompt_operation = spec["paths"]["/prompts"]["post"]
create_prompt_body = component(create_prompt_operation["requestBody"]["content"]["application/json"]["schema"])
assert {"prompt_id", "litellm_params"} <= set(create_prompt_body["required"])
catalog_operation = spec["paths"]["/mcp-rest/prompts/list"]["get"]
catalog_response = component(catalog_operation["responses"]["200"]["content"]["application/json"]["schema"])
catalog_prompt = component(catalog_response["properties"]["prompts"]["items"])
catalog_schema: Final = rest_endpoints.ListMCPPromptsRestAPIResponse.model_json_schema(
ref_template="#/components/schemas/{model}"
)
items_ref: Final = catalog_schema["properties"]["prompts"]["items"]["$ref"]
assert items_ref.rsplit("/", 1)[1] != PromptManagementRequest.__name__
catalog_prompt: Final = catalog_schema["$defs"][items_ref.rsplit("/", 1)[1]]
assert catalog_prompt["required"] == ["name"]
assert "arguments" in catalog_prompt["properties"]
management_schema: Final = PromptManagementRequest.model_json_schema()
assert {"prompt_id", "litellm_params"} <= set(management_schema["required"])
class TestCallToolRestAPI:
pytestmark = pytest.mark.asyncio