Updated test , especially for tool gen mocking

pull/615/head
Sambhav Dixit 2 months ago committed by GitHub
parent fe39cd5393
commit 750f9a1e14
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -5,6 +5,15 @@ from datetime import datetime
import unittest import unittest
from swarms.schemas.agent_step_schemas import ManySteps, Step from swarms.schemas.agent_step_schemas import ManySteps, Step
from swarms.structs.agent import Agent from swarms.structs.agent import Agent
from swarms.tools.tool_parse_exec import parse_and_execute_json
# Mock parse_and_execute_json for testing
parse_and_execute_json = MagicMock()
parse_and_execute_json.return_value = {
"tool_name": "calculator",
"args": {"numbers": [2, 2]},
"output": "4"
}
class TestAgentLogging(unittest.TestCase): class TestAgentLogging(unittest.TestCase):
def setUp(self): def setUp(self):

Loading…
Cancel
Save