(*model_ids: str)
| 884 | |
| 885 | |
| 886 | def _build_seed_mapper(*model_ids: str) -> ModelMapper: |
| 887 | mapper = ModelMapper("https://api.commonstack.ai/v1") |
| 888 | for model_id in model_ids: |
| 889 | provider = model_id.split("/", 1)[0] if "/" in model_id else "unknown" |
| 890 | mapper._pool[model_id] = DiscoveredModel( |
| 891 | id=model_id, |
| 892 | provider=provider, |
| 893 | owned_by=provider, |
| 894 | pricing=ModelPricing(0.2, 0.8), |
| 895 | capabilities=ModelCapabilities(tool_calling=True, vision=False, reasoning=False), |
| 896 | ) |
| 897 | mapper._upstream_models.add(model_id) |
| 898 | mapper._discovered = True |
| 899 | |
| 900 | async def _fake_discover(api_key: str | None = None) -> int: |
| 901 | return len(mapper._pool) |
| 902 | |
| 903 | mapper.discover = _fake_discover # type: ignore[method-assign] |
| 904 | return mapper |
| 905 | |
| 906 | |
| 907 | class TestMessagesEndpoint: |
no test coverage detected