fix: make chat input sorting logic exclude other components (#5760)
* fix: Adjust chat input handling in layer sorting logic * fix: Update chat input sorting logic and assertions in tests * [autofix.ci] apply automated fixes * fix: Update test assertion for chat input sorting to reflect changes in expected output * [autofix.ci] apply automated fixes * fix: update vertex sorting logic to handle start_component_id condition - Modified the condition for sorting layers to include a check for start_component_id being None, ensuring correct behavior when this parameter is not set. - This change improves the accuracy of the vertex sorting process in the graph utility functions. * fix: update test assertion for vertex IDs in test_get_vertices - Modified the test assertion in `test_get_vertices` to reflect the expected output, ensuring it only checks for "ChatInput" in the returned IDs. - This change improves the accuracy of the test by aligning it with the current expected behavior of the endpoint. * [autofix.ci] apply automated fixes * [autofix.ci] apply automated fixes (attempt 2/3) * [autofix.ci] apply automated fixes (attempt 3/3) --------- Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> Co-authored-by: Ítalo Johnny <italojohnnydosanjos@gmail.com>
This commit is contained in:
parent
feb6907909
commit
18e21cfec9
4 changed files with 15 additions and 12 deletions
|
|
@ -257,7 +257,7 @@ async def test_get_vertices(client, added_flow_webhook_test, logged_in_headers):
|
|||
# The important part is before the - (ConversationBufferMemory, PromptTemplate, ChatOpenAI, LLMChain)
|
||||
ids = [_id.split("-")[0] for _id in response.json()["ids"]]
|
||||
|
||||
assert set(ids) == {"ChatInput", "Webhook"}
|
||||
assert set(ids) == {"ChatInput"}
|
||||
|
||||
|
||||
async def test_build_vertex_invalid_flow_id(client, logged_in_headers):
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue