feat: add logs to response for tool run (#2205)

Co-authored-by: Caren Thomas <caren@caren-mac.local>
This commit is contained in:
cthomas
2024-12-10 13:24:05 -08:00
committed by GitHub
parent cf77cf79f9
commit 68792caec2
5 changed files with 38 additions and 8 deletions

View File

@@ -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):