feat: add new fields to test json

Author: andreiancuta-uipath

tests/agent/models/test_agent.py

@@ -38,10 +38,41 @@ def test_agent_with_all_tool_types_loads(self):
             "name": "Agent with All Tools",
             "metadata": {"isConversational": False, "storageVersion": "22.0.0"},
             "messages": [
-                {"role": "System", "content": "You are an agentic assistant."},
+                {
+                    "role": "System",
+                    "content": "You are an agentic assistant.",
+                    "contentTokens": [
+                        {
+                            "type": "simpleText",
+                            "rawString": "You are an agentic assistant.",
+                        }
+                    ],
+                },
                 {
                     "role": "User",
                     "content": "Use the provided tools. Execute {{task}} the number of {{times}}.",
+                    "contentTokens": [
+                        {
+                            "type": "simpleText",
+                            "rawString": "Use the provided tools. Execute ",
+                        },
+                        {
+                            "type": "variable",
+                            "rawString": "input.task",
+                        },
+                        {
+                            "type": "simpleText",
+                            "rawString": " the number of ",
+                        },
+                        {
+                            "type": "variable",
+                            "rawString": "input.times",
+                        },
+                        {
+                            "type": "simpleText",
+                            "rawString": ".",
+                        },
+                    ],
                 },
             ],
             "inputSchema": {
@@ -225,6 +256,13 @@ def test_agent_with_all_tool_types_loads(self):
                         "properties": {"output": {"type": "string"}},
                     },
                     "settings": {},
+                    "argumentProperties": {
+                        "task": {
+                            "variant": "argument",
+                            "argumentPath": "$['task']",
+                            "isSensitive": False,
+                        }
+                    },
                     "properties": {
                         "processName": "Basic RPA Process",
                         "folderPath": "TestFolder/Complete Solution 30 Sept",
@@ -270,6 +308,18 @@ def test_agent_with_all_tool_types_loads(self):
                                 },
                                 "required": ["timezone"],
                             },
+                            "argumentProperties": {
+                                "timezone": {
+                                    "variant": "textBuilder",
+                                    "tokens": [
+                                        {
+                                            "type": "simpleText",
+                                            "rawString": "Europe/London",
+                                        },
+                                    ],
+                                    "isSensitive": False,
+                                },
+                            },
                         },
                         {
                             "name": "convert_time",