File size: 734 Bytes
f580ad3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
"""
State Manager for CodeAct Agent.
Manages agent state creation and manipulation.
"""

import time
from typing import List, Dict, Optional


class StateManager:
    """Manages agent state creation and manipulation."""

    @staticmethod
    def create_state_dict(messages: List = None, step_count: int = 0,
                         error_count: int = 0, start_time: float = None,
                         current_plan: str = None) -> Dict:
        """Create a standardized state dictionary."""
        return {
            "messages": messages or [],
            "step_count": step_count,
            "error_count": error_count,
            "start_time": start_time or time.time(),
            "current_plan": current_plan
        }