Improved System Message (#7)

- feat: Update system message (a59bd602b051975ae4f46930fc0c3b8e597942d8)

tdagent/grchat.py

@@ -23,9 +23,24 @@ if TYPE_CHECKING:
 #### Constants ####
 
 SYSTEM_MESSAGE = SystemMessage(
-    "You a security analyst that analyses tickets. You will receives tickets to handle, you should enrich them with data from tools available, suggest if it's likely a true positive threat or not. You must also present your reasoning to arrive to that conclusion",
+    """
+You are a security analyst assistant responsible for collecting, analyzing
+and disseminating actionable intelligence related to cyber threats,
+vulnerabilities and threat actors.
+
+When presented with potential incidents information or tickets, you should
+evaluate the presented evidence, decide what is missing and gather
+additional data using any tool at your disposal. After gathering more
+information you must evaluate if the incident is a threat or
+not and, if possible, remediation actions.
+
+You must always present the conducted analysis and final conclusion.
+Never use external means of communication, like emails or SMS, unless
+instructed to do so.
+""".strip(),
 )
 
+
 GRADIO_ROLE_TO_LG_MESSAGE_TYPE = MappingProxyType(
     {
         "user": HumanMessage,
@@ -68,7 +83,7 @@ def create_bedrock_llm(
         llm = ChatBedrock(
             model_id=bedrock_model_id,
             client=bedrock_client,
-            model_kwargs={"temperature": 0.7},
+            model_kwargs={"temperature": 0.8},
         )
     except Exception as e:  # noqa: BLE001
         return None, str(e)

uv.lock

@@ -1938,7 +1938,7 @@ test = [
 
 [package.metadata]
 requires-dist = [
-    { name = "gradio", extras = ["mcp"], specifier = ">=5.32.1" },
+    { name = "gradio", extras = ["mcp"], specifier = "~=5.31" },
     { name = "huggingface-hub", specifier = ">=0.32.3" },
     { name = "langchain-aws", specifier = ">=0.2.24" },
     { name = "langchain-mcp-adapters", specifier = ">=0.1.1" },