Refactor imports and agent instantiation to use the netsecgame package, removing dynamic path additions and old AIDojoCoordinator references.

Author: ondrej-lukas

agents/attackers/llm_bert/llm_action_planner.py

@@ -53,13 +53,7 @@ def _safe_compile(fn=None, **kwargs):
 except ImportError:
     import validate_responses
 
-# Add parent directories dynamically
-sys.path.append(
-    path.dirname(path.dirname(path.dirname(path.dirname(path.dirname(path.abspath(__file__))))))
-)
-sys.path.append(path.dirname(path.dirname(path.dirname(path.abspath(__file__)))))
-
-from AIDojoCoordinator.game_components import ActionType, Observation
+from netsecgame import ActionType, Observation
 from NetSecGameAgents.agents.llm_utils import create_action_from_response, create_status_from_state
 
 

agents/attackers/llm_bert/llm_agent_qa.py

@@ -18,8 +18,8 @@
 except ImportError:
     from llm_action_planner import LLMActionPlanner
 
-from AIDojoCoordinator.game_components import AgentStatus, Action, ActionType
-from NetSecGameAgents.agents.base_agent import BaseAgent
+from netsecgame.game_components import AgentStatus, Action, ActionType, AgentRole
+from netsecgame import BaseAgent
 
 #mlflow.set_tracking_uri("http://147.32.83.60")
 #mlflow.set_experiment("LLM_QA_netsecgame_dec2024")
@@ -159,7 +159,7 @@
 
     logger = logging.getLogger("llm_react")
     logger.info("Start")
-    agent = BaseAgent(args.host, args.port, "Attacker")
+    agent = BaseAgent(args.host, args.port, role=AgentRole.Attacker)
 
     if not args.disable_mlflow:
         mlflow.set_tracking_uri(args.mlflow_tracking_uri)