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

fix: handle empty tool response (#8118)

Browse source 
* handle empty tool response

* Update src/backend/base/langflow/components/tools/mcp_component.py

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

* DataFrame

* fix test

---------

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
This commit is contained in:
Sebastián Estévez 2025-05-21 13:50:18 -04:00 • committed by GitHub
commit 076b228599
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 14 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/backend/tests/unit/components/tools/test_mcp_component.py
View file

@ -108,7 +108,10 @@ class TestMCPToolsComponent(ComponentTestBaseWithoutClient):
# Mock the coroutine response
mock_response = AsyncMock()
mock_response.content = [MagicMock(text="Test response")]
mock_content_item = MagicMock()
mock_content_item.text = "Test response"
mock_content_item.model_dump.return_value = {"text": "Test response"}
mock_response.content = [mock_content_item]
mock_create_coroutine.return_value = AsyncMock(return_value=mock_response)
# Create a mock tool and add it to the cache
@ -126,7 +129,8 @@ class TestMCPToolsComponent(ComponentTestBaseWithoutClient):
mock_get_inputs.return_value = {"test_tool": [mock_input]}
output = await component.build_output()
assert output.text == "Test response"
# Use iloc to access the first row's 'text' column value
assert output.iloc[0]["text"] == "Test response"
# Verify the mocks were called correctly
mock_get_inputs.assert_called_once_with(component.tools)
mock_structured_tool.coroutine.assert_called_once_with(test_param="test value")