feat: add logs to response for tool run (#2205)
Co-authored-by: Caren Thomas <caren@caren-mac.local>
This commit is contained in:
@@ -541,6 +541,8 @@ def test_tool_run(server, mock_e2b_api_key_none, user_id, agent_id):
|
||||
print(result)
|
||||
assert result.status == "success"
|
||||
assert result.function_return == "Ingested message Hello, world!", result.function_return
|
||||
assert result.stdout == ['']
|
||||
assert result.stderr == ['']
|
||||
|
||||
result = server.run_tool_from_source(
|
||||
user_id=user_id,
|
||||
@@ -552,6 +554,8 @@ def test_tool_run(server, mock_e2b_api_key_none, user_id, agent_id):
|
||||
print(result)
|
||||
assert result.status == "success"
|
||||
assert result.function_return == "Ingested message Well well well", result.function_return
|
||||
assert result.stdout == ['']
|
||||
assert result.stderr == ['']
|
||||
|
||||
result = server.run_tool_from_source(
|
||||
user_id=user_id,
|
||||
@@ -564,6 +568,8 @@ def test_tool_run(server, mock_e2b_api_key_none, user_id, agent_id):
|
||||
assert result.status == "error"
|
||||
assert "Error" in result.function_return, result.function_return
|
||||
assert "missing 1 required positional argument" in result.function_return, result.function_return
|
||||
assert result.stdout == ['']
|
||||
assert result.stderr != [''], "missing 1 required positional argument" in result.stderr[0]
|
||||
|
||||
# Test that we can still pull the tool out by default (pulls that last tool in the source)
|
||||
result = server.run_tool_from_source(
|
||||
@@ -576,6 +582,8 @@ def test_tool_run(server, mock_e2b_api_key_none, user_id, agent_id):
|
||||
print(result)
|
||||
assert result.status == "success"
|
||||
assert result.function_return == "Ingested message Well well well", result.function_return
|
||||
assert result.stdout != [''], "I'm a distractor" in result.stdout[0]
|
||||
assert result.stderr == ['']
|
||||
|
||||
# Test that we can pull the tool out by name
|
||||
result = server.run_tool_from_source(
|
||||
@@ -588,6 +596,8 @@ def test_tool_run(server, mock_e2b_api_key_none, user_id, agent_id):
|
||||
print(result)
|
||||
assert result.status == "success"
|
||||
assert result.function_return == "Ingested message Well well well", result.function_return
|
||||
assert result.stdout != [''], "I'm a distractor" in result.stdout[0]
|
||||
assert result.stderr == ['']
|
||||
|
||||
# Test that we can pull a different tool out by name
|
||||
result = server.run_tool_from_source(
|
||||
@@ -600,6 +610,8 @@ def test_tool_run(server, mock_e2b_api_key_none, user_id, agent_id):
|
||||
print(result)
|
||||
assert result.status == "success"
|
||||
assert result.function_return == str(None), result.function_return
|
||||
assert result.stdout != [''], "I'm a distractor" in result.stdout[0]
|
||||
assert result.stderr == ['']
|
||||
|
||||
|
||||
def test_composio_client_simple(server):
|
||||
|
||||
Reference in New Issue
Block a user