Hopiu/llm
1
0
Fork 0
mirror of https://github.com/Hopiu/llm.git synced 2026-04-05 05:51:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Test set_alias(..., alias) too, refs #154

Browse source
This commit is contained in:
Simon Willison 2023-08-19 22:02:58 -07:00
parent 8cdcf1e689
commit 2d73e6c8d1
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tests/test_aliases.py
View file

@ -5,10 +5,11 @@ import json
import pytest
def test_set_alias():
@pytest.mark.parametrize("model_id_or_alias", ("gpt-3.5-turbo", "chatgpt"))
def test_set_alias(model_id_or_alias):
with pytest.raises(llm.UnknownModelError):
llm.get_model("this-is-a-new-alias")
llm.set_alias("this-is-a-new-alias", "gpt-3.5-turbo")
llm.set_alias("this-is-a-new-alias", model_id_or_alias)
assert llm.get_model("this-is-a-new-alias").model_id == "gpt-3.5-turbo"