Update tdagent/grchat.py

tdagent/grchat.py

@@ -39,8 +39,9 @@ if TYPE_CHECKING:
 class AgentType(str, enum.Enum):
     """TDAgent type."""
 
-    INCIDENT_HANDLER = "Incident handler"
     DATA_ENRICHER = "Data enricher"
+    INCIDENT_HANDLER = "Incident handler"
+    PEN_TESTER = "PenTester"
 
     def __str__(self) -> str:  # noqa: D105
         return self.value
@@ -48,6 +49,14 @@ class AgentType(str, enum.Enum):
 
 AGENT_SYSTEM_MESSAGES = OrderedDict(
     (
+        (
+            AgentType.DATA_ENRICHER,
+            """
+You are a cybersecurity incidence data enriching assistant. Analysts
+will present information about security incidents and you must use
+all the tools at your disposal to enrich the data as much as possible.
+""".strip(),
+        ),
         (
             AgentType.INCIDENT_HANDLER,
             """
@@ -67,11 +76,10 @@ instructed to do so.
 """.strip(),
         ),
         (
-            AgentType.DATA_ENRICHER,
+            AgentType.PEN_TESTER,
             """
-You are a cybersecurity incidence data enriching assistant. Analysts
-will present information about security incidents and you must use
-all the tools at your disposal to enrich the data as much as possible.
+You are a cybersecurity pentester. You use tools to analyze domain to try to discover system vulnerabilities.
+Always report you findings and suggest next steps to deep dive where applicable.
 """.strip(),
         ),
     ),