mirror/temporal-ai-agent
1
0
Fork 0
mirror of https://github.com/temporal-community/temporal-ai-agent.git synced 2026-03-15 14:08:08 +01:00
Code Issues Packages Projects Releases Wiki Activity

straggler linted file

Browse Source
This commit is contained in:
Steve Androulakis
2025-06-01 08:58:19 -07:00
parent eb06cf5c8d
commit e52ddd3e5e
1 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tests/test_tool_activities.py
View File

@@ -98,7 +98,9 @@ class TestToolActivities:
# Mock the completion function # Mock the completion function
mock_response = MagicMock() mock_response = MagicMock()
mock_response.choices = [MagicMock()] mock_response.choices = [MagicMock()]
mock_response.choices[0].message.content = ( mock_response.choices[
0
].message.content = (
'{"next": "confirm", "tool": "TestTool", "response": "Test response"}' '{"next": "confirm", "tool": "TestTool", "response": "Test response"}'
) )
@@ -136,9 +138,9 @@ class TestToolActivities:
mock_response = MagicMock() mock_response = MagicMock()
mock_response.choices = [MagicMock()] mock_response.choices = [MagicMock()]
mock_response.choices[0].message.content = ( mock_response.choices[
'{"next": "done", "response": "Test"}' 0
) ].message.content = '{"next": "done", "response": "Test"}'
with patch("activities.tool_activities.completion") as mock_completion: with patch("activities.tool_activities.completion") as mock_completion:
mock_completion.return_value = mock_response mock_completion.return_value = mock_response
@@ -362,9 +364,9 @@ class TestEdgeCases:
# Mock the completion response # Mock the completion response
mock_response = MagicMock() mock_response = MagicMock()
mock_response.choices = [MagicMock()] mock_response.choices = [MagicMock()]
mock_response.choices[0].message.content = ( mock_response.choices[
'{"next": "done", "response": "Processed long prompt"}' 0
) ].message.content = '{"next": "done", "response": "Processed long prompt"}'
with patch("activities.tool_activities.completion", return_value=mock_response): with patch("activities.tool_activities.completion", return_value=mock_response):
activity_env = ActivityEnvironment() activity_env = ActivityEnvironment()