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

refactor(test_chains_template.py): make input_key and output_key required and show in all chains

Browse source 
feat(test_chains_template.py): add prompt to mid_journey_prompt_chain and time_travel_guide_chain chains
This commit is contained in:
Gabriel Almeida 2023-04-10 19:03:32 -03:00
commit 1d49cc16bd
1 changed files with 30 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

60
tests/test_chains_template.py
View file

@ -53,9 +53,9 @@ def test_conversation_chain(client: TestClient):
"list": False,
}
assert template["input_key"] == {
"required": False,
"required": True,
"placeholder": "",
"show": False,
"show": True,
"multiline": False,
"value": "input",
"password": False,
@ -64,9 +64,9 @@ def test_conversation_chain(client: TestClient):
"list": False,
}
assert template["output_key"] == {
"required": False,
"required": True,
"placeholder": "",
"show": False,
"show": True,
"multiline": False,
"value": "response",
"password": False,
@ -125,9 +125,9 @@ def test_llm_chain(client: TestClient):
"list": False,
}
assert template["output_key"] == {
"required": False,
"required": True,
"placeholder": "",
"show": False,
"show": True,
"multiline": False,
"value": "text",
"password": False,
@ -179,9 +179,9 @@ def test_llm_checker_chain(client: TestClient):
"list": False,
}
assert template["input_key"] == {
"required": False,
"required": True,
"placeholder": "",
"show": False,
"show": True,
"multiline": False,
"value": "query",
"password": False,
@ -190,9 +190,9 @@ def test_llm_checker_chain(client: TestClient):
"list": False,
}
assert template["output_key"] == {
"required": False,
"required": True,
"placeholder": "",
"show": False,
"show": True,
"multiline": False,
"value": "result",
"password": False,
@ -251,9 +251,9 @@ def test_llm_math_chain(client: TestClient):
"list": False,
}
assert template["input_key"] == {
"required": False,
"required": True,
"placeholder": "",
"show": False,
"show": True,
"multiline": False,
"value": "question",
"password": False,
@ -262,9 +262,9 @@ def test_llm_math_chain(client: TestClient):
"list": False,
}
assert template["output_key"] == {
"required": False,
"required": True,
"placeholder": "",
"show": False,
"show": True,
"multiline": False,
"value": "answer",
"password": False,
@ -338,9 +338,9 @@ def test_series_character_chain(client: TestClient):
"list": False,
}
assert template["input_key"] == {
"required": False,
"required": True,
"placeholder": "",
"show": False,
"show": True,
"multiline": False,
"value": "input",
"password": False,
@ -349,9 +349,9 @@ def test_series_character_chain(client: TestClient):
"list": False,
}
assert template["output_key"] == {
"required": False,
"required": True,
"placeholder": "",
"show": False,
"show": True,
"multiline": False,
"value": "response",
"password": False,
@ -462,13 +462,13 @@ def test_mid_journey_prompt_chain(client: TestClient):
assert template["prompt"] == {
"required": False,
"placeholder": "",
"show": False,
"show": True,
"multiline": False,
"value": {
"input_variables": ["history", "input"],
"output_parser": None,
"partial_variables": {},
"template": "The following is a friendly conversation between a human and an AI. The AI is talkative and provides lots of specific details from its context. If the AI does not know the answer to a question, it truthfully says it does not know.\n\nCurrent conversation:\n{history}\nHuman: {input}\nAI:", # noqa: E501
"template": "The following is a friendly conversation between a human and an AI. The AI is talkative and provides lots of specific details from its context. If the AI does not know the answer to a question, it truthfully says it does not know.\n\nCurrent conversation:\n{history}\nHuman: {input}\nAI:",
"template_format": "f-string",
"validate_template": True,
"_type": "prompt",
@ -489,9 +489,9 @@ def test_mid_journey_prompt_chain(client: TestClient):
"list": False,
}
assert template["output_key"] == {
"required": False,
"required": True,
"placeholder": "",
"show": False,
"show": True,
"multiline": False,
"value": "response",
"password": False,
@ -500,9 +500,9 @@ def test_mid_journey_prompt_chain(client: TestClient):
"list": False,
}
assert template["input_key"] == {
"required": False,
"required": True,
"placeholder": "",
"show": False,
"show": True,
"multiline": False,
"value": "input",
"password": False,
@ -590,13 +590,13 @@ def test_time_travel_guide_chain(client: TestClient):
assert template["prompt"] == {
"required": False,
"placeholder": "",
"show": False,
"show": True,
"multiline": False,
"value": {
"input_variables": ["history", "input"],
"output_parser": None,
"partial_variables": {},
"template": "The following is a friendly conversation between a human and an AI. The AI is talkative and provides lots of specific details from its context. If the AI does not know the answer to a question, it truthfully says it does not know.\n\nCurrent conversation:\n{history}\nHuman: {input}\nAI:", # noqa: E501
"template": "The following is a friendly conversation between a human and an AI. The AI is talkative and provides lots of specific details from its context. If the AI does not know the answer to a question, it truthfully says it does not know.\n\nCurrent conversation:\n{history}\nHuman: {input}\nAI:",
"template_format": "f-string",
"validate_template": True,
"_type": "prompt",
@ -617,9 +617,9 @@ def test_time_travel_guide_chain(client: TestClient):
"list": False,
}
assert template["output_key"] == {
"required": False,
"required": True,
"placeholder": "",
"show": False,
"show": True,
"multiline": False,
"value": "response",
"password": False,
@ -629,9 +629,9 @@ def test_time_travel_guide_chain(client: TestClient):
}
assert template["input_key"] == {
"required": False,
"required": True,
"placeholder": "",
"show": False,
"show": True,
"multiline": False,
"value": "input",
"password": False,