jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Tests fix

Browse source
This commit is contained in:
gustavoschaedler 2023-06-16 20:22:45 +01:00
commit 13d52e4987
3 changed files with 36 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

12
tests/test_llms_template.py
View file

@ -371,7 +371,16 @@ def test_chat_open_ai(client: TestClient):
"multiline": False,
"value": "gpt-3.5-turbo",
"password": False,
"options": ["gpt-3.5-turbo", "gpt-4", "gpt-4-32k"],
"options": [
"gpt-3.5-turbo-0613",
"gpt-3.5-turbo",
"gpt-3.5-turbo-16k-0613",
"gpt-3.5-turbo-16k",
"gpt-4-0613",
"gpt-4-32k-0613",
"gpt-4",
"gpt-4-32k"
],
"name": "model_name",
"type": "str",
"list": True,
@ -478,6 +487,7 @@ def test_chat_open_ai(client: TestClient):
== "Wrapper around OpenAI Chat large language models." # noqa E501
)
assert set(model["base_classes"]) == {
"Serializable",
"BaseChatModel",
"ChatOpenAI",
"BaseLanguageModel",