Merge pull request #1152 from NousResearch/hermes/hermes-f47f71c0
feat: concurrent tool execution with ThreadPoolExecutor
This commit is contained in:
commit
0157253145
3 changed files with 429 additions and 2 deletions
|
|
@ -91,8 +91,11 @@ class TestPreToolCheck:
|
|||
agent._persist_session = MagicMock()
|
||||
|
||||
# Import and call the method
|
||||
import types
|
||||
from run_agent import AIAgent
|
||||
# Bind the real method to our mock
|
||||
# Bind the real methods to our mock so dispatch works correctly
|
||||
agent._execute_tool_calls_sequential = types.MethodType(AIAgent._execute_tool_calls_sequential, agent)
|
||||
agent._execute_tool_calls_concurrent = types.MethodType(AIAgent._execute_tool_calls_concurrent, agent)
|
||||
AIAgent._execute_tool_calls(agent, assistant_msg, messages, "default")
|
||||
|
||||
# All 3 should be skipped
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue