andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-04-30 23:02:50 +00:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

fix: TokenCalcHandler hashability, test MinimalExecutor as instance

Browse Source
This commit is contained in:
Greyson LaLonde
2026-04-03 17:25:51 +08:00
parent 6504e39d47
commit 78fbe455b9
2 changed files with 22 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lib/crewai/src/crewai/utilities/token_counter_callback.py
View File

@@ -33,6 +33,8 @@ class TokenCalcHandler(BaseModel):
model_config = {"arbitrary_types_allowed": True}
__hash__ = object.__hash__
token_cost_process: TokenProcess | None = Field(default=None)
def __init__(

60
lib/crewai/tests/memory/test_memory_root_scope.py
View File

@@ -543,17 +543,12 @@ class TestAgentScopeExtension:
mock_task.description = "Research task"
mock_task.expected_output = "Report"
class MinimalExecutor(CrewAgentExecutorMixin):
crew = None
agent = mock_agent
task = mock_task
iterations = 0
max_iter = 1
messages = []
_i18n = MagicMock()
_printer = Printer()
executor = CrewAgentExecutorMixin(
crew=None,
agent=mock_agent,
task=mock_task,
)
executor = MinimalExecutor()
executor._save_to_memory(AgentFinish(thought="", output="Result", text="Result"))
mock_memory.remember_many.assert_called_once()
@@ -582,17 +577,12 @@ class TestAgentScopeExtension:
mock_task.description = "Task"
mock_task.expected_output = "Output"
class MinimalExecutor(CrewAgentExecutorMixin):
crew = None
agent = mock_agent
task = mock_task
iterations = 0
max_iter = 1
messages = []
_i18n = MagicMock()
_printer = Printer()
executor = CrewAgentExecutorMixin(
crew=None,
agent=mock_agent,
task=mock_task,
)
executor = MinimalExecutor()
executor._save_to_memory(AgentFinish(thought="", output="R", text="R"))
call_kwargs = mock_memory.remember_many.call_args.kwargs
@@ -1077,17 +1067,12 @@ class TestAgentExecutorBackwardCompat:
mock_task.description = "Task"
mock_task.expected_output = "Output"
class MinimalExecutor(CrewAgentExecutorMixin):
crew = None
agent = mock_agent
task = mock_task
iterations = 0
max_iter = 1
messages = []
_i18n = MagicMock()
_printer = Printer()
executor = CrewAgentExecutorMixin(
crew=None,
agent=mock_agent,
task=mock_task,
)
executor = MinimalExecutor()
executor._save_to_memory(AgentFinish(thought="", output="R", text="R"))
# Should NOT pass root_scope when memory has none
@@ -1117,17 +1102,12 @@ class TestAgentExecutorBackwardCompat:
mock_task.description = "Task"
mock_task.expected_output = "Output"
class MinimalExecutor(CrewAgentExecutorMixin):
crew = None
agent = mock_agent
task = mock_task
iterations = 0
max_iter = 1
messages = []
_i18n = MagicMock()
_printer = Printer()
executor = CrewAgentExecutorMixin(
crew=None,
agent=mock_agent,
task=mock_task,
)
executor = MinimalExecutor()
executor._save_to_memory(AgentFinish(thought="", output="R", text="R"))
# Should pass extended root_scope