Function _build_test_mapper

tests/test_proxy.py:64–77

(*model_ids: str)

def _build_test_mapper(*model_ids: str) -> ModelMapper:
    mapper = ModelMapper("https://api.example.test/v1")
    for model_id in model_ids:
        provider = model_id.split("/", 1)[0] if "/" in model_id else "unknown"
        mapper._pool[model_id] = DiscoveredModel(
            id=model_id,
            provider=provider,
            owned_by=provider,
            pricing=ModelPricing(1.0, 5.0),
            capabilities=ModelCapabilities(tool_calling=True, vision=False, reasoning=False),
        )
        mapper._upstream_models.add(model_id)
    mapper._discovered = True
    return mapper


class TestPromptExtraction: