Initial vLLM agent strategy
This commit is contained in:
21
tests/test_strategy_id_payload.py
Normal file
21
tests/test_strategy_id_payload.py
Normal file
@@ -0,0 +1,21 @@
|
||||
from app.clients.modelhub import ModelHubClient
|
||||
from app.settings import Settings
|
||||
|
||||
|
||||
def test_strategy_id_payload_field_configurable():
|
||||
settings = Settings(
|
||||
auth_token="a", hf_token="h", contest_api_token="c", email="e", user_id="1",
|
||||
contributors="u", strategy_id="sid", contest_task_strategy_field="strategy_id"
|
||||
)
|
||||
payload = ModelHubClient(settings).build_contest_payload("owner/model", "hygon_k100-ai", "cfg")
|
||||
assert payload["strategy_id"] == "sid"
|
||||
assert payload["taskType"] == "text-generation"
|
||||
|
||||
|
||||
def test_strategy_id_payload_can_disable_injection():
|
||||
settings = Settings(
|
||||
auth_token="a", hf_token="h", contest_api_token="c", email="e", user_id="1",
|
||||
contributors="u", strategy_id="sid", inject_strategy_id=False
|
||||
)
|
||||
payload = ModelHubClient(settings).build_contest_payload("owner/model", "hygon_k100-ai", "cfg")
|
||||
assert "strategyId" not in payload
|
||||
Reference in New Issue
Block a user