diff --git a/src/openjarvis/agents/native_react.py b/src/openjarvis/agents/native_react.py index bec88754..cfcc5023 100644 --- a/src/openjarvis/agents/native_react.py +++ b/src/openjarvis/agents/native_react.py @@ -109,7 +109,11 @@ class NativeReActAgent(ToolUsingAgent): all_tool_results: list[ToolResult] = [] turns = 0 - total_usage: dict[str, int] = {"prompt_tokens": 0, "completion_tokens": 0, "total_tokens": 0} + total_usage: dict[str, int] = { + "prompt_tokens": 0, + "completion_tokens": 0, + "total_tokens": 0, + } for _turn in range(self._max_turns): turns += 1 diff --git a/src/openjarvis/agents/operative.py b/src/openjarvis/agents/operative.py index e6a05270..7f88f840 100644 --- a/src/openjarvis/agents/operative.py +++ b/src/openjarvis/agents/operative.py @@ -104,7 +104,11 @@ class OperativeAgent(ToolUsingAgent): turns = 0 content = "" state_stored_by_tool = False - total_usage: dict[str, int] = {"prompt_tokens": 0, "completion_tokens": 0, "total_tokens": 0} + total_usage: dict[str, int] = { + "prompt_tokens": 0, + "completion_tokens": 0, + "total_tokens": 0, + } for _turn in range(self._max_turns): turns += 1 diff --git a/src/openjarvis/agents/rlm.py b/src/openjarvis/agents/rlm.py index 9efc99a0..0cc6e3a4 100644 --- a/src/openjarvis/agents/rlm.py +++ b/src/openjarvis/agents/rlm.py @@ -173,7 +173,11 @@ class RLMAgent(ToolUsingAgent): all_tool_results: list[ToolResult] = [] turns = 0 - total_usage: dict[str, int] = {"prompt_tokens": 0, "completion_tokens": 0, "total_tokens": 0} + total_usage: dict[str, int] = { + "prompt_tokens": 0, + "completion_tokens": 0, + "total_tokens": 0, + } for _turn in range(self._max_turns): turns += 1