18 lines
696 B
Python
18 lines
696 B
Python
from server.routers.model_provider_router import ModelProviderPayload
|
|
|
|
|
|
def test_model_provider_payload_accepts_embedding_and_rerank_urls():
|
|
payload = ModelProviderPayload(
|
|
provider_id="mixed-provider",
|
|
display_name="Mixed Provider",
|
|
base_url="https://api.example.com/v1",
|
|
embedding_base_url="https://api.example.com/v1/embeddings",
|
|
rerank_base_url="https://api.example.com/v1/rerank",
|
|
capabilities=["chat", "embedding", "rerank"],
|
|
)
|
|
|
|
data = payload.model_dump(exclude_none=True)
|
|
|
|
assert data["embedding_base_url"] == "https://api.example.com/v1/embeddings"
|
|
assert data["rerank_base_url"] == "https://api.example.com/v1/rerank"
|