Agent state codestyle fix

devika.py

@@ -42,7 +42,7 @@
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 
 manager = ProjectManager()
-AgentState = AgentState()
+agent_state = AgentState()
 config = Config()
 logger = Logger()
 
@@ -88,7 +88,7 @@ def handle_message(data):
         new_message['from_devika'] = False
         manager.add_message_from_user(project_name, new_message['message'])
 
-        if AgentState.is_agent_completed(project_name):
+        if agent_state.is_agent_completed(project_name):
             thread = Thread(target=lambda: agent.subsequent_execute(message, project_name))
             thread.start()
 
@@ -102,7 +102,7 @@ def handle_message(data):
 def is_agent_active():
     data = request.json
     project_name = data.get("project_name")
-    is_active = AgentState.is_agent_active(project_name)
+    is_active = agent_state.is_agent_active(project_name)
     return jsonify({"is_active": is_active})
 
 
@@ -111,8 +111,8 @@ def is_agent_active():
 def get_agent_state():
     data = request.json
     project_name = data.get("project_name")
-    agent_state = AgentState.get_latest_state(project_name)
-    return jsonify({"state": agent_state})
+    latest_state = agent_state.get_latest_state(project_name)
+    return jsonify({"state": latest_state})
 
 
 @app.route("/api/get-browser-snapshot", methods=["GET"])
@@ -126,23 +126,23 @@ def browser_snapshot():
 @route_logger(logger)
 def get_browser_session():
     project_name = request.args.get("project_name")
-    agent_state = AgentState.get_latest_state(project_name)
-    if not agent_state:
+    latest_state = agent_state.get_latest_state(project_name)
+    if not latest_state:
         return jsonify({"session": None})
     else:
-        browser_session = agent_state["browser_session"]
+        browser_session = latest_state["browser_session"]
         return jsonify({"session": browser_session})
 
 
 @app.route("/api/get-terminal-session", methods=["GET"])
 @route_logger(logger)
 def get_terminal_session():
     project_name = request.args.get("project_name")
-    agent_state = AgentState.get_latest_state(project_name)
-    if not agent_state:
+    latest_state = agent_state.get_latest_state(project_name)
+    if not latest_state:
         return jsonify({"terminal_state": None})
     else:
-        terminal_state = agent_state["terminal_session"]
+        terminal_state = latest_state["terminal_session"]
         return jsonify({"terminal_state": terminal_state})
 
 
@@ -169,7 +169,7 @@ def calculate_tokens():
 @route_logger(logger)
 def token_usage():
     project_name = request.args.get("project_name")
-    token_count = AgentState.get_latest_token_usage(project_name)
+    token_count = agent_state.get_latest_token_usage(project_name)
     return jsonify({"token_usage": token_count})
 
 

src/llm/llm.py

@@ -18,7 +18,7 @@
 
 ollama = Ollama()
 logger = Logger()
-agentState = AgentState()
+agent_state = AgentState()
 
 
 class LLM:
@@ -69,9 +69,9 @@ def model_id_to_enum_mapping(self) -> dict:
     @staticmethod
     def update_global_token_usage(string: str, project_name: str):
         token_usage = len(TIKTOKEN_ENC.encode(string))
-        agentState.update_token_usage(project_name, token_usage)
+        agent_state.update_token_usage(project_name, token_usage)
 
-        total = agentState.get_latest_token_usage(project_name) + token_usage
+        total = agent_state.get_latest_token_usage(project_name) + token_usage
         emit_agent("tokens", {"token_usage": total})
 
     def inference(self, prompt: str, project_name: str) -> str: