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

🔧 fix(test_endpoints.py): remove unused imports and refactor test_basic_chat_with_two_session_ids_and_names to use a loop for better readability and maintainability

Browse source 
✨ feat(test_endpoints.py): add test_basic_chat_with_two_session_ids_and_names to test the functionality of handling multiple session IDs and names in the chat endpoint
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-09-05 14:49:20 -03:00
commit 00290d0f9e
1 changed files with 18 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

37
tests/test_endpoints.py
View file

@ -8,6 +8,8 @@ from fastapi.testclient import TestClient
from langflow.interface.tools.constants import CUSTOM_TOOLS from langflow.interface.tools.constants import CUSTOM_TOOLS
from langflow.template.frontend_node.chains import TimeTravelGuideChainNode from langflow.template.frontend_node.chains import TimeTravelGuideChainNode
from tests.utils import run_post
PROMPT_REQUEST = { PROMPT_REQUEST = {
"name": "string", "name": "string",
@ -422,30 +424,27 @@ def test_basic_chat_different_session_ids(client, added_flow, created_api_key):
assert "Gabriel" not in response.json()["result"]["text"] assert "Gabriel" not in response.json()["result"]["text"]
@pytest.mark.parametrize("name", ["Gabriel", "John"]) def test_basic_chat_with_two_session_ids_and_names(client, added_flow, created_api_key):
def test_basic_chat_with_two_session_ids_and_names(
client, added_flow, created_api_key, name
):
headers = {"x-api-key": created_api_key.api_key} headers = {"x-api-key": created_api_key.api_key}
flow_id = added_flow.get("id")
names = ["Gabriel", "John"]
session_ids = []
def run_api_post(endpoint, headers, post_data): for name in names:
response = client.post(endpoint, headers=headers, json=post_data) post_data = {"inputs": {"text": f"Hi, My name is {name}"}}
assert response.status_code == 200, response.json() response_json = run_post(client, flow_id, headers, post_data)
return response.json()
def assert_name_and_session_in_response(name, response_json):
assert name in response_json["result"]["text"] assert name in response_json["result"]["text"]
assert "session_id" in response_json assert "session_id" in response_json
assert response_json["session_id"] is not None assert response_json["session_id"] is not None
# Run the /api/v1/process/{flow_id} endpoint session_ids.append(response_json["session_id"])
flow_id = added_flow.get("id")
post_data = {"inputs": {"text": f"Hi, My name is {name}"}}
response_json = run_api_post(f"api/v1/process/{flow_id}", headers, post_data)
assert_name_and_session_in_response(name, response_json)
session_id = response_json["session_id"]
# New request with the same session_id asking "What is my name?" should return name for i, name in enumerate(names):
post_data = {"inputs": {"text": "What is my name?"}, "session_id": session_id} post_data = {
response_json = run_api_post(f"api/v1/process/{flow_id}", headers, post_data) "inputs": {"text": "What is my name?"},
assert name in response_json["result"]["text"] "session_id": session_ids[i],
}
response_json = run_post(client, flow_id, headers, post_data)
assert name in response_json["result"]["text"]